Skip to content

Commit

Permalink
Fix QdrantVectorDB to use custom embedding_function when provided, de…
Browse files Browse the repository at this point in the history
…faulting to FastEmbedEmbeddingFunction() otherwise (#3396)

Co-authored-by: Li Jiang <[email protected]>
  • Loading branch information
eddyfidel and thinkall authored Aug 22, 2024
1 parent 395af4e commit 8aea367
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion autogen/agentchat/contrib/vectordb/qdrant.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def __init__(
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
Expand Down

0 comments on commit 8aea367

Please sign in to comment.