convexity-api-client 0.5.1.dev162__tar.gz → 0.5.1.dev164__tar.gz

{convexity_api_client-0.5.1.dev162 → convexity_api_client-0.5.1.dev164}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: convexity_api_client
-Version: 0.5.1.dev162
+Version: 0.5.1.dev164
 Summary: A client library for accessing Convexity API
 Requires-Dist: httpx>=0.23.0,<0.29.0
 Requires-Dist: attrs>=22.2.0

{convexity_api_client-0.5.1.dev162 → convexity_api_client-0.5.1.dev164}/convexity_api_client/models/__init__.py

@@ -86,6 +86,7 @@ from .create_run_request import CreateRunRequest
 from .create_run_request_parameter_input_type_0 import CreateRunRequestParameterInputType0
 from .create_run_request_parameter_type_0 import CreateRunRequestParameterType0
 from .create_run_response import CreateRunResponse
+from .create_run_task_payload import CreateRunTaskPayload
 from .create_s3_connection_request import CreateS3ConnectionRequest
 from .create_share_request import CreateShareRequest
 from .create_task_request import CreateTaskRequest
@@ -151,6 +152,8 @@ from .delete_workspace_request import DeleteWorkspaceRequest
 from .derived_dataset_config import DerivedDatasetConfig
 from .derived_dataset_config_transformations_item import DerivedDatasetConfigTransformationsItem
 from .execute_cell_request import ExecuteCellRequest
+from .execute_run_payload import ExecuteRunPayload
+from .execute_run_payload_extra_env_type_0 import ExecuteRunPayloadExtraEnvType0
 from .field_type import FieldType
 from .file_connection import FileConnection
 from .file_format import FileFormat
@@ -410,6 +413,7 @@ __all__ = (
     "CreateRunRequestParameterInputType0",
     "CreateRunRequestParameterType0",
     "CreateRunResponse",
+    "CreateRunTaskPayload",
     "CreateS3ConnectionRequest",
     "CreateShareRequest",
     "CreateTaskRequest",
@@ -473,6 +477,8 @@ __all__ = (
     "DerivedDatasetConfig",
     "DerivedDatasetConfigTransformationsItem",
     "ExecuteCellRequest",
+    "ExecuteRunPayload",
+    "ExecuteRunPayloadExtraEnvType0",
     "FieldType",
     "FileConnection",
     "FileFormat",

{convexity_api_client-0.5.1.dev162 → convexity_api_client-0.5.1.dev164}/convexity_api_client/models/create_run_request.py

@@ -11,7 +11,7 @@ from ..types import UNSET, Unset
 if TYPE_CHECKING:
     from ..models.create_run_request_parameter_input_type_0 import CreateRunRequestParameterInputType0
     from ..models.create_run_request_parameter_type_0 import CreateRunRequestParameterType0
-    from ..models.task_model import TaskModel
+    from ..models.create_run_task_payload import CreateRunTaskPayload
 
 
 T = TypeVar("T", bound="CreateRunRequest")
@@ -25,7 +25,7 @@ class CreateRunRequest:
         code (str):
         formulation_id (None | str | Unset):
         forecasting_id (None | str | Unset):
-        task (None | TaskModel | Unset):
+        task (CreateRunTaskPayload | None | Unset):
         parameter (CreateRunRequestParameterType0 | None | Unset):
         parameter_input (CreateRunRequestParameterInputType0 | None | Unset):
     """
@@ -33,7 +33,7 @@ class CreateRunRequest:
     code: str
     formulation_id: None | str | Unset = UNSET
     forecasting_id: None | str | Unset = UNSET
-    task: None | TaskModel | Unset = UNSET
+    task: CreateRunTaskPayload | None | Unset = UNSET
     parameter: CreateRunRequestParameterType0 | None | Unset = UNSET
     parameter_input: CreateRunRequestParameterInputType0 | None | Unset = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -41,7 +41,7 @@ class CreateRunRequest:
     def to_dict(self) -> dict[str, Any]:
         from ..models.create_run_request_parameter_input_type_0 import CreateRunRequestParameterInputType0
         from ..models.create_run_request_parameter_type_0 import CreateRunRequestParameterType0
-        from ..models.task_model import TaskModel
+        from ..models.create_run_task_payload import CreateRunTaskPayload
 
         code = self.code
 
@@ -60,7 +60,7 @@ class CreateRunRequest:
         task: dict[str, Any] | None | Unset
         if isinstance(self.task, Unset):
             task = UNSET
-        elif isinstance(self.task, TaskModel):
+        elif isinstance(self.task, CreateRunTaskPayload):
             task = self.task.to_dict()
         else:
             task = self.task
@@ -105,7 +105,7 @@ class CreateRunRequest:
     def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
         from ..models.create_run_request_parameter_input_type_0 import CreateRunRequestParameterInputType0
         from ..models.create_run_request_parameter_type_0 import CreateRunRequestParameterType0
-        from ..models.task_model import TaskModel
+        from ..models.create_run_task_payload import CreateRunTaskPayload
 
         d = dict(src_dict)
         code = d.pop("code")
@@ -128,7 +128,7 @@ class CreateRunRequest:
 
         forecasting_id = _parse_forecasting_id(d.pop("forecasting_id", UNSET))
 
-        def _parse_task(data: object) -> None | TaskModel | Unset:
+        def _parse_task(data: object) -> CreateRunTaskPayload | None | Unset:
             if data is None:
                 return data
             if isinstance(data, Unset):
@@ -136,12 +136,12 @@ class CreateRunRequest:
             try:
                 if not isinstance(data, dict):
                     raise TypeError()
-                task_type_0 = TaskModel.from_dict(data)
+                task_type_0 = CreateRunTaskPayload.from_dict(data)
 
                 return task_type_0
             except (TypeError, ValueError, AttributeError, KeyError):
                 pass
-            return cast(None | TaskModel | Unset, data)
+            return cast(CreateRunTaskPayload | None | Unset, data)
 
         task = _parse_task(d.pop("task", UNSET))
 

convexity_api_client-0.5.1.dev164/convexity_api_client/models/create_run_task_payload.py

@@ -0,0 +1,112 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.notebook_document import NotebookDocument
+
+
+T = TypeVar("T", bound="CreateRunTaskPayload")
+
+
+@_attrs_define
+class CreateRunTaskPayload:
+    """Minimal task payload required for run creation.
+
+    Attributes:
+        id (None | str | Unset):
+        notebook (None | NotebookDocument | Unset):
+    """
+
+    id: None | str | Unset = UNSET
+    notebook: None | NotebookDocument | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.notebook_document import NotebookDocument
+
+        id: None | str | Unset
+        if isinstance(self.id, Unset):
+            id = UNSET
+        else:
+            id = self.id
+
+        notebook: dict[str, Any] | None | Unset
+        if isinstance(self.notebook, Unset):
+            notebook = UNSET
+        elif isinstance(self.notebook, NotebookDocument):
+            notebook = self.notebook.to_dict()
+        else:
+            notebook = self.notebook
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if id is not UNSET:
+            field_dict["id"] = id
+        if notebook is not UNSET:
+            field_dict["notebook"] = notebook
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.notebook_document import NotebookDocument
+
+        d = dict(src_dict)
+
+        def _parse_id(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        id = _parse_id(d.pop("id", UNSET))
+
+        def _parse_notebook(data: object) -> None | NotebookDocument | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                notebook_type_0 = NotebookDocument.from_dict(data)
+
+                return notebook_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | NotebookDocument | Unset, data)
+
+        notebook = _parse_notebook(d.pop("notebook", UNSET))
+
+        create_run_task_payload = cls(
+            id=id,
+            notebook=notebook,
+        )
+
+        create_run_task_payload.additional_properties = d
+        return create_run_task_payload
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

convexity_api_client-0.5.1.dev164/convexity_api_client/models/execute_run_payload.py

@@ -0,0 +1,124 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Literal, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.execute_run_payload_extra_env_type_0 import ExecuteRunPayloadExtraEnvType0
+
+
+T = TypeVar("T", bound="ExecuteRunPayload")
+
+
+@_attrs_define
+class ExecuteRunPayload:
+    """Payload for executing a task run.
+
+    Executes the code associated with a run in a background worker process,
+    supporting long-running and CPU-intensive operations.
+
+        Attributes:
+            run_id (str): ID of the run to execute
+            user_id (str): User who initiated the run
+            type_ (Literal['execute_run'] | Unset):  Default: 'execute_run'.
+            extra_env (ExecuteRunPayloadExtraEnvType0 | None | Unset): Additional environment variables for run execution
+    """
+
+    run_id: str
+    user_id: str
+    type_: Literal["execute_run"] | Unset = "execute_run"
+    extra_env: ExecuteRunPayloadExtraEnvType0 | None | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.execute_run_payload_extra_env_type_0 import ExecuteRunPayloadExtraEnvType0
+
+        run_id = self.run_id
+
+        user_id = self.user_id
+
+        type_ = self.type_
+
+        extra_env: dict[str, Any] | None | Unset
+        if isinstance(self.extra_env, Unset):
+            extra_env = UNSET
+        elif isinstance(self.extra_env, ExecuteRunPayloadExtraEnvType0):
+            extra_env = self.extra_env.to_dict()
+        else:
+            extra_env = self.extra_env
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "runId": run_id,
+                "userId": user_id,
+            }
+        )
+        if type_ is not UNSET:
+            field_dict["type"] = type_
+        if extra_env is not UNSET:
+            field_dict["extraEnv"] = extra_env
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.execute_run_payload_extra_env_type_0 import ExecuteRunPayloadExtraEnvType0
+
+        d = dict(src_dict)
+        run_id = d.pop("runId")
+
+        user_id = d.pop("userId")
+
+        type_ = cast(Literal["execute_run"] | Unset, d.pop("type", UNSET))
+        if type_ != "execute_run" and not isinstance(type_, Unset):
+            raise ValueError(f"type must match const 'execute_run', got '{type_}'")
+
+        def _parse_extra_env(data: object) -> ExecuteRunPayloadExtraEnvType0 | None | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                extra_env_type_0 = ExecuteRunPayloadExtraEnvType0.from_dict(data)
+
+                return extra_env_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(ExecuteRunPayloadExtraEnvType0 | None | Unset, data)
+
+        extra_env = _parse_extra_env(d.pop("extraEnv", UNSET))
+
+        execute_run_payload = cls(
+            run_id=run_id,
+            user_id=user_id,
+            type_=type_,
+            extra_env=extra_env,
+        )
+
+        execute_run_payload.additional_properties = d
+        return execute_run_payload
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

convexity_api_client-0.5.1.dev164/convexity_api_client/models/execute_run_payload_extra_env_type_0.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ExecuteRunPayloadExtraEnvType0")
+
+
+@_attrs_define
+class ExecuteRunPayloadExtraEnvType0:
+    """ """
+
+    additional_properties: dict[str, str] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        execute_run_payload_extra_env_type_0 = cls()
+
+        execute_run_payload_extra_env_type_0.additional_properties = d
+        return execute_run_payload_extra_env_type_0
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> str:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: str) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

{convexity_api_client-0.5.1.dev162 → convexity_api_client-0.5.1.dev164}/convexity_api_client/models/worker_task_response.py

@@ -12,6 +12,7 @@ from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
     from ..models.delete_dataset_payload import DeleteDatasetPayload
+    from ..models.execute_run_payload import ExecuteRunPayload
     from ..models.refresh_dataset_payload import RefreshDatasetPayload
     from ..models.sync_catalog_to_s3_payload import SyncCatalogToS3Payload
     from ..models.vacuum_ducklake_payload import VacuumDucklakePayload
@@ -29,8 +30,8 @@ class WorkerTaskResponse:
         type_ (WorkerTaskType): Supported worker task types.
         status (WorkerTaskStatus):
         error (None | str | Unset):
-        payload (DeleteDatasetPayload | None | RefreshDatasetPayload | SyncCatalogToS3Payload | Unset |
-            VacuumDucklakePayload): Typed task payload with progress information
+        payload (DeleteDatasetPayload | ExecuteRunPayload | None | RefreshDatasetPayload | SyncCatalogToS3Payload |
+            Unset | VacuumDucklakePayload): Typed task payload with progress information
         user_id (None | str | Unset):
         dataset_id (None | str | Unset):
         project_id (None | str | Unset):
@@ -43,7 +44,13 @@ class WorkerTaskResponse:
     status: WorkerTaskStatus
     error: None | str | Unset = UNSET
     payload: (
-        DeleteDatasetPayload | None | RefreshDatasetPayload | SyncCatalogToS3Payload | Unset | VacuumDucklakePayload
+        DeleteDatasetPayload
+        | ExecuteRunPayload
+        | None
+        | RefreshDatasetPayload
+        | SyncCatalogToS3Payload
+        | Unset
+        | VacuumDucklakePayload
     ) = UNSET
     user_id: None | str | Unset = UNSET
     dataset_id: None | str | Unset = UNSET
@@ -54,6 +61,7 @@ class WorkerTaskResponse:
 
     def to_dict(self) -> dict[str, Any]:
         from ..models.delete_dataset_payload import DeleteDatasetPayload
+        from ..models.execute_run_payload import ExecuteRunPayload
         from ..models.refresh_dataset_payload import RefreshDatasetPayload
         from ..models.sync_catalog_to_s3_payload import SyncCatalogToS3Payload
         from ..models.vacuum_ducklake_payload import VacuumDucklakePayload
@@ -81,6 +89,8 @@ class WorkerTaskResponse:
             payload = self.payload.to_dict()
         elif isinstance(self.payload, SyncCatalogToS3Payload):
             payload = self.payload.to_dict()
+        elif isinstance(self.payload, ExecuteRunPayload):
+            payload = self.payload.to_dict()
         else:
             payload = self.payload
 
@@ -143,6 +153,7 @@ class WorkerTaskResponse:
     @classmethod
     def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
         from ..models.delete_dataset_payload import DeleteDatasetPayload
+        from ..models.execute_run_payload import ExecuteRunPayload
         from ..models.refresh_dataset_payload import RefreshDatasetPayload
         from ..models.sync_catalog_to_s3_payload import SyncCatalogToS3Payload
         from ..models.vacuum_ducklake_payload import VacuumDucklakePayload
@@ -166,7 +177,13 @@ class WorkerTaskResponse:
         def _parse_payload(
             data: object,
         ) -> (
-            DeleteDatasetPayload | None | RefreshDatasetPayload | SyncCatalogToS3Payload | Unset | VacuumDucklakePayload
+            DeleteDatasetPayload
+            | ExecuteRunPayload
+            | None
+            | RefreshDatasetPayload
+            | SyncCatalogToS3Payload
+            | Unset
+            | VacuumDucklakePayload
         ):
             if data is None:
                 return data
@@ -204,8 +221,17 @@ class WorkerTaskResponse:
                 return payload_type_0_type_3
             except (TypeError, ValueError, AttributeError, KeyError):
                 pass
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                payload_type_0_type_4 = ExecuteRunPayload.from_dict(data)
+
+                return payload_type_0_type_4
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
             return cast(
                 DeleteDatasetPayload
+                | ExecuteRunPayload
                 | None
                 | RefreshDatasetPayload
                 | SyncCatalogToS3Payload

{convexity_api_client-0.5.1.dev162 → convexity_api_client-0.5.1.dev164}/convexity_api_client/models/worker_task_type.py

@@ -3,6 +3,7 @@ from enum import Enum
 
 class WorkerTaskType(str, Enum):
     DELETE_DATASET = "delete_dataset"
+    EXECUTE_RUN = "execute_run"
     REFRESH_DATASET = "refresh_dataset"
     SYNC_CATALOG_TO_S3 = "sync_catalog_to_s3"
     VACUUM_DUCKLAKE = "vacuum_ducklake"

{convexity_api_client-0.5.1.dev162 → convexity_api_client-0.5.1.dev164}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "convexity_api_client"
-version = "0.5.1.dev162"
+version = "0.5.1.dev164"
 description = "A client library for accessing Convexity API"
 authors = []
 requires-python = ">=3.10"