Final_Assignment_Template

Sleeping

wishwakankanamg commited on May 24

Commit

af35baf

1 Parent(s): a8f5726

another change

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -178,7 +178,7 @@ def build_graph(provider: str = "huggingface"):
             # max_new_tokens=512,
             # model_kwargs={"top_k": 10}
         )
-        llm = ChatHuggingFace(llm=endpoint_llm)
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     # Bind tools to LLM

             # max_new_tokens=512,
             # model_kwargs={"top_k": 10}
         )
+        llm = ChatHuggingFace(llm=llm)
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     # Bind tools to LLM