UncountablePythonSDK 0.0.152__tar.gz → 0.0.154__tar.gz

{uncountablepythonsdk-0.0.152 → uncountablepythonsdk-0.0.154}/.github/workflows/documentation.yml

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

{uncountablepythonsdk-0.0.152 → uncountablepythonsdk-0.0.154}/.github/workflows/publish.yml

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

{uncountablepythonsdk-0.0.152 → uncountablepythonsdk-0.0.154}/PKG-INFO

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.152
+Version: 0.0.154
 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.152 → uncountablepythonsdk-0.0.154}/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

{uncountablepythonsdk-0.0.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/uncountable/core/client.py

@@ -1,7 +1,9 @@
 import base64
 import datetime
+import pathlib
 import re
 import typing
+import zipfile
 from dataclasses import dataclass
 from datetime import UTC, timedelta
 from enum import StrEnum
@@ -383,6 +385,26 @@ class Client(ClientMethods):
 
         content = response.content
         content_disposition = response.headers.get("Content-Disposition", None)
+        file_download_type = response.headers.get(
+            download_file_t.UNC_FILE_DOWNLOAD_TYPE_HEADER, None
+        )
+        if file_download_type == download_file_t.FileDownloadType.ARCHIVE:
+            archive = zipfile.ZipFile(BytesIO(content))
+            files: list[DownloadedFile] = []
+
+            for filepath in archive.namelist():
+                with archive.open(filepath, "r") as f:
+                    filename = pathlib.Path(filepath).name
+                    read_file = f.read()
+                    files.append(
+                        DownloadedFile(
+                            name=filename,
+                            size=len(read_file),
+                            data=BytesIO(read_file),
+                        )
+                    )
+            return files
+
         return [
             DownloadedFile(
                 name=self._get_downloaded_filename(cd=content_disposition),

{uncountablepythonsdk-0.0.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/uncountable/integration/construct_client.py

@@ -30,10 +30,10 @@ def _construct_auth_details(profile_meta: ProfileMetadata) -> AuthDetailsAll:
 
 
 def _construct_client_config(
-    profile_meta: ProfileMetadata, job_logger: JobLogger
-) -> ClientConfig | None:
+    *, profile_meta: ProfileMetadata, job_logger: JobLogger
+) -> ClientConfig:
     if profile_meta.client_options is None:
-        return None
+        return ClientConfig(logger=job_logger)
     return ClientConfig(
         allow_insecure_tls=profile_meta.client_options.allow_insecure_tls,
         extra_headers=profile_meta.client_options.extra_headers,
@@ -47,6 +47,6 @@ def construct_uncountable_client(
     return Client(
         base_url=profile_meta.base_url,
         auth_details=_construct_auth_details(profile_meta),
-        config=_construct_client_config(profile_meta, logger),
+        config=_construct_client_config(profile_meta=profile_meta, job_logger=logger),
         app_base_url=profile_meta.app_base_url,
     )

{uncountablepythonsdk-0.0.152 → uncountablepythonsdk-0.0.154}/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:
@@ -98,7 +97,7 @@ async def start_scheduler(
     result_queue: ResultQueue = asyncio.Queue()
 
     with ProcessPoolExecutor(
-        max_workers=_MAX_JOB_WORKERS, max_tasks_per_child=1
+        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.152 → uncountablepythonsdk-0.0.154}/uncountable/types/__init__.py

@@ -99,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
@@ -243,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.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/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.152 → uncountablepythonsdk-0.0.154}/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",
@@ -306,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"

uncountablepythonsdk-0.0.154/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.154/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