diff --git a/application/api/answer/routes/base.py b/application/api/answer/routes/base.py index e0118b58..8eb5aafc 100644 --- a/application/api/answer/routes/base.py +++ b/application/api/answer/routes/base.py @@ -110,6 +110,8 @@ class BaseAnswerResource: yield f"data: {data}\n\n" elif "tool_calls" in line: tool_calls = line["tool_calls"] + data = json.dumps({"type": "tool_calls", "tool_calls": tool_calls}) + yield f"data: {data}\n\n" elif "thought" in line: thought += line["thought"] data = json.dumps({"type": "thought", "thought": line["thought"]})