m

agent.py

@@ -170,15 +170,14 @@ def build_graph(provider: str = "huggingface"):
         if not hf_token:
             raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
         llm = HuggingFacePipeline.from_model_id(
-            model_id="HuggingFaceH4/zephyr-7b-beta",
+            repo_id="HuggingFaceH4/zephyr-7b-beta",
             task="text-generation",
-            pipeline_kwargs=dict(
-                max_new_tokens=512,
-                do_sample=False,
-                repetition_penalty=1.03,
-            ),
+            max_new_tokens=1000,
+            do_sample=False,
+            repetition_penalty=1.03,
+           
         )
-        llm = ChatHuggingFace(llm=llm)
+        llm = ChatHuggingFace(llm=llm, recursion_limit=50)
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     # Bind tools to LLM