zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-02 22:07:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle null response in LLM and improve JSON parsing in agent (#13187)

Browse Source 
Fixes AttributeError in _remove_reasoning_content() when LLM returns
None, and improves JSON parsing regex for markdown code fences in
agent_with_tools.py
This commit is contained in:
tuandang-diag
2026-02-24 12:15:09 +07:00
committed by GitHub
parent 67befc9119
commit d89ad8b79d
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
agent/component/agent_with_tools.py
View File

@@ -383,7 +383,10 @@ class Agent(LLM, ToolBase):
token_count += tk or 0
hist.append({"role": "assistant", "content": response})
try:
functions = json_repair.loads(re.sub(r"```.*", "", response))
# Remove markdown code fences properly
cleaned_response = re.sub(r"^.*```json\s*", "", response, flags=re.DOTALL)
cleaned_response = re.sub(r"```\s*$", "", cleaned_response, flags=re.DOTALL)
functions = json_repair.loads(cleaned_response)
if not isinstance(functions, list):
raise TypeError(f"List should be returned, but `{functions}`")
for f in functions: