Add missing types to weaviate.outputs.query

weaviate · Jun 20, 2024 · 5fc3a78 · 5fc3a78
1 parent 315f403
commit 5fc3a78
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/weaviate/outputs/query.py b/weaviate/outputs/query.py
@@ -10,13 +10,22 @@
 from weaviate.collections.classes.internal import (
     GenerativeNearMediaReturnType,
     GenerativeReturnType,
+    GenerativeGroupByReturnType,
+    GenerativeSearchReturnType,
     MetadataReturn,
     MetadataSingleObjectReturn,
     Object,
     ObjectSingleReturn,
     GroupByObject,
+    GroupByReturn,
+    GroupByReturnType,
     GenerativeObject,
+    GenerativeReturn,
+    GenerativeGroupByReturn,
+    GenerativeGroup,
+    Group,
     QueryNearMediaReturnType,
+    QueryReturn,
     QueryReturnType,
     QuerySingleReturn,
     ReferenceInput,
@@ -38,16 +47,25 @@
     "FilterReturn",
     "GenerativeNearMediaReturnType",
     "GenerativeReturnType",
+    "GenerativeGroupByReturnType",
+    "GenerativeSearchReturnType",
     "GeoCoordinate",
     "MetadataReturn",
     "MetadataSingleObjectReturn",
     "Object",
     "ObjectSingleReturn",
     "GroupByObject",
+    "GroupByReturn",
+    "Group",
+    "GroupByReturnType",
     "GenerativeObject",
+    "GenerativeReturn",
+    "GenerativeGroupByReturn",
+    "GenerativeGroup",
     "PhoneNumberType",
     "QueryNearMediaReturnType",
     "QueryReturnType",
+    "QueryReturn",
     "QuerySingleReturn",
     "ReferenceInput",
     "ReferenceInputs",