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

robosystems_client/extensions/sse_client.py

@@ -53,7 +53,7 @@ class SSEEvent:
   data: Any
   id: Optional[str] = None
   retry: Optional[int] = None
-  timestamp: datetime = None
+  timestamp: Optional[datetime] = None
 
   def __post_init__(self) -> None:
     if self.timestamp is None:

robosystems_client/extensions/table_ingest_client.py

@@ -11,20 +11,20 @@ import json
 import logging
 import httpx
 
-from ..api.tables.get_upload_url_v1_graphs_graph_id_tables_table_name_files_post import (
+from ..api.tables.get_upload_url import (
   sync_detailed as get_upload_url,
 )
-from ..api.tables.update_file_v1_graphs_graph_id_tables_files_file_id_patch import (
-  sync_detailed as update_file,
+from ..api.tables.update_file_status import (
+  sync_detailed as update_file_status,
 )
-from ..api.tables.list_tables_v1_graphs_graph_id_tables_get import (
+from ..api.tables.list_tables import (
   sync_detailed as list_tables,
 )
-from ..api.tables.ingest_tables_v1_graphs_graph_id_tables_ingest_post import (
+from ..api.tables.ingest_tables import (
   sync_detailed as ingest_tables,
 )
 from ..models.file_upload_request import FileUploadRequest
-from ..models.file_update_request import FileUpdateRequest
+from ..models.file_status_update import FileStatusUpdate
 from ..models.bulk_ingest_request import BulkIngestRequest
 
 logger = logging.getLogger(__name__)
@@ -95,7 +95,7 @@ class TableIngestClient:
     This method handles the complete 3-step upload process:
     1. Get presigned upload URL
     2. Upload file to S3
-    3. Update file metadata
+    3. Mark file as 'uploaded' (backend validates, calculates size/row count)
 
     Args:
         graph_id: The graph ID
@@ -104,7 +104,7 @@ class TableIngestClient:
         options: Upload options
 
     Returns:
-        UploadResult with upload details
+        UploadResult with upload details (size/row count calculated by backend)
     """
     if options is None:
       options = UploadOptions()
@@ -114,6 +114,9 @@ class TableIngestClient:
       file_or_buffer, "read"
     )
 
+    # Initialize file_path for type checking
+    file_path: Optional[Path] = None
+
     if is_buffer:
       # Handle buffer upload
       file_name = options.file_name or "data.parquet"
@@ -216,12 +219,12 @@ class TableIngestClient:
           # BinaryIO or file-like object
           file_or_buffer.seek(0)
           file_content = file_or_buffer.read()
-        file_size = len(file_content)
       else:
         # Read from file path
+        if file_path is None:
+          raise ValueError("file_path should not be None when not using buffer")
         with open(file_path, "rb") as f:
           file_content = f.read()
-        file_size = len(file_content)
 
       s3_response = self._http_client.put(
         upload_url,
@@ -230,54 +233,47 @@ class TableIngestClient:
       )
       s3_response.raise_for_status()
 
-      # Step 3: Get row count and update file metadata
+      # Step 3: Mark file as uploaded (backend validates and calculates size/row count)
       if options.on_progress:
-        options.on_progress(f"Updating file metadata for {file_name}...")
-
-      try:
-        import pyarrow.parquet as pq
-
-        if is_buffer:
-          # Read from buffer for row count
-          if hasattr(file_or_buffer, "seek"):
-            file_or_buffer.seek(0)
-          parquet_table = pq.read_table(file_or_buffer)
-        else:
-          # Read from file path
-          parquet_table = pq.read_table(file_path)
+        options.on_progress(f"Marking {file_name} as uploaded...")
 
-        row_count = parquet_table.num_rows
-      except ImportError:
-        logger.warning(
-          "pyarrow not installed, row count will be estimated from file size"
-        )
-        # Rough estimate: ~100 bytes per row for typical data
-        row_count = file_size // 100
-
-      metadata_update = FileUpdateRequest(
-        file_size_bytes=file_size, row_count=row_count
-      )
+      status_update = FileStatusUpdate(status="uploaded")
 
       kwargs = {
         "graph_id": graph_id,
         "file_id": file_id,
         "client": client,
-        "body": metadata_update,
+        "body": status_update,
       }
 
-      update_response = update_file(**kwargs)
+      update_response = update_file_status(**kwargs)
 
       if not update_response.parsed:
+        logger.error(
+          f"No parsed response from update_file_status. Status code: {update_response.status_code}"
+        )
         return UploadResult(
           file_id=file_id,
-          file_size=file_size,
-          row_count=row_count,
+          file_size=0,
+          row_count=0,
           table_name=table_name,
           file_name=file_name,
           success=False,
-          error="Failed to update file metadata",
+          error="Failed to complete file upload",
         )
 
+      response_data = update_response.parsed
+
+      if isinstance(response_data, dict):
+        file_size = response_data.get("file_size_bytes", 0)
+        row_count = response_data.get("row_count", 0)
+      elif hasattr(response_data, "additional_properties"):
+        file_size = response_data.additional_properties.get("file_size_bytes", 0)
+        row_count = response_data.additional_properties.get("row_count", 0)
+      else:
+        file_size = getattr(response_data, "file_size_bytes", 0)
+        row_count = getattr(response_data, "row_count", 0)
+
       if options.on_progress:
         options.on_progress(
           f"✅ Uploaded {file_name} ({file_size:,} bytes, {row_count:,} rows)"

robosystems_client/extensions/utils.py

@@ -144,7 +144,7 @@ class ResultProcessor:
             writer.writerow(headers)
 
           for row in data:
-            if isinstance(row, dict):
+            if isinstance(row, dict) and headers:
               writer.writerow([row.get(h, "") for h in headers])
             elif isinstance(row, (list, tuple)):
               writer.writerow(row)
@@ -159,7 +159,7 @@ class ResultProcessor:
           writer.writerow(headers)
 
         for row in data:
-          if isinstance(row, dict):
+          if isinstance(row, dict) and headers:
             writer.writerow([row.get(h, "") for h in headers])
           elif isinstance(row, (list, tuple)):
             writer.writerow(row)

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,
 )
@@ -127,9 +124,6 @@ from .get_graph_limits_response_getgraphlimits import (
 from .get_graph_monthly_bill_response_getgraphmonthlybill import (
   GetGraphMonthlyBillResponseGetgraphmonthlybill,
 )
-from .get_graph_schema_response_getgraphschema import (
-  GetGraphSchemaResponseGetgraphschema,
-)
 from .get_graph_usage_details_response_getgraphusagedetails import (
   GetGraphUsageDetailsResponseGetgraphusagedetails,
 )
@@ -164,9 +158,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
@@ -207,6 +199,8 @@ from .schema_export_response_data_stats_type_0 import SchemaExportResponseDataSt
 from .schema_export_response_schema_definition_type_0 import (
   SchemaExportResponseSchemaDefinitionType0,
 )
+from .schema_info_response import SchemaInfoResponse
+from .schema_info_response_schema import SchemaInfoResponseSchema
 from .schema_validation_request import SchemaValidationRequest
 from .schema_validation_request_schema_definition_type_0 import (
   SchemaValidationRequestSchemaDefinitionType0,
@@ -249,8 +243,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 +333,7 @@ __all__ = (
   "CypherQueryRequestParametersType0",
   "DatabaseHealthResponse",
   "DatabaseInfoResponse",
-  "DeleteFileV1GraphsGraphIdTablesFilesFileIdDeleteResponseDeleteFileV1GraphsGraphIdTablesFilesFileIdDelete",
+  "DeleteFileResponse",
   "DeleteSubgraphRequest",
   "DeleteSubgraphResponse",
   "DetailedTransactionsResponse",
@@ -351,7 +345,8 @@ __all__ = (
   "ErrorResponse",
   "ExchangeTokenRequest",
   "ExchangeTokenRequestMetadataType0",
-  "FileUpdateRequest",
+  "FileInfo",
+  "FileStatusUpdate",
   "FileUploadRequest",
   "FileUploadResponse",
   "ForgotPasswordRequest",
@@ -361,11 +356,10 @@ __all__ = (
   "GetBackupDownloadUrlResponseGetbackupdownloadurl",
   "GetCurrentAuthUserResponseGetcurrentauthuser",
   "GetCurrentGraphBillResponseGetcurrentgraphbill",
-  "GetFileInfoV1GraphsGraphIdTablesFilesFileIdGetResponseGetFileInfoV1GraphsGraphIdTablesFilesFileIdGet",
+  "GetFileInfoResponse",
   "GetGraphBillingHistoryResponseGetgraphbillinghistory",
   "GetGraphLimitsResponseGetgraphlimits",
   "GetGraphMonthlyBillResponseGetgraphmonthlybill",
-  "GetGraphSchemaResponseGetgraphschema",
   "GetGraphUsageDetailsResponseGetgraphusagedetails",
   "GetOperationStatusResponseGetoperationstatus",
   "GetSharedRepositoryLimitsResponseGetsharedrepositorylimits",
@@ -390,7 +384,7 @@ __all__ = (
   "LinkTokenRequestProviderType0",
   "ListConnectionsProviderType0",
   "ListSubgraphsResponse",
-  "ListTableFilesV1GraphsGraphIdTablesTableNameFilesGetResponseListTableFilesV1GraphsGraphIdTablesTableNameFilesGet",
+  "ListTableFilesResponse",
   "LoginRequest",
   "LogoutUserResponseLogoutuser",
   "MCPToolCall",
@@ -421,6 +415,8 @@ __all__ = (
   "SchemaExportResponse",
   "SchemaExportResponseDataStatsType0",
   "SchemaExportResponseSchemaDefinitionType0",
+  "SchemaInfoResponse",
+  "SchemaInfoResponseSchema",
   "SchemaValidationRequest",
   "SchemaValidationRequestSchemaDefinitionType0",
   "SchemaValidationResponse",
@@ -455,7 +451,7 @@ __all__ = (
   "TierUpgradeRequest",
   "TransactionSummaryResponse",
   "UpdateAPIKeyRequest",
-  "UpdateFileV1GraphsGraphIdTablesFilesFileIdPatchResponseUpdateFileV1GraphsGraphIdTablesFilesFileIdPatch",
+  "UpdateFileStatusResponseUpdatefilestatus",
   "UpdatePasswordRequest",
   "UpdateUserRequest",
   "UserAnalyticsResponse",

robosystems_client/models/create_graph_request.py

@@ -20,9 +20,10 @@ class CreateGraphRequest:
   """Request model for creating a new graph.
 
   Example:
-      {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'}, 'instance_tier':
-          'kuzu-standard', 'metadata': {'description': 'Main production graph', 'graph_name': 'Production System',
-          'schema_extensions': ['roboledger']}, 'tags': ['production', 'finance']}
+      {'initial_entity': {'cik': '0001234567', 'name': 'Acme Consulting LLC', 'uri': 'https://acmeconsulting.com'},
+          'instance_tier': 'kuzu-standard', 'metadata': {'description': 'Professional consulting services with full
+          accounting integration', 'graph_name': 'Acme Consulting LLC', 'schema_extensions': ['roboledger']}, 'tags':
+          ['consulting', 'professional-services']}
 
   Attributes:
       metadata (GraphMetadata): Metadata for graph creation.
@@ -31,6 +32,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 +42,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 +70,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 +89,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 +143,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 +152,7 @@ class CreateGraphRequest:
       instance_tier=instance_tier,
       custom_schema=custom_schema,
       initial_entity=initial_entity,
+      create_entity=create_entity,
       tags=tags,
     )
 

robosystems_client/models/cypher_query_request.py

@@ -2,7 +2,6 @@ from collections.abc import Mapping
 from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
-from attrs import field as _attrs_field
 
 from ..types import UNSET, Unset
 
@@ -20,15 +19,16 @@ class CypherQueryRequest:
   """Request model for Cypher query execution.
 
   Attributes:
-      query (str): The Cypher query to execute
-      parameters (Union['CypherQueryRequestParametersType0', None, Unset]): Optional parameters for the Cypher query
+      query (str): The Cypher query to execute. Use parameters ($param_name) for all dynamic values to prevent
+          injection attacks.
+      parameters (Union['CypherQueryRequestParametersType0', None, Unset]): Query parameters for safe value
+          substitution. ALWAYS use parameters instead of string interpolation.
       timeout (Union[None, Unset, int]): Query timeout in seconds (1-300) Default: 60.
   """
 
   query: str
   parameters: Union["CypherQueryRequestParametersType0", None, Unset] = UNSET
   timeout: Union[None, Unset, int] = 60
-  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
     from ..models.cypher_query_request_parameters_type_0 import (
@@ -52,7 +52,7 @@ class CypherQueryRequest:
       timeout = self.timeout
 
     field_dict: dict[str, Any] = {}
-    field_dict.update(self.additional_properties)
+
     field_dict.update(
       {
         "query": query,
@@ -108,21 +108,4 @@ class CypherQueryRequest:
       timeout=timeout,
     )
 
-    cypher_query_request.additional_properties = d
     return cypher_query_request
-
-  @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/{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