UncountablePythonSDK 0.0.128__tar.gz → 0.0.130__tar.gz

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/.github/workflows/documentation.yml

@@ -10,7 +10,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
-      - uses: actions/setup-python@82c7e631bb3cdc910f68e0081d67478d79c6982d # v5
+      - uses: actions/setup-python@e797f83bcb11b83ae66e0230d6156d7c80228e7c # v6
       - uses: extractions/setup-crate@4993624604c307fbca528d28a3c8b60fa5ecc859 # v1
         with:
           repo: casey/just@0.10

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/PKG-INFO

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

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

@@ -82,6 +82,7 @@ pkgs/strenum_compat/strenum_compat.py
 pkgs/type_spec/__init__.py
 pkgs/type_spec/__main__.py
 pkgs/type_spec/builder.py
+pkgs/type_spec/builder_types.py
 pkgs/type_spec/config.py
 pkgs/type_spec/cross_output_links.py
 pkgs/type_spec/emit_io_ts.py
@@ -207,6 +208,8 @@ uncountable/types/integrations.py
 uncountable/types/integrations_t.py
 uncountable/types/job_definition.py
 uncountable/types/job_definition_t.py
+uncountable/types/listing.py
+uncountable/types/listing_t.py
 uncountable/types/notifications.py
 uncountable/types/notifications_t.py
 uncountable/types/outputs.py
@@ -243,6 +246,8 @@ uncountable/types/secret_retrieval.py
 uncountable/types/secret_retrieval_t.py
 uncountable/types/sockets.py
 uncountable/types/sockets_t.py
+uncountable/types/structured_filters.py
+uncountable/types/structured_filters_t.py
 uncountable/types/units.py
 uncountable/types/units_t.py
 uncountable/types/uploader.py
@@ -268,6 +273,7 @@ uncountable/types/api/entity/create_or_update_entity.py
 uncountable/types/api/entity/export_entities.py
 uncountable/types/api/entity/get_entities_data.py
 uncountable/types/api/entity/grant_entity_permissions.py
+uncountable/types/api/entity/list_aggregate.py
 uncountable/types/api/entity/list_entities.py
 uncountable/types/api/entity/lock_entity.py
 uncountable/types/api/entity/lookup_entity.py
@@ -302,6 +308,8 @@ uncountable/types/api/integrations/push_notification.py
 uncountable/types/api/integrations/register_sockets_token.py
 uncountable/types/api/material_families/__init__.py
 uncountable/types/api/material_families/update_entity_material_families.py
+uncountable/types/api/notebooks/__init__.py
+uncountable/types/api/notebooks/add_notebook_content.py
 uncountable/types/api/outputs/__init__.py
 uncountable/types/api/outputs/get_output_data.py
 uncountable/types/api/outputs/get_output_names.py

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/docs/requirements.txt

@@ -1,4 +1,4 @@
-furo==2025.7.19
+furo==2025.9.25
 myst-parser==4.0.1
 sphinx-autoapi==3.6.0
 sphinx-copybutton==0.5.2

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/examples/integration-server/jobs/materials_auto/example_instrument.py

@@ -3,8 +3,6 @@ import time
 from dataclasses import dataclass
 from decimal import Decimal
 
-from pkgs.argument_parser.argument_parser import CachedParser
-from pkgs.serialization_util import serialize_for_api
 from uncountable.integration.job import JobArguments, WebhookJob, register_job
 from uncountable.types import (
     base_t,
@@ -16,6 +14,9 @@ from uncountable.types.integration_session_t import IntegrationSessionInstrument
 from websockets.sync.client import connect
 from websockets.typing import Data
 
+from pkgs.argument_parser.argument_parser import CachedParser
+from pkgs.serialization_util import serialize_for_api
+
 
 @dataclass(kw_only=True)
 class InstrumentPayload:
@@ -34,7 +35,7 @@ class InstrumentExample(WebhookJob[InstrumentPayload]):
         def parse_message(message: Data) -> sockets_t.SocketEventData | None:
             try:
                 return parser.parse_api(json.loads(message)).data
-            except ValueError as e:
+            except ValueError:
                 return None
 
         integration_session = IntegrationSessionInstrument(

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/examples/integration-server/jobs/materials_auto/example_parse.py

@@ -1,11 +1,14 @@
 from dataclasses import dataclass
+from decimal import Decimal
 
 from uncountable.integration.job import JobArguments, WebhookJob, register_job
 from uncountable.types import (
     base_t,
+    entity_t,
     generic_upload_t,
     identifier_t,
     job_definition_t,
+    notifications_t,
     uploader_t,
 )
 
@@ -20,9 +23,40 @@ class ParseExample(WebhookJob[ParsePayload]):
     def run(
         self, args: JobArguments, payload: ParsePayload
     ) -> job_definition_t.JobResult:
+        user_id: base_t.ObjectId | None = None
+        recipe_id: base_t.ObjectId | None = None
+        file_name: str | None = None
+        data = args.client.get_entities_data(
+            entity_ids=[payload.async_job_id], entity_type=entity_t.EntityType.ASYNC_JOB
+        )
+        for field_value in data.entity_details[0].field_values:
+            if field_value.field_ref_name == "core_async_job_jobData":
+                assert isinstance(field_value.value, dict)
+                assert isinstance(field_value.value["user_id"], int)
+                user_id = field_value.value["user_id"]
+            elif (
+                field_value.field_ref_name
+                == "unc_async_job_custom_parser_recipe_ids_in_view"
+            ):
+                if field_value.value is None:
+                    continue
+                assert isinstance(field_value.value, list)
+                if len(field_value.value) > 0:
+                    assert isinstance(field_value.value[0], int)
+                    recipe_id = field_value.value[0]
+            elif field_value.field_ref_name == "unc_async_job_custom_parser_input_file":
+                assert isinstance(field_value.value, list)
+                assert len(field_value.value) == 1
+                assert isinstance(field_value.value[0], dict)
+                assert isinstance(field_value.value[0]["name"], str)
+                file_name = field_value.value[0]["name"]
+
+        assert user_id is not None
+        assert file_name is not None
+
         dummy_parsed_file_data: list[uploader_t.ParsedFileData] = [
             uploader_t.ParsedFileData(
-                file_name="my_file_to_upload.xlsx",
+                file_name=file_name,
                 file_structures=[
                     uploader_t.DataChannel(
                         type=uploader_t.StructureElementType.CHANNEL,
@@ -68,18 +102,37 @@ class ParseExample(WebhookJob[ParsePayload]):
                             data=uploader_t.StringValue(value="my_file_to_upload.xlsx"),
                         ),
                     ),
+                    uploader_t.HeaderEntry(
+                        type=uploader_t.StructureElementType.HEADER,
+                        value=uploader_t.NumericHeaderData(
+                            name="file structure number",
+                            data=uploader_t.DecimalValue(value=Decimal(99)),
+                        ),
+                    ),
                 ],
             )
         ]
 
-        args.client.complete_async_parse(
+        complete_async_parse_req = args.batch_processor.complete_async_parse(
             parsed_file_data=dummy_parsed_file_data,
             async_job_key=identifier_t.IdentifierKeyId(id=payload.async_job_id),
             upload_destination=generic_upload_t.UploadDestinationRecipe(
-                recipe_key=identifier_t.IdentifierKeyId(id=1)
+                recipe_key=identifier_t.IdentifierKeyId(id=recipe_id or 1)
             ),
         )
 
+        args.batch_processor.push_notification(
+            depends_on=[complete_async_parse_req.batch_reference],
+            notification_targets=[
+                notifications_t.NotificationTargetUser(
+                    user_key=identifier_t.IdentifierKeyId(id=user_id)
+                )
+            ],
+            subject="Upload complete",
+            message="Your file has been uploaded",
+            display_notice=True,
+        )
+
         return job_definition_t.JobResult(success=True)
 
     @property

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/examples/integration-server/jobs/materials_auto/example_predictions.py

@@ -1,6 +1,6 @@
 import random
 from dataclasses import dataclass
-from datetime import datetime
+from datetime import UTC, datetime
 from decimal import Decimal
 
 from uncountable.integration.job import JobArguments, WebhookJob, register_job
@@ -25,7 +25,7 @@ class PredictionsExample(WebhookJob[PredictionsPayload]):
         self, args: JobArguments, payload: PredictionsPayload
     ) -> job_definition_t.JobResult:
         recipe_data = args.client.get_recipes_data(recipe_ids=payload.recipe_ids)
-        formatted_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        formatted_datetime = datetime.now(UTC).strftime("%Y-%m-%d %H:%M:%S")
 
         for recipe in recipe_data.recipes:
             test_sample_name = f"Predictions Model ({formatted_datetime})"

uncountablepythonsdk-0.0.130/examples/integration-server/jobs/materials_auto/example_runsheet_wh.py

@@ -0,0 +1,39 @@
+from io import BytesIO
+
+from uncountable.core.file_upload import DataFileUpload, FileUpload
+from uncountable.integration.job import JobArguments, RunsheetWebhookJob, register_job
+from uncountable.types import (
+    download_file_t,
+    entity_t,
+    identifier_t,
+    webhook_job_t,
+)
+
+
+@register_job
+class StandardRunsheetGenerator(RunsheetWebhookJob):
+    def build_runsheet(
+        self,
+        *,
+        args: JobArguments,
+        payload: webhook_job_t.RunsheetWebhookPayload,
+    ) -> FileUpload:
+        args.logger.log_info("Retrieving pre-exported runsheet file from async job")
+
+        file_query = download_file_t.FileDownloadQueryEntityField(
+            entity=entity_t.EntityIdentifier(
+                type=entity_t.EntityType.ASYNC_JOB,
+                identifier_key=identifier_t.IdentifierKeyId(id=payload.async_job_id),
+            ),
+            field_key=identifier_t.IdentifierKeyRefName(
+                ref_name="unc_async_job_export_runsheet_recipe_export"
+            ),
+        )
+
+        downloaded_files = args.client.download_files(file_query=file_query)
+
+        file_data = downloaded_files[0].data.read()
+        return DataFileUpload(
+            data=BytesIO(file_data),
+            name=downloaded_files[0].name,
+        )

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/examples/integration-server/pyproject.toml

@@ -115,7 +115,6 @@ lint.ignore = [
     "PD010", # .pivottable.  Should add
     "PD011", # use .to_numpy.  Skip
     "PD015", # use .merge.  Should add
-    "PD901", # avoid generic df name.  Skip
     "PERF203", # avoid try except in loop. Skip
     "PERF401", # use list comprehension. Skip
     "PERF402", # use list.copy.  Skip
@@ -213,6 +212,7 @@ exclude = [
 
 [tool.ruff.lint.isort]
 split-on-trailing-comma = true
+known-first-party = ["pkgs"]
 
 [tool.ruff.lint.mccabe]
 max-complexity = 130 # goal would be to bring this down to ~50 or so

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/argument_parser/argument_parser.py

@@ -168,6 +168,23 @@ def _invoke_membership_parser(
     raise ValueError(f"Expected value from {expected_values} but got value {value}")
 
 
+# Uses `is` to compare
+def _build_identity_parser(
+    identity_value: T,
+) -> ParserFunction[T]:
+    def parse(value: typing.Any) -> T:
+        if value is identity_value:
+            return identity_value
+        raise ValueError(
+            f"Expected value {identity_value} (type: {type(identity_value)}) but got value {value} (type: {type(value)})"
+        )
+
+    return parse
+
+
+NONE_IDENTITY_PARSER = _build_identity_parser(None)
+
+
 def _build_parser_discriminated_union(
     context: ParserContext,
     discriminator_raw: str,
@@ -243,8 +260,10 @@ def _build_parser_inner(
             for field_name, field_parser in field_parsers
         })
 
+    # IMPROVE: unclear why we need == here
     if parsed_type == type(None):  # noqa: E721
-        return lambda value: _invoke_membership_parser({None}, value)
+        # Need to convince type checker that parsed_type is type(None)
+        return typing.cast(ParserFunction[T], NONE_IDENTITY_PARSER)
 
     origin = typing.get_origin(parsed_type)
     if origin is tuple:

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/serialization_util/serialization_helpers.py

@@ -85,7 +85,9 @@ def _serialize_dataclass(d: Any) -> dict[str, JsonValue]:
 
 
 def _to_string_value(value: Any) -> str:
-    assert isinstance(value, (Decimal, int))
+    assert isinstance(value, (Decimal, int)), (
+        f"Expecting decimal or int, received: {value} (type={type(value)})"
+    )
     return str(value)
 
 

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/type_spec/builder.py

@@ -13,7 +13,7 @@ from enum import Enum, StrEnum, auto
 from typing import Any, Self
 
 from . import util
-from .cross_output_links import CrossOutputPaths
+from .builder_types import CrossOutputPaths
 from .non_discriminated_union_exceptions import NON_DISCRIMINATED_UNION_EXCEPTIONS
 from .util import parse_type_str
 
@@ -1101,7 +1101,7 @@ def _parse_const(
         elif const_type.defn_type.name == BaseTypeName.s_dict:
             assert isinstance(value, dict)
             builder.ensure(
-                len(const_type.parameters) == 2, "constant-dict-expects-one-type"
+                len(const_type.parameters) == 2, "constant-dict-expects-two-types"
             )
             key_type = const_type.parameters[0]
             value_type = const_type.parameters[1]
@@ -1150,6 +1150,11 @@ def _parse_const(
             )
             return value
 
+        if not const_type.is_base:
+            # IMPROVE: validate the object type properties before emission stage
+            builder.ensure(isinstance(value, dict), "invalid value for object constant")
+            return value
+
     raise Exception("unsupported-const-scalar-type", const_type)
 
 
@@ -1271,7 +1276,8 @@ class SpecNamespace:
 
             assert util.is_valid_type_name(name), f"{name} is not a valid type name"
             assert name not in self.types, f"{name} is duplicate"
-            defn_type = defn["type"]
+            defn_type = defn.get("type")
+            assert isinstance(defn_type, str), f"{name} requires a string type"
             spec_type: SpecTypeDefn
             if defn_type == DefnTypeName.s_alias:
                 spec_type = SpecTypeDefnAlias(self, name)

uncountablepythonsdk-0.0.130/pkgs/type_spec/builder_types.py

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+
+
+@dataclass(kw_only=True, frozen=True)
+class CrossOutputPaths:
+    python_types_output: str
+    typescript_types_output: str
+    typescript_routes_output_by_endpoint: dict[str, str]
+    typespec_files_input: list[str]

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/type_spec/cross_output_links.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
 import os
-from dataclasses import dataclass
 
-from pkgs.type_spec import builder
+from . import builder
+from .builder_types import CrossOutputPaths
 
 
 def get_python_stub_file_path(
@@ -17,14 +17,6 @@ def get_python_stub_file_path(
     return api_stub_file
 
 
-@dataclass(kw_only=True, frozen=True)
-class CrossOutputPaths:
-    python_types_output: str
-    typescript_types_output: str
-    typescript_routes_output_by_endpoint: dict[str, str]
-    typespec_files_input: list[str]
-
-
 def get_python_api_file_path(
     cross_output_paths: CrossOutputPaths,
     namespace: builder.SpecNamespace,

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/type_spec/emit_open_api.py

@@ -610,18 +610,6 @@ def _emit_type(
     ctx.types[stype.name] = final_type
 
 
-def _emit_constant(ctx: EmitOpenAPIContext, sconst: builder.SpecConstant) -> None:
-    if sconst.value_type.is_base_type(builder.BaseTypeName.s_string):
-        value = util.encode_common_string(cast(str, sconst.value))
-    elif sconst.value_type.is_base_type(builder.BaseTypeName.s_integer):
-        value = str(sconst.value)
-    else:
-        raise Exception("invalid constant type", sconst.name)
-
-    const_name = sconst.name.upper()
-    print("_emit_constant", value, const_name)
-
-
 def _emit_endpoint(
     gctx: EmitOpenAPIGlobalContext,
     ctx: EmitOpenAPIContext,

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/type_spec/emit_python.py

@@ -35,6 +35,11 @@ QUEUED_BATCH_REQUEST_STYPE = builder.SpecTypeDefnObject(
     namespace=ASYNC_BATCH_TYPE_NAMESPACE, name="QueuedAsyncBatchRequest"
 )
 
+CLIENT_CONFIG_TYPE_NAMESPACE = builder.SpecNamespace(name="client_config")
+REQUEST_OPTIONS_STYPE = builder.SpecTypeDefnObject(
+    namespace=CLIENT_CONFIG_TYPE_NAMESPACE, name="RequestOptions"
+)
+
 
 @dataclasses.dataclass(kw_only=True)
 class TrackingContext:
@@ -117,26 +122,36 @@ def _check_type_match(stype: builder.SpecType, value: Any) -> bool:
     raise Exception("invalid type", stype, value)
 
 
-def _emit_value(ctx: TrackingContext, stype: builder.SpecType, value: Any) -> str:
+def _emit_value(
+    ctx: TrackingContext, stype: builder.SpecType, value: Any, indent: int = 0
+) -> str:
     literal = builder.unwrap_literal_type(stype)
     if literal is not None:
         return _emit_value(ctx, literal.value_type, literal.value)
 
     if stype.is_base_type(builder.BaseTypeName.s_string):
-        assert isinstance(value, str)
+        assert isinstance(value, str), (
+            f"Expected str value for {stype.name} but got {value}"
+        )
         return util.encode_common_string(value)
     elif stype.is_base_type(builder.BaseTypeName.s_integer):
-        assert isinstance(value, int)
+        assert isinstance(value, int), (
+            f"Expected int value for {stype.name} but got {value}"
+        )
         return str(value)
     elif stype.is_base_type(builder.BaseTypeName.s_boolean):
-        assert isinstance(value, bool)
+        assert isinstance(value, bool), (
+            f"Expected bool value for {stype.name} but got {value}"
+        )
         return "True" if value else "False"
     elif stype.is_base_type(builder.BaseTypeName.s_decimal) or stype.is_base_type(
         builder.BaseTypeName.s_lossy_decimal
     ):
         # Note that decimal requires the `!decimal 123.12` style notation in the YAML
         # file since PyYaml parses numbers as float, unfortuantely
-        assert isinstance(value, (Decimal, int))
+        assert isinstance(value, (Decimal, int)), (
+            f"Expected decimal value for {stype.name} but got {value} (type: {type(value)})"
+        )
         if isinstance(value, int):
             # skip quotes for integers
             return f"Decimal({value})"
@@ -151,14 +166,14 @@ def _emit_value(ctx: TrackingContext, stype: builder.SpecType, value: Any) -> st
             key_type = stype.parameters[0]
             value_type = stype.parameters[1]
             return (
-                "{\n    "
-                + ",\n    ".join(
+                f"{{\n{INDENT * (indent + 1)}"
+                + f",\n{INDENT * (indent + 1)}".join(
                     _emit_value(ctx, key_type, dkey)
                     + ": "
-                    + _emit_value(ctx, value_type, dvalue)
+                    + _emit_value(ctx, value_type, dvalue, indent=indent + 1)
                     for dkey, dvalue in value.items()
                 )
-                + "\n}"
+                + f"\n{INDENT * indent}}}"
             )
 
         if stype.defn_type.is_base_type(builder.BaseTypeName.s_optional):
@@ -184,6 +199,34 @@ def _emit_value(ctx: TrackingContext, stype: builder.SpecType, value: Any) -> st
         return f"{refer_to(ctx, stype)}.{_resolve_enum_name(value, stype.name_case)}"
     elif isinstance(stype, builder.SpecTypeDefnAlias):
         return _emit_value(ctx, stype.alias, value)
+    elif isinstance(stype, builder.SpecTypeDefnObject):
+        assert isinstance(value, dict), (
+            f"Expected dict value for {stype.name} but got {value}"
+        )
+        if not stype.is_hashable:
+            raise Exception("invalid constant object type, non-hashable", value, stype)
+        obj_out = f"{refer_to(ctx, stype)}("
+        emitted_fields: set[str] = set()
+        for prop_name, prop in (stype.properties or {}).items():
+            if prop_name not in value:
+                continue
+            else:
+                value_to_emit = value[prop_name]
+            emitted_fields.add(prop_name)
+            py_name = python_field_name(prop.name, prop.name_case)
+            obj_out += f"\n{INDENT * (indent + 1)}{py_name}={_emit_value(ctx, prop.spec_type, value_to_emit, indent=indent + 1)},"
+        whitespace = f"\n{INDENT * indent}" if len(emitted_fields) > 0 else ""
+        obj_out += f"{whitespace})"
+
+        if emitted_fields != set(value.keys()):
+            raise Exception(
+                "invalid object type, extra fields found:",
+                value,
+                stype,
+                set(value.keys()) - emitted_fields,
+            )
+
+        return obj_out
 
     raise Exception("invalid constant type", value, stype)
 
@@ -471,6 +514,19 @@ def _emit_endpoint_invocation_function_signature(
         else []
     ) + (extra_params if extra_params is not None else [])
 
+    request_options_property = builder.SpecProperty(
+        name="_request_options",
+        label="_request_options",
+        spec_type=REQUEST_OPTIONS_STYPE,
+        extant=builder.PropertyExtant.optional,
+        convert_value=builder.PropertyConvertValue.auto,
+        name_case=builder.NameCase.convert,
+        default=None,
+        has_default=True,
+        desc=None,
+    )
+    all_arguments.append(request_options_property)
+
     # All endpoints share a function name
     function = endpoint.path_per_api_endpoint[endpoint.default_endpoint_key].function
     assert function is not None
@@ -641,6 +697,7 @@ def _emit_endpoint_invocation_function(
             method={refer_to(ctx=ctx, stype=endpoint_method_stype)},
             endpoint={refer_to(ctx=ctx, stype=endpoint_path_stype)},
             args=args,
+            request_options=_request_options,
         )
         return self.do_request(api_request=api_request, return_type={refer_to(ctx=ctx, stype=data_type)})"""
     )
@@ -1396,7 +1453,7 @@ CLIENT_CLASS_IMPORTS = [
     "import dataclasses",
 ]
 ASYNC_BATCH_PROCESSOR_FILENAME = "async_batch_processor"
-ASYNC_BATCH_PROCESSOR_IMPORTS = [
+ASYNC_BATCH_PROCESSOR_BASE_IMPORTS = [
     "import uuid",
     "from abc import ABC, abstractmethod",
     "from pkgs.serialization_util import serialize_for_api",
@@ -1434,8 +1491,11 @@ def _emit_async_batch_processor(
         config=config,
     )
 
+    imports = ASYNC_BATCH_PROCESSOR_BASE_IMPORTS.copy()
+    if ctx.use_dataclass:
+        imports.append("import dataclasses")
     async_batch_processor_out.write(
-        f"""{LINE_BREAK.join(ASYNC_BATCH_PROCESSOR_IMPORTS)}
+        f"""{LINE_BREAK.join(imports)}
 
 
 class AsyncBatchProcessorBase(ABC):
@@ -1498,6 +1558,7 @@ class APIRequest:
     method: str
     endpoint: str
     args: typing.Any
+    request_options: {refer_to(ctx=ctx, stype=REQUEST_OPTIONS_STYPE)} | None = None
 
 
 class ClientMethods(ABC):

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/type_spec/emit_typescript_util.py

@@ -3,7 +3,7 @@ import typing
 from dataclasses import dataclass, field
 
 from . import builder, util
-from .cross_output_links import CrossOutputPaths
+from .builder_types import CrossOutputPaths
 
 INDENT = "  "
 
@@ -51,7 +51,10 @@ def ts_name(name: str, name_case: builder.NameCase) -> str:
 
 
 def emit_value_ts(
-    ctx: EmitTypescriptContext, stype: builder.SpecType, value: typing.Any
+    ctx: EmitTypescriptContext,
+    stype: builder.SpecType,
+    value: typing.Any,
+    indent: int = 0,
 ) -> str:
     """Mimics emit_python even if not all types are used in TypeScript yet"""
     literal = builder.unwrap_literal_type(stype)
@@ -88,17 +91,17 @@ def emit_value_ts(
                 raise Exception("invalid dict keys -- dict keys must be string or enum")
 
             return (
-                "{\n\t"
-                + ",\n\t".join(
+                f"{{\n{INDENT * (indent + 1)}"
+                + f",\n{INDENT * (indent + 1)}".join(
                     (
                         f"[{emit_value_ts(ctx, key_type, dkey)}]: "
                         if not key_type.is_base_type(builder.BaseTypeName.s_string)
                         else f"{dkey}: "
                     )
-                    + emit_value_ts(ctx, value_type, dvalue)
+                    + emit_value_ts(ctx, value_type, dvalue, indent=indent + 1)
                     for dkey, dvalue in value.items()
                 )
-                + "\n}"
+                + f"\n{INDENT * (indent)}}}"
             )
 
         if stype.defn_type.is_base_type(builder.BaseTypeName.s_optional):
@@ -109,6 +112,25 @@ def emit_value_ts(
 
     elif isinstance(stype, builder.SpecTypeDefnStringEnum):
         return f"{refer_to(ctx, stype)}.{ts_enum_name(value, stype.name_case)}"
+    elif isinstance(stype, builder.SpecTypeDefnObject):
+        assert isinstance(value, dict), (
+            f"Expected dict value for {stype.name} but got {value}"
+        )
+        obj_out = "{"
+        did_emit = False
+        for prop_name, prop in (stype.properties or {}).items():
+            if prop_name not in value and prop.has_default:
+                value_to_emit = prop.default
+            elif prop_name not in value:
+                continue
+            else:
+                value_to_emit = value[prop_name]
+            did_emit = True
+            typescript_name = ts_name(prop.name, prop.name_case)
+            obj_out += f"\n{INDENT * (indent + 1)}{typescript_name}: {emit_value_ts(ctx, prop.spec_type, value_to_emit, indent=indent + 1)},"
+        whitespace = f"\n{INDENT * indent}" if did_emit else ""
+        obj_out += f"{whitespace}}} as const"
+        return obj_out
 
     raise Exception("invalid constant type", value, stype, type(stype))
 

{uncountablepythonsdk-0.0.128 → uncountablepythonsdk-0.0.130}/pkgs/type_spec/load_types.py

@@ -7,8 +7,8 @@ from shelljob import fs
 from pkgs.serialization import yaml
 
 from .builder import SpecBuilder
+from .builder_types import CrossOutputPaths
 from .config import Config
-from .cross_output_links import CrossOutputPaths
 
 ext_map = {
     ".ts": "typescript",