UncountablePythonSDK 0.0.145__tar.gz → 0.0.147__tar.gz

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.145
+Version: 0.0.147
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -39,14 +39,14 @@ Requires-Dist: grpcio==1.67.1
 Requires-Dist: protobuf>=4.21.1
 Requires-Dist: azure-storage-blob==12.*
 Requires-Dist: boto3-stubs[essential,ses]==1.*
-Requires-Dist: msgspec==0.19.*
+Requires-Dist: msgspec<0.21,>=0.19
 Requires-Dist: tabulate==0.9.0
-Requires-Dist: types-psutil~=7.0.0
+Requires-Dist: types-psutil~=7.1.3
 Requires-Dist: psutil>=5.9.0
 Provides-Extra: test
 Requires-Dist: mypy==1.*; extra == "test"
 Requires-Dist: ruff==0.*; extra == "test"
-Requires-Dist: pytest==8.*; extra == "test"
+Requires-Dist: pytest>=8; extra == "test"
 Requires-Dist: coverage[toml]==7.*; extra == "test"
 Requires-Dist: pytest-cov==7.*; extra == "test"
 Requires-Dist: pytest-xdist==3.*; extra == "test"

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/UncountablePythonSDK.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.145
+Version: 0.0.147
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git
@@ -39,14 +39,14 @@ Requires-Dist: grpcio==1.67.1
 Requires-Dist: protobuf>=4.21.1
 Requires-Dist: azure-storage-blob==12.*
 Requires-Dist: boto3-stubs[essential,ses]==1.*
-Requires-Dist: msgspec==0.19.*
+Requires-Dist: msgspec<0.21,>=0.19
 Requires-Dist: tabulate==0.9.0
-Requires-Dist: types-psutil~=7.0.0
+Requires-Dist: types-psutil~=7.1.3
 Requires-Dist: psutil>=5.9.0
 Provides-Extra: test
 Requires-Dist: mypy==1.*; extra == "test"
 Requires-Dist: ruff==0.*; extra == "test"
-Requires-Dist: pytest==8.*; extra == "test"
+Requires-Dist: pytest>=8; extra == "test"
 Requires-Dist: coverage[toml]==7.*; extra == "test"
 Requires-Dist: pytest-cov==7.*; extra == "test"
 Requires-Dist: pytest-xdist==3.*; extra == "test"

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/UncountablePythonSDK.egg-info/requires.txt

@@ -21,15 +21,15 @@ grpcio==1.67.1
 protobuf>=4.21.1
 azure-storage-blob==12.*
 boto3-stubs[essential,ses]==1.*
-msgspec==0.19.*
+msgspec<0.21,>=0.19
 tabulate==0.9.0
-types-psutil~=7.0.0
+types-psutil~=7.1.3
 psutil>=5.9.0
 
 [test]
 mypy==1.*
 ruff==0.*
-pytest==8.*
+pytest>=8
 coverage[toml]==7.*
 pytest-cov==7.*
 pytest-xdist==3.*

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/pkgs/argument_parser/argument_parser.py

@@ -296,7 +296,9 @@ def _build_parser_inner(
             )
 
     if dataclasses.is_dataclass(parsed_type):
-        return _build_parser_dataclass(parsed_type, context, type_var_map)
+        dc_merged_map: dict[typing.TypeVar, type] = {}
+        _resolve_base_type_vars(cls=parsed_type, type_var_map=dc_merged_map)
+        return _build_parser_dataclass(parsed_type, context, dc_merged_map)
 
     origin = typing.get_origin(parsed_type)
     type_args = typing.get_args(parsed_type)
@@ -365,7 +367,7 @@ def _build_parser_inner(
         args = typing.get_args(parsed_type)
         sorted_args = sorted(
             args,
-            key=lambda subtype: (0 if subtype == type(None) else 1),  # noqa: E721
+            key=lambda subtype: 0 if subtype == type(None) else 1,  # noqa: E721
         )
         arg_parsers = [
             _build_parser_inner(arg, context, type_var_map) for arg in sorted_args

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/pkgs/filesystem_utils/_gdrive_session.py

@@ -84,7 +84,8 @@ def list_gdrive_files(
     print("Listing files", end="", flush=True)
     paginated_files_in_folder = [
         (
-            gdrive_connection.files()
+            gdrive_connection
+            .files()
             .list(
                 q=query,
                 corpora="allDrives",
@@ -97,7 +98,8 @@ def list_gdrive_files(
     while paginated_files_in_folder[-1].get("nextPageToken") is not None:
         print(".", end="", flush=True)
         paginated_files_in_folder.append(
-            gdrive_connection.files()
+            gdrive_connection
+            .files()
             .list(
                 q=query,
                 corpora="allDrives",
@@ -152,7 +154,8 @@ def move_gdrive_file(
 ) -> None:
     # Retrieve the existing parents to remove
     file = (
-        gdrive_connection.files()
+        gdrive_connection
+        .files()
         .get(fileId=src_file_id, fields="parents, name", supportsTeamDrives=True)
         .execute()
     )

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/pyproject.toml

@@ -74,9 +74,9 @@ dependencies = [
     "protobuf >= 4.21.1",
     "azure-storage-blob == 12.*",
     "boto3-stubs[essential,ses]==1.*",
-    "msgspec == 0.19.*",
+    "msgspec>=0.19,<0.21",
     "tabulate == 0.9.0",
-    "types-psutil~=7.0.0",
+    "types-psutil~=7.1.3",
     "psutil>=5.9.0",
 ]
 dynamic = ["version"]
@@ -89,7 +89,7 @@ test = [
     "mypy == 1.*",
     "ruff == 0.*",
     # Testing
-    "pytest ==8.*",
+    "pytest >=8",
     "coverage[toml] ==7.*",
     "pytest-cov ==7.*",
     "pytest-xdist == 3.*",

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/uncountable/integration/telemetry.py

@@ -1,7 +1,6 @@
 import functools
 import json
 import os
-import resource
 import threading
 import time
 import traceback
@@ -36,6 +35,15 @@ from uncountable.core.environment import (
 )
 from uncountable.types import base_t, job_definition_t
 
+# Conditionally import resource module - not available on Windows
+if typing.TYPE_CHECKING:
+    import resource as resource
+else:
+    try:  # type: ignore[unreachable]
+        import resource as resource
+    except ImportError:
+        resource = None
+
 
 def _cast_attributes(attributes: dict[str, base_t.JsonValue]) -> Attributes:
     return cast(Attributes, attributes)
@@ -58,8 +66,8 @@ def get_otel_resource() -> Resource:
     unc_env = get_server_env()
     if unc_env is not None:
         attributes["deployment.environment"] = unc_env
-    resource = Resource.create(attributes=_cast_attributes(attributes))
-    return resource
+    otel_resource = Resource.create(attributes=_cast_attributes(attributes))
+    return otel_resource
 
 
 @functools.cache
@@ -178,17 +186,18 @@ class Logger:
         *,
         message: str | None = None,
         attributes: Attributes | None = None,
+        severity: LogSeverity = LogSeverity.ERROR,
     ) -> None:
         traceback_str = "".join(traceback.format_exception(exception))
         patched_attributes = self._patch_attributes(
-            message=message, severity=LogSeverity.ERROR, attributes=attributes
+            message=message, severity=severity, attributes=attributes
         )
         self.current_span.record_exception(
             exception=exception, attributes=patched_attributes
         )
-        self.log_error(
-            message=f"error: {message}\nexception: {exception}{traceback_str}",
-            attributes=patched_attributes,
+        log_message = f"error: {message}\nexception: {exception}{traceback_str}"
+        self._emit_log(
+            message=log_message, severity=severity, attributes=patched_attributes
         )
 
     @contextmanager
@@ -251,14 +260,15 @@ class PerJobResourceTracker:
         self.stop()
         stats = dict(self.summary())
         if isinstance(exc_value, MemoryError):
-            limit, _ = resource.getrlimit(resource.RLIMIT_AS)
-            self.logger.log_exception(
-                exc_value,
-                message=(
-                    f"Job '{self.logger.job_definition.name}' (ID={self.logger.job_definition.id}) "
-                    f"exceeded memory limit ({limit / (1024**3):.2f} GB)."
-                ),
-            )
+            if resource is not None:
+                limit, _ = resource.getrlimit(resource.RLIMIT_AS)
+                self.logger.log_exception(
+                    exc_value,
+                    message=(
+                        f"Job '{self.logger.job_definition.name}' (ID={self.logger.job_definition.id}) "
+                        f"exceeded memory limit ({limit / (1024**3):.2f} GB)."
+                    ),
+                )
             return
         self.logger.log_info("Job resource usage summary", attributes=stats)
         return

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/uncountable/types/api/notebooks/add_notebook_content.py

@@ -20,11 +20,16 @@ __all__: list[str] = [
     "ENDPOINT_METHOD",
     "ENDPOINT_PATH",
     "InputContent",
+    "InputContentTable",
     "InputContentTextMarkdown",
     "InputContentType",
     "Location",
     "LocationAppendToFirstPage",
     "LocationType",
+    "TableCell",
+    "TableCellContent",
+    "TableCellContentText",
+    "TableCellContentType",
 ]
 
 ENDPOINT_METHOD = "POST"
@@ -68,10 +73,66 @@ Location = typing.Annotated[
 @serial_string_enum(
     labels={
         "text_markdown": "Text Markdown",
+        "table": "Table",
     },
 )
 class InputContentType(StrEnum):
     TEXT_MARKDOWN = "text_markdown"
+    TABLE = "table"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_string_enum(
+    labels={
+        "text": "Text",
+    },
+)
+class TableCellContentType(StrEnum):
+    TEXT = "text"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.add_notebook_content.TableCellContentText",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class TableCellContentText:
+    type: typing.Literal[TableCellContentType.TEXT] = TableCellContentType.TEXT
+    text: str
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+TableCellContent = typing.Annotated[
+    typing.Union[TableCellContentText],
+    serial_union_annotation(
+    named_type_path="sdk.api.notebooks.add_notebook_content.TableCellContent",
+        discriminator="type",
+        discriminator_map={
+            "text": TableCellContentText,
+        },
+    ),
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.add_notebook_content.TableCell",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class TableCell:
+    content: TableCellContent
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.notebooks.add_notebook_content.InputContentTable",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class InputContentTable:
+    type: typing.Literal[InputContentType.TABLE] = InputContentType.TABLE
+    rows: list[list[TableCell]]
 
 
 # DO NOT MODIFY -- This file is generated by type_spec
@@ -87,12 +148,13 @@ class InputContentTextMarkdown:
 
 # DO NOT MODIFY -- This file is generated by type_spec
 InputContent = typing.Annotated[
-    typing.Union[InputContentTextMarkdown],
+    InputContentTextMarkdown | InputContentTable,
     serial_union_annotation(
     named_type_path="sdk.api.notebooks.add_notebook_content.InputContent",
         discriminator="type",
         discriminator_map={
             "text_markdown": InputContentTextMarkdown,
+            "table": InputContentTable,
         },
     ),
 ]

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/uncountable/types/api/recipes/edit_recipe_inputs.py

@@ -35,6 +35,7 @@ __all__: list[str] = [
     "RecipeInputEditInputBase",
     "RecipeInputEditPlaceholder",
     "RecipeInputEditSetLot",
+    "RecipeInputEditSetPropagatedPercentage",
     "RecipeInputEditSetRole",
     "RecipeInputEditType",
     "RecipeInputEditUpdateAnnotations",
@@ -54,6 +55,7 @@ class RecipeInputEditType(StrEnum):
     ADD_INPUT = "add_input"
     UPDATE_ANNOTATIONS = "update_annotations"
     SET_LOT = "set_lot"
+    SET_PROPAGATED_PERCENTAGE = "set_propagated_percentage"
     CHANGE_BASIS = "change_basis"
     ADD_INSTRUCTIONS = "add_instructions"
     SET_ROLE = "set_role"
@@ -147,6 +149,20 @@ class RecipeInputEditSetLot(RecipeInputEditBase):
     mix_step_match_strategy: MixStepMatchStrategy = MixStepMatchStrategy.FIRST
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputEditSetPropagatedPercentage",
+    to_string_values={"propagated_percentage"},
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class RecipeInputEditSetPropagatedPercentage(RecipeInputEditBase):
+    type: typing.Literal[RecipeInputEditType.SET_PROPAGATED_PERCENTAGE] = RecipeInputEditType.SET_PROPAGATED_PERCENTAGE
+    ingredient_key: identifier_t.IdentifierKey
+    propagated_percentage: Decimal
+    mix_step_match_strategy: MixStepMatchStrategy = MixStepMatchStrategy.FIRST
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputEditAddInstructions",
@@ -255,7 +271,7 @@ class RecipeInputEditPlaceholder(RecipeInputEditBase):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 RecipeInputEdit = typing.Annotated[
-    RecipeInputEditClearInputs | RecipeInputEditUpsertInput | RecipeInputEditAddInput | RecipeInputEditUpdateAnnotations | RecipeInputEditSetLot | RecipeInputEditChangeBasisViewed | RecipeInputEditAddInstructions | RecipeInputEditSetRole | RecipeInputEditPlaceholder,
+    RecipeInputEditClearInputs | RecipeInputEditUpsertInput | RecipeInputEditAddInput | RecipeInputEditUpdateAnnotations | RecipeInputEditSetLot | RecipeInputEditSetPropagatedPercentage | RecipeInputEditChangeBasisViewed | RecipeInputEditAddInstructions | RecipeInputEditSetRole | RecipeInputEditPlaceholder,
     serial_union_annotation(
     named_type_path="sdk.api.recipes.edit_recipe_inputs.RecipeInputEdit",
         discriminator="type",
@@ -265,6 +281,7 @@ RecipeInputEdit = typing.Annotated[
             "add_input": RecipeInputEditAddInput,
             "update_annotations": RecipeInputEditUpdateAnnotations,
             "set_lot": RecipeInputEditSetLot,
+            "set_propagated_percentage": RecipeInputEditSetPropagatedPercentage,
             "change_basis": RecipeInputEditChangeBasisViewed,
             "add_instructions": RecipeInputEditAddInstructions,
             "set_role": RecipeInputEditSetRole,

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/uncountable/types/api/recipes/get_recipes_data.py

@@ -208,6 +208,7 @@ class Recipe:
     recipe_lock_status: RecipeLockStatus
     creating_user_id: base_t.ObjectId | None = None
     barcode_value: str | None = None
+    assigned_spec_id: base_t.ObjectId | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/uncountable/types/entity_t.py

@@ -100,6 +100,8 @@ __all__: list[str] = [
         "inventory_amount": "Inventory Item",
         "inventory_limit": "Inventory Limit",
         "inventory_history": "Inventory History",
+        "inventory_reconciliation_action": "Inventory Reconciliation Action",
+        "inventory_reconciliation_session": "Inventory Reconciliation Session",
         "lab_location": "Lab Location",
         "lab_request": "Lab Request",
         "license": "License",
@@ -313,6 +315,8 @@ class EntityType(StrEnum):
     INVENTORY_AMOUNT = "inventory_amount"
     INVENTORY_LIMIT = "inventory_limit"
     INVENTORY_HISTORY = "inventory_history"
+    INVENTORY_RECONCILIATION_ACTION = "inventory_reconciliation_action"
+    INVENTORY_RECONCILIATION_SESSION = "inventory_reconciliation_session"
     LAB_LOCATION = "lab_location"
     LAB_REQUEST = "lab_request"
     LICENSE = "license"

{uncountablepythonsdk-0.0.145 → uncountablepythonsdk-0.0.147}/uncountable/types/job_definition_t.py

@@ -191,6 +191,7 @@ class JobDefinitionBase:
     enabled: bool = True
     logging_settings: JobLoggingSettings | None = None
     subqueue_name: str | None = None
+    tags: list[str] | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec