github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-21 12:01:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/chroma fix (#6390)

Browse Source 
Co-authored-by: Junu Moon(Fran) <francomoon7@gmail.com>
This commit is contained in:
Harrison Chase 2023-06-18 18:33:26 -07:00 committed by GitHub
parent 5c1fa3e70e
commit 10bff4ecc4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
langchain/vectorstores/chroma.py
View File

@ -109,30 +109,19 @@ class Chroma(VectorStore):
) -> List[Document]: ) -> List[Document]:
"""Query the chroma collection.""" """Query the chroma collection."""
try: try:
import chromadb import chromadb # noqa: F401
except ImportError: except ImportError:
raise ValueError( raise ValueError(
"Could not import chromadb python package. " "Could not import chromadb python package. "
"Please install it with `pip install chromadb`." "Please install it with `pip install chromadb`."
) )
for i in range(n_results, 0, -1):
try:
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=i, n_results=n_results,
where=where, where=where,
**kwargs, **kwargs,
) )
except chromadb.errors.NotEnoughElementsException:
logger.error(
f"Chroma collection {self._collection.name} "
f"contains fewer than {i} elements."
)
raise chromadb.errors.NotEnoughElementsException(
f"No documents found for Chroma collection {self._collection.name}"
)
def add_texts( def add_texts(
self, self,

2
pyproject.toml
View File

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "langchain" name = "langchain"
version = "0.0.203" version = "0.0.204"
description = "Building applications with LLMs through composability" description = "Building applications with LLMs through composability"
authors = [] authors = []
license = "MIT" license = "MIT"