wt002 commited on
Commit
7d15737
·
verified ·
1 Parent(s): 4edcb82

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -133,7 +133,7 @@ def create_retriever_tool(persist_directory="vector_store"):
133
  )
134
 
135
  # Create retriever tool
136
- create_retriever_tool = create_retriever_tool(
137
  retriever=vector_store.as_retriever(),
138
  name="Question Search",
139
  description="A tool to retrieve similar questions from a vector store.",
@@ -141,7 +141,7 @@ def create_retriever_tool(persist_directory="vector_store"):
141
 
142
  return retriever_tool, vector_store
143
 
144
-
145
 
146
  tools = [
147
  multiply,
 
133
  )
134
 
135
  # Create retriever tool
136
+ retriever_tool = create_retriever_tool(
137
  retriever=vector_store.as_retriever(),
138
  name="Question Search",
139
  description="A tool to retrieve similar questions from a vector store.",
 
141
 
142
  return retriever_tool, vector_store
143
 
144
+ tool, store = create_retriever_tool(persist_directory="/home/wendy/Downloads")
145
 
146
  tools = [
147
  multiply,