current-data-py 0.1.10__tar.gz → 0.1.12__tar.gz

{current_data_py-0.1.10 → current_data_py-0.1.12}/MANIFEST.in

@@ -3,3 +3,4 @@ include tests/check_environment.py
 include tests/check_remote.py
 include tests/check_management.py
 include tests/check_oss.py
+include tests/check_labeling.py

{current_data_py-0.1.10 → current_data_py-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: current-data-py
-Version: 0.1.10
+Version: 0.1.12
 Summary: Current Data protobuf messages and schema metadata
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
@@ -8,9 +8,9 @@ Requires-Dist: protobuf>=3.20
 Requires-Dist: turbodata==0.1.0
 Requires-Dist: grpcio>=1.51.1
 Requires-Dist: alibabacloud-oss-v2<2,>=1.4.0
-Provides-Extra: decode
-Requires-Dist: av>=12; extra == "decode"
-Requires-Dist: pillow>=9; extra == "decode"
+Requires-Dist: tos<3,>=2.9
+Requires-Dist: av>=12
+Requires-Dist: pillow>=9
 
 # current-data-py
 
@@ -170,9 +170,8 @@ retry or fall back to earlier RPC versions. Old servers return `UNIMPLEMENTED`;
 unconfigured destinations return `FAILED_PRECONDITION`. A response whose
 provider differs from the request raises `ValueError`.
 
-These methods only fetch credentials. Pass TOS credentials to a native TOS
-client and refresh them before expiration. `OSSClient`, `DataClient`, metadata
-registration and signed URLs have not been extended to TOS by this change.
+These methods only fetch credentials. For file transfers and TD reads with automatic
+credential refresh, see [TOS](#tos). Signed download URLs remain OSS-only.
 
 ## Remote TD reads
 
@@ -180,12 +179,11 @@ registration and signed URLs have not been extended to TOS by this change.
 
 > [!IMPORTANT]
 > **Video decoding requires PyAV, Pillow and the FFmpeg native libraries used by
-> PyAV.** `open_td()` enables decoding by default, but a base `current-data-py`
-> installation does not install the video dependencies. Install the `[decode]`
-> extra in every environment that decodes video, including DataLoader workers.
+> PyAV.** `current-data-py` installs PyAV and Pillow by default. Install the package
+> in every environment that decodes video, including DataLoader workers.
 
 ```bash
-python -m pip install --only-binary=av,pillow 'current-data-py[decode]'
+python -m pip install --only-binary=av,pillow current-data-py
 ```
 
 - **Prebuilt wheels:** PyAV wheels bundle FFmpeg libraries on supported platforms;
@@ -272,8 +270,9 @@ to each call.
 
 ### Storage and credentials
 
-`DataClient.open_td()` accepts an optional `bucket` argument. Without it, reads
-use the management service's `OSS_READ_BUCKET` and `OSS_READ_PREFIX` defaults.
+`DataClient.open_td()` accepts optional `bucket` and `provider` arguments. OSS
+references without either selection retain the service's `OSS_READ_BUCKET` and
+`OSS_READ_PREFIX` defaults. See [TOS](#tos) for provider selection.
 With it, the SDK uses `GetReadCredentialsV2` for whole-bucket credentials in the
 bucket's region, limited by the server's RAM role, and selects the first matching
 storage in the reference. An explicit bucket does not inherit the default prefix.
@@ -336,6 +335,69 @@ with ManagementClient(address=os.environ["MANAGEMENT_ADDRESS"]) as management:
   Deploy the V2 credential APIs and configure bucket regions before using reads
   or the upload `data_kind` option.
 
+## TOS
+
+`TOSClient` obtains and refreshes V3 credentials from Management. Keep
+`ManagementClient` open during transfers.
+
+```python
+import os
+from current_data_py.management import ManagementClient
+from current_data_py.tos import TOSClient
+
+with ManagementClient() as management:
+    with TOSClient(management, permission="read", bucket=os.environ["TOS_READ_BUCKET"],
+                   endpoint="https://tos-accelerate.volces.com") as source:
+        source.download_file(os.environ["SOURCE_OBJECT_KEY"], "/tmp/input.h5")
+```
+
+For uploads, use `permission="upload", data_type="YAM", data_kind="CONVERTED"`;
+register the returned bucket and uploaded object key in `Metadata.storages`. To read a
+TD file, use `DataClient.open_td(reference, provider="tos", bucket=bucket)`.
+`endpoint=` changes the network endpoint without changing the signing region;
+it cannot be combined with `use_internal_endpoint=True`.
+
+## Pre-annotation
+
+`current_data_py.labeling.LabelingClient` wraps the pre-annotation RPCs
+`CreateTask`, `ListTasks`, `ListItems`, `WriteResult`, and `UpdatePreAnnotation` with
+ManagementClient's TLS, package default address, caller-owned channel, per-call
+`timeout` (30 seconds by default, `None` disables it), and `metadata` conventions.
+Errors propagate without retries. Protobuf types are available through
+`service_pb2`, `task_pb2`, and `DataSelection` in the same module.
+
+```python
+from current_data_py.labeling import LabelingClient, service_pb2, task_pb2
+
+with LabelingClient() as labeling:
+    task = labeling.create_task(
+        "template-id", name="Pre-annotation",
+        input_sources=[task_pb2.InputSource(td_file=task_pb2.TdFileSource(data_id="data-id"))],
+        start_pre_annotation=True,
+    )
+    for item in labeling.list_items(task.task_id):
+        # Match question keys and option keys to the existing built-in template.
+        labeling.write_result(item.item_id, task_pb2.Result(question_results=[
+            task_pb2.QuestionResult(
+                question_key="quality",
+                choice=task_pb2.ChoiceResult(selected_option_keys=["ok"]),
+            ),
+        ]))
+        # On inference failure instead: write_result(item.item_id, pre_annotation_error="reason")
+    labeling.update_pre_annotation(
+        task.task_id, service_pb2.UpdatePreAnnotationRequest.ACTION_OPEN_HUMAN_REVIEW,
+    )
+```
+
+Alternatively, pass `data_selection=DataSelection(...)` when creating a task.
+Use `ACTION_ACCEPT_ALL_AND_COMPLETE` to accept all pre-annotations as final labels.
+Use an existing built-in template; custom templates do not support pre-annotation.
+The server validates inputs and task transitions.
+
+`list_tasks` and `list_items` return repeated protobuf collections.
+`create_task` and `update_pre_annotation` return full protobuf responses
+(including item count and task ID, or status). `write_result` returns `None`.
+
 ## 打包
 
 ```bash
@@ -355,8 +417,11 @@ python -m pip install --force-reinstall dist/*.whl
 python -I tests/check_package.py
 python -I tests/check_environment.py
 python -I tests/check_management.py
+python -I tests/check_labeling.py
 python -I tests/check_remote.py
 python -I tests/check_oss.py
+# With the tos extra installed:
+python -I tests/check_tos.py
 ```
 
 ## 推送

current_data_py-0.1.10/current_data_py.egg-info/PKG-INFO → current_data_py-0.1.12/README.md

@@ -1,17 +1,3 @@
-Metadata-Version: 2.4
-Name: current-data-py
-Version: 0.1.10
-Summary: Current Data protobuf messages and schema metadata
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-Requires-Dist: protobuf>=3.20
-Requires-Dist: turbodata==0.1.0
-Requires-Dist: grpcio>=1.51.1
-Requires-Dist: alibabacloud-oss-v2<2,>=1.4.0
-Provides-Extra: decode
-Requires-Dist: av>=12; extra == "decode"
-Requires-Dist: pillow>=9; extra == "decode"
-
 # current-data-py
 
 ## Environment
@@ -170,9 +156,8 @@ retry or fall back to earlier RPC versions. Old servers return `UNIMPLEMENTED`;
 unconfigured destinations return `FAILED_PRECONDITION`. A response whose
 provider differs from the request raises `ValueError`.
 
-These methods only fetch credentials. Pass TOS credentials to a native TOS
-client and refresh them before expiration. `OSSClient`, `DataClient`, metadata
-registration and signed URLs have not been extended to TOS by this change.
+These methods only fetch credentials. For file transfers and TD reads with automatic
+credential refresh, see [TOS](#tos). Signed download URLs remain OSS-only.
 
 ## Remote TD reads
 
@@ -180,12 +165,11 @@ registration and signed URLs have not been extended to TOS by this change.
 
 > [!IMPORTANT]
 > **Video decoding requires PyAV, Pillow and the FFmpeg native libraries used by
-> PyAV.** `open_td()` enables decoding by default, but a base `current-data-py`
-> installation does not install the video dependencies. Install the `[decode]`
-> extra in every environment that decodes video, including DataLoader workers.
+> PyAV.** `current-data-py` installs PyAV and Pillow by default. Install the package
+> in every environment that decodes video, including DataLoader workers.
 
 ```bash
-python -m pip install --only-binary=av,pillow 'current-data-py[decode]'
+python -m pip install --only-binary=av,pillow current-data-py
 ```
 
 - **Prebuilt wheels:** PyAV wheels bundle FFmpeg libraries on supported platforms;
@@ -272,8 +256,9 @@ to each call.
 
 ### Storage and credentials
 
-`DataClient.open_td()` accepts an optional `bucket` argument. Without it, reads
-use the management service's `OSS_READ_BUCKET` and `OSS_READ_PREFIX` defaults.
+`DataClient.open_td()` accepts optional `bucket` and `provider` arguments. OSS
+references without either selection retain the service's `OSS_READ_BUCKET` and
+`OSS_READ_PREFIX` defaults. See [TOS](#tos) for provider selection.
 With it, the SDK uses `GetReadCredentialsV2` for whole-bucket credentials in the
 bucket's region, limited by the server's RAM role, and selects the first matching
 storage in the reference. An explicit bucket does not inherit the default prefix.
@@ -336,6 +321,69 @@ with ManagementClient(address=os.environ["MANAGEMENT_ADDRESS"]) as management:
   Deploy the V2 credential APIs and configure bucket regions before using reads
   or the upload `data_kind` option.
 
+## TOS
+
+`TOSClient` obtains and refreshes V3 credentials from Management. Keep
+`ManagementClient` open during transfers.
+
+```python
+import os
+from current_data_py.management import ManagementClient
+from current_data_py.tos import TOSClient
+
+with ManagementClient() as management:
+    with TOSClient(management, permission="read", bucket=os.environ["TOS_READ_BUCKET"],
+                   endpoint="https://tos-accelerate.volces.com") as source:
+        source.download_file(os.environ["SOURCE_OBJECT_KEY"], "/tmp/input.h5")
+```
+
+For uploads, use `permission="upload", data_type="YAM", data_kind="CONVERTED"`;
+register the returned bucket and uploaded object key in `Metadata.storages`. To read a
+TD file, use `DataClient.open_td(reference, provider="tos", bucket=bucket)`.
+`endpoint=` changes the network endpoint without changing the signing region;
+it cannot be combined with `use_internal_endpoint=True`.
+
+## Pre-annotation
+
+`current_data_py.labeling.LabelingClient` wraps the pre-annotation RPCs
+`CreateTask`, `ListTasks`, `ListItems`, `WriteResult`, and `UpdatePreAnnotation` with
+ManagementClient's TLS, package default address, caller-owned channel, per-call
+`timeout` (30 seconds by default, `None` disables it), and `metadata` conventions.
+Errors propagate without retries. Protobuf types are available through
+`service_pb2`, `task_pb2`, and `DataSelection` in the same module.
+
+```python
+from current_data_py.labeling import LabelingClient, service_pb2, task_pb2
+
+with LabelingClient() as labeling:
+    task = labeling.create_task(
+        "template-id", name="Pre-annotation",
+        input_sources=[task_pb2.InputSource(td_file=task_pb2.TdFileSource(data_id="data-id"))],
+        start_pre_annotation=True,
+    )
+    for item in labeling.list_items(task.task_id):
+        # Match question keys and option keys to the existing built-in template.
+        labeling.write_result(item.item_id, task_pb2.Result(question_results=[
+            task_pb2.QuestionResult(
+                question_key="quality",
+                choice=task_pb2.ChoiceResult(selected_option_keys=["ok"]),
+            ),
+        ]))
+        # On inference failure instead: write_result(item.item_id, pre_annotation_error="reason")
+    labeling.update_pre_annotation(
+        task.task_id, service_pb2.UpdatePreAnnotationRequest.ACTION_OPEN_HUMAN_REVIEW,
+    )
+```
+
+Alternatively, pass `data_selection=DataSelection(...)` when creating a task.
+Use `ACTION_ACCEPT_ALL_AND_COMPLETE` to accept all pre-annotations as final labels.
+Use an existing built-in template; custom templates do not support pre-annotation.
+The server validates inputs and task transitions.
+
+`list_tasks` and `list_items` return repeated protobuf collections.
+`create_task` and `update_pre_annotation` return full protobuf responses
+(including item count and task ID, or status). `write_result` returns `None`.
+
 ## 打包
 
 ```bash
@@ -355,8 +403,11 @@ python -m pip install --force-reinstall dist/*.whl
 python -I tests/check_package.py
 python -I tests/check_environment.py
 python -I tests/check_management.py
+python -I tests/check_labeling.py
 python -I tests/check_remote.py
 python -I tests/check_oss.py
+# With the tos extra installed:
+python -I tests/check_tos.py
 ```
 
 ## 推送

current_data_py-0.1.10/README.md → current_data_py-0.1.12/current_data_py.egg-info/PKG-INFO

@@ -1,3 +1,17 @@
+Metadata-Version: 2.4
+Name: current-data-py
+Version: 0.1.12
+Summary: Current Data protobuf messages and schema metadata
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: protobuf>=3.20
+Requires-Dist: turbodata==0.1.0
+Requires-Dist: grpcio>=1.51.1
+Requires-Dist: alibabacloud-oss-v2<2,>=1.4.0
+Requires-Dist: tos<3,>=2.9
+Requires-Dist: av>=12
+Requires-Dist: pillow>=9
+
 # current-data-py
 
 ## Environment
@@ -156,9 +170,8 @@ retry or fall back to earlier RPC versions. Old servers return `UNIMPLEMENTED`;
 unconfigured destinations return `FAILED_PRECONDITION`. A response whose
 provider differs from the request raises `ValueError`.
 
-These methods only fetch credentials. Pass TOS credentials to a native TOS
-client and refresh them before expiration. `OSSClient`, `DataClient`, metadata
-registration and signed URLs have not been extended to TOS by this change.
+These methods only fetch credentials. For file transfers and TD reads with automatic
+credential refresh, see [TOS](#tos). Signed download URLs remain OSS-only.
 
 ## Remote TD reads
 
@@ -166,12 +179,11 @@ registration and signed URLs have not been extended to TOS by this change.
 
 > [!IMPORTANT]
 > **Video decoding requires PyAV, Pillow and the FFmpeg native libraries used by
-> PyAV.** `open_td()` enables decoding by default, but a base `current-data-py`
-> installation does not install the video dependencies. Install the `[decode]`
-> extra in every environment that decodes video, including DataLoader workers.
+> PyAV.** `current-data-py` installs PyAV and Pillow by default. Install the package
+> in every environment that decodes video, including DataLoader workers.
 
 ```bash
-python -m pip install --only-binary=av,pillow 'current-data-py[decode]'
+python -m pip install --only-binary=av,pillow current-data-py
 ```
 
 - **Prebuilt wheels:** PyAV wheels bundle FFmpeg libraries on supported platforms;
@@ -258,8 +270,9 @@ to each call.
 
 ### Storage and credentials
 
-`DataClient.open_td()` accepts an optional `bucket` argument. Without it, reads
-use the management service's `OSS_READ_BUCKET` and `OSS_READ_PREFIX` defaults.
+`DataClient.open_td()` accepts optional `bucket` and `provider` arguments. OSS
+references without either selection retain the service's `OSS_READ_BUCKET` and
+`OSS_READ_PREFIX` defaults. See [TOS](#tos) for provider selection.
 With it, the SDK uses `GetReadCredentialsV2` for whole-bucket credentials in the
 bucket's region, limited by the server's RAM role, and selects the first matching
 storage in the reference. An explicit bucket does not inherit the default prefix.
@@ -322,6 +335,69 @@ with ManagementClient(address=os.environ["MANAGEMENT_ADDRESS"]) as management:
   Deploy the V2 credential APIs and configure bucket regions before using reads
   or the upload `data_kind` option.
 
+## TOS
+
+`TOSClient` obtains and refreshes V3 credentials from Management. Keep
+`ManagementClient` open during transfers.
+
+```python
+import os
+from current_data_py.management import ManagementClient
+from current_data_py.tos import TOSClient
+
+with ManagementClient() as management:
+    with TOSClient(management, permission="read", bucket=os.environ["TOS_READ_BUCKET"],
+                   endpoint="https://tos-accelerate.volces.com") as source:
+        source.download_file(os.environ["SOURCE_OBJECT_KEY"], "/tmp/input.h5")
+```
+
+For uploads, use `permission="upload", data_type="YAM", data_kind="CONVERTED"`;
+register the returned bucket and uploaded object key in `Metadata.storages`. To read a
+TD file, use `DataClient.open_td(reference, provider="tos", bucket=bucket)`.
+`endpoint=` changes the network endpoint without changing the signing region;
+it cannot be combined with `use_internal_endpoint=True`.
+
+## Pre-annotation
+
+`current_data_py.labeling.LabelingClient` wraps the pre-annotation RPCs
+`CreateTask`, `ListTasks`, `ListItems`, `WriteResult`, and `UpdatePreAnnotation` with
+ManagementClient's TLS, package default address, caller-owned channel, per-call
+`timeout` (30 seconds by default, `None` disables it), and `metadata` conventions.
+Errors propagate without retries. Protobuf types are available through
+`service_pb2`, `task_pb2`, and `DataSelection` in the same module.
+
+```python
+from current_data_py.labeling import LabelingClient, service_pb2, task_pb2
+
+with LabelingClient() as labeling:
+    task = labeling.create_task(
+        "template-id", name="Pre-annotation",
+        input_sources=[task_pb2.InputSource(td_file=task_pb2.TdFileSource(data_id="data-id"))],
+        start_pre_annotation=True,
+    )
+    for item in labeling.list_items(task.task_id):
+        # Match question keys and option keys to the existing built-in template.
+        labeling.write_result(item.item_id, task_pb2.Result(question_results=[
+            task_pb2.QuestionResult(
+                question_key="quality",
+                choice=task_pb2.ChoiceResult(selected_option_keys=["ok"]),
+            ),
+        ]))
+        # On inference failure instead: write_result(item.item_id, pre_annotation_error="reason")
+    labeling.update_pre_annotation(
+        task.task_id, service_pb2.UpdatePreAnnotationRequest.ACTION_OPEN_HUMAN_REVIEW,
+    )
+```
+
+Alternatively, pass `data_selection=DataSelection(...)` when creating a task.
+Use `ACTION_ACCEPT_ALL_AND_COMPLETE` to accept all pre-annotations as final labels.
+Use an existing built-in template; custom templates do not support pre-annotation.
+The server validates inputs and task transitions.
+
+`list_tasks` and `list_items` return repeated protobuf collections.
+`create_task` and `update_pre_annotation` return full protobuf responses
+(including item count and task ID, or status). `write_result` returns `None`.
+
 ## 打包
 
 ```bash
@@ -341,8 +417,11 @@ python -m pip install --force-reinstall dist/*.whl
 python -I tests/check_package.py
 python -I tests/check_environment.py
 python -I tests/check_management.py
+python -I tests/check_labeling.py
 python -I tests/check_remote.py
 python -I tests/check_oss.py
+# With the tos extra installed:
+python -I tests/check_tos.py
 ```
 
 ## 推送

{current_data_py-0.1.10 → current_data_py-0.1.12}/current_data_py.egg-info/SOURCES.txt

@@ -2,14 +2,18 @@ MANIFEST.in
 README.md
 __init__.py
 decoding.py
+labeling.py
 management.py
 oss.py
 pyproject.toml
 setup.py
+tos.py
 ./__init__.py
 ./decoding.py
+./labeling.py
 ./management.py
 ./oss.py
+./tos.py
 ./examples/__init__.py
 ./examples/list_schemas.py
 ./examples/read_remote_td.py
@@ -18,6 +22,7 @@ setup.py
 ./remote/__init__.py
 ./remote/client.py
 ./remote/source.py
+./remote/tos_source.py
 ./schema/__init__.py
 ./schema/message_metadata.py
 current_data_py.egg-info/PKG-INFO
@@ -33,9 +38,11 @@ examples/write_td.py
 remote/__init__.py
 remote/client.py
 remote/source.py
+remote/tos_source.py
 schema/__init__.py
 schema/message_metadata.py
 tests/check_environment.py
+tests/check_labeling.py
 tests/check_management.py
 tests/check_oss.py
 tests/check_package.py

{current_data_py-0.1.10 → current_data_py-0.1.12}/current_data_py.egg-info/requires.txt

@@ -2,7 +2,6 @@ protobuf>=3.20
 turbodata==0.1.0
 grpcio>=1.51.1
 alibabacloud-oss-v2<2,>=1.4.0
-
-[decode]
+tos<3,>=2.9
 av>=12
 pillow>=9

{current_data_py-0.1.10 → current_data_py-0.1.12}/decoding.py

@@ -67,12 +67,9 @@ class DecodedMessage:
 class _VideoStream:
     """Hold one topic's H.264 decoder for the lifetime of a forward iterator."""
     def __init__(self, *, keep_timestamp=None):
-        """Load optional video dependencies and create a fresh decoder."""
-        try:
-            import av
-            from PIL import Image  # noqa: F401
-        except ImportError as exc:
-            raise ImportError("Video decoding requires pip install 'current-data-py[decode]'") from exc
+        """Load video dependencies and create a fresh decoder."""
+        import av
+        from PIL import Image  # noqa: F401
         self.av = av
         self.codec = av.CodecContext.create("h264", "r")
         self.codec.thread_count = 1

current_data_py-0.1.12/labeling.py

@@ -0,0 +1,102 @@
+"""Synchronous client for LabelingService pre-annotation RPCs."""
+
+from typing import Optional, Sequence, Tuple, Union
+
+import grpc
+
+import current_data_py as _package
+from . import DEFAULT_ADDRESS
+from ._proto.labeling.service import service_pb2, service_pb2_grpc
+from ._proto.labeling.task import task_pb2
+from ._proto.management.selection.selection_pb2 import DataSelection
+
+_CallMetadata = Optional[Sequence[Tuple[str, Union[str, bytes]]]]
+
+__all__ = [
+    "DEFAULT_ADDRESS", "LabelingClient", "DataSelection",
+    "service_pb2", "task_pb2",
+]
+
+
+class LabelingClient:
+    """Use TLS and current_data_py.DEFAULT_ADDRESS, or a caller-owned channel.
+
+    RPC errors propagate unchanged without retries. Call metadata carries headers
+    such as authorization; timeouts are in seconds. Server-side validation governs
+    pre-annotation results and task transitions.
+    """
+
+    def __init__(self, channel: Optional[grpc.Channel] = None, *, address: Optional[str] = None):
+        self._owns_channel = channel is None
+        self._channel = (
+            grpc.secure_channel(
+                _package.DEFAULT_ADDRESS if address is None else address, grpc.ssl_channel_credentials(),
+            )
+            if channel is None else channel
+        )
+        self.stub = service_pb2_grpc.LabelingServiceStub(self._channel)
+
+    def close(self) -> None:
+        """Close the channel only when this client created it."""
+        if self._owns_channel:
+            self._channel.close()
+
+    def __enter__(self) -> "LabelingClient":
+        return self
+
+    def __exit__(self, *args) -> None:
+        self.close()
+
+    def create_task(
+        self, template_id: str, *, input_sources: Optional[Sequence[task_pb2.InputSource]] = None,
+        data_selection: Optional[DataSelection] = None, name: str = "",
+        start_pre_annotation: bool = False,
+        timeout: Optional[float] = 30, metadata: _CallMetadata = None,
+    ) -> service_pb2.CreateTaskResponse:
+        return self.stub.CreateTask(
+            service_pb2.CreateTaskRequest(
+                template_id=template_id, input_sources=input_sources,
+                data_selection=data_selection, name=name, start_pre_annotation=start_pre_annotation,
+            ),
+            timeout=timeout, metadata=metadata,
+        )
+
+    def list_tasks(
+        self, *,
+        timeout: Optional[float] = 30, metadata: _CallMetadata = None,
+    ) -> Sequence[task_pb2.TaskSummary]:
+        return self.stub.ListTasks(
+            service_pb2.ListTasksRequest(),
+            timeout=timeout, metadata=metadata,
+        ).tasks
+
+    def list_items(
+        self, task_id: str, *,
+        timeout: Optional[float] = 30, metadata: _CallMetadata = None,
+    ) -> Sequence[task_pb2.Item]:
+        return self.stub.ListItems(
+            service_pb2.ListItemsRequest(task_id=task_id),
+            timeout=timeout, metadata=metadata,
+        ).items
+
+    def write_result(
+        self, item_id: str, result: Optional[task_pb2.Result] = None, *,
+        pre_annotation_error: str = "",
+        timeout: Optional[float] = 30, metadata: _CallMetadata = None,
+    ) -> None:
+        self.stub.WriteResult(
+            service_pb2.WriteResultRequest(
+                item_id=item_id, result=result,
+                pre_annotation_error=pre_annotation_error,
+            ),
+            timeout=timeout, metadata=metadata,
+        )
+
+    def update_pre_annotation(
+        self, task_id: str, action: int, *,
+        timeout: Optional[float] = 30, metadata: _CallMetadata = None,
+    ) -> service_pb2.UpdatePreAnnotationResponse:
+        return self.stub.UpdatePreAnnotation(
+            service_pb2.UpdatePreAnnotationRequest(task_id=task_id, action=action),
+            timeout=timeout, metadata=metadata,
+        )

{current_data_py-0.1.10 → current_data_py-0.1.12}/management.py

@@ -8,14 +8,13 @@ import current_data_py as _package
 from . import DEFAULT_ADDRESS  # Compatibility export; configure current_data_py.DEFAULT_ADDRESS.
 
 from ._proto.management.metadata import metadata_pb2
-from ._proto.management.metadata.metadata_pb2 import Metadata, MetadataFilter
+from ._proto.management.metadata.metadata_pb2 import Metadata, MetadataFilter, StorageProvider
 from ._proto.management.service import service_pb2, service_pb2_grpc
 from ._proto.management.service.service_pb2 import (
     DataKind,
     GetReadCredentialsResponse,
     GetUploadCredentialsResponse,
     StorageCredentials,
-    StorageProvider,
 )
 
 
@@ -30,9 +29,10 @@ __all__ = [
 
 def _storage_provider(value: Union[int, str]) -> int:
     if isinstance(value, str):
+        value = value.upper()
         name = value if value.startswith("STORAGE_PROVIDER_") else "STORAGE_PROVIDER_" + value
         value = StorageProvider.Value(name)
-    if isinstance(value, bool) or value not in (service_pb2.STORAGE_PROVIDER_OSS, service_pb2.STORAGE_PROVIDER_TOS):
+    if isinstance(value, bool) or value not in (metadata_pb2.STORAGE_PROVIDER_OSS, metadata_pb2.STORAGE_PROVIDER_TOS):
         raise ValueError("provider must explicitly select OSS or TOS")
     return value