sagar008 commited on
Commit
7bf713e
·
verified ·
1 Parent(s): 37b6f9c

Update vector_store.py

Browse files
Files changed (1) hide show
  1. vector_store.py +4 -5
vector_store.py CHANGED
@@ -128,18 +128,17 @@ class LegalDocumentVectorStore:
128
  text_key="text"
129
  )
130
 
131
- # More permissive search settings
132
  search_kwargs = {
133
- 'k': 10, # Increased from default 5
134
- 'include_metadata': True
135
  }
136
 
137
  if document_id:
138
  search_kwargs['filter'] = {'document_id': document_id}
139
 
140
- # Use similarity search without threshold initially
141
  return vectorstore.as_retriever(
142
- search_type="similarity", # Remove threshold for now
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