Update agent.py

agent.py

@@ -1,11 +1,9 @@
 import os
-import json
-
 from tools import TOOLS
-from metadata import load_metadata  
+from metadata import load_metadata
 from mistral_hf_wrapper import MistralInference
 
-# Load environment variables
+# Load Mistral endpoint and token from environment
 API_URL = os.getenv("HF_MISTRAL_URL")
 API_TOKEN = os.getenv("HF_TOKEN")
 
@@ -13,8 +11,8 @@ API_TOKEN = os.getenv("HF_TOKEN")
 def load_tasks():
     return load_metadata("metadata.jsonl")
 
-# Initialize Mistral
-mistral = MistralInference()
+# Initialize MistralInference wrapper
+mistral = MistralInference(api_url=API_URL, api_token=API_TOKEN)
 
 # Solve a single task
 def solve_task(task, tools=TOOLS):