commit

agent.py

@@ -150,6 +150,10 @@ tools = [
     arvix_search,
 ]
 
+hf_token = os.environ.get('HF_TOKEN')
+if not hf_token:
+    raise ValueError("Hugging Face API token (HF_TOKEN) not found in environment variables.")
+
 # Build graph function
 def build_graph(provider: str = "groq"):
     """Build the graph"""
@@ -162,10 +166,13 @@ def build_graph(provider: str = "groq"):
         llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
         # TODO: Add huggingface endpoint
+        repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
         llm = ChatHuggingFace(
+             
             llm=HuggingFaceEndpoint(
-                url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
+                repo_id=repo_id,
                 temperature=0,
+                huggingfacehub_api_token= hf_token
             ),
         )
     else: