Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -133,7 +133,7 @@ def create_retriever_tool(persist_directory="vector_store"):
|
|
133 |
)
|
134 |
|
135 |
# Create retriever tool
|
136 |
-
|
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,
|