mirror of https://github.com/kortix-ai/suna.git
feat: ready for deployment
This commit is contained in:
Soumyadas15
parent
1186ffefa3
commit
78b80bc2bc
|
|
@ -1,136 +0,0 @@
|
|||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script to verify enhanced response processor for agent builder tools.
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
from backend.agentpress.response_processor import ResponseProcessor
|
||||
from backend.agentpress.tool_registry import ToolRegistry
|
||||
from backend.agentpress.tool import ToolResult
|
||||
|
||||
class MockTool:
|
||||
"""Mock tool for testing."""
|
||||
|
||||
def success_response(self, data):
|
||||
return ToolResult(success=True, output=json.dumps(data, indent=2))
|
||||
|
||||
def fail_response(self, msg):
|
||||
return ToolResult(success=False, output=msg)
|
||||
|
||||
async def mock_add_message(thread_id, type, content, is_llm_message, metadata=None):
|
||||
"""Mock add message callback."""
|
||||
return {
|
||||
"message_id": "test-message-id",
|
||||
"thread_id": thread_id,
|
||||
"type": type,
|
||||
"content": content,
|
||||
"is_llm_message": is_llm_message,
|
||||
"metadata": metadata or {}
|
||||
}
|
||||
|
||||
def test_update_agent_response():
|
||||
"""Test update_agent tool response formatting."""
|
||||
|
||||
# Create mock tool result for update_agent
|
||||
mock_tool = MockTool()
|
||||
update_result = mock_tool.success_response({
|
||||
"message": "Agent updated successfully",
|
||||
"updated_fields": ["name", "description", "system_prompt"],
|
||||
"agent": {
|
||||
"agent_id": "test-agent-123",
|
||||
"name": "Research Assistant",
|
||||
"description": "An AI assistant specialized in research",
|
||||
"system_prompt": "You are a research assistant with expertise in gathering, analyzing, and synthesizing information from various sources.",
|
||||
"agentpress_tools": {
|
||||
"web_search": {"enabled": True, "description": "Search the web"},
|
||||
"sb_files": {"enabled": True, "description": "File operations"}
|
||||
},
|
||||
"configured_mcps": [
|
||||
{"name": "Exa Search", "qualifiedName": "exa", "enabledTools": ["search"]}
|
||||
],
|
||||
"avatar": "🔬",
|
||||
"avatar_color": "#4F46E5"
|
||||
}
|
||||
})
|
||||
|
||||
# Test with agent builder mode
|
||||
tool_registry = ToolRegistry()
|
||||
processor = ResponseProcessor(
|
||||
tool_registry=tool_registry,
|
||||
add_message_callback=mock_add_message,
|
||||
is_agent_builder=True,
|
||||
target_agent_id="test-agent-123"
|
||||
)
|
||||
|
||||
tool_call = {
|
||||
"function_name": "update_agent",
|
||||
"xml_tag_name": "update_agent",
|
||||
"arguments": {"name": "Research Assistant"}
|
||||
}
|
||||
|
||||
structured_result = processor._create_structured_tool_result(tool_call, update_result)
|
||||
|
||||
print("=== Agent Builder Mode - Update Agent Tool Response ===")
|
||||
print(structured_result["summary"])
|
||||
print("\n" + "="*60 + "\n")
|
||||
|
||||
# Test without agent builder mode
|
||||
processor_normal = ResponseProcessor(
|
||||
tool_registry=tool_registry,
|
||||
add_message_callback=mock_add_message,
|
||||
is_agent_builder=False
|
||||
)
|
||||
|
||||
structured_result_normal = processor_normal._create_structured_tool_result(tool_call, update_result)
|
||||
|
||||
print("=== Normal Mode - Update Agent Tool Response ===")
|
||||
print(structured_result_normal["summary"])
|
||||
print("\n" + "="*60 + "\n")
|
||||
|
||||
def test_get_current_agent_config_response():
|
||||
"""Test get_current_agent_config tool response formatting."""
|
||||
|
||||
mock_tool = MockTool()
|
||||
config_result = mock_tool.success_response({
|
||||
"summary": "Agent 'Research Assistant' has 2 tools enabled and 1 MCP servers configured.",
|
||||
"configuration": {
|
||||
"agent_id": "test-agent-123",
|
||||
"name": "Research Assistant",
|
||||
"description": "An AI assistant specialized in research",
|
||||
"system_prompt": "You are a research assistant with expertise in gathering, analyzing, and synthesizing information from various sources. Your approach is thorough and methodical.",
|
||||
"agentpress_tools": {
|
||||
"web_search": {"enabled": True, "description": "Search the web"},
|
||||
"sb_files": {"enabled": False, "description": "File operations"}
|
||||
},
|
||||
"configured_mcps": [],
|
||||
"avatar": "🔬",
|
||||
"avatar_color": "#4F46E5"
|
||||
}
|
||||
})
|
||||
|
||||
tool_registry = ToolRegistry()
|
||||
processor = ResponseProcessor(
|
||||
tool_registry=tool_registry,
|
||||
add_message_callback=mock_add_message,
|
||||
is_agent_builder=True,
|
||||
target_agent_id="test-agent-123"
|
||||
)
|
||||
|
||||
tool_call = {
|
||||
"function_name": "get_current_agent_config",
|
||||
"xml_tag_name": "get_current_agent_config",
|
||||
"arguments": {}
|
||||
}
|
||||
|
||||
structured_result = processor._create_structured_tool_result(tool_call, config_result)
|
||||
|
||||
print("=== Agent Builder Mode - Get Current Agent Config Response ===")
|
||||
print(structured_result["summary"])
|
||||
print("\n" + "="*60 + "\n")
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("Testing Enhanced Response Processor for Agent Builder Tools\n")
|
||||
test_update_agent_response()
|
||||
test_get_current_agent_config_response()
|
||||
print("✅ All tests completed!")
|
||||
Loading…
Reference in New Issue