Update agent.py

agent.py

@@ -56,12 +56,12 @@ tools = [
 ]
 
 # LLM
-llm = HfApiModel(
-            max_tokens=2096,
-            temperature=0.5,
-            model_id='Qwen/Qwen2-7B-Instruct',  # Smaller model as last resort
-            custom_role_conversions=None,
-        )
+llm = LiteLLMModel(
+    model_id="gemini/gemini-2.0-flash",
+    api_key=os.getenv("GEMINI_KEY"),
+    max_tokens=8192,
+)
+
 
 # Agent
 class BasicAgent: