Update agent.py

agent.py

@@ -16,8 +16,9 @@ from smolagents import InferenceClientModel, LiteLLMModel, CodeAgent, ToolCallin
 
 # Load environment and configure Gemini
 load_dotenv()
-configure(api_key=os.getenv("GOOGLE_API_KEY"))
-
+# configure(api_key=os.getenv("GOOGLE_API_KEY"))
+GOOGLE_API_KEY = userdata.get("GOOGLE_API_KEY")
+configure(api_key=GOOGLE_API_KEY)
 # Logging
 #logging.basicConfig(level=logging.INFO, format="%(asctime)s | %(levelname)s | %(message)s")
 #logger = logging.getLogger(__name__)
@@ -267,7 +268,7 @@ class BasicAgent:
         #     return InferenceClientModel()
         # else:
         #     return LiteLLMModel(model_id=GEMINI_MODEL_NAME, api_key=os.getenv("GOOGLE_API_KEY"))
-        return LiteLLMModel(model_id=GEMINI_MODEL_NAME, api_key=os.getenv("GOOGLE_API_KEY"))
+        return InferenceClientModel(model_id=HF_MODEL_NAME)
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
@@ -337,4 +338,4 @@ if __name__ == "__main__":
     if q == "dev":
         agent.evaluate_random_questions()
     else:
-        print(agent(q))
+        print(agent(q))