修改缺陷:获取会话时未记录选择的工程信息

This commit is contained in:
wanyaokun
2024-09-02 09:26:26 +08:00
parent 24c808d66d
commit a4dd385368
2 changed files with 3 additions and 2 deletions
+1 -1
View File
@@ -438,7 +438,7 @@ async def post_conversations(request: Request, data: ChatRequestData):
conversaObj = conversations() conversaObj = conversations()
conversationinfo = conversaObj.get(data.conversation_id) conversationinfo = conversaObj.get(data.conversation_id)
if conversationinfo is None: if conversationinfo is None:
conversationinfo = conversaObj.add(data.conversation_id, data.user, "新建会话") conversationinfo = conversaObj.add(data.conversation_id, data.user, "新建会话",inputs= data.inputs)
# 生成聊天参数 # 生成聊天参数
last_message_content = ChatMessage.from_str(data.query) last_message_content = ChatMessage.from_str(data.query)
+2 -1
View File
@@ -24,12 +24,13 @@ class conversations:
return records[0] return records[0]
return None return None
def add(self,id:str, user_id:str, name:str): def add(self,id:str, user_id:str, name:str,inputs:dict):
template = BaseConfig().ConversationCfg() template = BaseConfig().ConversationCfg()
template['id'] = id template['id'] = id
template['user_id'] = user_id template['user_id'] = user_id
template['name'] = name template['name'] = name
template['created_at'] = 1724399038 template['created_at'] = 1724399038
template['inputs'] = inputs
dbManage.addRecord(self._tableName,template) dbManage.addRecord(self._tableName,template)
def delete(self,id:str): def delete(self,id:str):