adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改BUG，读取接口的end_user_id获取不准确修复，statements检索内容为空需要给空列表 (#75)

Browse Source
This commit is contained in:
lixinyue11
2026-01-12 17:40:49 +08:00
committed by GitHub
parent 6dbc411adb
commit 5b05009123
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/app/services/memory_agent_service.py
View File

@@ -396,6 +396,7 @@ class MemoryAgentService:
import time import time
start_time = time.time() start_time = time.time()
ori_message=message ori_message=message
end_user_id=group_id
# Resolve config_id if None using end_user's connected config # Resolve config_id if None using end_user's connected config
if config_id is None: if config_id is None:
try: try:
@@ -602,18 +603,24 @@ class MemoryAgentService:
repo = ShortTermMemoryRepository(db) repo = ShortTermMemoryRepository(db)
if str(search_switch)!="2": if str(search_switch)!="2":
for intermediate in intermediate_outputs: for intermediate in intermediate_outputs:
print(intermediate)
intermediate_type=intermediate['type'] intermediate_type=intermediate['type']
if intermediate_type=="search_result": if intermediate_type=="search_result":
query=intermediate['query'] query=intermediate['query']
raw_results=intermediate['raw_results'] raw_results=intermediate['raw_results']
reranked_results=raw_results.get('reranked_results',[]) reranked_results=raw_results.get('reranked_results',[])
statements=[statement['statement'] for statement in reranked_results.get('statements', [])] try:
statements=[statement['statement'] for statement in reranked_results.get('statements', [])]
except Exception as e:
statements=[]
statements=list(set(statements)) statements=list(set(statements))
retrieved_content.append({query:statements}) retrieved_content.append({query:statements})
if '信息不足，无法回答' in str(final_answer) or retrieved_content!=[]: if retrieved_content==[]:
retrieved_content=''
if '信息不足，无法回答。' != str(final_answer) :#and retrieved_content!=[]
# 使用 upsert 方法 # 使用 upsert 方法
repo.upsert( repo.upsert(
end_user_id=group_id, # 确保这个变量在作用域内 end_user_id=end_user_id, # 确保这个变量在作用域内
messages=ori_message, messages=ori_message,
aimessages=final_answer, aimessages=final_answer,
retrieved_content=retrieved_content, retrieved_content=retrieved_content,