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

robosystems_client/models/__init__.py

@@ -36,6 +36,8 @@ from .backup_stats_response import BackupStatsResponse
 from .backup_stats_response_backup_formats import BackupStatsResponseBackupFormats
 from .batch_agent_request import BatchAgentRequest
 from .batch_agent_response import BatchAgentResponse
+from .bulk_ingest_request import BulkIngestRequest
+from .bulk_ingest_response import BulkIngestResponse
 from .cancel_operation_response_canceloperation import (
   CancelOperationResponseCanceloperation,
 )
@@ -50,9 +52,6 @@ from .connection_provider_info_provider import ConnectionProviderInfoProvider
 from .connection_response import ConnectionResponse
 from .connection_response_metadata import ConnectionResponseMetadata
 from .connection_response_provider import ConnectionResponseProvider
-from .copy_response import CopyResponse
-from .copy_response_error_details_type_0 import CopyResponseErrorDetailsType0
-from .copy_response_status import CopyResponseStatus
 from .create_api_key_request import CreateAPIKeyRequest
 from .create_api_key_response import CreateAPIKeyResponse
 from .create_connection_request import CreateConnectionRequest
@@ -74,10 +73,11 @@ from .custom_schema_definition_relationships_item import (
 )
 from .cypher_query_request import CypherQueryRequest
 from .cypher_query_request_parameters_type_0 import CypherQueryRequestParametersType0
-from .data_frame_copy_request import DataFrameCopyRequest
-from .data_frame_copy_request_format import DataFrameCopyRequestFormat
 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_subgraph_request import DeleteSubgraphRequest
 from .delete_subgraph_response import DeleteSubgraphResponse
 from .detailed_transactions_response import DetailedTransactionsResponse
@@ -93,6 +93,9 @@ 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_upload_request import FileUploadRequest
+from .file_upload_response import FileUploadResponse
 from .forgot_password_request import ForgotPasswordRequest
 from .forgot_password_response_forgotpassword import (
   ForgotPasswordResponseForgotpassword,
@@ -112,6 +115,9 @@ 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_graph_billing_history_response_getgraphbillinghistory import (
   GetGraphBillingHistoryResponseGetgraphbillinghistory,
 )
@@ -121,8 +127,8 @@ from .get_graph_limits_response_getgraphlimits import (
 from .get_graph_monthly_bill_response_getgraphmonthlybill import (
   GetGraphMonthlyBillResponseGetgraphmonthlybill,
 )
-from .get_graph_schema_info_response_getgraphschemainfo import (
-  GetGraphSchemaInfoResponseGetgraphschemainfo,
+from .get_graph_schema_response_getgraphschema import (
+  GetGraphSchemaResponseGetgraphschema,
 )
 from .get_graph_usage_details_response_getgraphusagedetails import (
   GetGraphUsageDetailsResponseGetgraphusagedetails,
@@ -157,10 +163,10 @@ from .link_token_request import LinkTokenRequest
 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_schema_extensions_response_listschemaextensions import (
-  ListSchemaExtensionsResponseListschemaextensions,
-)
 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 .login_request import LoginRequest
 from .logout_user_response_logoutuser import LogoutUserResponseLogoutuser
 from .mcp_tool_call import MCPToolCall
@@ -196,9 +202,6 @@ from .resend_verification_email_response_resendverificationemail import (
 from .reset_password_request import ResetPasswordRequest
 from .reset_password_validate_response import ResetPasswordValidateResponse
 from .response_mode import ResponseMode
-from .s3_copy_request import S3CopyRequest
-from .s3_copy_request_file_format import S3CopyRequestFileFormat
-from .s3_copy_request_s3_url_style_type_0 import S3CopyRequestS3UrlStyleType0
 from .schema_export_response import SchemaExportResponse
 from .schema_export_response_data_stats_type_0 import SchemaExportResponseDataStatsType0
 from .schema_export_response_schema_definition_type_0 import (
@@ -238,14 +241,19 @@ from .sync_connection_request_sync_options_type_0 import (
 from .sync_connection_response_syncconnection import (
   SyncConnectionResponseSyncconnection,
 )
+from .table_info import TableInfo
+from .table_ingest_result import TableIngestResult
+from .table_list_response import TableListResponse
+from .table_query_request import TableQueryRequest
+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_password_request import UpdatePasswordRequest
 from .update_user_request import UpdateUserRequest
-from .url_copy_request import URLCopyRequest
-from .url_copy_request_file_format import URLCopyRequestFileFormat
-from .url_copy_request_headers_type_0 import URLCopyRequestHeadersType0
 from .user_analytics_response import UserAnalyticsResponse
 from .user_analytics_response_api_usage import UserAnalyticsResponseApiUsage
 from .user_analytics_response_graph_usage import UserAnalyticsResponseGraphUsage
@@ -300,6 +308,8 @@ __all__ = (
   "BackupStatsResponseBackupFormats",
   "BatchAgentRequest",
   "BatchAgentResponse",
+  "BulkIngestRequest",
+  "BulkIngestResponse",
   "CancellationResponse",
   "CancelOperationResponseCanceloperation",
   "CheckCreditBalanceResponseCheckcreditbalance",
@@ -310,9 +320,6 @@ __all__ = (
   "ConnectionResponse",
   "ConnectionResponseMetadata",
   "ConnectionResponseProvider",
-  "CopyResponse",
-  "CopyResponseErrorDetailsType0",
-  "CopyResponseStatus",
   "CreateAPIKeyRequest",
   "CreateAPIKeyResponse",
   "CreateConnectionRequest",
@@ -332,8 +339,7 @@ __all__ = (
   "CypherQueryRequestParametersType0",
   "DatabaseHealthResponse",
   "DatabaseInfoResponse",
-  "DataFrameCopyRequest",
-  "DataFrameCopyRequestFormat",
+  "DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete",
   "DeleteSubgraphRequest",
   "DeleteSubgraphResponse",
   "DetailedTransactionsResponse",
@@ -345,6 +351,9 @@ __all__ = (
   "ErrorResponse",
   "ExchangeTokenRequest",
   "ExchangeTokenRequestMetadataType0",
+  "FileUpdateRequest",
+  "FileUploadRequest",
+  "FileUploadResponse",
   "ForgotPasswordRequest",
   "ForgotPasswordResponseForgotpassword",
   "GetAllCreditSummariesResponseGetallcreditsummaries",
@@ -352,10 +361,11 @@ __all__ = (
   "GetBackupDownloadUrlResponseGetbackupdownloadurl",
   "GetCurrentAuthUserResponseGetcurrentauthuser",
   "GetCurrentGraphBillResponseGetcurrentgraphbill",
+  "GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsGraphIdTablesFilesFileIdGet",
   "GetGraphBillingHistoryResponseGetgraphbillinghistory",
   "GetGraphLimitsResponseGetgraphlimits",
   "GetGraphMonthlyBillResponseGetgraphmonthlybill",
-  "GetGraphSchemaInfoResponseGetgraphschemainfo",
+  "GetGraphSchemaResponseGetgraphschema",
   "GetGraphUsageDetailsResponseGetgraphusagedetails",
   "GetOperationStatusResponseGetoperationstatus",
   "GetSharedRepositoryLimitsResponseGetsharedrepositorylimits",
@@ -379,8 +389,8 @@ __all__ = (
   "LinkTokenRequestOptionsType0",
   "LinkTokenRequestProviderType0",
   "ListConnectionsProviderType0",
-  "ListSchemaExtensionsResponseListschemaextensions",
   "ListSubgraphsResponse",
+  "ListTableFilesV1GraphsGraphIdTablesTableNameFilesGetResponseListTableFilesV1GraphsGraphIdTablesTableNameFilesGet",
   "LoginRequest",
   "LogoutUserResponseLogoutuser",
   "MCPToolCall",
@@ -408,9 +418,6 @@ __all__ = (
   "ResetPasswordRequest",
   "ResetPasswordValidateResponse",
   "ResponseMode",
-  "S3CopyRequest",
-  "S3CopyRequestFileFormat",
-  "S3CopyRequestS3UrlStyleType0",
   "SchemaExportResponse",
   "SchemaExportResponseDataStatsType0",
   "SchemaExportResponseSchemaDefinitionType0",
@@ -440,14 +447,17 @@ __all__ = (
   "SyncConnectionRequest",
   "SyncConnectionRequestSyncOptionsType0",
   "SyncConnectionResponseSyncconnection",
+  "TableInfo",
+  "TableIngestResult",
+  "TableListResponse",
+  "TableQueryRequest",
+  "TableQueryResponse",
   "TierUpgradeRequest",
   "TransactionSummaryResponse",
   "UpdateAPIKeyRequest",
+  "UpdateFileV1GraphsGraphIdTablesFilesFileIdPatchResponseUpdateFileV1GraphsGraphIdTablesFilesFileIdPatch",
   "UpdatePasswordRequest",
   "UpdateUserRequest",
-  "URLCopyRequest",
-  "URLCopyRequestFileFormat",
-  "URLCopyRequestHeadersType0",
   "UserAnalyticsResponse",
   "UserAnalyticsResponseApiUsage",
   "UserAnalyticsResponseGraphUsage",

robosystems_client/models/backup_restore_request.py

@@ -14,31 +14,23 @@ class BackupRestoreRequest:
   """Request model for restoring from a backup.
 
   Attributes:
-      backup_id (str): ID of backup to restore from
       create_system_backup (Union[Unset, bool]): Create a system backup of existing database before restore Default:
           True.
       verify_after_restore (Union[Unset, bool]): Verify database integrity after restore Default: True.
   """
 
-  backup_id: str
   create_system_backup: Union[Unset, bool] = True
   verify_after_restore: Union[Unset, bool] = True
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
-    backup_id = self.backup_id
-
     create_system_backup = self.create_system_backup
 
     verify_after_restore = self.verify_after_restore
 
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
-    field_dict.update(
-      {
-        "backup_id": backup_id,
-      }
-    )
+    field_dict.update({})
     if create_system_backup is not UNSET:
       field_dict["create_system_backup"] = create_system_backup
     if verify_after_restore is not UNSET:
@@ -49,14 +41,11 @@ class BackupRestoreRequest:
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     d = dict(src_dict)
-    backup_id = d.pop("backup_id")
-
     create_system_backup = d.pop("create_system_backup", UNSET)
 
     verify_after_restore = d.pop("verify_after_restore", UNSET)
 
     backup_restore_request = cls(
-      backup_id=backup_id,
       create_system_backup=create_system_backup,
       verify_after_restore=verify_after_restore,
     )

robosystems_client/models/bulk_ingest_request.py

@@ -0,0 +1,50 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="BulkIngestRequest")
+
+
+@_attrs_define
+class BulkIngestRequest:
+  """
+  Attributes:
+      ignore_errors (Union[Unset, bool]): Continue ingestion on row errors Default: True.
+      rebuild (Union[Unset, bool]): Rebuild graph database from scratch before ingestion. Safe operation - staged data
+          is the source of truth, graph can always be regenerated. Default: False.
+  """
+
+  ignore_errors: Union[Unset, bool] = True
+  rebuild: Union[Unset, bool] = False
+
+  def to_dict(self) -> dict[str, Any]:
+    ignore_errors = self.ignore_errors
+
+    rebuild = self.rebuild
+
+    field_dict: dict[str, Any] = {}
+
+    field_dict.update({})
+    if ignore_errors is not UNSET:
+      field_dict["ignore_errors"] = ignore_errors
+    if rebuild is not UNSET:
+      field_dict["rebuild"] = rebuild
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    ignore_errors = d.pop("ignore_errors", UNSET)
+
+    rebuild = d.pop("rebuild", UNSET)
+
+    bulk_ingest_request = cls(
+      ignore_errors=ignore_errors,
+      rebuild=rebuild,
+    )
+
+    return bulk_ingest_request

robosystems_client/models/bulk_ingest_response.py

@@ -0,0 +1,137 @@
+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.table_ingest_result import TableIngestResult
+
+
+T = TypeVar("T", bound="BulkIngestResponse")
+
+
+@_attrs_define
+class BulkIngestResponse:
+  """
+  Attributes:
+      status (str): Overall ingestion status
+      graph_id (str): Graph database identifier
+      total_tables (int): Total number of tables processed
+      successful_tables (int): Number of successfully ingested tables
+      failed_tables (int): Number of failed table ingestions
+      skipped_tables (int): Number of skipped tables (no files)
+      total_rows_ingested (int): Total rows ingested across all tables
+      total_execution_time_ms (float): Total execution time in milliseconds
+      results (list['TableIngestResult']): Per-table ingestion results
+  """
+
+  status: str
+  graph_id: str
+  total_tables: int
+  successful_tables: int
+  failed_tables: int
+  skipped_tables: int
+  total_rows_ingested: int
+  total_execution_time_ms: float
+  results: list["TableIngestResult"]
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    status = self.status
+
+    graph_id = self.graph_id
+
+    total_tables = self.total_tables
+
+    successful_tables = self.successful_tables
+
+    failed_tables = self.failed_tables
+
+    skipped_tables = self.skipped_tables
+
+    total_rows_ingested = self.total_rows_ingested
+
+    total_execution_time_ms = self.total_execution_time_ms
+
+    results = []
+    for results_item_data in self.results:
+      results_item = results_item_data.to_dict()
+      results.append(results_item)
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "status": status,
+        "graph_id": graph_id,
+        "total_tables": total_tables,
+        "successful_tables": successful_tables,
+        "failed_tables": failed_tables,
+        "skipped_tables": skipped_tables,
+        "total_rows_ingested": total_rows_ingested,
+        "total_execution_time_ms": total_execution_time_ms,
+        "results": results,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.table_ingest_result import TableIngestResult
+
+    d = dict(src_dict)
+    status = d.pop("status")
+
+    graph_id = d.pop("graph_id")
+
+    total_tables = d.pop("total_tables")
+
+    successful_tables = d.pop("successful_tables")
+
+    failed_tables = d.pop("failed_tables")
+
+    skipped_tables = d.pop("skipped_tables")
+
+    total_rows_ingested = d.pop("total_rows_ingested")
+
+    total_execution_time_ms = d.pop("total_execution_time_ms")
+
+    results = []
+    _results = d.pop("results")
+    for results_item_data in _results:
+      results_item = TableIngestResult.from_dict(results_item_data)
+
+      results.append(results_item)
+
+    bulk_ingest_response = cls(
+      status=status,
+      graph_id=graph_id,
+      total_tables=total_tables,
+      successful_tables=successful_tables,
+      failed_tables=failed_tables,
+      skipped_tables=skipped_tables,
+      total_rows_ingested=total_rows_ingested,
+      total_execution_time_ms=total_execution_time_ms,
+      results=results,
+    )
+
+    bulk_ingest_response.additional_properties = d
+    return bulk_ingest_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/create_graph_request.py

@@ -21,12 +21,13 @@ class CreateGraphRequest:
 
   Example:
       {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'}, 'instance_tier':
-          'standard', 'metadata': {'description': 'Main production graph', 'graph_name': 'Production System',
+          'kuzu-standard', 'metadata': {'description': 'Main production graph', 'graph_name': 'Production System',
           'schema_extensions': ['roboledger']}, 'tags': ['production', 'finance']}
 
   Attributes:
       metadata (GraphMetadata): Metadata for graph creation.
-      instance_tier (Union[Unset, str]): Instance tier: standard, enterprise, or premium Default: 'standard'.
+      instance_tier (Union[Unset, str]): Instance tier: kuzu-standard, kuzu-large, kuzu-xlarge, neo4j-community-large,
+          neo4j-enterprise-xlarge Default: 'kuzu-standard'.
       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.
@@ -34,7 +35,7 @@ class CreateGraphRequest:
   """
 
   metadata: "GraphMetadata"
-  instance_tier: Union[Unset, str] = "standard"
+  instance_tier: Union[Unset, str] = "kuzu-standard"
   custom_schema: Union["CustomSchemaDefinition", None, Unset] = UNSET
   initial_entity: Union["InitialEntityData", None, Unset] = UNSET
   tags: Union[Unset, list[str]] = UNSET

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

@@ -0,0 +1,47 @@
+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="DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete",
+)
+
+
+@_attrs_define
+class DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete:
+  """ """
+
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+
+    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()
+
+    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
+
+  @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_update_request.py

@@ -0,0 +1,62 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="FileUpdateRequest")
+
+
+@_attrs_define
+class FileUpdateRequest:
+  """
+  Attributes:
+      file_size_bytes (int): Actual uploaded file size in bytes
+      row_count (Union[None, Unset, int]): Number of rows in the file
+  """
+
+  file_size_bytes: int
+  row_count: Union[None, Unset, int] = UNSET
+
+  def to_dict(self) -> dict[str, Any]:
+    file_size_bytes = self.file_size_bytes
+
+    row_count: Union[None, Unset, int]
+    if isinstance(self.row_count, Unset):
+      row_count = UNSET
+    else:
+      row_count = self.row_count
+
+    field_dict: dict[str, Any] = {}
+
+    field_dict.update(
+      {
+        "file_size_bytes": file_size_bytes,
+      }
+    )
+    if row_count is not UNSET:
+      field_dict["row_count"] = row_count
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    file_size_bytes = d.pop("file_size_bytes")
+
+    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))
+
+    file_update_request = cls(
+      file_size_bytes=file_size_bytes,
+      row_count=row_count,
+    )
+
+    return file_update_request

robosystems_client/models/file_upload_request.py

@@ -0,0 +1,51 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="FileUploadRequest")
+
+
+@_attrs_define
+class FileUploadRequest:
+  """
+  Attributes:
+      file_name (str): File name to upload
+      content_type (Union[Unset, str]): File MIME type Default: 'application/x-parquet'.
+  """
+
+  file_name: str
+  content_type: Union[Unset, str] = "application/x-parquet"
+
+  def to_dict(self) -> dict[str, Any]:
+    file_name = self.file_name
+
+    content_type = self.content_type
+
+    field_dict: dict[str, Any] = {}
+
+    field_dict.update(
+      {
+        "file_name": file_name,
+      }
+    )
+    if content_type is not UNSET:
+      field_dict["content_type"] = content_type
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    file_name = d.pop("file_name")
+
+    content_type = d.pop("content_type", UNSET)
+
+    file_upload_request = cls(
+      file_name=file_name,
+      content_type=content_type,
+    )
+
+    return file_upload_request