Create the agent in a function

src/agent.py

@@ -0,0 +1,41 @@
+import yaml
+from smolagents import CodeAgent, HfApiModel
+
+from src.paths import PROMPT_FILE
+from src.tools.final_answer import FinalAnswerTool
+from src.tools.image_generation import image_generation_tool
+from src.tools.weather import get_current_weather_in_location
+
+
+def get_agent() -> CodeAgent:
+    final_answer = FinalAnswerTool()
+
+    # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
+    # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
+
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        custom_role_conversions=None,
+    )
+
+    with open(PROMPT_FILE, "r") as stream:
+        prompt_templates = yaml.safe_load(stream)
+
+    agent = CodeAgent(
+        model=model,
+        tools=[
+            image_generation_tool,
+            get_current_weather_in_location,
+            final_answer,
+        ],
+        max_steps=6,
+        verbosity_level=1,
+        grammar=None,
+        planning_interval=None,
+        name=None,
+        description=None,
+        prompt_templates=prompt_templates,
+    )
+    return agent

src/paths.py

@@ -0,0 +1,5 @@
+from pathlib import Path
+
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+PROMPTS_FOLDER = PROJECT_ROOT / "src" / "prompts"
+PROMPT_FILE = PROMPTS_FOLDER / "agent.yaml"