|
|
@ -2,6 +2,17 @@ from swarms import Agent
|
|
|
|
from swarms.tools.mcp_integration import MCPServerSseParams
|
|
|
|
from swarms.tools.mcp_integration import MCPServerSseParams
|
|
|
|
from swarms.prompts.agent_prompts import MATH_AGENT_PROMPT
|
|
|
|
from swarms.prompts.agent_prompts import MATH_AGENT_PROMPT
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Fallback in case the import fails
|
|
|
|
|
|
|
|
if 'MATH_AGENT_PROMPT' not in locals():
|
|
|
|
|
|
|
|
MATH_AGENT_PROMPT = """You are a specialized math agent that can perform calculations by calling external math service APIs.
|
|
|
|
|
|
|
|
Key responsibilities:
|
|
|
|
|
|
|
|
1. Understand mathematical queries and break them down into basic operations
|
|
|
|
|
|
|
|
2. Use available math tools (add, multiply, divide) appropriately
|
|
|
|
|
|
|
|
3. Provide clear explanations of calculations
|
|
|
|
|
|
|
|
4. Handle errors gracefully if operations fail
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Remember to use the available MCP tools for calculations rather than doing them directly."""
|
|
|
|
|
|
|
|
|
|
|
|
def main():
|
|
|
|
def main():
|
|
|
|
# Configure MCP server connection
|
|
|
|
# Configure MCP server connection
|
|
|
|
math_server = MCPServerSseParams(
|
|
|
|
math_server = MCPServerSseParams(
|
|
|
|