|
|
|
|
|
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 |
