diff --git a/bot/linkai/link_ai_bot.py b/bot/linkai/link_ai_bot.py index 9dd18b1a1..e7b9703d5 100644 --- 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)