Broomva commited on
Commit
6c0ccf6
·
1 Parent(s): cd4461d

Updated profiles names

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -95,11 +95,11 @@ async def chat_profile(current_user: cl.AppUser):
95
 
96
  return [
97
  cl.ChatProfile(
98
- name="Broomva Book Agent Lite",
99
  markdown_description="The underlying LLM model is **GPT-3.5**.",
100
  ),
101
  cl.ChatProfile(
102
- name="Broomva Book Agent Turbo",
103
  markdown_description="The underlying LLM model is **GPT-4 Turbo**.",
104
  ),
105
  ]
@@ -148,9 +148,9 @@ async def init():
148
 
149
  chat_profile = cl.user_session.get("chat_profile")
150
 
151
- if chat_profile == "Broomva Book Agent Lite":
152
  settings["model"] = "gpt-3.5-turbo"
153
- elif chat_profile == "Broomva Book Agent Turbo":
154
  settings["model"] = "gpt-4-1106-preview"
155
 
156
  chain = RetrievalQAWithSourcesChain.from_chain_type(
 
95
 
96
  return [
97
  cl.ChatProfile(
98
+ name="Turbo Agent",
99
  markdown_description="The underlying LLM model is **GPT-3.5**.",
100
  ),
101
  cl.ChatProfile(
102
+ name="GPT4 Agent",
103
  markdown_description="The underlying LLM model is **GPT-4 Turbo**.",
104
  ),
105
  ]
 
148
 
149
  chat_profile = cl.user_session.get("chat_profile")
150
 
151
+ if chat_profile == "Turbo Agent":
152
  settings["model"] = "gpt-3.5-turbo"
153
+ elif chat_profile == "GPT4 Agent":
154
  settings["model"] = "gpt-4-1106-preview"
155
 
156
  chain = RetrievalQAWithSourcesChain.from_chain_type(