Update agent.py

agent.py

@@ -6,17 +6,21 @@ from tools import ToolReadFiles, ToolReverseString, ToolDownloadImage, FinalAnsw
 
 dotenv.load_dotenv()
 
-HF_API_KEY = os.getenv("HF_API_KEY")
-OPENAI_KEY = os.getenv("OPEN_AI_KEY")
+# HF_API_KEY = os.getenv("HF_API_KEY")
+# OPENAI_KEY = os.getenv("OPEN_AI_KEY")
+
+# Create prompt template
+# with open("prompts.yaml", 'r') as stream:
+#    prompt_templates = yaml.safe_load(stream)
 
 
 class Agent:
     def __init__(self):
-        # self.hf_token = os.getenv("HF_API_KEY")
+        self.hf_token = os.getenv("HF_API_KEY")
         
         self.model = HfApiModel(
             model_id="Qwen/Qwen3-235B-A22B",
-            token=HF_API_KEY       
+            token=self.hf_token       
         )
         
         # self.model = OpenAIServerModel(