From 90ee3585cba1b97c21948536902b720e48cd7fbd Mon Sep 17 00:00:00 2001 From: Krishav Raj Singh Date: Thu, 9 Oct 2025 01:25:34 +0530 Subject: [PATCH] fiix: get usage info in case of tool --- backend/core/agentpress/response_processor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/core/agentpress/response_processor.py b/backend/core/agentpress/response_processor.py index aa960578..c42d4891 100644 --- a/backend/core/agentpress/response_processor.py +++ b/backend/core/agentpress/response_processor.py @@ -488,9 +488,9 @@ class ResponseProcessor: tool_index += 1 if finish_reason == "xml_tool_limit_reached": - logger.info("Stopping stream processing after loop due to XML tool call limit") - self.trace.event(name="stopping_stream_processing_after_loop_due_to_xml_tool_call_limit", level="DEFAULT", status_message=(f"Stopping stream processing after loop due to XML tool call limit")) - break + logger.info("XML tool call limit reached, continuing stream to capture usage data") + self.trace.event(name="xml_tool_call_limit_reached_continuing_stream", level="DEFAULT", status_message=(f"XML tool call limit reached, continuing stream to capture usage data")) + # Don't break - continue processing stream to capture final usage chunk logger.info(f"Stream complete. Total chunks: {chunk_count}")