new: extend signatures with images and inference objects (#842)

* new: extend signatures with images and inference objects * fix: remove draft code
qdrant · Nov 11, 2024 · 6d29cc6 · 6d29cc6
1 parent f9fbc90
commit 6d29cc6
Show file tree

Hide file tree

Showing 10 changed files with 32 additions and 0 deletions.
diff --git a/qdrant_client/async_client_base.py b/qdrant_client/async_client_base.py
@@ -105,6 +105,8 @@ async def query_points(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,
@@ -133,6 +135,8 @@ async def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/async_qdrant_client.py b/qdrant_client/async_qdrant_client.py
@@ -424,6 +424,8 @@ async def query_points(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,
@@ -564,6 +566,8 @@ async def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/async_qdrant_remote.py b/qdrant_client/async_qdrant_remote.py
@@ -436,6 +436,8 @@ async def query_points(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,
@@ -603,6 +605,8 @@ async def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/client_base.py b/qdrant_client/client_base.py
@@ -101,6 +101,8 @@ def query_points(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,
@@ -129,6 +131,8 @@ def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/conversions/common_types.py b/qdrant_client/conversions/common_types.py
@@ -115,6 +115,8 @@ def get_args_subscribed(tp):  # type: ignore
 Query: TypeAlias = rest.Query
 Prefetch: TypeAlias = rest.Prefetch
 Document: TypeAlias = rest.Document
+Image: TypeAlias = rest.Image
+InferenceObject: TypeAlias = rest.InferenceObject
 
 SearchRequest = Union[rest.SearchRequest, grpc.SearchPoints]
 RecommendRequest = Union[rest.RecommendRequest, grpc.RecommendPoints]

diff --git a/qdrant_client/local/async_qdrant_local.py b/qdrant_client/local/async_qdrant_local.py
@@ -470,6 +470,8 @@ async def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/local/local_collection.py b/qdrant_client/local/local_collection.py
@@ -926,6 +926,8 @@ def query_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/local/qdrant_local.py b/qdrant_client/local/qdrant_local.py
@@ -504,6 +504,8 @@ def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/qdrant_client.py b/qdrant_client/qdrant_client.py
@@ -473,6 +473,8 @@ def query_points(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,
@@ -618,6 +620,8 @@ def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,

diff --git a/qdrant_client/qdrant_remote.py b/qdrant_client/qdrant_remote.py
@@ -591,6 +591,8 @@ def query_points(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,
@@ -784,6 +786,8 @@ def query_points_groups(
             types.Query,
             types.NumpyArray,
             types.Document,
+            types.Image,
+            types.InferenceObject,
             None,
         ] = None,
         using: Optional[str] = None,