robosystems-client 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

robosystems_client/models/create_graph_request.py

@@ -31,6 +31,9 @@ class CreateGraphRequest:
       custom_schema (Union['CustomSchemaDefinition', None, Unset]): Custom schema definition to apply
       initial_entity (Union['InitialEntityData', None, Unset]): Optional initial entity to create in the graph. If
           provided, creates a entity-focused graph.
+      create_entity (Union[Unset, bool]): Whether to create the entity node and upload initial data. Only applies when
+          initial_entity is provided. Set to False to create graph without populating entity data (useful for file-based
+          ingestion workflows). Default: True.
       tags (Union[Unset, list[str]]): Optional tags for organization
   """
 
@@ -38,6 +41,7 @@ class CreateGraphRequest:
   instance_tier: Union[Unset, str] = "kuzu-standard"
   custom_schema: Union["CustomSchemaDefinition", None, Unset] = UNSET
   initial_entity: Union["InitialEntityData", None, Unset] = UNSET
+  create_entity: Union[Unset, bool] = True
   tags: Union[Unset, list[str]] = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -65,6 +69,8 @@ class CreateGraphRequest:
     else:
       initial_entity = self.initial_entity
 
+    create_entity = self.create_entity
+
     tags: Union[Unset, list[str]] = UNSET
     if not isinstance(self.tags, Unset):
       tags = self.tags
@@ -82,6 +88,8 @@ class CreateGraphRequest:
       field_dict["custom_schema"] = custom_schema
     if initial_entity is not UNSET:
       field_dict["initial_entity"] = initial_entity
+    if create_entity is not UNSET:
+      field_dict["create_entity"] = create_entity
     if tags is not UNSET:
       field_dict["tags"] = tags
 
@@ -134,6 +142,8 @@ class CreateGraphRequest:
 
     initial_entity = _parse_initial_entity(d.pop("initial_entity", UNSET))
 
+    create_entity = d.pop("create_entity", UNSET)
+
     tags = cast(list[str], d.pop("tags", UNSET))
 
     create_graph_request = cls(
@@ -141,6 +151,7 @@ class CreateGraphRequest:
       instance_tier=instance_tier,
       custom_schema=custom_schema,
       initial_entity=initial_entity,
+      create_entity=create_entity,
       tags=tags,
     )
 

robosystems_client/models/{delete_file_v1_graphs_graph_id_tables_files_file_id_delete_response_delete_file_v1_graphs_graph_id_tables_files_file_id_delete.py → delete_file_response.py}

@@ -4,31 +4,67 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar(
-  "T",
-  bound="DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete",
-)
+T = TypeVar("T", bound="DeleteFileResponse")
 
 
 @_attrs_define
-class DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete:
-  """ """
+class DeleteFileResponse:
+  """
+  Attributes:
+      status (str): Deletion status
+      file_id (str): Deleted file ID
+      file_name (str): Deleted file name
+      message (str): Operation message
+  """
 
+  status: str
+  file_id: str
+  file_name: str
+  message: str
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
+    status = self.status
+
+    file_id = self.file_id
+
+    file_name = self.file_name
+
+    message = self.message
+
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "status": status,
+        "file_id": file_id,
+        "file_name": file_name,
+        "message": message,
+      }
+    )
 
     return field_dict
 
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     d = dict(src_dict)
-    delete_file_v1_graphs_graph_id_tables_files_file_id_delete_response_delete_file_v1_graphs_graph_id_tables_files_file_id_delete = cls()
+    status = d.pop("status")
+
+    file_id = d.pop("file_id")
+
+    file_name = d.pop("file_name")
+
+    message = d.pop("message")
+
+    delete_file_response = cls(
+      status=status,
+      file_id=file_id,
+      file_name=file_name,
+      message=message,
+    )
 
-    delete_file_v1_graphs_graph_id_tables_files_file_id_delete_response_delete_file_v1_graphs_graph_id_tables_files_file_id_delete.additional_properties = d
-    return delete_file_v1_graphs_graph_id_tables_files_file_id_delete_response_delete_file_v1_graphs_graph_id_tables_files_file_id_delete
+    delete_file_response.additional_properties = d
+    return delete_file_response
 
   @property
   def additional_keys(self) -> list[str]:

robosystems_client/models/file_info.py

@@ -0,0 +1,169 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="FileInfo")
+
+
+@_attrs_define
+class FileInfo:
+  """
+  Attributes:
+      file_id (str): Unique file identifier
+      file_name (str): Original file name
+      file_format (str): File format (parquet, csv, etc.)
+      size_bytes (int): File size in bytes
+      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
+  """
+
+  file_id: str
+  file_name: str
+  file_format: str
+  size_bytes: int
+  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
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    file_id = self.file_id
+
+    file_name = self.file_name
+
+    file_format = self.file_format
+
+    size_bytes = self.size_bytes
+
+    upload_status = self.upload_status
+
+    upload_method = self.upload_method
+
+    s3_key = self.s3_key
+
+    row_count: Union[None, Unset, int]
+    if isinstance(self.row_count, Unset):
+      row_count = UNSET
+    else:
+      row_count = self.row_count
+
+    created_at: Union[None, Unset, str]
+    if isinstance(self.created_at, Unset):
+      created_at = UNSET
+    else:
+      created_at = self.created_at
+
+    uploaded_at: Union[None, Unset, str]
+    if isinstance(self.uploaded_at, Unset):
+      uploaded_at = UNSET
+    else:
+      uploaded_at = self.uploaded_at
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "file_id": file_id,
+        "file_name": file_name,
+        "file_format": file_format,
+        "size_bytes": size_bytes,
+        "upload_status": upload_status,
+        "upload_method": upload_method,
+        "s3_key": s3_key,
+      }
+    )
+    if row_count is not UNSET:
+      field_dict["row_count"] = row_count
+    if created_at is not UNSET:
+      field_dict["created_at"] = created_at
+    if uploaded_at is not UNSET:
+      field_dict["uploaded_at"] = uploaded_at
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    file_id = d.pop("file_id")
+
+    file_name = d.pop("file_name")
+
+    file_format = d.pop("file_format")
+
+    size_bytes = d.pop("size_bytes")
+
+    upload_status = d.pop("upload_status")
+
+    upload_method = d.pop("upload_method")
+
+    s3_key = d.pop("s3_key")
+
+    def _parse_row_count(data: object) -> Union[None, Unset, int]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, int], data)
+
+    row_count = _parse_row_count(d.pop("row_count", UNSET))
+
+    def _parse_created_at(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    created_at = _parse_created_at(d.pop("created_at", UNSET))
+
+    def _parse_uploaded_at(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    uploaded_at = _parse_uploaded_at(d.pop("uploaded_at", UNSET))
+
+    file_info = cls(
+      file_id=file_id,
+      file_name=file_name,
+      file_format=file_format,
+      size_bytes=size_bytes,
+      upload_status=upload_status,
+      upload_method=upload_method,
+      s3_key=s3_key,
+      row_count=row_count,
+      created_at=created_at,
+      uploaded_at=uploaded_at,
+    )
+
+    file_info.additional_properties = d
+    return file_info
+
+  @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

@@ -0,0 +1,41 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+
+T = TypeVar("T", bound="FileStatusUpdate")
+
+
+@_attrs_define
+class FileStatusUpdate:
+  """
+  Attributes:
+      status (str): File status: 'uploaded' (ready for ingest), 'disabled' (exclude from ingest), 'archived' (soft
+          deleted)
+  """
+
+  status: str
+
+  def to_dict(self) -> dict[str, Any]:
+    status = self.status
+
+    field_dict: dict[str, Any] = {}
+
+    field_dict.update(
+      {
+        "status": status,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    status = d.pop("status")
+
+    file_status_update = cls(
+      status=status,
+    )
+
+    return file_status_update

robosystems_client/models/get_file_info_response.py

@@ -0,0 +1,205 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="GetFileInfoResponse")
+
+
+@_attrs_define
+class GetFileInfoResponse:
+  """
+  Attributes:
+      file_id (str): Unique file identifier
+      graph_id (str): Graph database identifier
+      table_id (str): Table identifier
+      file_name (str): Original file name
+      file_format (str): File format (parquet, csv, etc.)
+      size_bytes (int): File size in bytes
+      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
+  """
+
+  file_id: str
+  graph_id: str
+  table_id: str
+  file_name: str
+  file_format: str
+  size_bytes: int
+  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
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    file_id = self.file_id
+
+    graph_id = self.graph_id
+
+    table_id = self.table_id
+
+    file_name = self.file_name
+
+    file_format = self.file_format
+
+    size_bytes = self.size_bytes
+
+    upload_status = self.upload_status
+
+    upload_method = self.upload_method
+
+    s3_key = self.s3_key
+
+    table_name: Union[None, Unset, str]
+    if isinstance(self.table_name, Unset):
+      table_name = UNSET
+    else:
+      table_name = self.table_name
+
+    row_count: Union[None, Unset, int]
+    if isinstance(self.row_count, Unset):
+      row_count = UNSET
+    else:
+      row_count = self.row_count
+
+    created_at: Union[None, Unset, str]
+    if isinstance(self.created_at, Unset):
+      created_at = UNSET
+    else:
+      created_at = self.created_at
+
+    uploaded_at: Union[None, Unset, str]
+    if isinstance(self.uploaded_at, Unset):
+      uploaded_at = UNSET
+    else:
+      uploaded_at = self.uploaded_at
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "file_id": file_id,
+        "graph_id": graph_id,
+        "table_id": table_id,
+        "file_name": file_name,
+        "file_format": file_format,
+        "size_bytes": size_bytes,
+        "upload_status": upload_status,
+        "upload_method": upload_method,
+        "s3_key": s3_key,
+      }
+    )
+    if table_name is not UNSET:
+      field_dict["table_name"] = table_name
+    if row_count is not UNSET:
+      field_dict["row_count"] = row_count
+    if created_at is not UNSET:
+      field_dict["created_at"] = created_at
+    if uploaded_at is not UNSET:
+      field_dict["uploaded_at"] = uploaded_at
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    file_id = d.pop("file_id")
+
+    graph_id = d.pop("graph_id")
+
+    table_id = d.pop("table_id")
+
+    file_name = d.pop("file_name")
+
+    file_format = d.pop("file_format")
+
+    size_bytes = d.pop("size_bytes")
+
+    upload_status = d.pop("upload_status")
+
+    upload_method = d.pop("upload_method")
+
+    s3_key = d.pop("s3_key")
+
+    def _parse_table_name(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    table_name = _parse_table_name(d.pop("table_name", UNSET))
+
+    def _parse_row_count(data: object) -> Union[None, Unset, int]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, int], data)
+
+    row_count = _parse_row_count(d.pop("row_count", UNSET))
+
+    def _parse_created_at(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    created_at = _parse_created_at(d.pop("created_at", UNSET))
+
+    def _parse_uploaded_at(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    uploaded_at = _parse_uploaded_at(d.pop("uploaded_at", UNSET))
+
+    get_file_info_response = cls(
+      file_id=file_id,
+      graph_id=graph_id,
+      table_id=table_id,
+      file_name=file_name,
+      file_format=file_format,
+      size_bytes=size_bytes,
+      upload_status=upload_status,
+      upload_method=upload_method,
+      s3_key=s3_key,
+      table_name=table_name,
+      row_count=row_count,
+      created_at=created_at,
+      uploaded_at=uploaded_at,
+    )
+
+    get_file_info_response.additional_properties = d
+    return get_file_info_response
+
+  @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/list_table_files_response.py

@@ -0,0 +1,105 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+  from ..models.file_info import FileInfo
+
+
+T = TypeVar("T", bound="ListTableFilesResponse")
+
+
+@_attrs_define
+class ListTableFilesResponse:
+  """
+  Attributes:
+      graph_id (str): Graph database identifier
+      table_name (str): Table name
+      files (list['FileInfo']): List of files in the table
+      total_files (int): Total number of files
+      total_size_bytes (int): Total size of all files in bytes
+  """
+
+  graph_id: str
+  table_name: str
+  files: list["FileInfo"]
+  total_files: int
+  total_size_bytes: int
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    graph_id = self.graph_id
+
+    table_name = self.table_name
+
+    files = []
+    for files_item_data in self.files:
+      files_item = files_item_data.to_dict()
+      files.append(files_item)
+
+    total_files = self.total_files
+
+    total_size_bytes = self.total_size_bytes
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "graph_id": graph_id,
+        "table_name": table_name,
+        "files": files,
+        "total_files": total_files,
+        "total_size_bytes": total_size_bytes,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.file_info import FileInfo
+
+    d = dict(src_dict)
+    graph_id = d.pop("graph_id")
+
+    table_name = d.pop("table_name")
+
+    files = []
+    _files = d.pop("files")
+    for files_item_data in _files:
+      files_item = FileInfo.from_dict(files_item_data)
+
+      files.append(files_item)
+
+    total_files = d.pop("total_files")
+
+    total_size_bytes = d.pop("total_size_bytes")
+
+    list_table_files_response = cls(
+      graph_id=graph_id,
+      table_name=table_name,
+      files=files,
+      total_files=total_files,
+      total_size_bytes=total_size_bytes,
+    )
+
+    list_table_files_response.additional_properties = d
+    return list_table_files_response
+
+  @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/{get_file_info_v1_graphs_graph_id_tables_files_file_id_get_response_get_file_info_v1_graphs_graph_id_tables_files_file_id_get.py → update_file_status_response_updatefilestatus.py}

@@ -4,14 +4,11 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar(
-  "T",
-  bound="GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsGraphIdTablesFilesFileIdGet",
-)
+T = TypeVar("T", bound="UpdateFileStatusResponseUpdatefilestatus")
 
 
 @_attrs_define
-class GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsGraphIdTablesFilesFileIdGet:
+class UpdateFileStatusResponseUpdatefilestatus:
   """ """
 
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -25,10 +22,10 @@ class GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsG
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     d = dict(src_dict)
-    get_file_info_v1_graphs_graph_id_tables_files_file_id_get_response_get_file_info_v1_graphs_graph_id_tables_files_file_id_get = cls()
+    update_file_status_response_updatefilestatus = cls()
 
-    get_file_info_v1_graphs_graph_id_tables_files_file_id_get_response_get_file_info_v1_graphs_graph_id_tables_files_file_id_get.additional_properties = d
-    return get_file_info_v1_graphs_graph_id_tables_files_file_id_get_response_get_file_info_v1_graphs_graph_id_tables_files_file_id_get
+    update_file_status_response_updatefilestatus.additional_properties = d
+    return update_file_status_response_updatefilestatus
 
   @property
   def additional_keys(self) -> list[str]:

{robosystems_client-0.2.1.dist-info → robosystems_client-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.1
+Version: 0.2.3
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT