Merge pull request #1792 from KrishavRajSingh/main

fiix: get usage info in case of tool
This commit is contained in:
Krishav 2025-10-09 01:26:06 +05:30 committed by GitHub
commit 2d5961d81f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions

View File

@ -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}")