Update agent.py

agent.py

@@ -1,5 +1,6 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, LiteLLMModel # HfApiModel, OpenAIServerModel
 import asyncio
+import os
 
 class MagAgent:
     def __init__(self):
@@ -7,7 +8,7 @@ class MagAgent:
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
             model_id="gemini/gemini-2.0-flash",
-            api_key=os.getenv("GEMINI_KEY"),
+            api_key= os.environ.get("GEMINI_KEY"),
             max_tokens=8192
 #        model = HfApiModel(
 #            max_tokens=2096,