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

feat: revert reponse change

This commit is contained in:
Vukasin 2025-08-08 09:59:57 +02:00
parent a948e63c4d
commit b18e1d65dd
1 changed files with 2 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
backend/agentpress/response_processor.py
View File

@ -1009,25 +1009,11 @@ class ResponseProcessor:
# --- Save and Yield assistant_response_end --- # --- Save and Yield assistant_response_end ---
if assistant_message_object: # Only save if assistant message was saved if assistant_message_object: # Only save if assistant message was saved
try: try:
# Ensure content is JSON-serializable # Save the full LiteLLM response object directly in content
serializable_response = None
try:
if hasattr(llm_response, "model_dump_json"):
import json as _json
serializable_response = _json.loads(llm_response.model_dump_json())
elif hasattr(llm_response, "model_dump"):
serializable_response = llm_response.model_dump() # type: ignore[attr-defined]
elif hasattr(llm_response, "dict"):
serializable_response = llm_response.dict() # type: ignore[attr-defined]
else:
serializable_response = str(llm_response)
except Exception:
serializable_response = str(llm_response)
await self.add_message( await self.add_message(
thread_id=thread_id, thread_id=thread_id,
type="assistant_response_end", type="assistant_response_end",
content=serializable_response, content=llm_response,
is_llm_message=False, is_llm_message=False,
metadata={"thread_run_id": thread_run_id} metadata={"thread_run_id": thread_run_id}
) )