UncountablePythonSDK 0.0.176__tar.gz → 0.0.178__tar.gz

{uncountablepythonsdk-0.0.176/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.178}/PKG-INFO

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: UncountablePythonSDK
-Version: 0.0.176
+Version: 0.0.178
 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.176 → uncountablepythonsdk-0.0.178}/docs/conf.py

@@ -7,6 +7,8 @@
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
 
 import datetime
+import importlib
+import inspect
 
 from docutils import nodes  # type: ignore[import-untyped]
 from sphinx.addnodes import pending_xref  # type: ignore[import-not-found]
@@ -63,6 +65,20 @@ favicons = [
 ]
 
 
+def _resolve_alias_to_real_module(target_module: str) -> str:
+    if "." not in target_module:
+        return target_module
+    parent_name, attr_name = target_module.rsplit(".", 1)
+    try:
+        parent = importlib.import_module(parent_name)
+    except ImportError:
+        return target_module
+    aliased = getattr(parent, attr_name, None)
+    if not inspect.ismodule(aliased) or aliased.__name__ == target_module:
+        return target_module
+    return aliased.__name__
+
+
 def _hook_missing_reference(
     _app: Sphinx, _env: BuildEnvironment, node: pending_xref, contnode: nodes.Text
 ) -> nodes.reference | None:
@@ -80,6 +96,10 @@ def _hook_missing_reference(
         return None
 
     target_module, target_name = target.rsplit(".", 1)
+    real_module = _resolve_alias_to_real_module(target_module)
+    if real_module != target_module:
+        target_module = real_module
+        target = f"{target_module}.{target_name}"
 
     # construct relative path from current doc page to target page
     relative_segments_to_root = [".." for _ in current_doc.split("/")]

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/pkgs/filesystem_utils/__init__.py

@@ -1,3 +1,5 @@
+# CLOSED MODULE
+
 from ._blob_session import BlobSession as BlobSession
 from ._file_share_session import FileShareSession as FileShareSession
 from ._gdrive_session import (
@@ -32,3 +34,32 @@ from .file_type_utils import (
     RemoteObjectReference as RemoteObjectReference,
 )
 from .filesystem_session import FileSystemSession as FileSystemSession
+
+__all__ = [
+    "BlobSession",
+    "FileObjectData",
+    "FileShareSession",
+    "FileSystemBlobConfig",
+    "FileSystemFileReference",
+    "FileSystemFileShareConfig",
+    "FileSystemObject",
+    "FileSystemS3Config",
+    "FileSystemSFTPConfig",
+    "FileSystemSession",
+    "FileTransfer",
+    "GDriveSession",
+    "IncompatibleFileReference",
+    "LocalSession",
+    "RemoteObjectReference",
+    "S3Session",
+    "SFTPConnection",
+    "SFTPCredentialError",
+    "SFTPSession",
+    "delete_gdrive_file",
+    "download_gdrive_file",
+    "list_gdrive_files",
+    "list_sftp_files",
+    "move_gdrive_file",
+    "move_sftp_files",
+    "upload_file_gdrive",
+]

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/pkgs/serialization/opaque_key.py

@@ -3,5 +3,7 @@ from typing import Self
 
 # Blocks a string key value from being interpreted for case conversion
 class OpaqueKey(str):  # noqa: FURB189
+    __slots__ = ()
+
     def __new__(cls, key: str) -> Self:
         return str.__new__(cls, key)

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/pkgs/strenum_compat/__init__.py

@@ -1,3 +1,5 @@
+# CLOSED MODULE
+
 from .strenum_compat import StrEnum
 
 __all__ = ["StrEnum"]

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/core/client.py

@@ -298,6 +298,10 @@ class Client(ClientMethods):
                 else None
             )
             response = self._send_request(request, timeout=timeout)
+            if self._cfg.logger is not None:
+                self._cfg.logger.set_span_attribute(
+                    "http.status_code", response.status_code
+                )
         response_data = self._get_response_json(response, request_id=request_id)
         cached_parser = self._get_cached_parser(return_type)
         try:

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/core/file_upload.py

@@ -17,6 +17,7 @@ from uncountable.types import uploader_t
 from .types import AuthDetailsAll, AuthDetailsApiKey
 
 _CHUNK_SIZE = 5 * 1024 * 1024  # s3 requires 5MiB minimum
+_PARSED_FILE_DATA_FILENAME = "parsed_file_data.json"
 
 
 class FileUploadType(StrEnum):
@@ -51,17 +52,27 @@ class UploaderFileUpload:
 FileUpload = MediaFileUpload | DataFileUpload | UploaderFileUpload
 
 
-@dataclass(kw_only=True)
-class _ParsedFileDataWrapper:
-    parsed_file_data: list[uploader_t.ParsedFileData]
-
-
 @dataclass(kw_only=True)
 class FileBytes:
     name: str
     bytes_data: BytesIO
 
 
+def _build_parsed_file_data_bytes(
+    *, parsed_file_data: list[uploader_t.ParsedFileData]
+) -> FileBytes:
+    buffer = BytesIO()
+    buffer.write(b'{"parsed_file_data": [')
+    for index, file_data in enumerate(parsed_file_data):
+        if index > 0:
+            buffer.write(b", ")
+        serialized_file_data = serialize_for_storage(file_data)
+        buffer.write(json.dumps(serialized_file_data).encode("utf-8"))
+    buffer.write(b"]}")
+    buffer.seek(0)
+    return FileBytes(name=_PARSED_FILE_DATA_FILENAME, bytes_data=buffer)
+
+
 @contextmanager
 def file_upload_data(file_upload: FileUpload) -> Generator[FileBytes, None, None]:
     match file_upload:
@@ -73,15 +84,9 @@ def file_upload_data(file_upload: FileUpload) -> Generator[FileBytes, None, None
         case DataFileUpload():
             yield FileBytes(name=file_upload.name, bytes_data=file_upload.data)
         case UploaderFileUpload():
-            wrapper = _ParsedFileDataWrapper(
+            yield _build_parsed_file_data_bytes(
                 parsed_file_data=file_upload.parsed_file_data
             )
-            serialized = serialize_for_storage(wrapper)
-            json_bytes = json.dumps(serialized).encode("utf-8")
-            yield FileBytes(
-                name="parsed_file_data.json",
-                bytes_data=BytesIO(json_bytes),
-            )
 
 
 @dataclass(kw_only=True)
@@ -135,7 +140,7 @@ class FileUploader:
                 case DataFileUpload():
                     attributes["file_name"] = file_upload.name
                 case UploaderFileUpload():
-                    attributes["file_name"] = "parsed_file_data.json"
+                    attributes["file_name"] = _PARSED_FILE_DATA_FILENAME
                 case _:
                     assert_never(file_upload)
             with push_scope_optional(

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/core/query/builder.py

@@ -13,6 +13,7 @@ from uncountable.core.query.column import (
     Column,
     ColumnFilter,
     EnumColumn,
+    JsonColumn,
     NullableColumn,
     PrimaryKeyColumn,
     get_column_base,
@@ -72,8 +73,16 @@ class QueryBuilder[QueryRowT: QueryRow]:
             column.identifier for column in column_definitions.values()
         )
         self._property_names: tuple[str, ...] = tuple(column_definitions.keys())
+        json_property_names = {
+            name
+            for name, column in column_definitions.items()
+            if isinstance(column, JsonColumn)
+        }
         self._parser: CachedParser = CachedParser(
-            serial_class(unconverted_keys={*self._property_names})(
+            serial_class(
+                unconverted_keys={*self._property_names},
+                unconverted_values=json_property_names,
+            )(
                 make_dataclass(
                     cls_name=self._model.__name__,
                     fields=[

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/core/query/column.py

@@ -465,6 +465,27 @@ class NullableTextsColumn[
     # IMPROVE: Implement "Not Exists" filter (when supported in SDK)
 
 
+class JsonColumn(Column[base_t.JsonValue, NullableColumn]):
+    __hash__ = Column.__hash__
+
+    def __init__(self, identifier: listing_t.ColumnIdentifier) -> None:
+        super().__init__(
+            identifier=identifier,
+            nullable=NullableColumn,
+            value_type=(
+                dict[str, base_t.JsonValue] | list[base_t.JsonValue] | base_t.JsonScalar
+            ),
+        )
+
+    def __ne__(self, other: None) -> ColumnFilter:  # type: ignore[override]
+        return ColumnFilter(
+            identifier=self.identifier,
+            filter=lambda identifier: listing_t.FilterSpecExists(column=identifier),
+        )
+
+    # IMPROVE: Implement "Not Exists" filter (when supported in SDK)
+
+
 class PrimaryKeyColumn(
     Column[base_t.ObjectId, NonNullableColumn],
 ):
@@ -684,6 +705,10 @@ class IdColumn[
         column: NullableTextsColumn[BaseNullableT],
     ) -> NullableTextsColumn[BaseNullableT]: ...
 
+    # Json
+    @overload
+    def __rshift__(self, column: JsonColumn) -> JsonColumn: ...
+
     # Ids
     @overload
     def __rshift__[
@@ -734,6 +759,7 @@ class IdColumn[
         | NumericColumn[BaseNullableT]
         | TextsColumn[BaseNullableT]
         | NullableTextsColumn[BaseNullableT]
+        | JsonColumn
         | IdsColumn[BaseNullableT]
         | IdColumn[BaseNullableT],
     ) -> (
@@ -768,6 +794,7 @@ class IdColumn[
         | NullableTextsColumn[BaseNullableT]
         | NullableTextsColumn[NonNullableColumn | NullableColumn]
         | NullableTextsColumn[NullableColumn]
+        | JsonColumn
         | IdsColumn[BaseNullableT]
         | IdsColumn[NonNullableColumn | NullableColumn]
         | IdsColumn[NullableColumn]
@@ -869,6 +896,8 @@ class IdColumn[
                     if self.nullable is NonNullableColumn
                     else NullableColumn,
                 )
+            case JsonColumn():
+                return JsonColumn(identifier=transitive_identifier)
             case IdsColumn():
                 return IdsColumn(
                     id_type=column.id_type,

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/integration/telemetry.py

@@ -27,6 +27,7 @@ from opentelemetry.sdk.trace.export import (
     SimpleSpanProcessor,
 )
 from opentelemetry.trace import Span, Tracer
+from opentelemetry.util.types import AttributeValue
 
 from pkgs.serialization_util import dict_fields
 from uncountable.core.environment import (
@@ -188,6 +189,9 @@ class Logger:
         )
         otel_logger.emit(log_record)
 
+    def set_span_attribute(self, key: str, value: AttributeValue) -> None:
+        self._current_span_with_fallback().set_attribute(key, value)
+
     def bind(self, *, attributes: Attributes) -> None:
         # Bound attributes flow into log records and the X-UNC-METADATA request
         # header on outbound API calls. The header is capped at 2048 bytes; if

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/types/api/integrations/publish_realtime_data.py

@@ -22,7 +22,7 @@ __all__: list[str] = [
 
 ENDPOINT_METHOD = "POST"
 ENDPOINT_PATH = "api/external/integrations/publish_realtime_data"
-ENDPOINT_DESCRIPTION = "Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. API is in beta \u2014 contact Uncountable for realtime integration guidance."
+ENDPOINT_DESCRIPTION = "Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. Contact Uncountable for realtime integration guidance."
 
 
 # DO NOT MODIFY -- This file is generated by type_spec

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/types/client_base.py

@@ -1713,7 +1713,7 @@ class ClientMethods(ABC):
         data_package: integrations_t.DataPackage,
         _request_options: client_config_t.RequestOptions | None = None,
     ) -> publish_realtime_data_t.Data:
-        """Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. API is in beta — contact Uncountable for realtime integration guidance.
+        """Publish a numeric reading for a realtime instrument integration. Currently only supports equipment entities. Resolves the target equipment via identifier key. Returns the integration session details and status. Contact Uncountable for realtime integration guidance.
 
         """
         args = publish_realtime_data_t.Arguments(

{uncountablepythonsdk-0.0.176 → uncountablepythonsdk-0.0.178}/uncountable/types/entity_t.py

@@ -48,9 +48,11 @@ __all__: list[str] = [
         "condition_parameter_matches": "Condition Parameter Matches",
         "condition_parameter_rule": "Condition Parameter Rule",
         "condition_parameter_value": "Condition Parameter Value",
+        "config_transfer_export": "Config Transfer Export",
         "constraint": "Constraint",
         "constraint_set": "Constraint Set",
         "control_type": "Control Type",
+        "credit_balance": "Credit Balance",
         "curve_settings": "Curve Settings",
         "custom_entity": "Custom Entity",
         "default_permissions_new_project": "Default Permissions New Project",
@@ -82,6 +84,7 @@ __all__: list[str] = [
         "external_ai_connector": "External AI Connector",
         "external_ai_model": "External AI Model",
         "external_ai_session": "External AI Session",
+        "external_ai_usage": "External AI Usage",
         "experiment_group": "Experiment Group",
         "experiment_group_member": "Experiment Group Member",
         "experiment_group_type": "Experiment Group Type",
@@ -299,9 +302,11 @@ class EntityType(StrEnum):
     CONDITION_PARAMETER_MATCHES = "condition_parameter_matches"
     CONDITION_PARAMETER_RULE = "condition_parameter_rule"
     CONDITION_PARAMETER_VALUE = "condition_parameter_value"
+    CONFIG_TRANSFER_EXPORT = "config_transfer_export"
     CONSTRAINT = "constraint"
     CONSTRAINT_SET = "constraint_set"
     CONTROL_TYPE = "control_type"
+    CREDIT_BALANCE = "credit_balance"
     CURVE_SETTINGS = "curve_settings"
     CUSTOM_ENTITY = "custom_entity"
     DEFAULT_PERMISSIONS_NEW_PROJECT = "default_permissions_new_project"
@@ -333,6 +338,7 @@ class EntityType(StrEnum):
     EXTERNAL_AI_CONNECTOR = "external_ai_connector"
     EXTERNAL_AI_MODEL = "external_ai_model"
     EXTERNAL_AI_SESSION = "external_ai_session"
+    EXTERNAL_AI_USAGE = "external_ai_usage"
     EXPERIMENT_GROUP = "experiment_group"
     EXPERIMENT_GROUP_MEMBER = "experiment_group_member"
     EXPERIMENT_GROUP_TYPE = "experiment_group_type"