소스 검색

fix: session save

master
zhayujie 1 년 전
부모
커밋
50e44dbb2a
1개의 변경된 파일3개의 추가작업 그리고 1개의 파일을 삭제
  1. +3
    -1
      bot/linkai/link_ai_bot.py

+ 3
- 1
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 @@ class LinkAIBot(Bot):

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 @@ class LinkAIBot(Bot):
# 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)


Loading…
취소
저장