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

robosystems_client/models/__init__.py

@@ -75,9 +75,7 @@ from .cypher_query_request import CypherQueryRequest
 from .cypher_query_request_parameters_type_0 import CypherQueryRequestParametersType0
 from .database_health_response import DatabaseHealthResponse
 from .database_info_response import DatabaseInfoResponse
-from .delete_file_v1_graphs_graph_id_tables_files_file_id_delete_response_delete_file_v1_graphs_graph_id_tables_files_file_id_delete import (
-  DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete,
-)
+from .delete_file_response import DeleteFileResponse
 from .delete_subgraph_request import DeleteSubgraphRequest
 from .delete_subgraph_response import DeleteSubgraphResponse
 from .detailed_transactions_response import DetailedTransactionsResponse
@@ -93,7 +91,8 @@ from .enhanced_credit_transaction_response_metadata import (
 from .error_response import ErrorResponse
 from .exchange_token_request import ExchangeTokenRequest
 from .exchange_token_request_metadata_type_0 import ExchangeTokenRequestMetadataType0
-from .file_update_request import FileUpdateRequest
+from .file_info import FileInfo
+from .file_status_update import FileStatusUpdate
 from .file_upload_request import FileUploadRequest
 from .file_upload_response import FileUploadResponse
 from .forgot_password_request import ForgotPasswordRequest
@@ -115,9 +114,7 @@ from .get_current_auth_user_response_getcurrentauthuser import (
 from .get_current_graph_bill_response_getcurrentgraphbill import (
   GetCurrentGraphBillResponseGetcurrentgraphbill,
 )
-from .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 import (
-  GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsGraphIdTablesFilesFileIdGet,
-)
+from .get_file_info_response import GetFileInfoResponse
 from .get_graph_billing_history_response_getgraphbillinghistory import (
   GetGraphBillingHistoryResponseGetgraphbillinghistory,
 )
@@ -164,9 +161,7 @@ from .link_token_request_options_type_0 import LinkTokenRequestOptionsType0
 from .link_token_request_provider_type_0 import LinkTokenRequestProviderType0
 from .list_connections_provider_type_0 import ListConnectionsProviderType0
 from .list_subgraphs_response import ListSubgraphsResponse
-from .list_table_files_v1_graphs_graph_id_tables_table_name_files_get_response_list_table_files_v1_graphs_graph_id_tables_table_name_files_get import (
-  ListTableFilesV1GraphsGraphIdTablesTableNameFilesGetResponseListTableFilesV1GraphsGraphIdTablesTableNameFilesGet,
-)
+from .list_table_files_response import ListTableFilesResponse
 from .login_request import LoginRequest
 from .logout_user_response_logoutuser import LogoutUserResponseLogoutuser
 from .mcp_tool_call import MCPToolCall
@@ -249,8 +244,8 @@ from .table_query_response import TableQueryResponse
 from .tier_upgrade_request import TierUpgradeRequest
 from .transaction_summary_response import TransactionSummaryResponse
 from .update_api_key_request import UpdateAPIKeyRequest
-from .update_file_v1_graphs_graph_id_tables_files_file_id_patch_response_update_file_v1_graphs_graph_id_tables_files_file_id_patch import (
-  UpdateFileV1GraphsGraphIdTablesFilesFileIdPatchResponseUpdateFileV1GraphsGraphIdTablesFilesFileIdPatch,
+from .update_file_status_response_updatefilestatus import (
+  UpdateFileStatusResponseUpdatefilestatus,
 )
 from .update_password_request import UpdatePasswordRequest
 from .update_user_request import UpdateUserRequest
@@ -339,7 +334,7 @@ __all__ = (
   "CypherQueryRequestParametersType0",
   "DatabaseHealthResponse",
   "DatabaseInfoResponse",
-  "DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete",
+  "DeleteFileResponse",
   "DeleteSubgraphRequest",
   "DeleteSubgraphResponse",
   "DetailedTransactionsResponse",
@@ -351,7 +346,8 @@ __all__ = (
   "ErrorResponse",
   "ExchangeTokenRequest",
   "ExchangeTokenRequestMetadataType0",
-  "FileUpdateRequest",
+  "FileInfo",
+  "FileStatusUpdate",
   "FileUploadRequest",
   "FileUploadResponse",
   "ForgotPasswordRequest",
@@ -361,7 +357,7 @@ __all__ = (
   "GetBackupDownloadUrlResponseGetbackupdownloadurl",
   "GetCurrentAuthUserResponseGetcurrentauthuser",
   "GetCurrentGraphBillResponseGetcurrentgraphbill",
-  "GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsGraphIdTablesFilesFileIdGet",
+  "GetFileInfoResponse",
   "GetGraphBillingHistoryResponseGetgraphbillinghistory",
   "GetGraphLimitsResponseGetgraphlimits",
   "GetGraphMonthlyBillResponseGetgraphmonthlybill",
@@ -390,7 +386,7 @@ __all__ = (
   "LinkTokenRequestProviderType0",
   "ListConnectionsProviderType0",
   "ListSubgraphsResponse",
-  "ListTableFilesV1GraphsGraphIdTablesTableNameFilesGetResponseListTableFilesV1GraphsGraphIdTablesTableNameFilesGet",
+  "ListTableFilesResponse",
   "LoginRequest",
   "LogoutUserResponseLogoutuser",
   "MCPToolCall",
@@ -455,7 +451,7 @@ __all__ = (
   "TierUpgradeRequest",
   "TransactionSummaryResponse",
   "UpdateAPIKeyRequest",
-  "UpdateFileV1GraphsGraphIdTablesFilesFileIdPatchResponseUpdateFileV1GraphsGraphIdTablesFilesFileIdPatch",
+  "UpdateFileStatusResponseUpdatefilestatus",
   "UpdatePasswordRequest",
   "UpdateUserRequest",
   "UserAnalyticsResponse",

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