Update agent.py

agent.py

@@ -6,12 +6,12 @@ class MagAgent:
         """Initialize the MagAgent with search tools."""
         print("Initializing MagAgent with search tools...")
         model = HfApiModel(
-            max_tokens=2096,
-            temperature=0.5,
-        #    model_id="gemini/gemini-2.0-flash-lite-001",
-            # api_key = os.environ["GEMINI_KEY"], #userdata.get("GEMINI_KEY")
-            model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/",
-            custom_role_conversions=None,
+#            max_tokens=2096,
+#            temperature=0.5,
+#        #    model_id="gemini/gemini-2.0-flash-lite-001",
+#            # api_key = os.environ["GEMINI_KEY"], #userdata.get("GEMINI_KEY")
+#            model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud/",
+#            custom_role_conversions=None,
         )
         self.agent = CodeAgent(
             model= model,  # OpenAIServerModel(model_id="gpt-4o"),