Files
qdrant-client/qdrant_client/fastembed_common.py
Andrey Vasnetsov 822aefa1c1 WIP: hybrid search with fastembed (#553)
* WIP: hybrid search with fastembed

* hybrid queries with fastembed

* test for hybrid

* fix typo

* new: extend hybrid search tests, fix mypy, small refactoring (#554)

* refactor: align model name parameters in setters, update tests

* fix: fix async

* fix: add a good test, fix sparse vectors in query batch

* refactoring: reduce branching, refactor fastembed tests

---------

Co-authored-by: George <george.panchuk@qdrant.tech>
2024-03-27 13:57:58 +01:00

15 lines
414 B
Python

from typing import Any, Dict, List, Optional, Union
from pydantic import BaseModel, Field
from qdrant_client.conversions.common_types import SparseVector
class QueryResponse(BaseModel, extra="forbid"): # type: ignore
id: Union[str, int]
embedding: Optional[List[float]]
sparse_embedding: Optional[SparseVector] = Field(default=None)
metadata: Dict[str, Any]
document: str
score: float