Also add objects

weaviate · Feb 20, 2024 · 19222bc · 19222bc
1 parent 7697dc5
commit 19222bc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/weaviate/collections/classes/internal.py b/weaviate/collections/classes/internal.py
@@ -598,3 +598,12 @@ def from_input(
     List[Object[TProperties, References]],
     List[Object[TProperties, CrossReferences]],
 ]
+
+QueryObjectReturnType = Union[
+    Object[Properties, References],
+    Object[TProperties, TReferences],
+    Object[Properties, CrossReferences],
+    Object[Properties, TReferences],
+    Object[TProperties, References],
+    Object[TProperties, CrossReferences],
+]
diff --git a/weaviate/outputs/query.py b/weaviate/outputs/query.py
@@ -18,6 +18,7 @@
     ReferenceInput,
     ReferenceInputs,
     QueryObjectsReturnType,
+    QueryObjectReturnType,
 )
 from weaviate.collections.classes.types import (
     GeoCoordinate,
@@ -40,6 +41,7 @@
     "MetadataSingleObjectReturn",
     "PhoneNumberType",
     "QueryNearMediaReturnType",
+    "QueryObjectReturnType",
     "QueryObjectsReturnType",
     "QueryReturnType",
     "QuerySingleReturn",