upstream-archive/fastembed
1
0
Fork 0
mirror of https://github.com/qdrant/fastembed.git synced 2026-08-07 02:20:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

* chore(qdrant_mixin.py): refactor QdrantClientMixin to use DefaultEmbedding instead of SentenceTransformersEmbedding

Browse Source
This commit is contained in:
NirantK
2023-07-31 16:28:32 +05:30
parent 81956c371b
commit 538c63d2ee
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
fastembed/qdrant_mixin.py
View File

@@ -4,7 +4,7 @@ from typing import Any, Dict, Generator, List, Optional
from pydantic import BaseModel
from qdrant_client.models import Distance, PointStruct, SearchParams, VectorParams
from .embedding import Embedding, SentenceTransformersEmbedding
from .embedding import DefaultEmbedding, Embedding
class QueryResponse(BaseModel):
@@ -47,7 +47,7 @@ class QdrantClientMixin:
docs: Dict[str, List[Any]],
batch_size: int = 512,
wait: bool = True,
embedding_model: Embedding = SentenceTransformersEmbedding(),
embedding_model: Embedding = DefaultEmbedding(),
**kwargs,
) -> None:
"""
@@ -56,7 +56,7 @@ class QdrantClientMixin:
docs (Dict[str, List[Any]]): _description_
batch_size (int, optional): _description_. Defaults to 512.
wait (bool, optional): _description_. Defaults to True.
embedding_model (Embedding, optional): Defaults to SentenceTransformersEmbedding.
embedding_model (Embedding, optional): Defaults to DefaultEmbedding() with all-MiniLM-L6-v2.
"""
# Iterate over documents and metadatas in batches
@@ -90,7 +90,7 @@ class QdrantClientMixin:
batch_size: int = 512,
query_filter: Optional[Dict[str, Any]] = None,
search_params: SearchParams = SearchParams(hnsw_ef=128, exact=False),
embedding_model: Embedding = SentenceTransformersEmbedding(),
embedding_model: Embedding = DefaultEmbedding(),
**kwargs,
) -> List[QueryResponse]:
"""