UncountablePythonSDK 0.0.116__tar.gz → 0.0.117__tar.gz

{uncountablepythonsdk-0.0.116 → uncountablepythonsdk-0.0.117}/PKG-INFO

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

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

@@ -194,6 +194,8 @@ uncountable/types/inputs.py
 uncountable/types/inputs_t.py
 uncountable/types/integration_server.py
 uncountable/types/integration_server_t.py
+uncountable/types/integrations.py
+uncountable/types/integrations_t.py
 uncountable/types/job_definition.py
 uncountable/types/job_definition_t.py
 uncountable/types/outputs.py
@@ -279,6 +281,8 @@ uncountable/types/api/inputs/set_input_attribute_values.py
 uncountable/types/api/inputs/set_input_category.py
 uncountable/types/api/inputs/set_input_subcategories.py
 uncountable/types/api/inputs/set_intermediate_type.py
+uncountable/types/api/integrations/__init__.py
+uncountable/types/api/integrations/publish_realtime_data.py
 uncountable/types/api/material_families/__init__.py
 uncountable/types/api/material_families/update_entity_material_families.py
 uncountable/types/api/outputs/__init__.py

{uncountablepythonsdk-0.0.116 → uncountablepythonsdk-0.0.117}/pkgs/type_spec/emit_open_api.py

@@ -479,9 +479,18 @@ def _emit_type(
         return
 
     if isinstance(stype, builder.SpecTypeDefnUnion):
+        converted_discriminator_map: dict[str, OpenAPIRefType] = dict()
+        if stype.discriminator_map is not None:
+            for discriminator_value, base_type in stype.discriminator_map.items():
+                converted_base_type = open_api_type(ctx, base_type, config=config)
+                assert isinstance(converted_base_type, OpenAPIRefType)
+                converted_discriminator_map[discriminator_value] = converted_base_type
         ctx.types[stype.name] = OpenAPIUnionType(
             [open_api_type(ctx, p, config=config) for p in stype.types],
-            discrimnator=stype.discriminator,
+            discriminator=stype.discriminator,
+            discriminator_map=converted_discriminator_map
+            if stype.discriminator_map is not None
+            else None,
         )
         return
 

{uncountablepythonsdk-0.0.116 → uncountablepythonsdk-0.0.117}/pkgs/type_spec/open_api_util.py

@@ -223,19 +223,26 @@ class OpenAPIUnionType(OpenAPIType):
         base_types: list[OpenAPIType],
         description: str | None = None,
         nullable: bool = False,
-        discrimnator: str | None = None,
-        discriminator_map: dict[str, OpenAPIType] | None = None,
+        discriminator: str | None = None,
+        discriminator_map: dict[str, OpenAPIRefType] | None = None,
     ) -> None:
         self.base_types = base_types
-        self._discriminator = discrimnator
+        self._discriminator = discriminator
+        self._discriminator_map = discriminator_map
         super().__init__(description=description, nullable=nullable)
 
     def asdict(self) -> dict[str, object]:
         # TODO: use parents description and nullable
         return {
             "oneOf": [base_type.asdict() for base_type in self.base_types],
-            "discriminator": {"propertyName": self._discriminator}
-            if self._discriminator is not None
+            "discriminator": {
+                "propertyName": self._discriminator,
+                "mapping": {
+                    discriminator_value: base_type.source
+                    for discriminator_value, base_type in self._discriminator_map.items()
+                },
+            }
+            if self._discriminator is not None and self._discriminator_map is not None
             else None,
         }
 

{uncountablepythonsdk-0.0.116 → uncountablepythonsdk-0.0.117}/uncountable/types/__init__.py

@@ -63,6 +63,7 @@ from . import identifier_t as identifier_t
 from . import input_attributes_t as input_attributes_t
 from . import inputs_t as inputs_t
 from . import integration_server_t as integration_server_t
+from . import integrations_t as integrations_t
 from .api.uploader import invoke_uploader as invoke_uploader_t
 from . import job_definition_t as job_definition_t
 from .api.entity import list_entities as list_entities_t
@@ -76,6 +77,7 @@ from . import overrides_t as overrides_t
 from . import permissions_t as permissions_t
 from . import phases_t as phases_t
 from . import post_base_t as post_base_t
+from .api.integrations import publish_realtime_data as publish_realtime_data_t
 from . import queued_job_t as queued_job_t
 from . import recipe_identifiers_t as recipe_identifiers_t
 from . import recipe_inputs_t as recipe_inputs_t
@@ -180,6 +182,7 @@ __all__: list[str] = [
     "input_attributes_t",
     "inputs_t",
     "integration_server_t",
+    "integrations_t",
     "invoke_uploader_t",
     "job_definition_t",
     "list_entities_t",
@@ -193,6 +196,7 @@ __all__: list[str] = [
     "permissions_t",
     "phases_t",
     "post_base_t",
+    "publish_realtime_data_t",
     "queued_job_t",
     "recipe_identifiers_t",
     "recipe_inputs_t",

uncountablepythonsdk-0.0.117/uncountable/types/api/integrations/publish_realtime_data.py

@@ -0,0 +1,41 @@
+# DO NOT MODIFY -- This file is generated by type_spec
+# ruff: noqa: E402 Q003
+# fmt: off
+# isort: skip_file
+from __future__ import annotations
+import typing  # noqa: F401
+import datetime  # noqa: F401
+from decimal import Decimal  # noqa: F401
+import dataclasses
+from pkgs.serialization import serial_class
+from ... import base_t
+from ... import integrations_t
+
+__all__: list[str] = [
+    "Arguments",
+    "Data",
+    "ENDPOINT_METHOD",
+    "ENDPOINT_PATH",
+]
+
+ENDPOINT_METHOD = "POST"
+ENDPOINT_PATH = "api/external/integrations/publish_realtime_data"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.integrations.publish_realtime_data.Arguments",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Arguments:
+    data_package: integrations_t.DataPackage
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.api.integrations.publish_realtime_data.Data",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class Data:
+    pass
+# DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.117/uncountable/types/api/uploader/__init__.py

@@ -0,0 +1 @@
+# DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.117/uncountable/types/integrations.py

@@ -0,0 +1,10 @@
+# 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 .integrations_t import DataPackageType as DataPackageType
+from .integrations_t import DataPackageBase as DataPackageBase
+from .integrations_t import DataPackageNumericReading as DataPackageNumericReading
+from .integrations_t import DataPackage as DataPackage
+# DO NOT MODIFY -- This file is generated by type_spec

uncountablepythonsdk-0.0.117/uncountable/types/integrations_t.py

@@ -0,0 +1,62 @@
+# 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
+import dataclasses
+from pkgs.serialization import serial_class
+from pkgs.serialization import serial_union_annotation
+from . import base_t
+from . import entity_t
+
+__all__: list[str] = [
+    "DataPackage",
+    "DataPackageBase",
+    "DataPackageNumericReading",
+    "DataPackageType",
+]
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+class DataPackageType(StrEnum):
+    NUMERIC_READING = "numeric_reading"
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.integrations.DataPackageBase",
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class DataPackageBase:
+    type: DataPackageType
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+@serial_class(
+    named_type_path="sdk.integrations.DataPackageNumericReading",
+    to_string_values={"value"},
+    parse_require={"type"},
+)
+@dataclasses.dataclass(slots=base_t.ENABLE_SLOTS, kw_only=True)  # type: ignore[literal-required]
+class DataPackageNumericReading(DataPackageBase):
+    type: typing.Literal[DataPackageType.NUMERIC_READING] = DataPackageType.NUMERIC_READING
+    value: Decimal
+    target_entity: entity_t.EntityIdentifier
+
+
+# DO NOT MODIFY -- This file is generated by type_spec
+DataPackage = typing.Annotated[
+    typing.Union[DataPackageNumericReading],
+    serial_union_annotation(
+    named_type_path="sdk.integrations.DataPackage",
+        discriminator="type",
+        discriminator_map={
+            "numeric_reading": DataPackageNumericReading,
+        },
+    ),
+]
+# DO NOT MODIFY -- This file is generated by type_spec