modify some code and update chainlit 1.3.0

Mgrsc · Oct 25, 2024 · 79fdbeb · 79fdbeb
1 parent 94a16a3
commit 79fdbeb
Show file tree

Hide file tree

Showing 3 changed files with 114 additions and 103 deletions.
diff --git a/src/app.py b/src/app.py
@@ -7,10 +7,6 @@
 from utils.logger import logger
 from chainlit.config import config
 
-# setting default tag and history
-@cl.on_chat_start
-def start_chat():
-    cl.user_session.set("is_new_session", True)
 
 # Construct a new conversation setting system prompt.
 async def handle_new_session(user_input: str):
@@ -28,6 +24,11 @@ async def handle_new_session(user_input: str):
 
     cl.user_session.set("is_new_session", False)
 
+# setting default tag and history
+@cl.on_chat_start
+def start_chat():
+    cl.user_session.set("is_new_session", True)
+
 # main logic ,receive message will call this function
 @cl.on_message
 async def main(message: cl.Message):

diff --git a/src/config.py b/src/config.py
@@ -76,10 +76,5 @@ async def chat_profile():
 
 # get model for profile
 def get_model_for_profile(chat_profile: str) -> str:
-    model_map = config.get('MODEL_MAP', {})
-    if model_map.get(chat_profile):
-        model_name = model_map.get(chat_profile).get('model_name', 'gpt-4o-mini')
-        return model_name
-    return "gpt-4o-mini"
-
+    return config.get('MODEL_MAP', {}).get(chat_profile, {}).get('model_name', 'gpt-4o-mini')