Update agent.py

agent.py

@@ -16,7 +16,7 @@ from langchain.chat_models import ChatHuggingFace
 
 from langchain_google_genai import ChatGoogleGenerativeAI
 
-from langchain.llms import HuggingFaceEndpoint
+from langchain.llms import HuggingFaceHub
 #load_dotenv()
 google_api_key = os.environ["GOOGLE_API_KEY"]
 hf_api_key = os.environ["HF_TOKEN"]
@@ -108,13 +108,10 @@ def build_graph(provider: str = "huggingface"):
         # Google Gemini
         llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0,api_key=google_api_key)
     elif provider == "huggingface":
-        # TODO: Add huggingface endpoint
-        llm = ChatHuggingFace(
-            llm=HuggingFaceEndpoint(
-                endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
-                model_kwargs={"temperature": 0},
-                huggingfacehub_api_token=hf_api_key
-            ),
+        llm = HuggingFaceHub(
+            repo_id="meta-llama/Llama-2-7b-chat-hf",  
+            huggingfacehub_api_token=hf_api_key,        
+            model_kwargs={"temperature": 0}
         )
     def assistant(state: MessagesState):
         """ Use the tools to answer the query. you have add,subtract,multiply,divide,web_search,wikipedia_search,arxiv_search tools."""