agent tools

app.py

@@ -7,6 +7,7 @@ from langchain.tools import Tool
 import os
 import datasets
 from langchain.docstore.document import Document
+from langgraph.graph import START, StateGraph
 from langchain_community.retrievers import BM25Retriever
 from retriever import extract_text
 
@@ -72,4 +73,16 @@ tools = [guest_info_tool]
 chat_with_tools = model.bind_tools(tools)
 
 
+# Generate the AgentState and Agent graph
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+
+def assistant(state: AgentState):
+    return {
+        "messages": [chat_with_tools.invoke(state["messages"])],
+    }
+
+## The graph
+builder = StateGraph(AgentState)
+
 demo.launch()