agent.py

from smolagents import LiteLLMModel, ToolCallingAgent, CodeAgent
from tools import DuckDuckGoSearchTool
import os
# also call HfApiModel in case we need it
# ...nevermind, it fives me a runtime error, no HfApiModel to be used then...

# Agents go here, let's try some

# Agent 1: BasicAgent - GEMINI
class BasicAgent:
    def __init__(self):
        print("BasicAgent initialized.")

    def __call__(self, question: str) -> str:
        print(f"Agent received question (first 50 chars): {question[:50]}...")

        # now for the agent:
        
        model = LiteLLMModel(model_id="gemini/gemini-2.5-pro-preview-06-05", api_key=os.getenv(key="MY_GEMINI_API_KEY"))
        
        # my_gemini_model = os.getenv(key="GEMINI_MODEL")
        # model = LiteLLMModel(model_id=f"gemini/{my_gemini_model}", api_key=os.getenv(key="MY_GEMINI_API_KEY"))
        
        agent = CodeAgent(
            model=model,
            tools=[DuckDuckGoSearchTool()]
        )

        prompt = f"""You are a general AI assistant. I will ask you a question. Report your thoughts, and 
    finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should 
    be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked 
    for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified 
    otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and 
    write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply 
    the above rules depending of whether the element to be put in the list is a number or a string.

    {question}
    """
        answer = agent.run(prompt)
        
        print(f"Agent returning answer: {answer}")
        return answer