| |
| import os |
| import yaml |
| import importlib |
|
|
| from smolagents import ( |
| CodeAgent, |
| DuckDuckGoSearchTool, |
| VisitWebpageTool, |
| WikipediaSearchTool, |
| OpenAIServerModel, |
| SpeechToTextTool, |
| FinalAnswerTool, |
| ) |
|
|
| |
| from tools import GetTaskFileTool, LoadXlsxFileTool, LoadTextFileTool, AnalyzeImageTool |
|
|
| sys_instruction = ( |
| "You are a general-purpose AI assistant. For each question, first reason through your answer, " |
| "then use the 'final_answer' tool to submit ONLY the final output.\n" |
| "Your final answer MUST be one of the following:\n" |
| "- A number (without commas or units, unless explicitly instructed),\n" |
| "- A short phrase,\n" |
| "- A comma-separated list of numbers or strings.\n" |
| "For strings, avoid articles and abbreviations, and write digits fully unless told otherwise. " |
| "Apply these formatting rules consistently to each item in a list." |
| ) |
|
|
| prompts = yaml.safe_load( |
| importlib.resources.files("smolagents.prompts").joinpath("code_agent.yaml").read_text() |
| ) |
| prompts["system_prompt"] = sys_instruction + prompts["system_prompt"] |
|
|
| |
| req_instruction = ( |
| "You are a highly capable and autonomous agent named {{name}}, designed to solve complex tasks efficiently.\n" |
| "A valued client has assigned you the following task:\n" |
| "---\n" |
| "Task:\n" |
| "{{task}}\n" |
| "---\n" |
| "To complete this task successfully, follow these steps carefully:\n" |
| " 1. Comprehend the task and identify the intended goal.\n" |
| " 2. Break the task into clear, logical steps.\n" |
| " 3. Select and prepare the tools or resources you need.\n" |
| " 4. Set up the required environment or context.\n" |
| " 5. Execute each step methodically.\n" |
| " 6. Monitor outcomes and identify any deviations.\n" |
| " 7. Revise your plan if necessary based on feedback.\n" |
| " 8. Maintain internal state and track progress.\n" |
| " 9. Verify that the goal has been fully achieved.\n" |
| " 10. Present the final result clearly and concisely.\n" |
| "If you succeed, you will be rewarded with a significant bonus.\n\n" |
| "Your final_answer MUST be:\n" |
| "- a number (retain its original type; do not include units),\n" |
| "- a concise phrase,\n" |
| "- or a comma-separated list of numbers or strings (no articles, no abbreviations).\n\n" |
| "Only the content passed to the final_answer tool will be preserved. Any other content will be discarded." |
| ) |
|
|
| prompts['managed_agent']['task'] = req_instruction |
| prompts['managed_agent']['report'] = "{{final_answer}}" |
|
|
|
|
| def get_model( |
| model_id: str = "gpt-4.1-mini", |
| model_temperature: float = 0.7, |
| ): |
| """ |
| Create and return an OpenAIServerModel instance with the specified model ID and temperature. |
| """ |
| |
| if "gpt" in model_id: |
| model = OpenAIServerModel( |
| model_id=model_id, |
| api_key=os.getenv("OPENAI_API_KEY"), |
| temperature=model_temperature |
| ) |
| elif "gemini" in model_id: |
| model = OpenAIServerModel( |
| model_id=model_id, |
| api_key=os.getenv("GOOGLEAI_API_KEY"), |
| api_base="https://generativelanguage.googleapis.com/v1beta/openai/", |
| temperature=model_temperature |
| ) |
| else: |
| raise ValueError(f"Unknown model_id: {model_id}. Supported models are gpt and gemini.") |
| return model |
|
|
| def get_agent( |
| model_id: str = "gpt-4.1-mini", |
| model_temperature: float = 0.7, |
| agent_max_steps: int = 15, |
| ): |
| """ |
| Create and return a CodeAgent instance with the specified model and tools. |
| """ |
|
|
| |
| agent = CodeAgent( |
| tools=[ |
| DuckDuckGoSearchTool(), |
| VisitWebpageTool(), |
| WikipediaSearchTool(), |
| GetTaskFileTool(), |
| SpeechToTextTool(), |
| LoadXlsxFileTool(), |
| LoadTextFileTool(), |
| AnalyzeImageTool(model_id=model_id), |
| FinalAnswerTool(), |
| ], |
| model=get_model( |
| model_id=model_id, |
| model_temperature=model_temperature, |
| ), |
| prompt_templates=prompts, |
| max_steps=agent_max_steps, |
| additional_authorized_imports = ["pandas"], |
| name="GAIAAgent", |
| ) |
| return agent |
|
|
| def test(): |
| """test tests something. """ |
| |
| return True |
