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

fix: Return async generator directly for non-streaming responses

This commit is contained in:
LE Quoc Dat 2025-04-16 19:08:44 +01:00
parent bc7662a484
commit 7de2756b44
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
backend/agentpress/thread_manager.py
View File

@ -325,16 +325,13 @@ Here are the XML tools available with examples:
return response_generator
else:
logger.debug("Processing non-streaming response")
try:
response = await self.response_processor.process_non_streaming_response(
llm_response=llm_response,
thread_id=thread_id,
config=processor_config
)
return response
except Exception as e:
logger.error(f"Error in non-streaming response: {str(e)}", exc_info=True)
raise
# Return the async generator directly, don't await it
response_generator = self.response_processor.process_non_streaming_response(
llm_response=llm_response,
thread_id=thread_id,
config=processor_config
)
return response_generator # Return the generator
except Exception as e:
logger.error(f"Error in run_thread: {str(e)}", exc_info=True)