Update agent.py

agent.py

@@ -12,7 +12,7 @@ from typing import Any
 from dotenv import load_dotenv
 from google.generativeai import types, configure
 
-from smolagents import InferenceClientModel, LiteLLMModel, CodeAgent, ToolCallingAgent, Tool, DuckDuckGoSearchTool, HfApiModel
+from smolagents import InferenceClientModel, LiteLLMModel, CodeAgent, ToolCallingAgent, Tool, DuckDuckGoSearchTool, HfApiModel, OpenAIServerModel
 
 # Logging
 #logging.basicConfig(level=logging.INFO, format="%(asctime)s | %(levelname)s | %(message)s")
@@ -20,6 +20,7 @@ from smolagents import InferenceClientModel, LiteLLMModel, CodeAgent, ToolCallin
 
 # --- Model Configuration ---
 HF_MODEL_NAME = "Qwen/Qwen2.5-Coder-32B-Instruct"
+OPENROUTER_API_KEY = "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"
 
 # --- Tool Definitions ---
 class MathSolver(Tool):
@@ -91,6 +92,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         #model = HF_MODEL_NAME
+        """
         model = HfApiModel(
             max_tokens=2096,
             temperature=0.5,
@@ -98,6 +100,14 @@ class BasicAgent:
             #model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
             custom_role_conversions=None,
             )
+        """
+        model = OpenAIServerModel(
+            # You can use any model ID available on OpenRouter
+            model_id="mistralai/mistral-small-3.2-24b-instruct:free",
+            # OpenRouter API base URL
+            api_base="https://openrouter.ai/api/v1",
+            api_key=OPENROUTER_API_KEY,
+        )
         tools = [
             DuckDuckGoSearchTool(),
             WikiTitleFinder(),