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

robosystems_client/models/execute_cypher_query_response_200.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
@@ -19,35 +21,35 @@ T = TypeVar("T", bound="ExecuteCypherQueryResponse200")
 class ExecuteCypherQueryResponse200:
   """
   Attributes:
-      success (Union[Unset, bool]):
-      data (Union[Unset, list['ExecuteCypherQueryResponse200DataItem']]):
-      columns (Union[Unset, list[str]]):
-      row_count (Union[Unset, int]):
-      execution_time_ms (Union[Unset, float]):
-      graph_id (Union[Unset, str]):
-      timestamp (Union[Unset, str]):
+      success (bool | Unset):
+      data (list[ExecuteCypherQueryResponse200DataItem] | Unset):
+      columns (list[str] | Unset):
+      row_count (int | Unset):
+      execution_time_ms (float | Unset):
+      graph_id (str | Unset):
+      timestamp (str | Unset):
   """
 
-  success: Union[Unset, bool] = UNSET
-  data: Union[Unset, list["ExecuteCypherQueryResponse200DataItem"]] = UNSET
-  columns: Union[Unset, list[str]] = UNSET
-  row_count: Union[Unset, int] = UNSET
-  execution_time_ms: Union[Unset, float] = UNSET
-  graph_id: Union[Unset, str] = UNSET
-  timestamp: Union[Unset, str] = UNSET
+  success: bool | Unset = UNSET
+  data: list[ExecuteCypherQueryResponse200DataItem] | Unset = UNSET
+  columns: list[str] | Unset = UNSET
+  row_count: int | Unset = UNSET
+  execution_time_ms: float | Unset = UNSET
+  graph_id: str | Unset = UNSET
+  timestamp: str | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
     success = self.success
 
-    data: Union[Unset, list[dict[str, Any]]] = UNSET
+    data: list[dict[str, Any]] | Unset = UNSET
     if not isinstance(self.data, Unset):
       data = []
       for data_item_data in self.data:
         data_item = data_item_data.to_dict()
         data.append(data_item)
 
-    columns: Union[Unset, list[str]] = UNSET
+    columns: list[str] | Unset = UNSET
     if not isinstance(self.columns, Unset):
       columns = self.columns
 
@@ -88,12 +90,14 @@ class ExecuteCypherQueryResponse200:
     d = dict(src_dict)
     success = d.pop("success", UNSET)
 
-    data = []
     _data = d.pop("data", UNSET)
-    for data_item_data in _data or []:
-      data_item = ExecuteCypherQueryResponse200DataItem.from_dict(data_item_data)
+    data: list[ExecuteCypherQueryResponse200DataItem] | Unset = UNSET
+    if _data is not UNSET:
+      data = []
+      for data_item_data in _data:
+        data_item = ExecuteCypherQueryResponse200DataItem.from_dict(data_item_data)
 
-      data.append(data_item)
+        data.append(data_item)
 
     columns = cast(list[str], d.pop("columns", UNSET))
 

robosystems_client/models/execute_cypher_query_response_200_data_item.py

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

robosystems_client/models/fact_detail.py

@@ -0,0 +1,109 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="FactDetail")
+
+
+@_attrs_define
+class FactDetail:
+  """
+  Attributes:
+      fact_id (str):
+      element_uri (str):
+      element_name (str):
+      numeric_value (float):
+      unit (str):
+      period_start (str):
+      period_end (str):
+  """
+
+  fact_id: str
+  element_uri: str
+  element_name: str
+  numeric_value: float
+  unit: str
+  period_start: str
+  period_end: str
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    fact_id = self.fact_id
+
+    element_uri = self.element_uri
+
+    element_name = self.element_name
+
+    numeric_value = self.numeric_value
+
+    unit = self.unit
+
+    period_start = self.period_start
+
+    period_end = self.period_end
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "fact_id": fact_id,
+        "element_uri": element_uri,
+        "element_name": element_name,
+        "numeric_value": numeric_value,
+        "unit": unit,
+        "period_start": period_start,
+        "period_end": period_end,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    fact_id = d.pop("fact_id")
+
+    element_uri = d.pop("element_uri")
+
+    element_name = d.pop("element_name")
+
+    numeric_value = d.pop("numeric_value")
+
+    unit = d.pop("unit")
+
+    period_start = d.pop("period_start")
+
+    period_end = d.pop("period_end")
+
+    fact_detail = cls(
+      fact_id=fact_id,
+      element_uri=element_uri,
+      element_name=element_name,
+      numeric_value=numeric_value,
+      unit=unit,
+      period_start=period_start,
+      period_end=period_end,
+    )
+
+    fact_detail.additional_properties = d
+    return fact_detail
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties

robosystems_client/models/file_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
@@ -20,9 +22,9 @@ class FileInfo:
       upload_status (str): Current upload status
       upload_method (str): Upload method used
       s3_key (str): S3 object key
-      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
+      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
   """
 
   file_id: str
@@ -32,9 +34,9 @@ class FileInfo:
   upload_status: str
   upload_method: str
   s3_key: str
-  row_count: Union[None, Unset, int] = UNSET
-  created_at: Union[None, Unset, str] = UNSET
-  uploaded_at: Union[None, Unset, str] = UNSET
+  row_count: int | None | Unset = UNSET
+  created_at: None | str | Unset = UNSET
+  uploaded_at: None | str | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -52,19 +54,19 @@ class FileInfo:
 
     s3_key = self.s3_key
 
-    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:
@@ -109,30 +111,30 @@ class FileInfo:
 
     s3_key = d.pop("s3_key")
 
-    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))
 

robosystems_client/models/file_layer_status.py

@@ -0,0 +1,123 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="FileLayerStatus")
+
+
+@_attrs_define
+class FileLayerStatus:
+  """
+  Attributes:
+      status (str): Layer status
+      timestamp (None | str | Unset): Status timestamp
+      row_count (int | None | Unset): Row count (if available)
+      size_bytes (int | None | Unset): Size in bytes (S3 layer only)
+  """
+
+  status: str
+  timestamp: None | str | Unset = UNSET
+  row_count: int | None | Unset = UNSET
+  size_bytes: int | None | Unset = UNSET
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    status = self.status
+
+    timestamp: None | str | Unset
+    if isinstance(self.timestamp, Unset):
+      timestamp = UNSET
+    else:
+      timestamp = self.timestamp
+
+    row_count: int | None | Unset
+    if isinstance(self.row_count, Unset):
+      row_count = UNSET
+    else:
+      row_count = self.row_count
+
+    size_bytes: int | None | Unset
+    if isinstance(self.size_bytes, Unset):
+      size_bytes = UNSET
+    else:
+      size_bytes = self.size_bytes
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "status": status,
+      }
+    )
+    if timestamp is not UNSET:
+      field_dict["timestamp"] = timestamp
+    if row_count is not UNSET:
+      field_dict["row_count"] = row_count
+    if size_bytes is not UNSET:
+      field_dict["size_bytes"] = size_bytes
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    status = d.pop("status")
+
+    def _parse_timestamp(data: object) -> None | str | Unset:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(None | str | Unset, data)
+
+    timestamp = _parse_timestamp(d.pop("timestamp", UNSET))
+
+    def _parse_row_count(data: object) -> int | None | Unset:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(int | None | Unset, data)
+
+    row_count = _parse_row_count(d.pop("row_count", UNSET))
+
+    def _parse_size_bytes(data: object) -> int | None | Unset:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(int | None | Unset, data)
+
+    size_bytes = _parse_size_bytes(d.pop("size_bytes", UNSET))
+
+    file_layer_status = cls(
+      status=status,
+      timestamp=timestamp,
+      row_count=row_count,
+      size_bytes=size_bytes,
+    )
+
+    file_layer_status.additional_properties = d
+    return file_layer_status
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties

robosystems_client/models/file_status_update.py

@@ -1,8 +1,12 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 
 from attrs import define as _attrs_define
 
+from ..types import UNSET, Unset
+
 T = TypeVar("T", bound="FileStatusUpdate")
 
 
@@ -12,13 +16,18 @@ class FileStatusUpdate:
   Attributes:
       status (str): File status: 'uploaded' (ready for ingest), 'disabled' (exclude from ingest), 'archived' (soft
           deleted)
+      ingest_to_graph (bool | Unset): Auto-ingest to graph after DuckDB staging. Default=false (batch mode). Set to
+          true for real-time incremental updates. Default: False.
   """
 
   status: str
+  ingest_to_graph: bool | Unset = False
 
   def to_dict(self) -> dict[str, Any]:
     status = self.status
 
+    ingest_to_graph = self.ingest_to_graph
+
     field_dict: dict[str, Any] = {}
 
     field_dict.update(
@@ -26,6 +35,8 @@ class FileStatusUpdate:
         "status": status,
       }
     )
+    if ingest_to_graph is not UNSET:
+      field_dict["ingest_to_graph"] = ingest_to_graph
 
     return field_dict
 
@@ -34,8 +45,11 @@ class FileStatusUpdate:
     d = dict(src_dict)
     status = d.pop("status")
 
+    ingest_to_graph = d.pop("ingest_to_graph", UNSET)
+
     file_status_update = cls(
       status=status,
+      ingest_to_graph=ingest_to_graph,
     )
 
     return file_status_update

robosystems_client/models/file_upload_request.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 
@@ -13,17 +15,25 @@ class FileUploadRequest:
   """
   Attributes:
       file_name (str): File name to upload
-      content_type (Union[Unset, str]): File MIME type Default: 'application/x-parquet'.
+      content_type (str | Unset): File MIME type Default: 'application/x-parquet'.
+      table_name (None | str | Unset): Table name to associate file with (required for first-class /files endpoint)
   """
 
   file_name: str
-  content_type: Union[Unset, str] = "application/x-parquet"
+  content_type: str | Unset = "application/x-parquet"
+  table_name: None | str | Unset = UNSET
 
   def to_dict(self) -> dict[str, Any]:
     file_name = self.file_name
 
     content_type = self.content_type
 
+    table_name: None | str | Unset
+    if isinstance(self.table_name, Unset):
+      table_name = UNSET
+    else:
+      table_name = self.table_name
+
     field_dict: dict[str, Any] = {}
 
     field_dict.update(
@@ -33,6 +43,8 @@ class FileUploadRequest:
     )
     if content_type is not UNSET:
       field_dict["content_type"] = content_type
+    if table_name is not UNSET:
+      field_dict["table_name"] = table_name
 
     return field_dict
 
@@ -43,9 +55,19 @@ class FileUploadRequest:
 
     content_type = d.pop("content_type", UNSET)
 
+    def _parse_table_name(data: object) -> None | str | Unset:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(None | str | Unset, data)
+
+    table_name = _parse_table_name(d.pop("table_name", UNSET))
+
     file_upload_request = cls(
       file_name=file_name,
       content_type=content_type,
+      table_name=table_name,
     )
 
     return file_upload_request

robosystems_client/models/file_upload_response.py

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

robosystems_client/models/forgot_password_request.py

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

robosystems_client/models/forgot_password_response_forgotpassword.py

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

robosystems_client/models/get_current_auth_user_response_getcurrentauthuser.py

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