Files
qdrant-client/qdrant_client/embed/models.py
George d2771afdfb Local inference image support (#836)
* new: add backbone for image support

* new: convert b64 to pil, embed images, add test

* tests: add test file

* refactor: replace 3 different inference object vars with a common one

* fix: fix type hints

* fix: fix type hints

* tests: add tests

* fix: remove redundant imports

* new: propagate image options

* Custom inference object (#837)

* new: add inference object support

* new: add inference object support

* fix: remove redundant import

* refactor: return newline

* fix: fix propagate options test
2024-11-04 08:37:53 +01:00

26 lines
562 B
Python

from typing import Union, List, Dict
from pydantic import StrictFloat, StrictStr
from qdrant_client.http.models import ExtendedPointId, SparseVector
NumericVector = Union[
List[StrictFloat],
SparseVector,
List[List[StrictFloat]],
]
NumericVectorInput = Union[
List[StrictFloat],
SparseVector,
List[List[StrictFloat]],
ExtendedPointId,
]
NumericVectorStruct = Union[
List[StrictFloat],
List[List[StrictFloat]],
Dict[StrictStr, NumericVector],
]
__all__ = ["NumericVector", "NumericVectorInput", "NumericVectorStruct"]