github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix QdrantVectorDB to use custom embedding_function when provided, defaulting to FastEmbedEmbeddingFunction() otherwise (#3396)

Browse Source 
Co-authored-by: Li Jiang <bnujli@gmail.com>
This commit is contained in:
Eddy Fidel
2024-08-22 05:38:34 -04:00
committed by GitHub
parent 395af4e258
commit 8aea367558
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
autogen/agentchat/contrib/vectordb/qdrant.py
View File

@@ -93,7 +93,7 @@ class QdrantVectorDB(VectorDB):
kwargs: dict | Additional keyword arguments.
"""
self.client: QdrantClient = client or QdrantClient(location=":memory:")
self.embedding_function = FastEmbedEmbeddingFunction() or embedding_function
self.embedding_function = embedding_function or FastEmbedEmbeddingFunction()
self.collection_options = collection_options
self.content_payload_key = content_payload_key
self.metadata_payload_key = metadata_payload_key