Enforce kwargs for export

Author: dirkkul

weaviate/collections/collections/executor.py

@@ -44,13 +44,13 @@
     _check_references_generic,
 )
 from weaviate.collections.collection import Collection, CollectionAsync
+from weaviate.collections.config.executor import _any_property_has_text_analyzer
 from weaviate.connect import executor
 from weaviate.connect.v4 import (
     ConnectionAsync,
     ConnectionType,
     _ExpectedStatusCodes,
 )
-from weaviate.collections.config.executor import _any_property_has_text_analyzer
 from weaviate.exceptions import WeaviateInvalidInputError, WeaviateUnsupportedFeatureError
 from weaviate.util import _capitalize_first_letter, _decode_json_response_dict
 from weaviate.validator import _validate_input, _ValidateArgument

weaviate/collections/config/executor.py

@@ -5,8 +5,8 @@
     Generic,
     List,
     Literal,
-    Sequence,
     Optional,
+    Sequence,
     Tuple,
     Union,
     cast,

weaviate/export/async_.pyi

@@ -14,23 +14,24 @@ class _ExportAsync(_ExportExecutor[ConnectionAsync]):
     @overload
     async def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
         include_collections: Union[List[str], str, None] = None,
         exclude_collections: Union[List[str], str, None] = None,
-        *,
         wait_for_completion: Literal[True],
     ) -> ExportStatusReturn: ...
     @overload
     async def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
         include_collections: Union[List[str], str, None] = None,
         exclude_collections: Union[List[str], str, None] = None,
         wait_for_completion: Literal[False] = False,
     ) -> ExportCreateReturn: ...
-    async def get_status(self, export_id: str, backend: ExportStorage) -> ExportStatusReturn: ...
-    async def cancel(self, export_id: str, backend: ExportStorage) -> bool: ...
+    async def get_status(self, *, export_id: str, backend: ExportStorage) -> ExportStatusReturn: ...
+    async def cancel(self, *, export_id: str, backend: ExportStorage) -> bool: ...

weaviate/export/executor.py

@@ -40,18 +40,19 @@ def __init__(self, connection: Connection):
     @overload
     def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
         include_collections: Union[List[str], str, None] = None,
         exclude_collections: Union[List[str], str, None] = None,
-        *,
         wait_for_completion: Literal[True],
     ) -> executor.Result[ExportStatusReturn]: ...
 
     @overload
     def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
@@ -62,6 +63,7 @@ def create(
 
     def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
@@ -180,6 +182,7 @@ async def _execute() -> Union[ExportCreateReturn, ExportStatusReturn]:
 
     def get_status(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
     ) -> executor.Result[ExportStatusReturn]:
@@ -220,6 +223,7 @@ def resp(res: Response) -> ExportStatusReturn:
 
     def cancel(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
     ) -> executor.Result[bool]:

weaviate/export/sync.pyi

@@ -16,23 +16,24 @@ class _Export(_ExportExecutor[ConnectionSync]):
     @overload
     def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
         include_collections: Union[List[str], str, None] = None,
         exclude_collections: Union[List[str], str, None] = None,
-        *,
         wait_for_completion: Literal[True],
     ) -> ExportStatusReturn: ...
     @overload
     def create(
         self,
+        *,
         export_id: str,
         backend: ExportStorage,
         file_format: ExportFileFormat,
         include_collections: Union[List[str], str, None] = None,
         exclude_collections: Union[List[str], str, None] = None,
         wait_for_completion: Literal[False] = False,
     ) -> ExportCreateReturn: ...
-    def get_status(self, export_id: str, backend: ExportStorage) -> ExportStatusReturn: ...
-    def cancel(self, export_id: str, backend: ExportStorage) -> bool: ...
+    def get_status(self, *, export_id: str, backend: ExportStorage) -> ExportStatusReturn: ...
+    def cancel(self, *, export_id: str, backend: ExportStorage) -> bool: ...