asdf

agent.py

@@ -6,7 +6,7 @@ from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_groq import ChatGroq
-from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
+from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings,  HuggingFacePipeline
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.document_loaders import ArxivLoader
@@ -169,13 +169,14 @@ def build_graph(provider: str = "huggingface"):
         repo_id = "togethercomputer/evo-1-131k-base"
         if not hf_token:
             raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
-        llm = HuggingFaceEndpoint(
+        llm = HuggingFacePipeline.from_model_id(
             repo_id="HuggingFaceH4/zephyr-7b-beta",
             task="text-generation",
-            max_new_tokens=512,
-            do_sample=False,
-            repetition_penalty=1.03,
-            recursion_limit=50
+            pipeline_kwargs=dict(
+                max_new_tokens=512,
+                do_sample=False,
+                repetition_penalty=1.03,
+            ),
         )
         llm = ChatHuggingFace(llm=llm)
     else: