Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Chroma integration failing when there are less than 4 items in the collection (#3674)

Browse Source 
The code was failing to decrement the `n_results` kwarg passed to
`query(...)`
This commit is contained in:
uyhcire 2023-04-28 21:18:19 -07:00 committed by GitHub
parent 57e028549a
commit 0a7a2b99b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/vectorstores/chroma.py
View File

@ -111,7 +111,7 @@ class Chroma(VectorStore):
return self._collection.query( return self._collection.query(
query_texts=query_texts, query_texts=query_texts,
query_embeddings=query_embeddings, query_embeddings=query_embeddings,
n_results=n_results, n_results=i,
where=where, where=where,
) )
except chromadb.errors.NotEnoughElementsException: except chromadb.errors.NotEnoughElementsException: