fix: session save

tudou2 · May 28, 2023 · 50e44db · 50e44db
1 parent 34277a3
commit 50e44db
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bot/linkai/link_ai_bot.py b/bot/linkai/link_ai_bot.py
@@ -16,6 +16,7 @@ class LinkAIBot(Bot):
 
     def __init__(self):
         self.base_url = "https://api.link-ai.chat/v1"
+        self.sessions = SessionManager(ChatGPTSession, model=conf().get("model") or "gpt-3.5-turbo")
 
     def reply(self, query, context: Context = None) -> Reply:
         return self._chat(query, context)
@@ -28,7 +29,7 @@ def _chat(self, query, context, retry_count=0):
 
         try:
             session_id = context["session_id"]
-            self.sessions = SessionManager(ChatGPTSession, model=conf().get("model") or "gpt-3.5-turbo")
+
             session = self.sessions.session_query(query, session_id)
 
             # remove system message
@@ -61,6 +62,7 @@ def _chat(self, query, context, retry_count=0):
             # execute success
             reply_content = res["data"]["content"]
             logger.info(f"[LINKAI] reply={reply_content}")
+            self.sessions.session_reply(reply_content, session_id)
             return Reply(ReplyType.TEXT, reply_content)
         except Exception as e:
             logger.exception(e)