UncountablePythonSDK 0.0.182__py3-none-any.whl → 0.0.184__py3-none-any.whl

docs/requirements.txt

@@ -1,6 +1,6 @@
 furo==2025.12.19
 myst-parser==5.1.0
-sphinx-autoapi==3.8.0
+sphinx-autoapi==3.8.1
 sphinx-copybutton==0.5.2
 Sphinx==9.1.0
 sphinx_design==0.7.0

examples/integration-server/pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
   "types-openpyxl == 3.*",
   "types-pytz ==2026.*",
   "types-requests == 2.*",
-  "types-simplejson == 3.*",
+  "types-simplejson ==4.*",
   "pandas-stubs",
   "xlrd == 2.*",
   "msgspec ==0.21.*",

pkgs/filesystem_utils/_blob_session.py

@@ -2,7 +2,7 @@ from io import BytesIO
 from typing import Self
 
 from azure.core.credentials import AzureSasCredential
-from azure.storage.blob import BlobServiceClient, ContainerClient
+from azure.storage.blob import BlobPrefix, BlobServiceClient, ContainerClient
 
 from pkgs.filesystem_utils.file_type_utils import (
     FileObjectData,
@@ -78,6 +78,24 @@ class BlobSession(FileSystemSession):
 
         return filesystem_file_references
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to list_directories on uninitialized blob session"
+        )
+
+        prefix = _add_slash(dir_path.filepath)
+        directories: list[FileSystemObject] = []
+        for item in self.container_client.walk_blobs(
+            name_starts_with=prefix, delimiter="/"
+        ):
+            if isinstance(item, BlobPrefix):
+                directories.append(FileSystemFileReference(item.name.rstrip("/")))
+
+        return directories
+
     def download_files(
         self,
         filepaths: list[FileSystemObject],
@@ -108,6 +126,22 @@ class BlobSession(FileSystemSession):
 
         return downloaded_files
 
+    def _move_blob(self, src_name: str, dest_name: str) -> None:
+        assert self.container_client is not None, (
+            "call to _move_blob on uninitialized blob session"
+        )
+        source_blob_client = self.container_client.get_blob_client(src_name)
+        dest_blob_client = self.container_client.get_blob_client(dest_name)
+
+        source_url = (
+            f"{source_blob_client.url}?{self.config.credential.signature}"
+            if isinstance(self.config.credential, AzureSasCredential)
+            else source_blob_client.url
+        )
+
+        dest_blob_client.start_copy_from_url(source_url)
+        source_blob_client.delete_blob()
+
     def move_files(self, file_mappings: list[FileTransfer]) -> None:
         assert self.service_client is not None and self.container_client is not None, (
             "call to move_files on uninitialized blob session"
@@ -119,19 +153,25 @@ class BlobSession(FileSystemSession):
             ):
                 raise IncompatibleFileReferenceError()
 
-            source_blob_client = self.container_client.get_blob_client(
-                src_file.filepath
-            )
-            dest_blob_client = self.container_client.get_blob_client(dest_file.filepath)
+            self._move_blob(src_file.filepath, dest_file.filepath)
 
-            source_url = (
-                f"{source_blob_client.url}?{self.config.credential.signature}"
-                if isinstance(self.config.credential, AzureSasCredential)
-                else source_blob_client.url
-            )
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        if not isinstance(src_dir, FileSystemFileReference) or not isinstance(
+            dest_dir, FileSystemFileReference
+        ):
+            raise IncompatibleFileReferenceError()
+
+        assert self.service_client is not None and self.container_client is not None, (
+            "call to move_directory on uninitialized blob session"
+        )
 
-            dest_blob_client.start_copy_from_url(source_url)
-            source_blob_client.delete_blob()
+        src_prefix = _add_slash(src_dir.filepath)
+        dest_prefix = _add_slash(dest_dir.filepath)
+        for blob in self.container_client.list_blobs(name_starts_with=src_prefix):
+            dest_name = dest_prefix + blob.name[len(src_prefix) :]
+            self._move_blob(blob.name, dest_name)
 
     def delete_files(self, filepaths: list[FileSystemObject]) -> None:
         assert self.service_client is not None and self.container_client is not None, (

pkgs/filesystem_utils/_file_share_session.py

@@ -79,6 +79,49 @@ class FileShareSession(FileSystemSession):
             dir_client, recursive=recursive, valid_extensions=valid_extensions
         )
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+        assert self._share_client is not None, (
+            "call to list_directories on uninitialized file share session"
+        )
+        dir_client = self._share_client.get_directory_client(dir_path.filepath)
+        directories: list[FileSystemObject] = []
+        for item in dir_client.list_directories_and_files():
+            if item["is_directory"]:
+                item_path = f"{dir_client.directory_path}/{item['name']}".lstrip("/")
+                directories.append(FileSystemFileReference(filepath=item_path))
+        return directories
+
+    def create_directory(self, dir_path: FileSystemObject) -> None:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+        assert self._share_client is not None, (
+            "call to create_directory on uninitialized file share session"
+        )
+        current = ""
+        for segment in dir_path.filepath.strip("/").split("/"):
+            if not segment:
+                continue
+            current = f"{current}/{segment}".lstrip("/")
+            dir_client = self._share_client.get_directory_client(current)
+            if not dir_client.exists():
+                dir_client.create_directory()
+
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        if not isinstance(src_dir, FileSystemFileReference) or not isinstance(
+            dest_dir, FileSystemFileReference
+        ):
+            raise IncompatibleFileReferenceError()
+        assert self._share_client is not None, (
+            "call to move_directory on uninitialized file share session"
+        )
+        self._share_client.get_directory_client(src_dir.filepath).rename_directory(
+            dest_dir.filepath
+        )
+
     def download_files(self, filepaths: list[FileSystemObject]) -> list[FileObjectData]:
         assert self._share_client is not None, (
             "call to download_files on uninitialized file share session"

pkgs/filesystem_utils/_s3_session.py

@@ -86,6 +86,28 @@ class S3Session(FileSystemSession):
 
         return filesystem_references
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        if not isinstance(dir_path, FileSystemFileReference):
+            raise IncompatibleFileReferenceError()
+
+        assert self.bucket is not None, (
+            "call to list_directories on uninitialized s3 session"
+        )
+
+        prefix = _add_slash(dir_path.filepath)
+        paginator = self.bucket.meta.client.get_paginator("list_objects_v2")
+
+        directories: list[FileSystemObject] = []
+        for page in paginator.paginate(
+            Bucket=self.bucket.name, Prefix=prefix, Delimiter="/"
+        ):
+            for common_prefix in page.get("CommonPrefixes", []):
+                directories.append(
+                    FileSystemFileReference(common_prefix["Prefix"].rstrip("/"))
+                )
+
+        return directories
+
     def download_files(
         self,
         filepaths: list[FileSystemObject],
@@ -115,6 +137,18 @@ class S3Session(FileSystemSession):
 
         return downloaded_files
 
+    def _move_object(self, src_key: str, dest_key: str) -> None:
+        assert self.bucket is not None, (
+            "call to _move_object on uninitialized s3 session"
+        )
+        self.bucket.Object(dest_key).copy_from(
+            CopySource={
+                "Bucket": self.bucket.name,
+                "Key": src_key,
+            }
+        )
+        self.bucket.Object(src_key).delete()
+
     def move_files(self, file_mappings: list[FileTransfer]) -> None:
         assert self.bucket is not None, "call to move_files on uninitialized s3 session"
 
@@ -123,10 +157,22 @@ class S3Session(FileSystemSession):
                 dest_file, FileSystemFileReference
             ):
                 raise IncompatibleFileReferenceError()
-            self.bucket.Object(dest_file.filepath).copy_from(
-                CopySource={
-                    "Bucket": self.bucket.name,
-                    "Key": src_file.filepath,
-                }
-            )
-            self.bucket.Object(src_file.filepath).delete()
+            self._move_object(src_file.filepath, dest_file.filepath)
+
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        if not isinstance(src_dir, FileSystemFileReference) or not isinstance(
+            dest_dir, FileSystemFileReference
+        ):
+            raise IncompatibleFileReferenceError()
+
+        assert self.bucket is not None, (
+            "call to move_directory on uninitialized s3 session"
+        )
+
+        src_prefix = _add_slash(src_dir.filepath)
+        dest_prefix = _add_slash(dest_dir.filepath)
+        for obj in self.bucket.objects.filter(Prefix=src_prefix):
+            dest_key = dest_prefix + obj.key[len(src_prefix) :]
+            self._move_object(obj.key, dest_key)

pkgs/filesystem_utils/filesystem_session.py

@@ -32,6 +32,17 @@ class FileSystemSession(ABC):
     def delete_files(self, filepaths: list[FileSystemObject]) -> None:
         raise NotImplementedError
 
+    def list_directories(self, dir_path: FileSystemObject) -> list[FileSystemObject]:
+        raise NotImplementedError
+
+    def create_directory(self, dir_path: FileSystemObject) -> None:
+        raise NotImplementedError
+
+    def move_directory(
+        self, src_dir: FileSystemObject, dest_dir: FileSystemObject
+    ) -> None:
+        raise NotImplementedError
+
     @abstractmethod
     def __enter__(self) -> "FileSystemSession": ...
 

pkgs/type_spec/actions_registry/__main__.py

@@ -82,6 +82,7 @@ def main() -> None:
                 short_description=definition.short_description,
                 description=definition.description,
                 icon=definition.icon,
+                action=definition.action,
                 ref_name=ref_name,
                 module=actions_registry_t.ActionsRegistryModule(
                     ts_name(module_str, name_case=builder.NameCase.convert)

pkgs/type_spec/actions_registry/emit_python.py

@@ -51,6 +51,8 @@ def _emit_action_definition(
     out.write(f"{sub_indent}description={_py_str(action.description)},\n")
     if action.icon is not None:
         out.write(f"{sub_indent}icon={_py_str(action.icon)},\n")
+    if action.action is not None:
+        out.write(f"{sub_indent}action={_py_str(action.action)},\n")
     out.write(f"{sub_indent}ref_name={_py_str(action.ref_name)},\n")
     out.write(
         f"{sub_indent}module=actions_registry_t.ActionsRegistryModule.{action.module.name},\n"

pkgs/type_spec/actions_registry/emit_typescript.py

@@ -130,6 +130,10 @@ def _emit_action_definition(
         out.write(
             f"{sub_indent}icon: {encode_common_string(action_definition.icon)},\n"
         )
+    if action_definition.action is not None:
+        out.write(
+            f"{sub_indent}action: {encode_common_string(action_definition.action)},\n"
+        )
     out.write(
         f"{sub_indent}shortDescription: {encode_common_string(action_definition.short_description)},\n"
     )

pkgs/type_spec/builder.py

@@ -1106,6 +1106,7 @@ class SpecEndpoint:
     desc: str | None = None
     account_type: str | None
     route_group: str | None
+    allow_non_materials_path: bool = False
 
     # function, path details per api endpoint
     path_per_api_endpoint: dict[str, EndpointSpecificPath]
@@ -1135,6 +1136,7 @@ class SpecEndpoint:
                 "has_attachment",
                 "account_type",
                 "route_group",
+                "allow_non_materials_path",
             ]
             + list(builder.api_endpoints.keys()),
         )
@@ -1188,6 +1190,10 @@ class SpecEndpoint:
         assert isinstance(suppress_ts, bool)
         self.suppress_ts = suppress_ts
 
+        allow_non_materials_path = data.get("allow_non_materials_path", False)
+        assert isinstance(allow_non_materials_path, bool)
+        self.allow_non_materials_path = allow_non_materials_path
+
         self.result_type = ResultType(data.get("result_type", ResultType.json.value))
         self.has_attachment = data.get("has_attachment", False)
         self.desc = data.get("desc")

pkgs/type_spec/emit_python.py

@@ -1166,13 +1166,25 @@ def _emit_routes(*, builder: builder.SpecBuilder, config: PythonConfig) -> None:
                 continue
 
             endpoints.append(endpoint)
+
+        has_route_opts = any(
+            endpoint.allow_non_materials_path for endpoint in endpoints
+        )
+        route_opts_import = ["RouteOpts"] if has_route_opts else []
+        static_route_imports = ", ".join(
+            sorted(["StaticRouteType", *route_opts_import])
+        )
+        dynamic_route_imports = ", ".join(
+            sorted(["DynamicRouteType", *route_opts_import])
+        )
+
         static_out = io.StringIO()
         static_out.write(
             f"""{MODIFY_NOTICE}{LINT_HEADER}{ROUTE_NOTICE}
 
 # Static routes for use locally and to ensure linting/mypy finds all modules
 
-from main.site.framework.types import StaticRouteType
+from main.site.framework.types import {static_route_imports}
 """
         )
 
@@ -1218,7 +1230,7 @@ ROUTES: list[StaticRouteType] = [
 
 # Dynamic loaded routes to provide faster startup times
 
-from main.site.framework.types import DynamicRouteType
+from main.site.framework.types import {dynamic_route_imports}
 
 ROUTE_PREFIX = "/{last_endpoint.path_per_api_endpoint[endpoint_root].path_root}"
 
@@ -1226,19 +1238,35 @@ ROUTES: list[DynamicRouteType] = [
 """
         )
 
+        route_opts_out = io.StringIO()
         for endpoint in sorted_endpoints:
             endpoint_function_path = endpoint.path_per_api_endpoint[endpoint_root]
+            route = f"{endpoint_function_path.path_dirname}/{endpoint_function_path.path_basename}"
             dynamic_out.write(
-                f'{INDENT}("{endpoint_function_path.path_dirname}/{endpoint_function_path.path_basename}", "{endpoint_function_path.function}", ["{endpoint.method.upper()}"]),\n'
+                f'{INDENT}("{route}", "{endpoint_function_path.function}", ["{endpoint.method.upper()}"]),\n'
             )
 
             assert endpoint_function_path.function
             static_out.write(
-                f'{INDENT}("{endpoint_function_path.path_dirname}/{endpoint_function_path.path_basename}", {endpoint_function_path.function}, ["{endpoint.method.upper()}"]),\n'
+                f'{INDENT}("{route}", {endpoint_function_path.function}, ["{endpoint.method.upper()}"]),\n'
             )
 
-        dynamic_out.write(f"{MODIFY_NOTICE}]\n")
-        static_out.write(f"{MODIFY_NOTICE}]\n")
+            if endpoint.allow_non_materials_path:
+                route_opts_out.write(
+                    f'{INDENT}"{route}": RouteOpts(allow_non_materials_path=True),\n'
+                )
+
+        routes_close = f"{MODIFY_NOTICE}]\n"
+        dynamic_out.write(routes_close)
+        static_out.write(routes_close)
+
+        if has_route_opts:
+            route_opts_emit = f"""
+ROUTE_OPTS: dict[str, RouteOpts] = {{
+{route_opts_out.getvalue()}{MODIFY_NOTICE}}}
+"""
+            dynamic_out.write(route_opts_emit)
+            static_out.write(route_opts_emit)
 
         util.rewrite_file(f"{output}/routes_static.py", static_out.getvalue())
         util.rewrite_file(f"{output}/routes_dynamic.py", dynamic_out.getvalue())

uncountable/core/environment.py

@@ -26,10 +26,6 @@ def get_local_admin_server_port() -> int:
     return int(os.environ.get("UNC_ADMIN_SERVER_PORT", "50051"))
 
 
-def get_integration_admin_token() -> str | None:
-    return os.environ.get("UNC_INTEGRATION_ADMIN_TOKEN")
-
-
 def get_otel_enabled() -> bool:
     return os.environ.get("UNC_OTEL_ENABLED") == "true"
 

uncountable/integration/webhook_server/entrypoint.py

@@ -12,7 +12,6 @@ from uncountable.integration.http_server import GenericHttpRequest, HttpExceptio
 from uncountable.integration.job import CustomHttpJob, WebhookJob
 from uncountable.integration.scan_profiles import load_profiles
 from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_routes import register_admin_routes
 from uncountable.types import job_definition_t
 
 app = flask.Flask(__name__)
@@ -99,7 +98,6 @@ def register_route(
 
 def main() -> None:
     app.add_url_rule("/health", "health", lambda: ("OK", 200))
-    register_admin_routes(app)
 
     profiles = load_profiles()
     for profile_metadata in profiles:

uncountable/types/api/recipes/set_recipe_inputs.py

@@ -21,7 +21,7 @@ __all__: list[str] = [
 
 ENDPOINT_METHOD = "POST"
 ENDPOINT_PATH = "api/external/recipes/external_set_recipe_inputs"
-ENDPOINT_DESCRIPTION = "Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Supports setting actual values, associating lots (creating them if needed), and removing inputs. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided."
+ENDPOINT_DESCRIPTION = "Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Each entry edits one input on one recipe with a single operation: set the value (the actual value when set_actual_value is true, otherwise the set value), associate a lot (lot_recipe_id, created if the recipe is not already a lot), or remove the input (remove). A lot association targets the set value; combining it with set_actual_value fails. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided."
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

uncountable/types/client_base.py

@@ -2177,7 +2177,7 @@ class ClientMethods(ABC):
         input_data: list[set_recipe_inputs_t.RecipeInputValue],
         _request_options: client_config_t.RequestOptions | None = None,
     ) -> set_recipe_inputs_t.Data:
-        """Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Supports setting actual values, associating lots (creating them if needed), and removing inputs. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided.
+        """Sets input values for experiments. Supports numeric, text, and categorical types. Validates that the recipe step belongs to the correct recipe. Each entry edits one input on one recipe with a single operation: set the value (the actual value when set_actual_value is true, otherwise the set value), associate a lot (lot_recipe_id, created if the recipe is not already a lot), or remove the input (remove). A lot association targets the set value; combining it with set_actual_value fails. Writes all values to the first workflow step when no step is specified. Fails if both a numeric and string value are provided.
 
         :param input_data: The inputs to set. Must be at most 100 entries long
         """

uncountable/types/entity_t.py

@@ -27,6 +27,7 @@ __all__: list[str] = [
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_string_enum(
     labels={
+        "agent_memory": "Agent Memory",
         "aggregated_data_view": "Aggregated Data View",
         "analytical_method": "Analytical Method",
         "analytical_method_category": "Analytical Method Category",
@@ -35,6 +36,8 @@ __all__: list[str] = [
         "approval": "Approval",
         "async_job": "Async Job",
         "barcode": "Barcode",
+        "cad_document": "CAD Document",
+        "cad_relationship": "CAD Relationship",
         "calculation": "Calculation",
         "calendar": "Calendar",
         "calendar_event": "Calendar Event",
@@ -57,6 +60,7 @@ __all__: list[str] = [
         "credit_balance": "Credit Balance",
         "curve_settings": "Curve Settings",
         "custom_entity": "Custom Entity",
+        "data_pipeline": "Data Pipeline",
         "dataset_binding": "Dataset Binding",
         "default_permissions_new_project": "Default Permissions New Project",
         "document": "Document",
@@ -90,6 +94,7 @@ __all__: list[str] = [
         "external_ai_model": "External AI Model",
         "external_ai_session": "External AI Session",
         "external_ai_usage": "External AI Usage",
+        "external_ai_plugin": "External AI Plugin",
         "external_oauth_provider": "External OAuth Provider",
         "experiment_group": "Experiment Group",
         "experiment_group_member": "Experiment Group Member",
@@ -206,6 +211,7 @@ __all__: list[str] = [
         "regional_entity_group": "Regional Entity Group",
         "regulation": "Regulation",
         "regulation_cas": "Regulation CAS",
+        "regulation_product_category": "Regulation Product Category",
         "metadata_value": "Metadata Value",
         "report_definition": "Report Definition",
         "reusable_field_condition": "Reusable Field Condition",
@@ -297,6 +303,7 @@ __all__: list[str] = [
     },
 )
 class EntityType(StrEnum):
+    AGENT_MEMORY = "agent_memory"
     AGGREGATED_DATA_VIEW = "aggregated_data_view"
     ANALYTICAL_METHOD = "analytical_method"
     ANALYTICAL_METHOD_CATEGORY = "analytical_method_category"
@@ -305,6 +312,8 @@ class EntityType(StrEnum):
     APPROVAL = "approval"
     ASYNC_JOB = "async_job"
     BARCODE = "barcode"
+    CAD_DOCUMENT = "cad_document"
+    CAD_RELATIONSHIP = "cad_relationship"
     CALCULATION = "calculation"
     CALENDAR = "calendar"
     CALENDAR_EVENT = "calendar_event"
@@ -327,6 +336,7 @@ class EntityType(StrEnum):
     CREDIT_BALANCE = "credit_balance"
     CURVE_SETTINGS = "curve_settings"
     CUSTOM_ENTITY = "custom_entity"
+    DATA_PIPELINE = "data_pipeline"
     DATASET_BINDING = "dataset_binding"
     DEFAULT_PERMISSIONS_NEW_PROJECT = "default_permissions_new_project"
     DOCUMENT = "document"
@@ -360,6 +370,7 @@ class EntityType(StrEnum):
     EXTERNAL_AI_MODEL = "external_ai_model"
     EXTERNAL_AI_SESSION = "external_ai_session"
     EXTERNAL_AI_USAGE = "external_ai_usage"
+    EXTERNAL_AI_PLUGIN = "external_ai_plugin"
     EXTERNAL_OAUTH_PROVIDER = "external_oauth_provider"
     EXPERIMENT_GROUP = "experiment_group"
     EXPERIMENT_GROUP_MEMBER = "experiment_group_member"
@@ -476,6 +487,7 @@ class EntityType(StrEnum):
     REGIONAL_ENTITY_GROUP = "regional_entity_group"
     REGULATION = "regulation"
     REGULATION_CAS = "regulation_cas"
+    REGULATION_PRODUCT_CATEGORY = "regulation_product_category"
     METADATA_VALUE = "metadata_value"
     REPORT_DEFINITION = "report_definition"
     REUSABLE_FIELD_CONDITION = "reusable_field_condition"
@@ -568,7 +580,7 @@ class EntityType(StrEnum):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 LimitedEntityType = typing.Annotated[
-    typing.Literal[EntityType.LAB_REQUEST] | typing.Literal[EntityType.APPROVAL] | typing.Literal[EntityType.CUSTOM_ENTITY] | typing.Literal[EntityType.INGREDIENT_ATTRIBUTE] | typing.Literal[EntityType.INVENTORY_AMOUNT] | typing.Literal[EntityType.TASK] | typing.Literal[EntityType.PROJECT] | typing.Literal[EntityType.EQUIPMENT] | typing.Literal[EntityType.INV_LOCAL_LOCATIONS] | typing.Literal[EntityType.FIELD_OPTION_SET] | typing.Literal[EntityType.WEBHOOK] | typing.Literal[EntityType.SPECS] | typing.Literal[EntityType.GOAL] | typing.Literal[EntityType.INGREDIENT_TAG_MAP] | typing.Literal[EntityType.INGREDIENT_TAG] | typing.Literal[EntityType.CONDITION_PARAMETER] | typing.Literal[EntityType.NOTEBOOK] | typing.Literal[EntityType.OUTPUT] | typing.Literal[EntityType.OUTPUT_CONDITION_PARAMETER] | typing.Literal[EntityType.ASYNC_JOB] | typing.Literal[EntityType.CONSTRAINT] | typing.Literal[EntityType.CONSTRAINT_SET] | typing.Literal[EntityType.INGREDIENT_CATEGORY_ALL] | typing.Literal[EntityType.TIME_SERIES_SEGMENT] | typing.Literal[EntityType.EQUIPMENT_MAINTENANCE] | typing.Literal[EntityType.MAINTENANCE_SCHEDULE] | typing.Literal[EntityType.CONDITION_PARAMETER_RULE] | typing.Literal[EntityType.INGREDIENT] | typing.Literal[EntityType.TIMESHEET_ENTRY] | typing.Literal[EntityType.SAVE] | typing.Literal[EntityType.RECIPE_CHECK] | typing.Literal[EntityType.EXPERIMENT_GROUP_MEMBER] | typing.Literal[EntityType.CALENDAR_EVENT],
+    typing.Literal[EntityType.LAB_REQUEST] | typing.Literal[EntityType.APPROVAL] | typing.Literal[EntityType.CUSTOM_ENTITY] | typing.Literal[EntityType.INGREDIENT_ATTRIBUTE] | typing.Literal[EntityType.INVENTORY_AMOUNT] | typing.Literal[EntityType.TASK] | typing.Literal[EntityType.PROJECT] | typing.Literal[EntityType.EQUIPMENT] | typing.Literal[EntityType.INV_LOCAL_LOCATIONS] | typing.Literal[EntityType.FIELD_OPTION_SET] | typing.Literal[EntityType.WEBHOOK] | typing.Literal[EntityType.SPECS] | typing.Literal[EntityType.GOAL] | typing.Literal[EntityType.INGREDIENT_TAG_MAP] | typing.Literal[EntityType.INGREDIENT_TAG] | typing.Literal[EntityType.CONDITION_PARAMETER] | typing.Literal[EntityType.NOTEBOOK] | typing.Literal[EntityType.OUTPUT] | typing.Literal[EntityType.OUTPUT_CONDITION_PARAMETER] | typing.Literal[EntityType.ASYNC_JOB] | typing.Literal[EntityType.CONSTRAINT] | typing.Literal[EntityType.CONSTRAINT_SET] | typing.Literal[EntityType.INGREDIENT_CATEGORY_ALL] | typing.Literal[EntityType.TIME_SERIES_SEGMENT] | typing.Literal[EntityType.EQUIPMENT_MAINTENANCE] | typing.Literal[EntityType.MAINTENANCE_SCHEDULE] | typing.Literal[EntityType.CONDITION_PARAMETER_RULE] | typing.Literal[EntityType.INGREDIENT] | typing.Literal[EntityType.TIMESHEET_ENTRY] | typing.Literal[EntityType.SAVE] | typing.Literal[EntityType.RECIPE_CHECK] | typing.Literal[EntityType.EXPERIMENT_GROUP_MEMBER] | typing.Literal[EntityType.INPUT_GROUP] | typing.Literal[EntityType.INPUT_GROUP_MEMBER] | typing.Literal[EntityType.CALENDAR_EVENT],
     serial_alias_annotation(
     named_type_path="sdk.entity.LimitedEntityType",
     ),

uncountable/types/uploader.py

@@ -8,8 +8,10 @@ from .uploader_t import ChannelType as ChannelType
 from .uploader_t import StructureElementType as StructureElementType
 from .uploader_t import StructureElementBase as StructureElementBase
 from .uploader_t import DecimalValue as DecimalValue
+from .uploader_t import IdValue as IdValue
 from .uploader_t import StringValue as StringValue
 from .uploader_t import BaseData as BaseData
+from .uploader_t import FileHeaderData as FileHeaderData
 from .uploader_t import NumericHeaderData as NumericHeaderData
 from .uploader_t import TextHeaderData as TextHeaderData
 from .uploader_t import HeaderValue as HeaderValue

uncountable/types/uploader_t.py

@@ -10,6 +10,7 @@ from enum import StrEnum
 import dataclasses
 from pkgs.serialization import serial_class
 from pkgs.serialization import serial_union_annotation
+from . import base_t
 
 __all__: list[str] = [
     "BaseData",
@@ -17,9 +18,11 @@ __all__: list[str] = [
     "ChannelType",
     "DataChannel",
     "DecimalValue",
+    "FileHeaderData",
     "HeaderEntry",
     "HeaderType",
     "HeaderValue",
+    "IdValue",
     "NumericChannelData",
     "NumericHeaderData",
     "ParsedFileData",
@@ -35,6 +38,7 @@ __all__: list[str] = [
 
 # DO NOT MODIFY -- This file is generated by type_spec
 class HeaderType(StrEnum):
+    FILE_HEADER = "file_header"
     NUMERIC_HEADER = "numeric_header"
     TEXT_HEADER = "text_header"
 
@@ -71,6 +75,15 @@ class DecimalValue:
     value: Decimal
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.uploader.IdValue",
+)
+@dataclasses.dataclass(slots=True, kw_only=True)
+class IdValue:
+    value: base_t.ObjectId
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.uploader.StringValue",
@@ -90,6 +103,17 @@ class BaseData:
     type: str
 
 
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.uploader.FileHeaderData",
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=True, kw_only=True)
+class FileHeaderData(BaseData):
+    type: typing.Literal[HeaderType.FILE_HEADER] = HeaderType.FILE_HEADER
+    data: IdValue | None
+
+
 # DO NOT MODIFY -- This file is generated by type_spec
 @serial_class(
     named_type_path="sdk.uploader.NumericHeaderData",
@@ -114,11 +138,12 @@ class TextHeaderData(BaseData):
 
 # DO NOT MODIFY -- This file is generated by type_spec
 HeaderValue = typing.Annotated[
-    NumericHeaderData | TextHeaderData,
+    FileHeaderData | NumericHeaderData | TextHeaderData,
     serial_union_annotation(
     named_type_path="sdk.uploader.HeaderValue",
         discriminator="type",
         discriminator_map={
+            "file_header": FileHeaderData,
             "numeric_header": NumericHeaderData,
             "text_header": TextHeaderData,
         },

{uncountablepythonsdk-0.0.182.dist-info → uncountablepythonsdk-0.0.184.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.182
+Version: 0.0.184
 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.182.dist-info → uncountablepythonsdk-0.0.184.dist-info}/RECORD

@@ -3,7 +3,7 @@ docs/.gitignore,sha256=_ebkZUcwfvfnGEJ95rfj1lxoBNd6EE9ZvtOc7FsbfFE,7
 docs/conf.py,sha256=KivJNvQWboXqJ07NMgcYY5tsV6CoIfyTpZh8jdn_5Qk,4239
 docs/index.md,sha256=g4Yi5831fEkywYkkcFohYLkKzSI91SOZF7DxKsm9zgI,3193
 docs/justfile,sha256=WymCEQ6W2A8Ak79iUPmecmuaUNN2htb7STUrz5K7ELE,273
-docs/requirements.txt,sha256=8aqtVaNV4AlzUZ7nAM8kxOF1ccAcujhQ_ayU9Whksf8,172
+docs/requirements.txt,sha256=_GyKrMFrmjp2hKhTLlhGdKqD7kXs_bomeY4yGt1oI7c,172
 docs/integration_examples/add_file_to_folder.md,sha256=ZCv8miP7ZlPjor-ZGDwy2oucyopqMPa3aXlmOEfbX10,1556
 docs/integration_examples/create_ingredient.md,sha256=ufzAzPyGaDr9rCBDMGQAE8xahsCMkVfG8VLs1pb3qr0,1549
 docs/integration_examples/create_output.md,sha256=aDn2TjzKgY-HnxnvgsZS578cvajmHpF1y2HKkHfdtd4,2104
@@ -33,7 +33,7 @@ examples/set_recipe_metadata_file.py,sha256=cRVXGz4UN4aqnNrNSzyBmikYHpe63lMIuzOp
 examples/set_recipe_output_file_sdk.py,sha256=Lz1amqppnWTX83z-C090wCJ4hcKmCD3kb-4v0uBRi0Y,782
 examples/stream_dataset_measurements.py,sha256=iLmDn7Yv-deeUsfDY9DhYckheoTfLhj5_ORjO0oLHng,5890
 examples/upload_files.py,sha256=qMaSvMSdTMPOOP55y1AwEurc0SOdZAMvEydlqJPsGpg,432
-examples/integration-server/pyproject.toml,sha256=ATAn8vGr9cRRKtceYjwQHParKwfzz_j46rIEEOq4SvE,11591
+examples/integration-server/pyproject.toml,sha256=CxNt8vQ17Po7h4fOXT8jtbrFJosxBVTVY8kTuRyxtjY,11590
 examples/integration-server/jobs/materials_auto/concurrent_cron.py,sha256=xsK3H9ZEaniedC2nJUB0rqOcFI8y-ojfl_nLSJb9AMM,312
 examples/integration-server/jobs/materials_auto/example_cron.py,sha256=y1nAtGwbPJfIrfQsrHVmJLAHmQtCEHIy1g-0PjeXx04,735
 examples/integration-server/jobs/materials_auto/example_http.py,sha256=hA--iNztlTwLaEBzlmoRZ6-hRTUExUUqLV0-ipU7_mo,1560
@@ -57,15 +57,15 @@ pkgs/argument_parser/parser_inner.py,sha256=0zepgTqQm6yFp9WMlHJmWV3S9WsxzRiN_nBw
 pkgs/argument_parser/parser_options.py,sha256=NAwNZSIH8EjyfJXJSxbn4kYmF3K7D3HuOedlSjjsF28,1332
 pkgs/argument_parser/type_predicates.py,sha256=CkQexiseN05XDz4e34K3gIOlER5LG5Ke8u0UHK4gC04,1157
 pkgs/filesystem_utils/__init__.py,sha256=b8NblpFX4rwyRSY3d8GGJRk9x3ET5jpsxDz0M2VnGlw,2071
-pkgs/filesystem_utils/_blob_session.py,sha256=2xcelJlME7bCNmBiTMnwycGDkwo0UtYfbGQEkCj0Ndo,5183
-pkgs/filesystem_utils/_file_share_session.py,sha256=27QmvifiwMYQ25eRVKWdr3VZDdZmBtYXTF6lhPUcld4,5155
+pkgs/filesystem_utils/_blob_session.py,sha256=pdRWflJAQux06ZZZLjbH1kic4djG8YBTiAN3fZ8nSVA,6870
+pkgs/filesystem_utils/_file_share_session.py,sha256=GyC9w762CbnvpC0ooZumZnzqrIULYOkHN7UIz3I7jXc,7131
 pkgs/filesystem_utils/_gdrive_session.py,sha256=pxZNTR5t8gJFS-HwMCZiQ-ZwCJV_ij5PWiBf1BUy9DE,11122
 pkgs/filesystem_utils/_local_session.py,sha256=FXxLn7CqmYZZib3mCGg0BuItIGwlvOIsKEJcZHfJ_iw,2359
-pkgs/filesystem_utils/_s3_session.py,sha256=jmc3UdOog_4cpnRvoqJgkzSAgBGYjmoMN92GIgUX-zw,4523
+pkgs/filesystem_utils/_s3_session.py,sha256=0MS1RXPA7wnoLKG8eDV_ONFS4f_rhA5Ey3yn65-JZ_g,6250
 pkgs/filesystem_utils/_sftp_connection.py,sha256=XeRNdfdXPjpRYiirp7GGsmQOQ4W4WMZihuDKFqy9nRE,4854
 pkgs/filesystem_utils/_sftp_session.py,sha256=62E2OWHWvtAcKjxq2g5L4cCWiXiOFwRoyQX9Hr6d3RU,4927
 pkgs/filesystem_utils/file_type_utils.py,sha256=Hf-NP1E8lKn9QAx9LH2a6Fc5jSXfmE8l0CMCtqvjyFA,2140
-pkgs/filesystem_utils/filesystem_session.py,sha256=BQ2Go8Mu9-GcnaWh2Pm4x7ugLVsres6XrOQ8RoiEpcE,1045
+pkgs/filesystem_utils/filesystem_session.py,sha256=zOddRkBZC5tsTiwquySeKTkKM5yNwW3Q_jDq9U0XcuI,1411
 pkgs/serialization/__init__.py,sha256=HWzdZ5CtMnZ-svy08eiDn6K9P_IveCjkjQ3whxPoCs4,1093
 pkgs/serialization/annotation.py,sha256=JXP2caXNoj8h_Pvx_26iarPlsaPcIqe1qssTksJiKjM,3408
 pkgs/serialization/missing_sentry.py,sha256=VCXiYs7s5MkfG8kbmwDAswdq0fL4zFRj4B2uSQzmTos,819
@@ -86,14 +86,14 @@ pkgs/strenum_compat/__init__.py,sha256=Bq1kcPIQsXRiLUGWc63QGnVa5BlhvuFWpMMvdNcHy
 pkgs/strenum_compat/strenum_compat.py,sha256=uOUAgpYTjHs1MX8dG81jRlyTkt3KNbkV_25zp7xTX2s,36
 pkgs/type_spec/__init__.py,sha256=h5DmJTca4QVV10sZR1x0-MlkZfuGYDfapR3zHvXfzto,19
 pkgs/type_spec/__main__.py,sha256=snBwPNvzU46Rqp4HlYb5Urv10Rs36gUbsPoFZfQcWRY,1162
-pkgs/type_spec/builder.py,sha256=yDtMWRd1EeHlcS9SqR1RKZ0ovuasmIgjpg0v4DvZ4mg,66086
+pkgs/type_spec/builder.py,sha256=i8uMhsG7cJa5Dxn25RQyJk9pgwvOd1mc-bO0O49tYxA,66376
 pkgs/type_spec/builder_types.py,sha256=EFXvwd3DhuFcqMtG12U9RHNYXHxi_g6kY5AVPBo3fCg,253
 pkgs/type_spec/config.py,sha256=8SQkCRAKzrcycR-9Tjt6J4-kNalN8IvbE7zo1FMlrZc,7226
 pkgs/type_spec/cross_output_links.py,sha256=exmHrdr-Mrq6FyMXzWgmuKhE-uqo_rN5k_PcirQgrGU,3037
 pkgs/type_spec/emit_io_ts.py,sha256=sIdS5f_C79SiaSqVS4HtQ8SFxA87ehXkj9CrmrpjTzg,5862
 pkgs/type_spec/emit_open_api.py,sha256=C3PRRSQJpHUZkS4CeEMGywhbLYs_MD3t6u-DlVIe2WA,28638
 pkgs/type_spec/emit_open_api_util.py,sha256=bTmRvrGP82-eB75hwf9ySI7pDEC87FNQTF18VKEWSXY,2367
-pkgs/type_spec/emit_python.py,sha256=K9fH9wUw1h3y5Uxv4Xg2KFEMhDySuhDmV0v0xSPssHM,58681
+pkgs/type_spec/emit_python.py,sha256=R66jEWVL0RcN15xDUvjNBe4YcT15hBfKyWU-k5ipbWo,59567
 pkgs/type_spec/emit_scss.py,sha256=6psmBAJyu7cvfQAIvIIMk2U6VFG_xeWy1Ea4afv0XyM,1497
 pkgs/type_spec/emit_typescript.py,sha256=Qu9f1Yv_wc449KHcBGO58hatbzvPvgMbZ11jferi3oY,11567
 pkgs/type_spec/emit_typescript_util.py,sha256=dRiOm38M4DR5NtvJyaEoGXE8Zzrfnae6yuDC76OzCeU,14915
@@ -103,9 +103,9 @@ pkgs/type_spec/open_api_util.py,sha256=LX6ny4xNOWXQO3qWVM8elrSDTqWsj_7jFplax2POS
 pkgs/type_spec/test.py,sha256=4ueujBq-pEgnX3Z69HyPmD-bullFXmpixcpVzfOkhP4,489
 pkgs/type_spec/util.py,sha256=Yc7QTZDjlu4ndOu3BwxLy5gRwG2mhnXN_s-iY9SYnIo,4834
 pkgs/type_spec/actions_registry/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pkgs/type_spec/actions_registry/__main__.py,sha256=tnVGrh8stDuhMGimx8dtjz8lPQ89sjHqezR6beTDlF8,4681
-pkgs/type_spec/actions_registry/emit_python.py,sha256=6vidMRCQk3qqo9Akz07fy_c2TUkkX_Pb8nXJhyBdDNE,3650
-pkgs/type_spec/actions_registry/emit_typescript.py,sha256=F0BRJpXVO8OJzqJgW_mv35D3scdhSpU0PN8HKO0cQIM,6626
+pkgs/type_spec/actions_registry/__main__.py,sha256=hWUnmiG68xeW5GhnIolfHOvPY2GI_kxMgqwtOSgLlvo,4723
+pkgs/type_spec/actions_registry/emit_python.py,sha256=atY1_Z9Ue6r1nBsMeoeX02vQCc_UpAnDLK7gdVNemZ4,3753
+pkgs/type_spec/actions_registry/emit_typescript.py,sha256=XqmbX5OrgjdCGQf4-iNc_aZ7JiXgaVGt1wJcpAUh8SA,6787
 pkgs/type_spec/parts/base.py.prepart,sha256=1BCL7DCPoot5CYR3O-IpcOvMVthe1cbTQkTiUeZvuk8,2316
 pkgs/type_spec/parts/base.ts.prepart,sha256=m5aM7AU_42w3Lpbm_s3GYnuFxH3EcUlnFq7zOIE1fDo,1820
 pkgs/type_spec/type_info/__main__.py,sha256=TLNvCHGcmaj_8Sj5bAQNpuNaaw2dpDzoFDWZds0V4Qo,1002
@@ -122,7 +122,7 @@ uncountable/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/core/__init__.py,sha256=RFv0kO6rKFf1PtBPu83hCGmxqkJamRtsgQ9_-ztw7tA,341
 uncountable/core/async_batch.py,sha256=9pYGFzVCQXt8059qFHgutweGIFPquJ5Xfq6NT5P-1K0,1206
 uncountable/core/client.py,sha256=53Voo2Y-q9iipwwJoHKb-idW41JUOeV8QeWnaLGFf48,16715
-uncountable/core/environment.py,sha256=OsUC5URj4VkYFbmZ89MF4gD3vMtIrFKKHSxljXi4oeI,1240
+uncountable/core/environment.py,sha256=Z9vu7JtnSDgQB_KKcZnjTFNyARXjRr_PDW9krwxNNAo,1132
 uncountable/core/file_upload.py,sha256=BOixJ09uWjZBTrNAlJqNvJEFvaNS2dt7eFq2rG6xP5g,5987
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
 uncountable/core/query/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -180,13 +180,7 @@ uncountable/integration/queue_runner/datastore/model.py,sha256=06_fGS0QTALEA_381
 uncountable/integration/secret_retrieval/__init__.py,sha256=sisf8cxCYcJmCXCKjEPoT2xIxm1j-Ru9iTfMVxH6NM0,171
 uncountable/integration/secret_retrieval/basic_secret.py,sha256=PfpeaEDwdi8nFlHUJnzCob9GYgGG5O_3mmcaxPPY6zI,3272
 uncountable/integration/secret_retrieval/retrieve_secret.py,sha256=c0VkCvppWKXwgxdblQuj2WhRaEGBPa7vLDoS6mlNOAQ,3919
-uncountable/integration/webhook_server/admin_auth.py,sha256=RYyK54wH5SHfzAVS4CeGgaKHeD4qAhqsvNZqkWuN34Q,1234
-uncountable/integration/webhook_server/entrypoint.py,sha256=LT0IDQtCqNoZBtzb-x6Ev96e1Xug3GRv4hn5NUbG_A4,4385
-uncountable/integration/webhook_server/admin_routes/__init__.py,sha256=iAgtrlljGe-yBCFNMwz8ZkH_KCC_DgW_hazAZgHb8I8,129
-uncountable/integration/webhook_server/admin_routes/_boundary.py,sha256=RKM5LfnMKslRvOBFVi_5iIKDB-3t1RpAVDO5sV1wLIo,1371
-uncountable/integration/webhook_server/admin_routes/health.py,sha256=OJ3fwzchLLeS33rIU23cdKnB3rtTxKlHZqqcpUA3Frc,963
-uncountable/integration/webhook_server/admin_routes/jobs.py,sha256=TR9VeHB1nZQ2KHTipSXWTxvELWRj9f1KxIOOtRTO3Q0,5112
-uncountable/integration/webhook_server/admin_routes/register.py,sha256=Jr1qVEhQgzp8T8zLifAQ5dDAlorFad-1su5kdyvtYWg,501
+uncountable/integration/webhook_server/entrypoint.py,sha256=_1360zLIvEVQVf8ToyK-E2fnaszGKs2HcxcZjNraADc,4268
 uncountable/types/__init__.py,sha256=7fIYhDsz995jFGK7wBXIVMzIxUJbltUmdqm9U6HA_QQ,14367
 uncountable/types/async_batch.py,sha256=yCCWrrLQfxXVqZp-KskxLBNkNmuELdz4PJjx8ULppgs,662
 uncountable/types/async_batch_processor.py,sha256=6Z1qx_H9IqLqp0rzpCtrRvw0DqQ2O5ctkEnw0yNVaV8,48727
@@ -201,7 +195,7 @@ uncountable/types/calculations.py,sha256=fApOFpgBemt_t7IVneVR0VdI3X5EOxiG6Xhzr6R
 uncountable/types/calculations_t.py,sha256=_K4MDVoxa6AgxlxPrEDBujWhEGWEhUQuE2-ZgiLg9OU,660
 uncountable/types/chemical_structure.py,sha256=ujyragaD26-QG5jgKnWhO7TN3N1V9b_04T2WhqNYxxo,281
 uncountable/types/chemical_structure_t.py,sha256=tbA7NBO-vv7Hnk3-ETxc6Pw3wqWeenBG27Y9_bdPCu4,785
-uncountable/types/client_base.py,sha256=X_1ccYzdrQHyn4tCyIGhxMlLHdoThx2lcav2tVvQE1A,138269
+uncountable/types/client_base.py,sha256=eM3WCohb2OESAE6OoV7MMWbXV4S6vGMZrEC0Bw4wXGU,138517
 uncountable/types/client_config.py,sha256=M7FZ0m_lGmBsIYcMn8pm92DdoVzrLpzd8sH6DqTQLKo,456
 uncountable/types/client_config_t.py,sha256=-YNp-zFvk5OL6_WGwGSx30ToK2bGIOh0kv_HN7KcCys,1796
 uncountable/types/condition_match.py,sha256=ekDzij7-e1PtVwIslSjD1T9fuBIDr5c_7QPdaasiGJw,262
@@ -213,7 +207,7 @@ uncountable/types/data_t.py,sha256=U_dIh5GcyaJlkrHEI4Qf4TOo1uJX07PYO70sYzzKtTo,2
 uncountable/types/dataset_measurements.py,sha256=CVwWePUC02B_R_UB86yfNBwnclR-5xpZaiCs2jWtu2A,512
 uncountable/types/dataset_measurements_t.py,sha256=iVwh-3Z23aXmPJcUJ3s9axPeOs627DElE_dhfj_PRlQ,1711
 uncountable/types/entity.py,sha256=Zclk1LYcRaYrMDhqyCjMSLEg0fE6_q8LHvV22Qvscgs,566
-uncountable/types/entity_t.py,sha256=s8qf6fRnpjyWTveVc0fzs3zL4rUMXD7YrKfwyqgcD6I,29331
+uncountable/types/entity_t.py,sha256=1ysk_fygPjWr3MQlWQYLCZkcpxHGEiwDGebqMTGfaDo,29968
 uncountable/types/experiment_groups.py,sha256=qUpFOx1AKgzaT_4khCOv5Xs6jwiQGbvHH-GUh3v1nv4,288
 uncountable/types/experiment_groups_t.py,sha256=QjZBdjnbsD1D-VpPSSV6hBg-Q7mujjKvYtdMewtXNVU,712
 uncountable/types/exports.py,sha256=VMmxUO2PpV1Y63hZ2AnVor4H-B6aswJ7YpSru_u89lU,334
@@ -298,8 +292,8 @@ uncountable/types/step_relationships.py,sha256=I6t7tj-VmHAhH8YpJPIUFBxPaRiXU_Zg-
 uncountable/types/step_relationships_t.py,sha256=pS9NJxYUab0XsHnninHXR8V-4ohofSP67iIDIhPR_QI,5306
 uncountable/types/units.py,sha256=yxuddayiE8cnzrjQiIsURisWc-Vm1F37uyS3fjM--Ao,254
 uncountable/types/units_t.py,sha256=6u4JAQFzVtaLcKo15JHsDXZ09no5nlHRQKAhGfiySPE,655
-uncountable/types/uploader.py,sha256=odT7wkBfXUf1MoFy6W5JzZ-WY8JX0vO6odGOS_C2Voo,1222
-uncountable/types/uploader_t.py,sha256=6YAJ-SqBLuzlP3e9VHtS3B2taIoqZO4UEk92d-P2-Y0,6138
+uncountable/types/uploader.py,sha256=k9OEe6jxjlBcc3iiMl5LoLFEIuI-Z0z6-ec3OYMa5nQ,1322
+uncountable/types/uploader_t.py,sha256=YrC8XQpMy6avkOTeBv-ma9iHE3pClJx6ZxgOVykKP3Q,6830
 uncountable/types/users.py,sha256=R-bFIh07mMl6HyxP8hKmlT-QMbBXZPZ7mVuOIeOlCsg,254
 uncountable/types/users_t.py,sha256=n2Ldc78cMU8gKynf2u9GvcKdelcKDRkjaOpAUO6Q5m4,678
 uncountable/types/webhook.py,sha256=ZLXFWaup9ueboI8D4-V3sRGZyb0pFXjzOTywlk8ZfF0,264
@@ -416,7 +410,7 @@ uncountable/types/api/recipes/get_recipe_output_metadata.py,sha256=IYm4DA7RN8gvx
 uncountable/types/api/recipes/get_recipes_data.py,sha256=B_rYOq7TfHVi4pyFN6r5zi5QDzMXhRDw2thlldwO0wQ,8859
 uncountable/types/api/recipes/lock_recipes.py,sha256=Ez5oWRwq_n16RHli56uaEURGrRYxyIS2VLOJEUQ9kaY,1885
 uncountable/types/api/recipes/remove_recipe_from_project.py,sha256=h1I3cLUb_zZ3QjpK7IKPT_1gTDPTPbGWg0OJN_rUn6o,1186
-uncountable/types/api/recipes/set_recipe_inputs.py,sha256=2NhVAfkIjVTtUKeV1aPLb42wP6XZBerbYKJ_K_HH3kA,2040
+uncountable/types/api/recipes/set_recipe_inputs.py,sha256=Qf3Ny6EkUKZ2QmRba8-K4AfpKHTwGexlxyzCOCjIQsM,2288
 uncountable/types/api/recipes/set_recipe_metadata.py,sha256=8BT03B4LC5bAJ7lu5LSxdAML5Q58Rejagpm_EVuJUYk,1356
 uncountable/types/api/recipes/set_recipe_output_annotations.py,sha256=058uueGk-0ILboQy29nnj_jTD6suROEXuXuYsoAib2w,3823
 uncountable/types/api/recipes/set_recipe_output_file.py,sha256=Jv6zkjvIxWM8o9RiX5p2SRkw1UJt9pzYC-KgG0BZoVY,1915
@@ -439,7 +433,7 @@ uncountable/types/api/uploader/complete_async_parse.py,sha256=Os1HAubxTlbut6Gylj
 uncountable/types/api/uploader/invoke_uploader.py,sha256=QU1KmAFgGHSWskZEKfaAww8rCj-hVNP0i4kMQE3x3Fc,1822
 uncountable/types/api/user/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/user/get_current_user_info.py,sha256=BT7bGp5SOHz3XdKWhhdZ6acnInafT2PgdJ-H4YH3W7c,1181
-uncountablepythonsdk-0.0.182.dist-info/METADATA,sha256=oGJSxcjq_RKLaOjhKMpBEnj3_JRLoewFIegeuf9_ZWI,2211
-uncountablepythonsdk-0.0.182.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
-uncountablepythonsdk-0.0.182.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-uncountablepythonsdk-0.0.182.dist-info/RECORD,,
+uncountablepythonsdk-0.0.184.dist-info/METADATA,sha256=mC6m3lEpjQAUsOqEx1Vcc7aoLKXtvnvRXmSUTJpJ1Rg,2211
+uncountablepythonsdk-0.0.184.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+uncountablepythonsdk-0.0.184.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+uncountablepythonsdk-0.0.184.dist-info/RECORD,,

uncountable/integration/webhook_server/admin_auth.py

@@ -1,38 +0,0 @@
-import functools
-import hmac
-from collections.abc import Callable
-
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.core.environment import get_integration_admin_token
-from uncountable.integration.http_server import HttpException
-
-_BEARER_PREFIX = "Bearer "
-
-
-def _extract_bearer_token(authorization_header: str | None) -> str | None:
-    if authorization_header is None:
-        return None
-    if not authorization_header.startswith(_BEARER_PREFIX):
-        return None
-    return authorization_header[len(_BEARER_PREFIX) :]
-
-
-def require_admin_auth(
-    view: Callable[..., ResponseReturnValue],
-) -> Callable[..., ResponseReturnValue]:
-    @functools.wraps(view)
-    def wrapper(*args: object, **kwargs: object) -> ResponseReturnValue:
-        admin_token = get_integration_admin_token()
-        if admin_token is None:
-            raise HttpException.service_unavailable()
-        provided_token = _extract_bearer_token(
-            flask.request.headers.get("Authorization")
-        )
-        if provided_token is None or not hmac.compare_digest(
-            provided_token, admin_token
-        ):
-            raise HttpException.unauthorized()
-        return view(*args, **kwargs)
-
-    return wrapper

uncountable/integration/webhook_server/admin_routes/__init__.py

@@ -1,3 +0,0 @@
-from uncountable.integration.webhook_server.admin_routes.register import (
-    register_admin_routes as register_admin_routes,
-)

uncountable/integration/webhook_server/admin_routes/_boundary.py

@@ -1,39 +0,0 @@
-import functools
-from collections.abc import Callable
-
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.integration.http_server import HttpException
-from uncountable.integration.queue_runner.command_server.types import (
-    CommandServerException,
-)
-from uncountable.integration.telemetry import Logger
-
-
-def register_admin_route(
-    *,
-    app: flask.Flask,
-    logger: Logger,
-    route: str,
-    endpoint: str,
-    view: Callable[..., ResponseReturnValue],
-    methods: list[str],
-) -> None:
-    @functools.wraps(view)
-    def wrapped(*args: object, **kwargs: object) -> ResponseReturnValue:
-        with logger.push_scope(route):
-            try:
-                return view(*args, **kwargs)
-            except HttpException as e:
-                return e.make_error_response()
-            except CommandServerException as e:
-                logger.log_exception(
-                    e,
-                    message=f"admin route {route} could not reach the command server",
-                )
-                return HttpException.bad_gateway().make_error_response()
-            except Exception as e:
-                logger.log_exception(e, message=f"unexpected error on {route}")
-                return HttpException.unknown_error().make_error_response()
-
-    app.add_url_rule(route, endpoint=endpoint, view_func=wrapped, methods=methods)

uncountable/integration/webhook_server/admin_routes/health.py

@@ -1,30 +0,0 @@
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.core.environment import get_local_admin_server_port
-from uncountable.integration.queue_runner.command_server.command_client import (
-    check_health,
-)
-from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_auth import require_admin_auth
-from uncountable.integration.webhook_server.admin_routes._boundary import (
-    register_admin_route,
-)
-
-_ADMIN_HEALTH_ROUTE = "/admin/health"
-
-
-@require_admin_auth
-def _handle_admin_health() -> ResponseReturnValue:
-    healthy = check_health(port=get_local_admin_server_port())
-    return flask.jsonify({"healthy": healthy})
-
-
-def register_health_routes(*, app: flask.Flask, logger: Logger) -> None:
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_HEALTH_ROUTE,
-        endpoint="admin_health",
-        view=_handle_admin_health,
-        methods=["GET"],
-    )

uncountable/integration/webhook_server/admin_routes/jobs.py

@@ -1,147 +0,0 @@
-import datetime
-
-import flask
-from flask.typing import ResponseReturnValue
-from uncountable.core.environment import get_local_admin_server_port
-from uncountable.integration.http_server import HttpException
-from uncountable.integration.queue_runner.command_server.command_client import (
-    send_job_cancellation_message,
-    send_job_queue_message,
-    send_list_queued_jobs_message,
-    send_retry_job_message,
-)
-from uncountable.integration.queue_runner.command_server.protocol.command_server_pb2 import (
-    ListQueuedJobsResult,
-)
-from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_auth import require_admin_auth
-from uncountable.integration.webhook_server.admin_routes._boundary import (
-    register_admin_route,
-)
-from uncountable.types import base_t, queued_job_t
-
-from pkgs.serialization_util import serialize_for_api
-
-_ADMIN_JOBS_ROUTE = "/admin/jobs"
-_ADMIN_ENQUEUE_JOB_ROUTE = "/admin/jobs/enqueue"
-_ADMIN_RETRY_JOB_ROUTE = "/admin/jobs/<job_uuid>/retry"
-_ADMIN_CANCEL_JOB_ROUTE = "/admin/jobs/<job_uuid>/cancel"
-_DEFAULT_JOBS_LIMIT = 100
-_VALID_JOB_STATUSES = {status.value for status in queued_job_t.JobStatus}
-
-
-def _parse_job_status(raw_status: str | None) -> queued_job_t.JobStatus | None:
-    if raw_status is None:
-        return None
-    if raw_status not in _VALID_JOB_STATUSES:
-        raise HttpException.bad_request(f"invalid status: {raw_status}")
-    return queued_job_t.JobStatus(raw_status)
-
-
-def _parse_non_negative_int(raw_value: str | None, *, default: int) -> int:
-    if raw_value is None:
-        return default
-    if not raw_value.isdigit():
-        raise HttpException.bad_request(
-            f"expected a non-negative integer, got: {raw_value}"
-        )
-    return int(raw_value)
-
-
-def _serialize_queued_job(
-    item: ListQueuedJobsResult.ListQueuedJobsResultItem,
-) -> dict[str, base_t.JsonValue]:
-    metadata = queued_job_t.QueuedJobMetadata(
-        queued_job_uuid=item.uuid,
-        job_ref_name=item.job_ref_name,
-        num_attempts=item.num_attempts,
-        status=queued_job_t.JobStatus(item.status),
-        submitted_at=item.submitted_at.ToDatetime(datetime.timezone.utc),
-    )
-    return serialize_for_api(metadata)
-
-
-@require_admin_auth
-def _handle_list_jobs() -> ResponseReturnValue:
-    status = _parse_job_status(flask.request.args.get("status"))
-    offset = _parse_non_negative_int(flask.request.args.get("offset"), default=0)
-    limit = _parse_non_negative_int(
-        flask.request.args.get("limit"), default=_DEFAULT_JOBS_LIMIT
-    )
-    jobs = send_list_queued_jobs_message(
-        status=status, offset=offset, limit=limit, port=get_local_admin_server_port()
-    )
-    return flask.jsonify({"jobs": [_serialize_queued_job(job) for job in jobs]})
-
-
-@require_admin_auth
-def _handle_enqueue_job() -> ResponseReturnValue:
-    body = flask.request.get_json(silent=True)
-    if not isinstance(body, dict):
-        raise HttpException.bad_request("expected a JSON object body")
-    job_ref_name = body.get("job_ref_name")
-    if not isinstance(job_ref_name, str):
-        raise HttpException.bad_request("job_ref_name is required")
-    webhook_payload = body.get("webhook_payload")
-    invocation_context: queued_job_t.InvocationContext = (
-        queued_job_t.InvocationContextManual()
-        if webhook_payload is None
-        else queued_job_t.InvocationContextWebhook(webhook_payload=webhook_payload)
-    )
-    queued_job_uuid = send_job_queue_message(
-        job_ref_name=job_ref_name,
-        payload=queued_job_t.QueuedJobPayload(invocation_context=invocation_context),
-        port=get_local_admin_server_port(),
-    )
-    return flask.jsonify({"queuedJobUuid": queued_job_uuid})
-
-
-@require_admin_auth
-def _handle_retry_job(job_uuid: str) -> ResponseReturnValue:
-    queued_job_uuid = send_retry_job_message(
-        job_uuid=job_uuid, port=get_local_admin_server_port()
-    )
-    return flask.jsonify({"queuedJobUuid": queued_job_uuid})
-
-
-@require_admin_auth
-def _handle_cancel_job(job_uuid: str) -> ResponseReturnValue:
-    status = send_job_cancellation_message(
-        queued_job_uuid=job_uuid, port=get_local_admin_server_port()
-    )
-    return flask.jsonify({"status": status.value})
-
-
-def register_job_routes(*, app: flask.Flask, logger: Logger) -> None:
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_JOBS_ROUTE,
-        endpoint="admin_list_jobs",
-        view=_handle_list_jobs,
-        methods=["GET"],
-    )
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_ENQUEUE_JOB_ROUTE,
-        endpoint="admin_enqueue_job",
-        view=_handle_enqueue_job,
-        methods=["POST"],
-    )
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_RETRY_JOB_ROUTE,
-        endpoint="admin_retry_job",
-        view=_handle_retry_job,
-        methods=["POST"],
-    )
-    register_admin_route(
-        app=app,
-        logger=logger,
-        route=_ADMIN_CANCEL_JOB_ROUTE,
-        endpoint="admin_cancel_job",
-        view=_handle_cancel_job,
-        methods=["POST"],
-    )

uncountable/integration/webhook_server/admin_routes/register.py

@@ -1,13 +0,0 @@
-import flask
-from opentelemetry.trace import get_current_span
-from uncountable.integration.telemetry import Logger
-from uncountable.integration.webhook_server.admin_routes.health import (
-    register_health_routes,
-)
-from uncountable.integration.webhook_server.admin_routes.jobs import register_job_routes
-
-
-def register_admin_routes(app: flask.Flask) -> None:
-    logger = Logger(get_current_span())
-    register_health_routes(app=app, logger=logger)
-    register_job_routes(app=app, logger=logger)