robosystems-client 0.2.15__py3-none-any.whl → 0.2.16__py3-none-any.whl

robosystems_client/models/get_file_info_response.py

@@ -1,11 +1,17 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
+if TYPE_CHECKING:
+  from ..models.enhanced_file_status_layers import EnhancedFileStatusLayers
+
+
 T = TypeVar("T", bound="GetFileInfoResponse")
 
 
@@ -22,10 +28,12 @@ class GetFileInfoResponse:
       upload_status (str): Current upload status
       upload_method (str): Upload method used
       s3_key (str): S3 object key
-      table_name (Union[None, Unset, str]): Table name
-      row_count (Union[None, Unset, int]): Estimated row count
-      created_at (Union[None, Unset, str]): File creation timestamp
-      uploaded_at (Union[None, Unset, str]): File upload completion timestamp
+      table_name (None | str | Unset): Table name
+      row_count (int | None | Unset): Estimated row count
+      created_at (None | str | Unset): File creation timestamp
+      uploaded_at (None | str | Unset): File upload completion timestamp
+      layers (EnhancedFileStatusLayers | None | Unset): Multi-layer pipeline status (S3 → DuckDB → Graph). Shows
+          status, timestamps, and row counts for each layer independently.
   """
 
   file_id: str
@@ -37,13 +45,16 @@ class GetFileInfoResponse:
   upload_status: str
   upload_method: str
   s3_key: str
-  table_name: Union[None, Unset, str] = UNSET
-  row_count: Union[None, Unset, int] = UNSET
-  created_at: Union[None, Unset, str] = UNSET
-  uploaded_at: Union[None, Unset, str] = UNSET
+  table_name: None | str | Unset = UNSET
+  row_count: int | None | Unset = UNSET
+  created_at: None | str | Unset = UNSET
+  uploaded_at: None | str | Unset = UNSET
+  layers: EnhancedFileStatusLayers | None | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
+    from ..models.enhanced_file_status_layers import EnhancedFileStatusLayers
+
     file_id = self.file_id
 
     graph_id = self.graph_id
@@ -62,30 +73,38 @@ class GetFileInfoResponse:
 
     s3_key = self.s3_key
 
-    table_name: Union[None, Unset, str]
+    table_name: None | str | Unset
     if isinstance(self.table_name, Unset):
       table_name = UNSET
     else:
       table_name = self.table_name
 
-    row_count: Union[None, Unset, int]
+    row_count: int | None | Unset
     if isinstance(self.row_count, Unset):
       row_count = UNSET
     else:
       row_count = self.row_count
 
-    created_at: Union[None, Unset, str]
+    created_at: None | str | Unset
     if isinstance(self.created_at, Unset):
       created_at = UNSET
     else:
       created_at = self.created_at
 
-    uploaded_at: Union[None, Unset, str]
+    uploaded_at: None | str | Unset
     if isinstance(self.uploaded_at, Unset):
       uploaded_at = UNSET
     else:
       uploaded_at = self.uploaded_at
 
+    layers: dict[str, Any] | None | Unset
+    if isinstance(self.layers, Unset):
+      layers = UNSET
+    elif isinstance(self.layers, EnhancedFileStatusLayers):
+      layers = self.layers.to_dict()
+    else:
+      layers = self.layers
+
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
     field_dict.update(
@@ -109,11 +128,15 @@ class GetFileInfoResponse:
       field_dict["created_at"] = created_at
     if uploaded_at is not UNSET:
       field_dict["uploaded_at"] = uploaded_at
+    if layers is not UNSET:
+      field_dict["layers"] = layers
 
     return field_dict
 
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.enhanced_file_status_layers import EnhancedFileStatusLayers
+
     d = dict(src_dict)
     file_id = d.pop("file_id")
 
@@ -133,42 +156,59 @@ class GetFileInfoResponse:
 
     s3_key = d.pop("s3_key")
 
-    def _parse_table_name(data: object) -> Union[None, Unset, str]:
+    def _parse_table_name(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     table_name = _parse_table_name(d.pop("table_name", UNSET))
 
-    def _parse_row_count(data: object) -> Union[None, Unset, int]:
+    def _parse_row_count(data: object) -> int | None | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, int], data)
+      return cast(int | None | Unset, data)
 
     row_count = _parse_row_count(d.pop("row_count", UNSET))
 
-    def _parse_created_at(data: object) -> Union[None, Unset, str]:
+    def _parse_created_at(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     created_at = _parse_created_at(d.pop("created_at", UNSET))
 
-    def _parse_uploaded_at(data: object) -> Union[None, Unset, str]:
+    def _parse_uploaded_at(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     uploaded_at = _parse_uploaded_at(d.pop("uploaded_at", UNSET))
 
+    def _parse_layers(data: object) -> EnhancedFileStatusLayers | None | Unset:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      try:
+        if not isinstance(data, dict):
+          raise TypeError()
+        layers_type_0 = EnhancedFileStatusLayers.from_dict(data)
+
+        return layers_type_0
+      except (TypeError, ValueError, AttributeError, KeyError):
+        pass
+      return cast(EnhancedFileStatusLayers | None | Unset, data)
+
+    layers = _parse_layers(d.pop("layers", UNSET))
+
     get_file_info_response = cls(
       file_id=file_id,
       graph_id=graph_id,
@@ -183,6 +223,7 @@ class GetFileInfoResponse:
       row_count=row_count,
       created_at=created_at,
       uploaded_at=uploaded_at,
+      layers=layers,
     )
 
     get_file_info_response.additional_properties = d

robosystems_client/models/get_operation_status_response_getoperationstatus.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/get_storage_usage_response_getstorageusage.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/graph_info.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -19,8 +21,8 @@ class GraphInfo:
       role (str): User's role/access level
       is_selected (bool): Whether this is the currently selected graph
       created_at (str): Creation timestamp
-      is_repository (Union[Unset, bool]): Whether this is a shared repository (vs user graph) Default: False.
-      repository_type (Union[None, Unset, str]): Repository type if isRepository=true
+      is_repository (bool | Unset): Whether this is a shared repository (vs user graph) Default: False.
+      repository_type (None | str | Unset): Repository type if isRepository=true
   """
 
   graph_id: str
@@ -28,8 +30,8 @@ class GraphInfo:
   role: str
   is_selected: bool
   created_at: str
-  is_repository: Union[Unset, bool] = False
-  repository_type: Union[None, Unset, str] = UNSET
+  is_repository: bool | Unset = False
+  repository_type: None | str | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -45,7 +47,7 @@ class GraphInfo:
 
     is_repository = self.is_repository
 
-    repository_type: Union[None, Unset, str]
+    repository_type: None | str | Unset
     if isinstance(self.repository_type, Unset):
       repository_type = UNSET
     else:
@@ -84,12 +86,12 @@ class GraphInfo:
 
     is_repository = d.pop("isRepository", UNSET)
 
-    def _parse_repository_type(data: object) -> Union[None, Unset, str]:
+    def _parse_repository_type(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     repository_type = _parse_repository_type(d.pop("repositoryType", UNSET))
 

robosystems_client/models/graph_limits_response.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -32,19 +34,19 @@ class GraphLimitsResponse:
       copy_operations (CopyOperationLimits): Copy/ingestion operation limits.
       backups (BackupLimits): Backup operation limits.
       rate_limits (RateLimits): API rate limits.
-      credits_ (Union['CreditLimits', None, Unset]): AI credit limits (if applicable)
+      credits_ (CreditLimits | None | Unset): AI credit limits (if applicable)
   """
 
   graph_id: str
   subscription_tier: str
   graph_tier: str
   is_shared_repository: bool
-  storage: "StorageLimits"
-  queries: "QueryLimits"
-  copy_operations: "CopyOperationLimits"
-  backups: "BackupLimits"
-  rate_limits: "RateLimits"
-  credits_: Union["CreditLimits", None, Unset] = UNSET
+  storage: StorageLimits
+  queries: QueryLimits
+  copy_operations: CopyOperationLimits
+  backups: BackupLimits
+  rate_limits: RateLimits
+  credits_: CreditLimits | None | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -68,7 +70,7 @@ class GraphLimitsResponse:
 
     rate_limits = self.rate_limits.to_dict()
 
-    credits_: Union[None, Unset, dict[str, Any]]
+    credits_: dict[str, Any] | None | Unset
     if isinstance(self.credits_, Unset):
       credits_ = UNSET
     elif isinstance(self.credits_, CreditLimits):
@@ -124,7 +126,7 @@ class GraphLimitsResponse:
 
     rate_limits = RateLimits.from_dict(d.pop("rate_limits"))
 
-    def _parse_credits_(data: object) -> Union["CreditLimits", None, Unset]:
+    def _parse_credits_(data: object) -> CreditLimits | None | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
@@ -135,9 +137,9 @@ class GraphLimitsResponse:
         credits_type_0 = CreditLimits.from_dict(data)
 
         return credits_type_0
-      except:  # noqa: E722
+      except (TypeError, ValueError, AttributeError, KeyError):
         pass
-      return cast(Union["CreditLimits", None, Unset], data)
+      return cast(CreditLimits | None | Unset, data)
 
     credits_ = _parse_credits_(d.pop("credits", UNSET))
 

robosystems_client/models/graph_metadata.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -15,31 +17,31 @@ class GraphMetadata:
 
   Attributes:
       graph_name (str): Display name for the graph
-      description (Union[None, Unset, str]): Optional description
-      schema_extensions (Union[Unset, list[str]]): Schema extensions to enable
-      tags (Union[Unset, list[str]]): Tags for organizing graphs
+      description (None | str | Unset): Optional description
+      schema_extensions (list[str] | Unset): Schema extensions to enable
+      tags (list[str] | Unset): Tags for organizing graphs
   """
 
   graph_name: str
-  description: Union[None, Unset, str] = UNSET
-  schema_extensions: Union[Unset, list[str]] = UNSET
-  tags: Union[Unset, list[str]] = UNSET
+  description: None | str | Unset = UNSET
+  schema_extensions: list[str] | Unset = UNSET
+  tags: list[str] | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
     graph_name = self.graph_name
 
-    description: Union[None, Unset, str]
+    description: None | str | Unset
     if isinstance(self.description, Unset):
       description = UNSET
     else:
       description = self.description
 
-    schema_extensions: Union[Unset, list[str]] = UNSET
+    schema_extensions: list[str] | Unset = UNSET
     if not isinstance(self.schema_extensions, Unset):
       schema_extensions = self.schema_extensions
 
-    tags: Union[Unset, list[str]] = UNSET
+    tags: list[str] | Unset = UNSET
     if not isinstance(self.tags, Unset):
       tags = self.tags
 
@@ -64,12 +66,12 @@ class GraphMetadata:
     d = dict(src_dict)
     graph_name = d.pop("graph_name")
 
-    def _parse_description(data: object) -> Union[None, Unset, str]:
+    def _parse_description(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     description = _parse_description(d.pop("description", UNSET))
 

robosystems_client/models/graph_metrics_response.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -35,20 +37,20 @@ class GraphMetricsResponse:
       relationship_counts (GraphMetricsResponseRelationshipCounts): Relationship counts by type
       estimated_size (GraphMetricsResponseEstimatedSize): Database size estimates
       health_status (GraphMetricsResponseHealthStatus): Database health information
-      graph_name (Union[None, Unset, str]): Display name for the graph
-      user_role (Union[None, Unset, str]): User's role in this graph
+      graph_name (None | str | Unset): Display name for the graph
+      user_role (None | str | Unset): User's role in this graph
   """
 
   graph_id: str
   timestamp: str
   total_nodes: int
   total_relationships: int
-  node_counts: "GraphMetricsResponseNodeCounts"
-  relationship_counts: "GraphMetricsResponseRelationshipCounts"
-  estimated_size: "GraphMetricsResponseEstimatedSize"
-  health_status: "GraphMetricsResponseHealthStatus"
-  graph_name: Union[None, Unset, str] = UNSET
-  user_role: Union[None, Unset, str] = UNSET
+  node_counts: GraphMetricsResponseNodeCounts
+  relationship_counts: GraphMetricsResponseRelationshipCounts
+  estimated_size: GraphMetricsResponseEstimatedSize
+  health_status: GraphMetricsResponseHealthStatus
+  graph_name: None | str | Unset = UNSET
+  user_role: None | str | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -68,13 +70,13 @@ class GraphMetricsResponse:
 
     health_status = self.health_status.to_dict()
 
-    graph_name: Union[None, Unset, str]
+    graph_name: None | str | Unset
     if isinstance(self.graph_name, Unset):
       graph_name = UNSET
     else:
       graph_name = self.graph_name
 
-    user_role: Union[None, Unset, str]
+    user_role: None | str | Unset
     if isinstance(self.user_role, Unset):
       user_role = UNSET
     else:
@@ -137,21 +139,21 @@ class GraphMetricsResponse:
 
     health_status = GraphMetricsResponseHealthStatus.from_dict(d.pop("health_status"))
 
-    def _parse_graph_name(data: object) -> Union[None, Unset, str]:
+    def _parse_graph_name(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     graph_name = _parse_graph_name(d.pop("graph_name", UNSET))
 
-    def _parse_user_role(data: object) -> Union[None, Unset, str]:
+    def _parse_user_role(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     user_role = _parse_user_role(d.pop("user_role", UNSET))
 

robosystems_client/models/graph_metrics_response_estimated_size.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/graph_metrics_response_health_status.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/graph_metrics_response_node_counts.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/graph_metrics_response_relationship_counts.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/graph_subscription_response.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -22,11 +24,11 @@ class GraphSubscriptionResponse:
       status (str): Subscription status
       base_price_cents (int): Base price in cents
       created_at (str): Creation timestamp
-      current_period_start (Union[None, Unset, str]): Current billing period start
-      current_period_end (Union[None, Unset, str]): Current billing period end
-      started_at (Union[None, Unset, str]): Subscription start date
-      canceled_at (Union[None, Unset, str]): Cancellation date
-      ends_at (Union[None, Unset, str]): Subscription end date (when access will be revoked, especially relevant for
+      current_period_start (None | str | Unset): Current billing period start
+      current_period_end (None | str | Unset): Current billing period end
+      started_at (None | str | Unset): Subscription start date
+      canceled_at (None | str | Unset): Cancellation date
+      ends_at (None | str | Unset): Subscription end date (when access will be revoked, especially relevant for
           cancelled subscriptions)
   """
 
@@ -38,11 +40,11 @@ class GraphSubscriptionResponse:
   status: str
   base_price_cents: int
   created_at: str
-  current_period_start: Union[None, Unset, str] = UNSET
-  current_period_end: Union[None, Unset, str] = UNSET
-  started_at: Union[None, Unset, str] = UNSET
-  canceled_at: Union[None, Unset, str] = UNSET
-  ends_at: Union[None, Unset, str] = UNSET
+  current_period_start: None | str | Unset = UNSET
+  current_period_end: None | str | Unset = UNSET
+  started_at: None | str | Unset = UNSET
+  canceled_at: None | str | Unset = UNSET
+  ends_at: None | str | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -62,31 +64,31 @@ class GraphSubscriptionResponse:
 
     created_at = self.created_at
 
-    current_period_start: Union[None, Unset, str]
+    current_period_start: None | str | Unset
     if isinstance(self.current_period_start, Unset):
       current_period_start = UNSET
     else:
       current_period_start = self.current_period_start
 
-    current_period_end: Union[None, Unset, str]
+    current_period_end: None | str | Unset
     if isinstance(self.current_period_end, Unset):
       current_period_end = UNSET
     else:
       current_period_end = self.current_period_end
 
-    started_at: Union[None, Unset, str]
+    started_at: None | str | Unset
     if isinstance(self.started_at, Unset):
       started_at = UNSET
     else:
       started_at = self.started_at
 
-    canceled_at: Union[None, Unset, str]
+    canceled_at: None | str | Unset
     if isinstance(self.canceled_at, Unset):
       canceled_at = UNSET
     else:
       canceled_at = self.canceled_at
 
-    ends_at: Union[None, Unset, str]
+    ends_at: None | str | Unset
     if isinstance(self.ends_at, Unset):
       ends_at = UNSET
     else:
@@ -138,50 +140,50 @@ class GraphSubscriptionResponse:
 
     created_at = d.pop("created_at")
 
-    def _parse_current_period_start(data: object) -> Union[None, Unset, str]:
+    def _parse_current_period_start(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     current_period_start = _parse_current_period_start(
       d.pop("current_period_start", UNSET)
     )
 
-    def _parse_current_period_end(data: object) -> Union[None, Unset, str]:
+    def _parse_current_period_end(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     current_period_end = _parse_current_period_end(d.pop("current_period_end", UNSET))
 
-    def _parse_started_at(data: object) -> Union[None, Unset, str]:
+    def _parse_started_at(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     started_at = _parse_started_at(d.pop("started_at", UNSET))
 
-    def _parse_canceled_at(data: object) -> Union[None, Unset, str]:
+    def _parse_canceled_at(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     canceled_at = _parse_canceled_at(d.pop("canceled_at", UNSET))
 
-    def _parse_ends_at(data: object) -> Union[None, Unset, str]:
+    def _parse_ends_at(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     ends_at = _parse_ends_at(d.pop("ends_at", UNSET))