From d62e84c4f5853ed75c12f9a28607d82799d2092f Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Mon, 1 Apr 2024 10:10:44 -0700 Subject: [PATCH] =?UTF-8?q?community[patch]:=20Revert=20"=20Fix=20the=20bu?= =?UTF-8?q?g=20that=20Chroma=20does=20not=20specify=20`e=E2=80=A6=20(#1986?= =?UTF-8?q?6)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit …mbedding_function` (#19277)" This reverts commit 7042934b5f346e1dabff40ad184d1d808032c252. Fixes #19848 --- libs/community/langchain_community/vectorstores/chroma.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/libs/community/langchain_community/vectorstores/chroma.py b/libs/community/langchain_community/vectorstores/chroma.py index 95d3fcc0c5..7723285faf 100644 --- a/libs/community/langchain_community/vectorstores/chroma.py +++ b/libs/community/langchain_community/vectorstores/chroma.py @@ -28,7 +28,6 @@ if TYPE_CHECKING: import chromadb.config from chromadb.api.types import ID, OneOrMany, Where, WhereDocument - logger = logging.getLogger() DEFAULT_K = 4 # Number of Documents to return. @@ -81,7 +80,6 @@ class Chroma(VectorStore): try: import chromadb import chromadb.config - from chromadb.utils import embedding_functions except ImportError: raise ImportError( "Could not import chromadb python package. " @@ -124,12 +122,10 @@ class Chroma(VectorStore): _client_settings.persist_directory or persist_directory ) - self._embedding_function = ( - embedding_function or embedding_functions.DefaultEmbeddingFunction() - ) + self._embedding_function = embedding_function self._collection = self._client.get_or_create_collection( name=collection_name, - embedding_function=self._embedding_function, + embedding_function=None, metadata=collection_metadata, ) self.override_relevance_score_fn = relevance_score_fn