zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-04 14:50:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Agent session lost <think></think> tag (#16670)

Browse Source
This commit is contained in:
Wang Qi
2026-07-06 17:47:38 +08:00
committed by GitHub
parent 57625c919a
commit ed52255868
2 changed files with 57 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/apps/restful_apis/agent_api.py
View File

@@ -315,6 +315,10 @@ async def _run_workflow_session(
ans["session_id"] = session_id
if ans.get("event") == "message":
full_content += ans.get("data", {}).get("content", "")
if ans.get("data", {}).get("start_to_think", False):
full_content += "<think>"
elif ans.get("data", {}).get("end_to_think", False):
full_content += "</think>"
if ans.get("data", {}).get("reference", None):
reference.update(ans["data"]["reference"])
if ans.get("event") == "node_finished":
@@ -371,6 +375,10 @@ async def _run_workflow_session(
ans["session_id"] = session_id
if ans.get("event") == "message":
full_content += ans.get("data", {}).get("content", "")
if ans.get("data", {}).get("start_to_think", False):
full_content += "<think>"
elif ans.get("data", {}).get("end_to_think", False):
full_content += "</think>"
if ans.get("data", {}).get("reference", None):
reference.update(ans["data"]["reference"])
if ans.get("event") == "node_finished":
@@ -2467,6 +2475,7 @@ async def webhook_trace(agent_id: str):
}
)
def _attachment_request_metadata():
ext = request.args.get("ext")
mime_type = request.args.get("mime_type")