Update agent.py

agent.py

@@ -16,18 +16,19 @@ dotenv.load_dotenv()
 
 class Agent:
     def __init__(self):
-        self.hf_token = os.getenv("HF_API_KEY")
+        # self.hf_token = os.getenv("HF_API_KEY")
+        self.openai_key = os.getenv("OPEN_AI_KEY")
         
-        self.model = HfApiModel(
-            model_id="Qwen/Qwen3-235B-A22B",
-            token=self.hf_token       
-        )
-        
-        # self.model = OpenAIServerModel(
-        #    model_id="gpt-4o-mini-2024-07-18",
-        #    api_key=OPENAI_KEY
+        # self.model = HfApiModel(
+        #    model_id="Qwen/Qwen3-235B-A22B",
+        #    token=self.hf_token       
         # )
         
+        self.model = OpenAIServerModel(
+            model_id="gpt-4o-mini-2024-07-18",
+            api_key=self.openai_key
+        )
+        
         # self.model_local = LiteLLMModel(
         #    model_id="ollama/qwen3:8b",  
         #    api_base="http://localhost:11434",