Applied gemini

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIServerModel, LiteLLMModel
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -16,8 +16,11 @@ class BasicAgent:
     def __init__(self):
         print("Initializing the BasicAgent")
         # model = OpenAIServerModel(model_id="gpt-4o")
-        model = OpenAIServerModel(model_id="gpt-3.5-turbo")
-        print("GPT-4o LLM is instantiated")
+        # model = OpenAIServerModel(model_id="gpt-3.5-turbo")
+        gemini_model_id = "gemini/gemini-2.0-flash"
+        model = LiteLLMModel(model_id=gemini_model_id)
+        # print("GPT-4o LLM is instantiated")
+        print(f"{gemini_model_id} LLM instantiated using LiteLLM")
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
         print("DuckDuckSearchTool instantiated")