migrated to langchain from graph

app.py

@@ -5,7 +5,7 @@ import requests
 import inspect
 import pandas as pd
 import datetime
-# from dotenv import load_dotenv
+from dotenv import load_dotenv
 
 from langchain.tools import tool
 # from langchain_community.tools import get_all_tools
@@ -13,9 +13,12 @@ from typing import TypedDict, Annotated
 # from langgraph.graph.message import add_messages
 # from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
 from langgraph.prebuilt import ToolNode
-from langgraph.graph import START, StateGraph, END
+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
 
 ## # Load environment variables from .env file
 # --- Constants ---
@@ -105,12 +108,12 @@ def convert_units(args: str) -> str:
 
 # --- TOOL 5: Date & Time Tool ---
 @tool
-def get_time(_: str = "") -> str:
+def get_time(input: str) -> str:
     """Get current UTC time as HH:MM."""
     return datetime.datetime.utc().strftime("%H:%M")
 
 @tool
-def get_date(_: str = "") -> str:
+def get_date(input: str) -> str:
     """Get current date as YYYY-MM-DD."""
     return datetime.datetime.utc().strftime("%Y-%m-%d")
 
@@ -238,53 +241,22 @@ Instructions:
 
 ## --- Initialize Hugging Face Model ---
 # Generate the chat interface, including the tools
-llm = HuggingFaceEndpoint(
+llm = HuggingFaceHub(
     repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     huggingfacehub_api_token=HF_ACCESS_KEY,
-    model_kwargs={'prompt': system_prompt}
-    # system_prompt=system_prompt,
+    # model_kwargs={'prompt': system_prompt}
 )
+chat_llm = ChatHuggingFace(llm=llm)
 # chat = ChatHuggingFace(llm=llm, verbose=True)
 # tools = [search_tool, fetch_weather]
 # chat_with_tools = chat.bind_tools(tools)
 
-
-##
-# --- LANGGRAPH AGENT SETUP ---
-
-# Define the state for the graph
-class AgentState(dict):
-    pass
-
-# Define the main node (agent logic)
-def agent_node(state: AgentState) -> AgentState:
-    question = state["question"]
-    # The LLM will decide which tool to use based on the prompt and tools
-    # response = chat_with_tools.invoke(question) # use this if using ChatHuggingFace with binding option to tools
-    response = llm.invoke(question, tools=tools_list)
-    return AgentState({"question": question, "answer": response})
-
-# Build the graph
-graph = StateGraph(AgentState)
-graph.add_node("agent", agent_node)
-# graph.add_node("tools", ToolNode(tools)) #use this when using ChatHuggingFace with binding option to tools
-# graph.add_edge(START, "agent") #alternatively use the below with set_entry_point
-graph.set_entry_point("agent")
-graph.add_edge("agent", END)
-my_agent = graph.compile()
-
-# Or try simply with Graph instead of StateGraph
-# from langgraph.graph import Graph
-# graph = Graph(llm=llm, tools=tools_list)
-# def agent(question: str) -> str:
-#     return graph.run(question)
-
-## --- AGENT CALL FUNCTION ---
-def agent(question: str) -> str:
-    state = AgentState({"question": question})
-    result = my_agent.invoke(state)
-    return result["answer"]
-
+agent = initialize_agent(
+    tools=tools_list,
+    llm=chat_llm,
+    agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+    verbose=True,
+)
 
 
 ## --
@@ -351,7 +323,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            full_prompt = f"{system_prompt}\n Input Question: {question_text}"
+            submitted_answer = agent(full_prompt)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:

requirements.txt

@@ -4,4 +4,5 @@ langchain
 langgraph
 langchainhub
 huggingface-hub
-langchain-huggingface
+langchain-huggingface
+langchain-community