Ling0925
/
suna
mirror of https://github.com/kortix-ai/suna.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'PRODUCTION'

This commit is contained in:
sharath 2025-06-05 12:17:59 +00:00
parent 90af2d0bf7 a1d5ed00cb
commit 524e1beb67
No known key found for this signature in database
2 changed files with 24 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/agentpress/response_processor.py
View File

@ -357,6 +357,7 @@ class ResponseProcessor:
): ):
logger.info("🔥 No usage data from provider, counting with litellm.token_counter") logger.info("🔥 No usage data from provider, counting with litellm.token_counter")
try:
# prompt side # prompt side
prompt_tokens = token_counter( prompt_tokens = token_counter(
model=llm_model, model=llm_model,
@ -377,6 +378,10 @@ class ResponseProcessor:
f"🔥 Estimated tokens – prompt: {prompt_tokens}, " f"🔥 Estimated tokens – prompt: {prompt_tokens}, "
f"completion: {completion_tokens}, total: {prompt_tokens + completion_tokens}" f"completion: {completion_tokens}, total: {prompt_tokens + completion_tokens}"
) )
self.trace.event(name="usage_calculated_with_litellm_token_counter", level="DEFAULT", status_message=(f"Usage calculated with litellm.token_counter"))
except Exception as e:
logger.warning(f"Failed to calculate usage: {str(e)}")
self.trace.event(name="failed_to_calculate_usage", level="WARNING", status_message=(f"Failed to calculate usage: {str(e)}"))
# Wait for pending tool executions from streaming phase # Wait for pending tool executions from streaming phase
@ -1273,7 +1278,7 @@ class ResponseProcessor:
"arguments": params # The extracted parameters "arguments": params # The extracted parameters
} }
logger.debug(f"Parsed old format tool call: {tool_call}") logger.debug(f"Parsed old format tool call: {tool_call["function_name"]}")
return tool_call, parsing_details # Return both dicts return tool_call, parsing_details # Return both dicts
except Exception as e: except Exception as e:

2
backend/run_agent_background.py
View File

@ -243,7 +243,7 @@ async def run_agent_background(
# Wait for all pending redis operations to complete, with timeout # Wait for all pending redis operations to complete, with timeout
try: try:
await asyncio.wait_for(asyncio.gather(*pending_redis_operations), timeout=5.0) await asyncio.wait_for(asyncio.gather(*pending_redis_operations), timeout=30.0)
except asyncio.TimeoutError: except asyncio.TimeoutError:
logger.warning(f"Timeout waiting for pending Redis operations for {agent_run_id}") logger.warning(f"Timeout waiting for pending Redis operations for {agent_run_id}")