modified hugginfaceendpoint

app.py

@@ -15,10 +15,10 @@ from typing import TypedDict, Annotated
 from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph, END, Graph
 # from langgraph.prebuilt import tools_condition
-# from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain.agents import initialize_agent, AgentType
-from langchain_community.llms import HuggingFaceHub
-from langchain_community.chat_models import ChatHuggingFace
+# from langchain_community.llms import HuggingFaceHub
+# from langchain_community.chat_models import ChatHuggingFace
 
 ## # Load environment variables from .env file
 # --- Constants ---
@@ -241,11 +241,13 @@ Instructions:
 
 ## --- Initialize Hugging Face Model ---
 # Generate the chat interface, including the tools
-llm = HuggingFaceHub(
+llm = HuggingFaceEndpoint(
     repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     huggingfacehub_api_token=HF_ACCESS_KEY,
-    # model_kwargs={'prompt': system_prompt}
+    model_kwargs={'prompt': system_prompt}
+    # system_prompt=system_prompt,
 )
+
 chat_llm = ChatHuggingFace(llm=llm)
 # chat = ChatHuggingFace(llm=llm, verbose=True)
 # tools = [search_tool, fetch_weather]