applied fix

app.py

@@ -17,10 +17,23 @@ class BasicAgent:
         print("Initializing the BasicAgent")
         # model = OpenAIServerModel(model_id="gpt-4o")
         # 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")
+        google_api_key = os.getenv('GOOGLE_API_KEY')
+        if google_api_key:
+            print(f"GOOGLE_API_KEY found. Last 10 characters: ...{google_api_key[-10:]}")
+        else:
+            print("GOOGLE_API_KEY environment variable not set.")
+            
+        try:
+            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")
+        except Exception as e:
+            print(f"Error instantiating LLM: {e}")
+            print("Please ensure you have installed smolagents[litellm] and google-generativeai,")
+            print("and that your GOOGLE_API_KEY environment variable is set.")
+            raise e # Re-raise the exception to fail initialization if model fails
+
         # Initialize the search tool
         search_tool = DuckDuckGoSearchTool()
         print("DuckDuckSearchTool instantiated")