Skip to content

Commit 2000d47

Browse files
authored
Fix chromadb get_collection ignores custom embedding_function (#2854)
1 parent 1e5c03f commit 2000d47

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

autogen/agentchat/contrib/vectordb/chromadb.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ def create_collection(
8383
if self.active_collection and self.active_collection.name == collection_name:
8484
collection = self.active_collection
8585
else:
86-
collection = self.client.get_collection(collection_name)
86+
collection = self.client.get_collection(collection_name, embedding_function=self.embedding_function)
8787
except ValueError:
8888
collection = None
8989
if collection is None:
@@ -126,7 +126,9 @@ def get_collection(self, collection_name: str = None) -> Collection:
126126
)
127127
else:
128128
if not (self.active_collection and self.active_collection.name == collection_name):
129-
self.active_collection = self.client.get_collection(collection_name)
129+
self.active_collection = self.client.get_collection(
130+
collection_name, embedding_function=self.embedding_function
131+
)
130132
return self.active_collection
131133

132134
def delete_collection(self, collection_name: str) -> None:

0 commit comments

Comments
 (0)