Update agent.py

agent.py

@@ -161,11 +161,17 @@ 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
+        # llm = ChatHuggingFace(
+        #     llm=HuggingFaceEndpoint(
+        #         repo_id="Meta-DeepLearning/llama-2-7b-chat-hf",
+        #         temperature=0
+        #     )
+        # )
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(
-                repo_id="Meta-DeepLearning/llama-2-7b-chat-hf",
-                temperature=0
-            )
+                url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
+                temperature=0,
+            ),
         )
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")