UncountablePythonSDK 0.0.151__tar.gz → 0.0.153__tar.gz

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/.github/workflows/documentation.yml

@@ -17,7 +17,7 @@ jobs:
       - name: Set up UV
         uses: astral-sh/setup-uv@f0ec1fc3b38f5e7cd731bb6ce540c5af426746bb
         with:
-          version: "0.9.25"
+          version: "0.10.0"
           activate-environment: true
           enable-cache: true
       - name: Install dependencies

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/.github/workflows/publish.yml

@@ -26,7 +26,7 @@ jobs:
       - name: Set up UV
         uses: astral-sh/setup-uv@f0ec1fc3b38f5e7cd731bb6ce540c5af426746bb
         with:
-          version: "0.9.25"
+          version: "0.10.0"
           activate-environment: true
           enable-cache: true
       - name: Install build dependencies

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.151
+Version: 0.0.153
 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.151 → uncountablepythonsdk-0.0.153}/UncountablePythonSDK.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.151
+Version: 0.0.153
 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.151 → uncountablepythonsdk-0.0.153}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -239,6 +239,8 @@ uncountable/types/recipe_metadata.py
 uncountable/types/recipe_metadata_t.py
 uncountable/types/recipe_output_metadata.py
 uncountable/types/recipe_output_metadata_t.py
+uncountable/types/recipe_step_relationships.py
+uncountable/types/recipe_step_relationships_t.py
 uncountable/types/recipe_tags.py
 uncountable/types/recipe_tags_t.py
 uncountable/types/recipe_workflow_steps.py
@@ -320,6 +322,7 @@ uncountable/types/api/integrations/publish_realtime_data.py
 uncountable/types/api/integrations/push_notification.py
 uncountable/types/api/integrations/register_sockets_token.py
 uncountable/types/api/listing/__init__.py
+uncountable/types/api/listing/export_listing.py
 uncountable/types/api/listing/fetch_listing.py
 uncountable/types/api/material_families/__init__.py
 uncountable/types/api/material_families/update_entity_material_families.py

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/pkgs/filesystem_utils/_sftp_session.py

@@ -68,7 +68,7 @@ class SFTPSession(FileSystemSession):
         super().__init__()
         self.host: str = sftp_config.ip
         self.username: str = sftp_config.username
-        self.key_file: str | paramiko.RSAKey | None = (
+        self.key_file: str | paramiko.PKey | None = (
             sftp_config.pem_path
             if sftp_config.pem_path is not None
             else sftp_config.pem_key

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/pkgs/filesystem_utils/file_type_utils.py

@@ -61,7 +61,7 @@ class FileSystemSFTPConfig:
     ip: str
     username: str
     pem_path: str | None
-    pem_key: paramiko.RSAKey | None = None
+    pem_key: paramiko.PKey | None = None
     password: str | None = None
     valid_extensions: tuple[str] | None = None
     recursive: bool = True

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/pkgs/type_spec/emit_python.py

@@ -240,7 +240,11 @@ def _emit_value(
             value,
             indent=indent,
         )
+
     elif isinstance(stype, builder.SpecTypeDefnObject):
+        if stype.name is builder.BaseTypeName.s_none:
+            assert value is None
+            return base_name_map[stype.name]
         assert isinstance(value, dict), (
             f"Expected dict value for {stype.name} but got {value}"
         )

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/core/environment.py

@@ -39,3 +39,7 @@ def get_integration_envs() -> list[integration_server_t.IntegrationEnvironment]:
         integration_server_t.IntegrationEnvironment(env)
         for env in os.environ.get("UNC_INTEGRATION_ENVS", "prod").split(",")
     ]
+
+
+def get_max_job_workers() -> int:
+    return int(os.environ.get("UNC_MAX_JOB_WORKERS", "5"))

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/integration/queue_runner/command_server/command_server.py

@@ -148,12 +148,16 @@ async def serve(command_queue: CommandQueue, datastore: DatastoreSqlite) -> None
                     f"Invalid status: '{request.status}'",
                 )
 
-            submitted_from = None
-            submitted_to = None
-            if request.submitted_from:
-                submitted_from = request.submitted_from.ToDatetime(tzinfo=timezone.utc)
-            if request.submitted_to:
-                submitted_to = request.submitted_to.ToDatetime(tzinfo=timezone.utc)
+            submitted_from = (
+                request.submitted_from.ToDatetime(tzinfo=timezone.utc)
+                if request.HasField("submitted_from")
+                else None
+            )
+            submitted_to = (
+                request.submitted_to.ToDatetime(tzinfo=timezone.utc)
+                if request.HasField("submitted_to")
+                else None
+            )
 
             queued_job_metadata = datastore.list_queued_job_metadata(
                 status=job_status,

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/integration/queue_runner/job_scheduler.py

@@ -8,6 +8,7 @@ from dataclasses import dataclass
 
 from opentelemetry.trace import get_current_span
 
+from uncountable.core.environment import get_max_job_workers
 from uncountable.integration.queue_runner.command_server import (
     CommandEnqueueJob,
     CommandEnqueueJobResponse,
@@ -31,8 +32,6 @@ from uncountable.types import job_definition_t, queued_job_t
 
 from .types import RESTART_EXIT_CODE, ResultQueue, ResultTask
 
-_MAX_JOB_WORKERS = 5
-
 
 @dataclass(kw_only=True, frozen=True)
 class JobListenerKey:
@@ -97,7 +96,9 @@ async def start_scheduler(
     logger = Logger(get_current_span())
     result_queue: ResultQueue = asyncio.Queue()
 
-    with ProcessPoolExecutor(max_workers=_MAX_JOB_WORKERS) as process_pool:
+    with ProcessPoolExecutor(
+        max_workers=get_max_job_workers(), max_tasks_per_child=1
+    ) as process_pool:
         job_worker_lookup = _start_workers(
             process_pool, result_queue, datastore=datastore
         )

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/types/__init__.py

@@ -40,6 +40,7 @@ from .api.batch import execute_batch_load_async as execute_batch_load_async_t
 from . import experiment_groups_t as experiment_groups_t
 from .api.runsheet import export_default_runsheet as export_default_runsheet_t
 from .api.entity import export_entities as export_entities_t
+from .api.listing import export_listing as export_listing_t
 from . import exports_t as exports_t
 from .api.listing import fetch_listing as fetch_listing_t
 from . import field_values_t as field_values_t
@@ -98,6 +99,7 @@ from . import recipe_inputs_t as recipe_inputs_t
 from . import recipe_links_t as recipe_links_t
 from . import recipe_metadata_t as recipe_metadata_t
 from . import recipe_output_metadata_t as recipe_output_metadata_t
+from . import recipe_step_relationships_t as recipe_step_relationships_t
 from . import recipe_tags_t as recipe_tags_t
 from . import recipe_workflow_steps_t as recipe_workflow_steps_t
 from . import recipes_t as recipes_t
@@ -183,6 +185,7 @@ __all__: list[str] = [
     "experiment_groups_t",
     "export_default_runsheet_t",
     "export_entities_t",
+    "export_listing_t",
     "exports_t",
     "fetch_listing_t",
     "field_values_t",
@@ -241,6 +244,7 @@ __all__: list[str] = [
     "recipe_links_t",
     "recipe_metadata_t",
     "recipe_output_metadata_t",
+    "recipe_step_relationships_t",
     "recipe_tags_t",
     "recipe_workflow_steps_t",
     "recipes_t",

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/types/api/files/download_file.py

@@ -23,12 +23,20 @@ __all__: list[str] = [
     "FileDownloadQueryEntityField",
     "FileDownloadQueryTextDocumentId",
     "FileDownloadQueryType",
+    "FileDownloadType",
+    "UNC_FILE_DOWNLOAD_TYPE_HEADER",
 ]
 
 ENDPOINT_METHOD = "GET"
 ENDPOINT_PATH = "api/external/files/download_file"
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+class FileDownloadType(StrEnum):
+    RAW = "raw"
+    ARCHIVE = "archive"
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 class FileDownloadQueryType(StrEnum):
     ENTITY_FIELD = "entity_field"
@@ -88,4 +96,8 @@ FileDownloadQuery = typing.Annotated[
 @dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class Arguments:
     file_query: FileDownloadQuery
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+UNC_FILE_DOWNLOAD_TYPE_HEADER = "X-File-Download-Type"
 # DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.153/uncountable/types/api/listing/export_listing.py

@@ -0,0 +1,45 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# 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
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import async_batch_t
+from ... import base_t
+from ... import entity_t
+from ... import listing_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/listing/export_listing"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.listing.export_listing.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    entity_type: entity_t.EntityType
+    columns: list[listing_t.ColumnIdentifier]
+    filters: listing_t.FilterNode | None = None
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.listing.export_listing.Data",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Data(async_batch_t.AsyncBatchActionReturn):
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/types/api/recipes/get_recipes_data.py

@@ -15,6 +15,7 @@ from ... import experiment_groups_t
 from ... import inputs_t
 from ... import outputs_t
 from ... import recipe_metadata_t
+from ... import recipe_step_relationships_t
 from ... import recipe_tags_t
 from ... import workflows_t
 
@@ -24,6 +25,7 @@ __all__: list[str] = [
     "Data",
     "ENDPOINT_METHOD",
     "ENDPOINT_PATH",
+    "FeedInfo",
     "Recipe",
     "RecipeInput",
     "RecipeLockStatus",
@@ -161,10 +163,27 @@ class RecipeWorkflowStep:
     recipe_step_groups: list[RecipeStepGroup]
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.recipes.get_recipes_data.FeedInfo",
+    to_string_values={"feed_rate", "feed_rate_calculation_quantity", "feed_time", "start_time"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class FeedInfo:
+    id: base_t.ObjectId
+    recipe_step_relationship_id: base_t.ObjectId
+    start_time: Decimal
+    feed_time_unit_id: base_t.ObjectId
+    feed_time: Decimal | None = None
+    feed_rate: Decimal | None = None
+    feed_rate_calculation_quantity: Decimal | None = None
+    feed_free_variable: recipe_step_relationships_t.FeedFreeVariable | None = None
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.api.recipes.get_recipes_data.RecipeStepRelationship",
-    to_string_values={"actual_quantity_dec", "quantity_dec"},
+    to_string_values={"actual_quantity_dec", "quantity_dec", "quantity_mass"},
 )
 @dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
 class RecipeStepRelationship:
@@ -174,6 +193,8 @@ class RecipeStepRelationship:
     target_recipe_step_id: base_t.ObjectId
     quantity_dec: Decimal | None = None
     actual_quantity_dec: Decimal | None = None
+    feed_info: FeedInfo | None = None
+    quantity_mass: Decimal | None = None
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/types/client_base.py

@@ -722,7 +722,7 @@ class ClientMethods(ABC):
         requests: list[async_batch_t.AsyncBatchRequest],
         _request_options: client_config_t.RequestOptions | None = None,
     ) -> execute_batch_load_async_t.Data:
-        """Run multiple API calls via one request
+        """Run multiple API calls asynchronously via one request. WARNING: Batch action execution order is randomized by default. If execution order is important, use the dependsOn parameter to specify the actions that must be completed before a given action is executed.
 
         """
         args = execute_batch_load_async_t.Arguments(

{uncountablepythonsdk-0.0.151 → uncountablepythonsdk-0.0.153}/uncountable/types/entity_t.py

@@ -82,6 +82,7 @@ __all__: list[str] = [
         "experiment_group_member": "Experiment Group Member",
         "experiment_group_type": "Experiment Group Type",
         "favorite_link": "Favorite Link",
+        "oauth_client": "OAuth Client",
         "field_inheritance_mapping": "Field Inheritance Mapping",
         "field_option_set": "Field Option Set",
         "field_option": "Field Option",
@@ -124,6 +125,7 @@ __all__: list[str] = [
         "navigation_process_step": "Navigation Process Step",
         "notebook": "Notebook",
         "notebook_cell": "Notebook Cell",
+        "notebook_section": "Notebook Section",
         "output": "Output",
         "output_calculation_entity_mapping": "Output Calculation Entity Mapping",
         "output_category_all": "Output Category",
@@ -305,6 +307,7 @@ class EntityType(StrEnum):
     EXPERIMENT_GROUP_MEMBER = "experiment_group_member"
     EXPERIMENT_GROUP_TYPE = "experiment_group_type"
     FAVORITE_LINK = "favorite_link"
+    OAUTH_CLIENT = "oauth_client"
     FIELD_INHERITANCE_MAPPING = "field_inheritance_mapping"
     FIELD_OPTION_SET = "field_option_set"
     FIELD_OPTION = "field_option"
@@ -347,6 +350,7 @@ class EntityType(StrEnum):
     NAVIGATION_PROCESS_STEP = "navigation_process_step"
     NOTEBOOK = "notebook"
     NOTEBOOK_CELL = "notebook_cell"
+    NOTEBOOK_SECTION = "notebook_section"
     OUTPUT = "output"
     OUTPUT_CALCULATION_ENTITY_MAPPING = "output_calculation_entity_mapping"
     OUTPUT_CATEGORY_ALL = "output_category_all"

uncountablepythonsdk-0.0.153/uncountable/types/recipe_step_relationships.py

@@ -0,0 +1,7 @@
+# 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 .recipe_step_relationships_t import FeedFreeVariable as FeedFreeVariable
+# DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.153/uncountable/types/recipe_step_relationships_t.py

@@ -0,0 +1,20 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# 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 enum import StrEnum
+
+__all__: list[str] = [
+    "FeedFreeVariable",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class FeedFreeVariable(StrEnum):
+    FEED_RATE = "feed_rate"
+    FEED_TIME = "feed_time"
+# DO NOT MODIFY -- This file is generated by type_spec