UncountablePythonSDK 0.0.33__tar.gz → 0.0.35__tar.gz

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/.github/workflows/publish.yml

@@ -50,4 +50,4 @@ jobs:
         name: package-distributions
         path: dist/
     - name: Publish package distributions to PyPI
-      uses: pypa/gh-action-pypi-publish@ec4db0b4ddc65acdf4bff5fa45ac92d78b56bdf0
+      uses: pypa/gh-action-pypi-publish@fb9fc6a4e67ca27a7a76b17bbf90be83c2d3c716

{uncountablepythonsdk-0.0.33/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.33
+Version: 0.0.35
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.33
+Version: 0.0.35
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -26,6 +26,7 @@ docs/static/favicons/mstile-150x150.png
 docs/static/favicons/safari-pinned-tab.svg
 examples/async_batch.py
 examples/create_entity.py
+examples/edit_recipe_inputs.py
 examples/upload_files.py
 pkgs/__init__.py
 pkgs/argument_parser/__init__.py
@@ -182,9 +183,11 @@ uncountable/types/api/recipes/get_recipe_links.py
 uncountable/types/api/recipes/get_recipe_names.py
 uncountable/types/api/recipes/get_recipe_output_metadata.py
 uncountable/types/api/recipes/get_recipes_data.py
+uncountable/types/api/recipes/lock_recipes.py
 uncountable/types/api/recipes/remove_recipe_from_project.py
 uncountable/types/api/recipes/set_recipe_inputs.py
 uncountable/types/api/recipes/set_recipe_metadata.py
+uncountable/types/api/recipes/set_recipe_output_annotations.py
 uncountable/types/api/recipes/set_recipe_outputs.py
 uncountable/types/api/recipes/set_recipe_tags.py
 uncountable/types/api/recipes/unarchive_recipes.py

uncountablepythonsdk-0.0.35/examples/edit_recipe_inputs.py

@@ -0,0 +1,45 @@
+import os
+from decimal import Decimal
+
+from uncountable.core import AsyncBatchProcessor, AuthDetailsApiKey, Client
+from uncountable.types import (
+    edit_recipe_inputs_t,
+    recipe_workflow_steps_t,
+)
+from uncountable.types.identifier import IdentifierKeyBatchReference, IdentifierKeyId
+from uncountable.types.recipe_identifiers import (
+    RecipeIdentifiers,
+)
+from uncountable.types.recipe_inputs import QuantityBasis
+
+client = Client(
+    base_url=os.environ["UNC_BASE_URL"],
+    auth_details=AuthDetailsApiKey(
+        api_id=os.environ["UNC_API_ID"], api_secret_key=os.environ["UNC_API_SECRET_KEY"]
+    ),
+)
+batch_loader = AsyncBatchProcessor(client=client)
+recipe_identifiers: RecipeIdentifiers = []
+req = batch_loader.create_recipe(
+    material_family_id=1, workflow_id=1, identifiers=recipe_identifiers
+)
+created_recipe_reference = req.batch_reference
+edits: list[edit_recipe_inputs_t.RecipeInputEdit] = []
+edits.append(
+    edit_recipe_inputs_t.RecipeInputEditAddInput(
+        quantity_basis=QuantityBasis.MASS,
+        ingredient_key=IdentifierKeyId(id=1),
+        value_numeric=Decimal(56.7),
+    )
+)
+edits.append(
+    edit_recipe_inputs_t.RecipeInputEditChangeBasisViewed(
+        quantity_basis=QuantityBasis.VOLUME, ingredient_key=IdentifierKeyId(id=1)
+    )
+)
+batch_loader.edit_recipe_inputs(
+    recipe_key=IdentifierKeyBatchReference(reference=created_recipe_reference),
+    edits=edits,
+    recipe_workflow_step_identifier=recipe_workflow_steps_t.RecipeWorkflowStepIdentifierDefault(),
+)
+job_id = batch_loader.send()

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/pkgs/serialization_util/serialization_helpers.py

@@ -1,5 +1,10 @@
+# warnings -- types here assume that keys to dictionaries are strings
+# this is true most of the time, but there are cases where we have integer indexes
+
+import dataclasses
 import datetime
 import enum
+import functools
 from collections.abc import Callable, Mapping, Sequence
 from decimal import Decimal
 from typing import (
@@ -39,14 +44,25 @@ def identity(x: T) -> T:
     return x
 
 
-def key_convert_to_camelcase(o: Any) -> str:
+@overload
+def key_convert_to_camelcase(o: str) -> str: ...
+
+
+@overload
+def key_convert_to_camelcase(o: int) -> int: ...
+
+
+def key_convert_to_camelcase(o: Any) -> str | int:
     if isinstance(o, OpaqueKey):
         return o
-    if isinstance(o, enum.Enum):
-        return o.value  # type: ignore[no-any-return]
+    if isinstance(o, enum.StrEnum):
+        return o.value
     if isinstance(o, str):
         return snake_to_camel_case(o)
-    return o  # type: ignore[no-any-return]
+    if isinstance(o, int):
+        # we allow dictionaries to use integer keys
+        return o
+    raise ValueError("Unexpected key type", o)
 
 
 def _convert_dict(d: dict[str, Any]) -> dict[str, JsonValue]:
@@ -61,27 +77,50 @@ def _serialize_dict(d: dict[str, Any]) -> dict[str, JsonValue]:
     return {k: serialize_for_storage(v) for k, v in d.items() if v != MISSING_SENTRY}
 
 
-def _convert_dataclass(d: Dataclass) -> dict[str, JsonValue]:
-    dct = type(d)
-    scd = get_serial_class_data(dct)
+def _to_string_value(value: Any) -> str:
+    assert isinstance(value, (Decimal, int))
+    return str(value)
 
-    def key_convert(key: str) -> str:
+
+@dataclasses.dataclass(kw_only=True)
+class DataclassConversions:
+    key_conversions: dict[str, str]
+    value_conversion_functions: dict[str, Callable[[Any], JsonValue]]
+
+
+@functools.lru_cache(maxsize=10000)
+def _get_dataclass_conversion_lookups(dataclass_type: Any) -> DataclassConversions:
+    scd = get_serial_class_data(dataclass_type)
+
+    key_conversions: dict[str, str] = {}
+    value_conversion_functions: dict[str, Callable[[Any], JsonValue]] = {}
+
+    for field in dataclasses.fields(dataclass_type):
+        key = field.name
         if scd.has_unconverted_key(key):
-            return key
-        return key_convert_to_camelcase(key)
+            key_conversions[key] = key
+        else:
+            key_conversions[key] = key_convert_to_camelcase(key)
 
-    def value_convert(key: str, value: Any) -> JsonValue:
         if scd.has_to_string_value(key):
-            # Limit to types we know we need to support to avoid surprises
-            # Generics, like List/Dict would need to be per-value stringified
-            assert isinstance(value, (Decimal, int))
-            return str(value)
-        if scd.has_unconverted_value(key):
-            return value  # type: ignore[no-any-return]
-        return serialize_for_api(value)  # type: ignore[no-any-return,unused-ignore]
+            value_conversion_functions[key] = _to_string_value
+        elif scd.has_unconverted_value(key):
+            value_conversion_functions[key] = identity
+        else:
+            value_conversion_functions[key] = serialize_for_api
+
+    return DataclassConversions(
+        key_conversions=key_conversions,
+        value_conversion_functions=value_conversion_functions,
+    )
+
 
+def _convert_dataclass(d: Any) -> dict[str, JsonValue]:
+    conversions = _get_dataclass_conversion_lookups(type(d))  # type: ignore[arg-type]
     return {
-        key_convert(k): (value_convert(k, v) if v is not None else None)
+        conversions.key_conversions[k]: (
+            conversions.value_conversion_functions[k](v) if v is not None else None
+        )
         for k, v in d.__dict__.items()
         if v != MISSING_SENTRY
     }
@@ -126,6 +165,10 @@ def serialize_for_api(obj: Any) -> JsonValue:
     Use the CachedParser.parse_api to parse this data.
     """
     serialization_type = get_serialization_type(type(obj))  # type: ignore
+    if (
+        serialization_type == SerializationType.UNKNOWN
+    ):  # performance optimization to not do function lookup
+        return obj  # type: ignore
     return _CONVERSION_SERIALIZATION_FUNCS[serialization_type](obj)
 
 
@@ -153,6 +196,10 @@ def serialize_for_storage(obj: Any) -> JsonValue:
     Use the CachedParser.parse_storage to parse this data.
     """
     serialization_type = get_serialization_type(type(obj))  # type: ignore
+    if (
+        serialization_type == SerializationType.UNKNOWN
+    ):  # performance optimization to not do function lookup
+        return obj  # type: ignore
     return _SERIALIZATION_FUNCS[serialization_type](obj)
 
 

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/pkgs/type_spec/emit_typescript_util.py

@@ -18,7 +18,7 @@ class EmitTypescriptContext:
 
 
 def ts_type_name(name: str) -> str:
-    return "".join([x.capitalize() for x in name.split("_")])
+    return "".join([x.title() for x in name.split("_")])
 
 
 def resolve_namespace_ref(namespace: builder.SpecNamespace) -> str:
@@ -29,4 +29,4 @@ def ts_name(name: str, name_case: builder.NameCase) -> str:
     if name_case == builder.NameCase.preserve:
         return name
     bits = util.split_any_name(name)
-    return "".join([bits[0], *[x.capitalize() for x in bits[1:]]])
+    return "".join([bits[0], *[x.title() for x in bits[1:]]])

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/uncountable/types/__init__.py

@@ -50,6 +50,7 @@ from . import inputs as inputs_t
 from .api.entity import list_entities as list_entities_t
 from .api.id_source import list_id_source as list_id_source_t
 from .api.entity import lock_entity as lock_entity_t
+from .api.recipes import lock_recipes as lock_recipes_t
 from .api.id_source import match_id_source as match_id_source_t
 from . import outputs as outputs_t
 from . import permissions as permissions_t
@@ -75,6 +76,7 @@ from .api.inputs import set_input_subcategories as set_input_subcategories_t
 from .api.inputs import set_intermediate_type as set_intermediate_type_t
 from .api.recipes import set_recipe_inputs as set_recipe_inputs_t
 from .api.recipes import set_recipe_metadata as set_recipe_metadata_t
+from .api.recipes import set_recipe_output_annotations as set_recipe_output_annotations_t
 from .api.recipes import set_recipe_outputs as set_recipe_outputs_t
 from .api.recipes import set_recipe_tags as set_recipe_tags_t
 from .api.entity import set_values as set_values_t
@@ -136,6 +138,7 @@ __all__: list[str] = [
     "list_entities_t",
     "list_id_source_t",
     "lock_entity_t",
+    "lock_recipes_t",
     "match_id_source_t",
     "outputs_t",
     "permissions_t",
@@ -161,6 +164,7 @@ __all__: list[str] = [
     "set_intermediate_type_t",
     "set_recipe_inputs_t",
     "set_recipe_metadata_t",
+    "set_recipe_output_annotations_t",
     "set_recipe_outputs_t",
     "set_recipe_tags_t",
     "set_values_t",

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/uncountable/types/api/entity/list_entities.py

@@ -32,8 +32,8 @@ ENDPOINT_PATH = "api/external/entity/external_list_entities"
 )
 @dataclass(kw_only=True)
 class Arguments:
-    entity_type: entity_t.EntityType
     config_reference: str
+    entity_type: typing.Optional[entity_t.EntityType] = None
     attributes: typing.Optional[dict[OpaqueKey, base_t.JsonValue]] = None
     offset: typing.Optional[typing.Optional[int]] = None
     limit: typing.Optional[typing.Optional[int]] = None

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/uncountable/types/api/id_source/match_id_source.py

@@ -8,7 +8,6 @@ import typing  # noqa: F401
 import datetime  # noqa: F401
 from decimal import Decimal  # noqa: F401
 from dataclasses import dataclass
-from pkgs.serialization import serial_class
 from ... import base as base_t
 from ... import id_source as id_source_t
 
@@ -39,9 +38,6 @@ class Match:
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
-@serial_class(
-    unconverted_values={"results"},
-)
 @dataclass(kw_only=True)
 class Data:
     results: list[Match]

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/uncountable/types/api/recipes/edit_recipe_inputs.py

@@ -24,8 +24,10 @@ __all__: list[str] = [
     "RecipeInputEdit",
     "RecipeInputEditAddInput",
     "RecipeInputEditBase",
+    "RecipeInputEditChangeBasisViewed",
     "RecipeInputEditClearInputs",
     "RecipeInputEditInputBase",
+    "RecipeInputEditSetLot",
     "RecipeInputEditType",
     "RecipeInputEditUpdateAnnotations",
     "RecipeInputEditUpsertInput",
@@ -41,6 +43,8 @@ class RecipeInputEditType(StrEnum):
     UPSERT_INPUT = "upsert_input"
     ADD_INPUT = "add_input"
     UPDATE_ANNOTATIONS = "update_annotations"
+    SET_LOT = "set_lot"
+    CHANGE_BASIS = "change_basis"
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -72,6 +76,15 @@ class RecipeInputEditInputBase(RecipeInputEditBase):
     calculation_key: typing.Optional[identifier_t.IdentifierKey] = None
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclass(kw_only=True)
+class RecipeInputEditChangeBasisViewed(RecipeInputEditInputBase):
+    type: typing.Literal[RecipeInputEditType.CHANGE_BASIS] = RecipeInputEditType.CHANGE_BASIS
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     parse_require={"type"},
@@ -91,6 +104,17 @@ class RecipeInputEditAddInput(RecipeInputEditInputBase):
     type: typing.Literal[RecipeInputEditType.ADD_INPUT] = RecipeInputEditType.ADD_INPUT
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclass(kw_only=True)
+class RecipeInputEditSetLot(RecipeInputEditBase):
+    type: typing.Literal[RecipeInputEditType.SET_LOT] = RecipeInputEditType.SET_LOT
+    ingredient_key: identifier_t.IdentifierKey
+    ingredient_lot_recipe_key: identifier_t.IdentifierKey
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     to_string_values={"lower_value", "upper_value"},
@@ -116,7 +140,7 @@ class RecipeInputEditUpdateAnnotations(RecipeInputEditInputBase):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 RecipeInputEdit = typing.Annotated[
-    typing.Union[RecipeInputEditClearInputs, RecipeInputEditUpsertInput, RecipeInputEditAddInput, RecipeInputEditUpdateAnnotations],
+    typing.Union[RecipeInputEditClearInputs, RecipeInputEditUpsertInput, RecipeInputEditAddInput, RecipeInputEditUpdateAnnotations, RecipeInputEditSetLot, RecipeInputEditChangeBasisViewed],
     serial_union_annotation(
         discriminator="type",
         discriminator_map={
@@ -124,6 +148,8 @@ RecipeInputEdit = typing.Annotated[
             "upsert_input": RecipeInputEditUpsertInput,
             "add_input": RecipeInputEditAddInput,
             "update_annotations": RecipeInputEditUpdateAnnotations,
+            "set_lot": RecipeInputEditSetLot,
+            "change_basis": RecipeInputEditChangeBasisViewed,
         },
     ),
 ]

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/uncountable/types/api/recipes/get_recipes_data.py

@@ -15,7 +15,6 @@ from ... import inputs as inputs_t
 from ... import outputs as outputs_t
 from ... import recipe_metadata as recipe_metadata_t
 from ... import recipe_tags as recipe_tags_t
-from ... import users as users_t
 from ... import workflows as workflows_t
 
 __all__: list[str] = [
@@ -97,7 +96,6 @@ class RecipeInput:
     quantity_json: base_t.JsonValue
     curve_id: typing.Optional[base_t.ObjectId]
     actual_quantity_json: base_t.JsonValue
-    input_type: str
     behavior: str
     quantity_dec: typing.Optional[Decimal] = None
     actual_quantity_dec: typing.Optional[Decimal] = None
@@ -170,7 +168,6 @@ class Data:
     inputs: list[inputs_t.SimpleInput]
     outputs: list[outputs_t.SimpleOutput]
     output_conditions: list[SimpleOutputCondition]
-    users: list[users_t.SimpleUser]
     recipe_tags: list[recipe_tags_t.SimpleRecipeTag]
     experiment_groups: list[experiment_groups_t.SimpleExperimentGroup]
 # DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.35/uncountable/types/api/recipes/lock_recipes.py

@@ -0,0 +1,89 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402
+# 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
+from dataclasses import dataclass
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from ... import identifier as identifier_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "RecipeLock",
+    "RecipeLockAll",
+    "RecipeLockBase",
+    "RecipeLockInputs",
+    "RecipeLockType",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/recipes/lock_recipes"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class RecipeLockType(StrEnum):
+    ALL = "all"
+    INPUTS_ONLY = "inputs_only"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class RecipeLockBase:
+    recipe: identifier_t.IdentifierKey
+    globally_removable: bool
+    type: RecipeLockType
+    lock_samples: typing.Optional[bool] = None
+    comments: typing.Optional[str] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclass(kw_only=True)
+class RecipeLockAll(RecipeLockBase):
+    type: typing.Literal[RecipeLockType.ALL] = RecipeLockType.ALL
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclass(kw_only=True)
+class RecipeLockInputs(RecipeLockBase):
+    type: typing.Literal[RecipeLockType.INPUTS_ONLY] = RecipeLockType.INPUTS_ONLY
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+RecipeLock = typing.Annotated[
+    typing.Union[RecipeLockAll, RecipeLockInputs],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "all": RecipeLockAll,
+            "inputs_only": RecipeLockInputs,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Arguments:
+    locks: list[RecipeLock]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Data:
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.35/uncountable/types/api/recipes/set_recipe_output_annotations.py

@@ -0,0 +1,102 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# flake8: noqa: F821
+# ruff: noqa: E402
+# 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
+from dataclasses import dataclass
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from ... import base as base_t
+from ... import identifier as identifier_t
+
+__all__: list[str] = [
+    "AnnotationEdit",
+    "AnnotationUpdateType",
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+    "RecipeOutputEditBase",
+    "RecipeOutputMergeAnnotations",
+    "RecipeOutputReplaceAnnotations",
+    "RecipeOutputUpdateAnnotations",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/recipes/set_recipe_output_annotations"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    to_string_values={"lower_value", "upper_value"},
+)
+@dataclass(kw_only=True)
+class AnnotationEdit:
+    annotation_type_key: identifier_t.IdentifierKey
+    lower_value: typing.Optional[Decimal] = None
+    upper_value: typing.Optional[Decimal] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class AnnotationUpdateType(StrEnum):
+    MERGE = "merge"
+    REPLACE = "replace"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class RecipeOutputEditBase:
+    recipe_id: base_t.ObjectId
+    output_id: base_t.ObjectId
+    experiment_num: int
+    annotations: list[AnnotationEdit]
+    condition_id: typing.Optional[base_t.ObjectId] = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclass(kw_only=True)
+class RecipeOutputMergeAnnotations(RecipeOutputEditBase):
+    type: typing.Literal[AnnotationUpdateType.MERGE] = AnnotationUpdateType.MERGE
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    parse_require={"type"},
+)
+@dataclass(kw_only=True)
+class RecipeOutputReplaceAnnotations(RecipeOutputEditBase):
+    type: typing.Literal[AnnotationUpdateType.REPLACE] = AnnotationUpdateType.REPLACE
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+RecipeOutputUpdateAnnotations = typing.Annotated[
+    typing.Union[RecipeOutputMergeAnnotations, RecipeOutputReplaceAnnotations],
+    serial_union_annotation(
+        discriminator="type",
+        discriminator_map={
+            "merge": RecipeOutputMergeAnnotations,
+            "replace": RecipeOutputReplaceAnnotations,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Arguments:
+    updates: list[RecipeOutputUpdateAnnotations]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@dataclass(kw_only=True)
+class Data:
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.33 → uncountablepythonsdk-0.0.35}/uncountable/types/client_base.py

@@ -49,6 +49,7 @@ from uncountable.types import identifier as identifier_t
 import uncountable.types.api.entity.list_entities as list_entities_t
 import uncountable.types.api.id_source.list_id_source as list_id_source_t
 import uncountable.types.api.entity.lock_entity as lock_entity_t
+import uncountable.types.api.recipes.lock_recipes as lock_recipes_t
 import uncountable.types.api.id_source.match_id_source as match_id_source_t
 from uncountable.types import permissions as permissions_t
 from uncountable.types import post_base as post_base_t
@@ -67,6 +68,7 @@ import uncountable.types.api.inputs.set_input_subcategories as set_input_subcate
 import uncountable.types.api.inputs.set_intermediate_type as set_intermediate_type_t
 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_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
@@ -826,15 +828,15 @@ class ClientMethods(ABC):
     def list_entities(
         self,
         *,
-        entity_type: entity_t.EntityType,
         config_reference: str,
+        entity_type: typing.Optional[entity_t.EntityType] = None,
         attributes: typing.Optional[dict[OpaqueKey, base_t.JsonValue]] = None,
         offset: typing.Optional[typing.Optional[int]] = None,
         limit: typing.Optional[typing.Optional[int]] = None,
     ) -> list_entities_t.Data:
         """Uses a structured loading configuration to list entities in the system
 
-        :param entity_type: The type of the entities requested, e.g. lab_request, recipe
+        :param entity_type: DEPRECATED: The type of the entities requested, e.g. lab_request, recipe
         :param config_reference: The configuration reference name for the listing config
         :param attributes: Attributes to pass to the configuration for parameterizing filters
         :param offset: Used for pagination. Pagination is done based on the sorting of the config. [Pagination More Info](#pagination)
@@ -906,6 +908,25 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=lock_entity_t.Data)
 
+    def lock_recipes(
+        self,
+        *,
+        locks: list[lock_recipes_t.RecipeLock],
+    ) -> lock_recipes_t.Data:
+        """Lock experiments. Experiments will require unlocking to be editable. Edits to the experiments are blocked while they are locked.
+
+        :param locks: The recipe locks to apply, a maximum of 100 can be sent
+        """
+        args = lock_recipes_t.Arguments(
+            locks=locks,
+        )
+        api_request = APIRequest(
+            method=lock_recipes_t.ENDPOINT_METHOD,
+            endpoint=lock_recipes_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=lock_recipes_t.Data)
+
     def match_id_source(
         self,
         *,
@@ -1172,6 +1193,25 @@ class ClientMethods(ABC):
         )
         return self.do_request(api_request=api_request, return_type=set_recipe_metadata_t.Data)
 
+    def set_recipe_output_annotations(
+        self,
+        *,
+        updates: list[set_recipe_output_annotations_t.RecipeOutputUpdateAnnotations],
+    ) -> set_recipe_output_annotations_t.Data:
+        """Update annotations for an experiments outputs
+
+        :param updates: The output edits to perform. Must be at most 100 entries long
+        """
+        args = set_recipe_output_annotations_t.Arguments(
+            updates=updates,
+        )
+        api_request = APIRequest(
+            method=set_recipe_output_annotations_t.ENDPOINT_METHOD,
+            endpoint=set_recipe_output_annotations_t.ENDPOINT_PATH,
+            args=args,
+        )
+        return self.do_request(api_request=api_request, return_type=set_recipe_output_annotations_t.Data)
+
     def set_recipe_outputs(
         self,
         *,