| |
| import os |
| import yaml |
| import importlib |
|
|
| from smolagents import ( |
| CodeAgent, |
| DuckDuckGoSearchTool, |
| VisitWebpageTool, |
| WikipediaSearchTool, |
| OpenAIServerModel, |
| SpeechToTextTool, |
| FinalAnswerTool, |
| ) |
|
|
| |
| from tools import GetTaskFileTool, LoadXlsxFileTool, LoadTextFileTool |
|
|
| sys_instruction = ( |
| "You are a general AI assistant. Answer each question by reporting your thoughts, " |
| "then submit ONLY a concise text using the 'final_answer' tool. " |
| "Final answer MUST be a number, a few words, or a comma-separated list of numbers and/or strings. " |
| "For numbers, avoid commas and units unless specified. For strings, avoid articles and abbreviations, " |
| "and write digits in full unless stated otherwise. Apply these rules for list elements as well." |
| ) |
|
|
| 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 an expert and helpful agent named {{name}}.\n" |
| "A valued client has assigned you the following task:\n" |
| "---\n" |
| "Task:\n" |
| "{{task}}\n" |
| "---\n" |
| "Use your tools as needed. Before completing the task, plan your actions carefully.\n" |
| "While completing the task, think step by step. And after completing the task, carefully double check your solution.\n\n" |
| "If you respond correctly, you will be rewarded with a very high bonus.\n\n" |
| "Your final_answer MUST be:\n" |
| "- a number,\n" |
| "- a short phrase,\n" |
| "- or a comma-separated list of numbers or strings (no articles or abbreviations).\n\n" |
| "Only the content passed to the final_answer tool will be preserved—everything else 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(), |
| 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 |
