mirror of https://github.com/kortix-ai/suna.git
87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
from typing import Dict, Any, List, Optional
|
|
from agentpress.processor.base_processors import ResultsAdderBase
|
|
|
|
# --- Standard Results Adder Implementation ---
|
|
|
|
class StandardResultsAdder(ResultsAdderBase):
|
|
"""Standard implementation for handling tool results and message processing.
|
|
|
|
Provides straightforward implementations for adding and updating messages
|
|
in a conversation thread, maintaining proper message ordering and structure.
|
|
|
|
Methods:
|
|
add_initial_response: Add the first response in a sequence
|
|
update_response: Update an existing response
|
|
add_tool_result: Add a tool execution result
|
|
"""
|
|
|
|
def __init__(self, thread_manager):
|
|
"""Initialize with ThreadManager instance.
|
|
|
|
Args:
|
|
thread_manager: Instance providing message management capabilities
|
|
"""
|
|
super().__init__(thread_manager)
|
|
|
|
async def add_initial_response(self, thread_id: str, content: str, tool_calls: Optional[List[Dict[str, Any]]] = None):
|
|
"""Add the initial assistant response to the thread.
|
|
|
|
Args:
|
|
thread_id: ID of the conversation thread
|
|
content: Text content of the response
|
|
tool_calls: Optional list of tool calls to include
|
|
|
|
Notes:
|
|
- Sets message_added flag to True after successful addition
|
|
- Includes tool_calls in message if provided
|
|
"""
|
|
message = {
|
|
"role": "assistant",
|
|
"content": content
|
|
}
|
|
if tool_calls:
|
|
message["tool_calls"] = tool_calls
|
|
|
|
await self.add_message(thread_id, message)
|
|
self.message_added = True
|
|
|
|
async def update_response(self, thread_id: str, content: str, tool_calls: Optional[List[Dict[str, Any]]] = None):
|
|
"""Update an existing assistant response in the thread.
|
|
|
|
Args:
|
|
thread_id: ID of the conversation thread
|
|
content: Updated text content
|
|
tool_calls: Optional updated list of tool calls
|
|
|
|
Notes:
|
|
- Creates initial message if none exists
|
|
- Updates existing message with new content and tool calls
|
|
"""
|
|
if not self.message_added:
|
|
await self.add_initial_response(thread_id, content, tool_calls)
|
|
return
|
|
|
|
message = {
|
|
"role": "assistant",
|
|
"content": content
|
|
}
|
|
if tool_calls:
|
|
message["tool_calls"] = tool_calls
|
|
|
|
await self.update_message(thread_id, message)
|
|
|
|
async def add_tool_result(self, thread_id: str, result: Dict[str, Any]):
|
|
"""Add a tool execution result to the thread.
|
|
|
|
Args:
|
|
thread_id: ID of the conversation thread
|
|
result: Tool execution result to add
|
|
|
|
Notes:
|
|
- Checks for duplicate tool results before adding
|
|
- Adds result only if tool_call_id is unique
|
|
"""
|
|
messages = await self.get_messages(thread_id)
|
|
if not any(msg.get('tool_call_id') == result['tool_call_id'] for msg in messages):
|
|
await self.add_message(thread_id, result)
|