Spaces:
Sleeping
Sleeping
File size: 1,569 Bytes
d853cbf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | """
WebSearch Agent - MCP client using Agents SDK
"""
import logging
from agents import Agent
from agents.mcp import MCPServerStdio, create_static_tool_filter
from agents.model_settings import ModelSettings
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
# System prompt for the WebSearchAgent
WEB_SEARCH_INSTRUCTIONS = """You are a research assistant specialized in Python, pandas, matplotlib, and data analysis.
Your role is to search the web for:
- Documentation and API references
- Solutions to Python/pandas errors
- Code examples and best practices
- Matplotlib/seaborn visualization techniques
Use the MCP tool `search_web(query)` to find relevant information.
Guidelines:
- Provide concise, actionable summaries
- Include concrete code snippets when available
- Focus on authoritative sources (official docs, Stack Overflow, etc.)
- Return your findings in a clear, structured format
- The code you provide should be formatted as
```python
YOUR CODE HERE
```
"""
def create_web_search_agent(mcp_server: MCPServerStdio, model: str = "gpt-4o-mini") -> Agent:
"""
Create an Agent SDK for web search
Args:
mcp_server: MCP server already configured with search_web tool filter
model: OpenAI model to use
Returns:
Agent configured for web search
"""
return Agent(
name="WebSearchAgent",
instructions=WEB_SEARCH_INSTRUCTIONS,
mcp_servers=[mcp_server],
model=model,
model_settings=ModelSettings(tool_choice="required"),
)
|