UncountablePythonSDK 0.0.68__py3-none-any.whl → 0.0.70__py3-none-any.whl

uncountable/types/client_base.py

@@ -74,6 +74,7 @@ import uncountable.types.api.inputs.set_intermediate_type as set_intermediate_ty
 import uncountable.types.api.recipes.set_recipe_inputs as set_recipe_inputs_t
 import uncountable.types.api.recipes.set_recipe_metadata as set_recipe_metadata_t
 import uncountable.types.api.recipes.set_recipe_output_annotations as set_recipe_output_annotations_t
+import uncountable.types.api.recipes.set_recipe_output_file as set_recipe_output_file_t
 import uncountable.types.api.recipes.set_recipe_outputs as set_recipe_outputs_t
 import uncountable.types.api.recipes.set_recipe_tags as set_recipe_tags_t
 import uncountable.types.api.entity.set_values as set_values_t
@@ -1321,6 +1322,25 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=set_recipe_output_annotations_t.Data)
 
+    def set_recipe_output_file(
+        self,
+        *,
+        output_file_data: set_recipe_output_file_t.RecipeOutputFileValue,
+    ) -> set_recipe_output_file_t.Data:
+        """Sets output file value for an experiment.  Include a single file as part of the FormData of the request with the filename as the key
+
+        :param output_file_data: The output file to set
+        """
+        args = set_recipe_output_file_t.Arguments(
+            output_file_data=output_file_data,
+        )
+        api_request = APIRequest(
+            method=set_recipe_output_file_t.ENDPOINT_METHOD,
+            endpoint=set_recipe_output_file_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=set_recipe_output_file_t.Data)
+
     def set_recipe_outputs(
         self,
         *,

uncountable/types/entity_t.py

@@ -119,6 +119,7 @@ __all__: list[str] = [
         "recipe_permission_view": "Recipe Permission View",
         "recipe_project": "Recipe Project",
         "recipe_step": "Recipe Step",
+        "recipe_workflow_step": "Recipe Workflow Step",
         "recipe_tag": "Recipe Tag",
         "recipe_metadata": "Experiment Metadata",
         "metadata_value": "Metadata Value",
@@ -271,6 +272,7 @@ class EntityType(StrEnum):
     RECIPE_PERMISSION_VIEW = "recipe_permission_view"
     RECIPE_PROJECT = "recipe_project"
     RECIPE_STEP = "recipe_step"
+    RECIPE_WORKFLOW_STEP = "recipe_workflow_step"
     RECIPE_TAG = "recipe_tag"
     RECIPE_METADATA = "recipe_metadata"
     METADATA_VALUE = "metadata_value"

uncountable/types/queued_job.py

@@ -0,0 +1,16 @@
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+# DO NOT MODIFY -- This file is generated by type_spec
+# Kept only for SDK backwards compatibility
+from .queued_job_t import InvocationContextType as InvocationContextType
+from .queued_job_t import InvocationContextBase as InvocationContextBase
+from .queued_job_t import InvocationContextCron as InvocationContextCron
+from .queued_job_t import InvocationContextManual as InvocationContextManual
+from .queued_job_t import InvocationContextWebhook as InvocationContextWebhook
+from .queued_job_t import InvocationContext as InvocationContext
+from .queued_job_t import QueuedJobPayload as QueuedJobPayload
+from .queued_job_t import QueuedJobResult as QueuedJobResult
+from .queued_job_t import QueuedJob as QueuedJob
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/queued_job_t.py

@@ -0,0 +1,107 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+from pkgs.strenum_compat import StrEnum
+import dataclasses
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from . import base_t
+from . import job_definition_t
+
+__all__: list[str] = [
+    "InvocationContext",
+    "InvocationContextBase",
+    "InvocationContextCron",
+    "InvocationContextManual",
+    "InvocationContextType",
+    "InvocationContextWebhook",
+    "QueuedJob",
+    "QueuedJobPayload",
+    "QueuedJobResult",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class InvocationContextType(StrEnum):
+    CRON = "cron"
+    MANUAL = "manual"
+    WEBHOOK = "webhook"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class InvocationContextBase:
+    type: InvocationContextType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class InvocationContextCron(InvocationContextBase):
+    type: typing.Literal[InvocationContextType.CRON] = InvocationContextType.CRON
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class InvocationContextManual(InvocationContextBase):
+    type: typing.Literal[InvocationContextType.MANUAL] = InvocationContextType.MANUAL
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    unconverted_values={"webhook_payload"},
+    parse_require={"type"},
+)
+@dataclasses.dataclass(kw_only=True)
+class InvocationContextWebhook(InvocationContextBase):
+    type: typing.Literal[InvocationContextType.WEBHOOK] = InvocationContextType.WEBHOOK
+    webhook_payload: base_t.JsonValue
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+InvocationContext = typing.Annotated[
+    typing.Union[InvocationContextCron, InvocationContextManual, InvocationContextWebhook],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "cron": InvocationContextCron,
+            "manual": InvocationContextManual,
+            "webhook": InvocationContextWebhook,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class QueuedJobPayload:
+    invocation_context: InvocationContext
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class QueuedJobResult:
+    queued_job_uuid: str
+    job_result: job_definition_t.JobResult
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclasses.dataclass(kw_only=True)
+class QueuedJob:
+    queued_job_uuid: str
+    job_ref_name: str
+    num_attempts: int
+    submitted_at: datetime.datetime
+    payload: QueuedJobPayload
+# DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/recipe_metadata_t.py

@@ -29,6 +29,7 @@ class MetadataValue:
     value_numeric: typing.Optional[Decimal] = None
     value_str: typing.Optional[str] = None
     value_json: typing.Optional[base_t.JsonValue] = None
+    value_file_ids: typing.Optional[list[base_t.ObjectId]] = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec