Update vector_store.py
Browse files- vector_store.py +4 -5
vector_store.py
CHANGED
@@ -128,18 +128,17 @@ class LegalDocumentVectorStore:
|
|
128 |
text_key="text"
|
129 |
)
|
130 |
|
131 |
-
#
|
132 |
search_kwargs = {
|
133 |
-
'k': 10
|
134 |
-
'include_metadata': True
|
135 |
}
|
136 |
|
137 |
if document_id:
|
138 |
search_kwargs['filter'] = {'document_id': document_id}
|
139 |
|
140 |
-
|
141 |
return vectorstore.as_retriever(
|
142 |
-
search_type="similarity",
|
143 |
search_kwargs=search_kwargs
|
144 |
)
|
145 |
|
|
|
128 |
text_key="text"
|
129 |
)
|
130 |
|
131 |
+
# ⭐ FIX 2: Remove include_metadata (not supported by PineconeVectorStore)
|
132 |
search_kwargs = {
|
133 |
+
'k': 10 # Increased from default 5
|
|
|
134 |
}
|
135 |
|
136 |
if document_id:
|
137 |
search_kwargs['filter'] = {'document_id': document_id}
|
138 |
|
139 |
+
# Use similarity search without threshold initially
|
140 |
return vectorstore.as_retriever(
|
141 |
+
search_type="similarity",
|
142 |
search_kwargs=search_kwargs
|
143 |
)
|
144 |
|