Spaces:
Build error
Build error
Refactor agent structure by modularizing agent implementations into separate directories for web, data analysis, and media agents. Remove legacy code from agents.py, prompts.py, and tools.py, enhancing maintainability. Update main_v2.py to reflect new import paths and agent initialization. Add new tools for enhanced functionality, including web searching and data extraction. Update requirements.txt to include necessary dependencies for new tools.
837e221
unverified
| import importlib | |
| import yaml | |
| from smolagents import CodeAgent | |
| from tools import parse_csv, perform_calculation | |
| def create_data_agent(model): | |
| """ | |
| Create a specialized agent for data analysis tasks. | |
| Args: | |
| model: The model to use for the agent | |
| Returns: | |
| Configured CodeAgent for data analysis | |
| """ | |
| # Load default prompts | |
| prompt_templates = yaml.safe_load( | |
| importlib.resources.files("smolagents.prompts") | |
| .joinpath("code_agent.yaml") | |
| .read_text() | |
| ) | |
| data_agent = CodeAgent( | |
| tools=[parse_csv, perform_calculation], | |
| model=model, | |
| name="data_agent", | |
| description="Specialized agent for data analysis. Use this agent to analyze data, perform calculations, and extract insights from structured data.", | |
| add_base_tools=True, | |
| additional_authorized_imports=["pandas", "numpy", "math", "csv", "io"], | |
| prompt_templates=prompt_templates, | |
| ) | |
| return data_agent |