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

读取接口内层嵌套BUG修复

Browse Source
This commit is contained in:
lixinyue
2026-01-20 16:32:52 +08:00
parent fe79978f88
commit a5ecbec9a6
1 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
api/app/core/memory/agent/langgraph_graph/routing/routers.py
View File

@@ -45,18 +45,17 @@ def Retrieve_continue(state) -> Literal["Verify", "Retrieve_Summary"]:
return 'Retrieve_Summary' # Default based on business logic return 'Retrieve_Summary' # Default based on business logic
def Verify_continue(state: ReadState) -> Literal["Summary", "Summary_fails", "content_input"]: def Verify_continue(state: ReadState) -> Literal["Summary", "Summary_fails", "content_input"]:
status=state.get('verify', '')['status'] status=state.get('verify', '')['status']
loop_count = counter.get_total() # loop_count = counter.get_total()
print(status)
if "success" in status: if "success" in status:
counter.reset() # counter.reset()
return "Summary" return "Summary"
elif "failed" in status: elif "failed" in status:
if loop_count < 2: # Maximum loop count is 3 # if loop_count < 2: # Maximum loop count is 3
return "content_input" # return "content_input"
else: # else:
counter.reset() # counter.reset()
return "Summary_fails" return "Summary_fails"
# else: else:
# # Add default return value to avoid returning None # Add default return value to avoid returning None
# counter.reset() # counter.reset()
# return "Summary" # Default based on business requirements return "Summary" # Default based on business requirements