debug

app.py

@@ -13,7 +13,6 @@ 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
 
 HUGGINGFACEHUB_API_TOKEN = os.environ["HUGGINGFACEHUB_API_TOKEN"]
 login(token=HUGGINGFACEHUB_API_TOKEN)
@@ -38,6 +37,13 @@ docs = [
 
 bm25_retriever = BM25Retriever.from_documents(docs)
 
+def extract_text(query: str) -> str:
+    """Retrieves detailed information about gala guests based on their name or relation."""
+    results = bm25_retriever.invoke(query)
+    if results:
+        return "\n\n".join([doc.page_content for doc in results[:3]])
+    else:
+        return "No matching guest information found."
 
 llm = HuggingFaceEndpoint(
     repo_id="HuggingFaceH4/zephyr-7b-beta",
@@ -76,7 +82,6 @@ guest_info_tool = Tool(
 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]

retriever.py

@@ -1,11 +0,0 @@
-from langchain_community.retrievers import BM25Retriever
-from langchain.tools import Tool
-
-def extract_text(query: str) -> str:
-    """Retrieves detailed information about gala guests based on their name or relation."""
-    results = bm25_retriever.invoke(query)
-    if results:
-        return "\n\n".join([doc.page_content for doc in results[:3]])
-    else:
-        return "No matching guest information found."
-