Update agent.py

agent.py

@@ -1,12 +1,12 @@
 import os
 import dotenv
 import yaml
-from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel
+from smolagents import CodeAgent, LiteLLMModel, OpenAIServerModel, HfApiModel
 from tools import tool_read_files, tool_reverse_string, tool_download_image
 
 dotenv.load_dotenv()
 
-HF_TOKEN = os.getenv("HF_TOKEN")
+HF_API_KEY = os.getenv("HF_API_KEY")
 OPENAI_KEY = os.getenv("OPEN_AI_KEY")
 
 
@@ -14,21 +14,21 @@ class Agent:
     def __init__(self):
         self.hf_token = os.getenv("HF_TOKEN")
         
-        # self.model = InferenceClientModel(
-        #     model_id="Qwen/Qwen3-235B-A22B",
-        #     token=HF_TOKEN       
-        # )
-        
-        self.model = OpenAIServerModel(
-            model_id="gpt-4o-mini-2024-07-18",
-            api_key=OPENAI_KEY
+        self.model = InferenceClientModel(
+            model_id="Qwen/Qwen3-235B-A22B",
+            token=HF_TOKEN       
         )
         
-        self.model_local = LiteLLMModel(
-            model_id="ollama/qwen3:8b",  
-            api_base="http://localhost:11434", 
-            num_ctx=40000, # Context set for Mistral Nemo
-        )
+        #self.model = OpenAIServerModel(
+        #    model_id="gpt-4o-mini-2024-07-18",
+        #    api_key=OPENAI_KEY
+        # )
+        
+        # self.model_local = LiteLLMModel(
+        #    model_id="ollama/qwen3:8b",  
+        #    api_base="http://localhost:11434", 
+        #    num_ctx=40000, # Context set for Mistral Nemo
+        # )