Allowing outside parameters for Qdrant. (#7910)

@baskaryan @rlancemartin, @eyurtsev
This commit is contained in:
Hanit 2023-07-20 16:58:54 +03:00 committed by GitHub
parent d6493590da
commit ea149dbd89
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -149,8 +149,7 @@ class Qdrant(VectorStore):
texts, metadatas, ids, batch_size texts, metadatas, ids, batch_size
): ):
self.client.upsert( self.client.upsert(
collection_name=self.collection_name, collection_name=self.collection_name, points=points, **kwargs
points=points,
) )
added_ids.extend(batch_ids) added_ids.extend(batch_ids)
@ -266,7 +265,7 @@ class Qdrant(VectorStore):
Returns: Returns:
List of Documents most similar to the query. List of Documents most similar to the query.
""" """
results = await self.asimilarity_search_with_score(query, k, filter) results = await self.asimilarity_search_with_score(query, k, filter, **kwargs)
return list(map(itemgetter(0), results)) return list(map(itemgetter(0), results))
def similarity_search_with_score( def similarity_search_with_score(