Update agent.py

agent.py

@@ -1,13 +1,19 @@
 from tools import all_tools
 from dotenv import load_dotenv
-from smolagents import InferenceClientModel, CodeAgent, PromptTemplates
+from smolagents import InferenceClientModel, CodeAgent, PromptTemplates, LiteLLMModel
 load_dotenv()
-
+'''
 model = InferenceClientModel(
     max_tokens=2096,
     temperature=0.0,
     model_id='Qwen/Qwen3-235B-A22B'
 )
+'''
+model = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash",
+    api_key=os.getenv("GEMINI_API_KEY"),
+    max_tokens=8192
+)
 
 # Loads system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f: