Update agent.py

agent.py

@@ -18,6 +18,7 @@ class Agent:
     def __init__(self):
         # self.hf_token = os.getenv("HF_API_KEY")
         self.openai_key = os.getenv("OPEN_AI_KEY")
+        # self.gemini_key = os.getenv("GEMINI_API_KEY")
         
         # self.model = HfApiModel(
         #    model_id="Qwen/Qwen3-235B-A22B",
@@ -30,9 +31,8 @@ class Agent:
         )
         
         # self.model_local = LiteLLMModel(
-        #    model_id="ollama/qwen3:8b",  
-        #    api_base="http://localhost:11434", 
-        #    num_ctx=40000, # Context set for Mistral Nemo
+        #    model_id="gemini/gemini-2.0-flash-exp",  
+        #    api_key=self.gemini_key 
         # )