Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -135,7 +135,7 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
135 |
|
136 |
# Initialize SentenceTransformer and set max_seq_length
|
137 |
sentence_transformer = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
|
138 |
-
sentence_transformer.max_seq_length =
|
139 |
|
140 |
# Initialize embeddings with the model name (dim=768)
|
141 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
|
@@ -155,7 +155,7 @@ vector_store = SupabaseVectorStore(
|
|
155 |
)
|
156 |
|
157 |
# Create retriever tool
|
158 |
-
|
159 |
retriever=vector_store.as_retriever(),
|
160 |
name="Question Search",
|
161 |
description="A tool to retrieve similar questions from a vector store."
|
|
|
135 |
|
136 |
# Initialize SentenceTransformer and set max_seq_length
|
137 |
sentence_transformer = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
|
138 |
+
sentence_transformer.max_seq_length = 512 # Set max sequence length
|
139 |
|
140 |
# Initialize embeddings with the model name (dim=768)
|
141 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
|
|
|
155 |
)
|
156 |
|
157 |
# Create retriever tool
|
158 |
+
retriever_tool = create_retriever_tool(
|
159 |
retriever=vector_store.as_retriever(),
|
160 |
name="Question Search",
|
161 |
description="A tool to retrieve similar questions from a vector store."
|