Stefan888 commited on
Commit
48af48e
·
1 Parent(s): 23d14cd

fixed some bugs

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. retriever.py +1 -1
app.py CHANGED
@@ -6,6 +6,8 @@ from langgraph.prebuilt import ToolNode
6
  from langgraph.graph import START, StateGraph
7
  from langgraph.prebuilt import tools_condition
8
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 
 
9
 
10
  # Generate the chat interface, including the tools
11
  llm = HuggingFaceEndpoint(
 
6
  from langgraph.graph import START, StateGraph
7
  from langgraph.prebuilt import tools_condition
8
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
9
+ from langchain.tools import Tool
10
+ from retriever import load_guest_dataset
11
 
12
  # Generate the chat interface, including the tools
13
  llm = HuggingFaceEndpoint(
retriever.py CHANGED
@@ -1,5 +1,5 @@
1
  from langchain_community.retrievers import BM25Retriever
2
- from langchain.tools import Tool
3
 
4
  import datasets
5
  from langchain.docstore.document import Document
 
1
  from langchain_community.retrievers import BM25Retriever
2
+
3
 
4
  import datasets
5
  from langchain.docstore.document import Document