Adding gemini model if needed to llm only agent

llm_only_agent.py

@@ -1,3 +1,4 @@
+import os
 import re
 from smolagents import (
     AgentMemory,
@@ -9,6 +10,7 @@ from smolagents import (
 from collections.abc import Callable
 
 from smolagents.default_tools import VisitWebpageTool, WikipediaSearchTool
+from smolagents.models import OpenAIModel
 
 
 class LLMOnlyAgent:
@@ -21,13 +23,23 @@ class LLMOnlyAgent:
         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."""
 
         # Basic inference model
-        model = InferenceClientModel(
+        qwen_model = InferenceClientModel(
             max_tokens=8096,
             model_id="Qwen/Qwen3-Coder-30B-A3B-Instruct",
             custom_role_conversions=None,
             provider="nebius",
         )
 
+        gemini_model = OpenAIModel(
+            max_tokens=8096,
+            model_id="gemini-2.5-flash",
+            # Google Gemini OpenAI-compatible API base URL
+            api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
+            api_key=os.environ["GEMINI_API_KEY"],
+        )
+
+        model = gemini_model
+
         # Code Agent
         self.agent = CodeAgent(
             model=model,

requirements.txt

@@ -7,6 +7,7 @@ wikipedia-api
 markdownify
 requests
 smolagents[telemetry,toolkit]
+smolagents[openai]
 chess
 pandas
 ipykernel