Update agent.py

agent.py

@@ -690,8 +690,9 @@ def build_graph(provider: str = "huggingface"):
         """
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(
-                repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
-                task="text-generation",  # for chat‐style use “text-generation”
+                #repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
+                endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
+                #task="text-generation",  # for chat‐style use “text-generation”
                 #max_new_tokens=1024,
                 #do_sample=False,
                 #repetition_penalty=1.03,