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

robosystems_client/api/tables/list_table_files.py

@@ -1,329 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union, cast
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.error_response import ErrorResponse
-from ...models.http_validation_error import HTTPValidationError
-from ...models.list_table_files_response import ListTableFilesResponse
-from ...types import Response
-
-
-def _get_kwargs(
-  graph_id: str,
-  table_name: str,
-) -> dict[str, Any]:
-  _kwargs: dict[str, Any] = {
-    "method": "get",
-    "url": f"/v1/graphs/{graph_id}/tables/{table_name}/files",
-  }
-
-  return _kwargs
-
-
-def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]:
-  if response.status_code == 200:
-    response_200 = ListTableFilesResponse.from_dict(response.json())
-
-    return response_200
-
-  if response.status_code == 401:
-    response_401 = cast(Any, None)
-    return response_401
-
-  if response.status_code == 403:
-    response_403 = ErrorResponse.from_dict(response.json())
-
-    return response_403
-
-  if response.status_code == 404:
-    response_404 = ErrorResponse.from_dict(response.json())
-
-    return response_404
-
-  if response.status_code == 422:
-    response_422 = HTTPValidationError.from_dict(response.json())
-
-    return response_422
-
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
-
-  if client.raise_on_unexpected_status:
-    raise errors.UnexpectedStatus(response.status_code, response.content)
-  else:
-    return None
-
-
-def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]:
-  return Response(
-    status_code=HTTPStatus(response.status_code),
-    content=response.content,
-    headers=response.headers,
-    parsed=_parse_response(client=client, response=response),
-  )
-
-
-def sync_detailed(
-  graph_id: str,
-  table_name: str,
-  *,
-  client: AuthenticatedClient,
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]:
-  """List Files in Staging Table
-
-   List all files uploaded to a staging table with comprehensive metadata.
-
-  Get a complete inventory of all files in a staging table, including upload status,
-  file sizes, row counts, and S3 locations. Essential for monitoring upload progress
-  and validating data before ingestion.
-
-  **Use Cases:**
-  - Monitor file upload progress
-  - Verify files are ready for ingestion
-  - Check file formats and sizes
-  - Track storage usage per table
-  - Identify failed or incomplete uploads
-  - Pre-ingestion validation
-
-  **Returned Metadata:**
-  - File ID, name, and format (parquet, csv, json)
-  - Size in bytes and row count (if available)
-  - Upload status and method
-  - Creation and upload timestamps
-  - S3 key for reference
-
-  **Upload Status Values:**
-  - `pending`: Upload URL generated, awaiting upload
-  - `uploaded`: Successfully uploaded, ready for ingestion
-  - `disabled`: Excluded from ingestion
-  - `archived`: Soft deleted
-  - `failed`: Upload failed
-
-  **Important Notes:**
-  - Only `uploaded` files are ingested
-  - Check `row_count` to estimate data volume
-  - Use `total_size_bytes` for storage monitoring
-  - Files with `failed` status should be deleted and re-uploaded
-  - File listing is included - no credit consumption
-
-  Args:
-      graph_id (str):
-      table_name (str): Table name
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]
-  """
-
-  kwargs = _get_kwargs(
-    graph_id=graph_id,
-    table_name=table_name,
-  )
-
-  response = client.get_httpx_client().request(
-    **kwargs,
-  )
-
-  return _build_response(client=client, response=response)
-
-
-def sync(
-  graph_id: str,
-  table_name: str,
-  *,
-  client: AuthenticatedClient,
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]:
-  """List Files in Staging Table
-
-   List all files uploaded to a staging table with comprehensive metadata.
-
-  Get a complete inventory of all files in a staging table, including upload status,
-  file sizes, row counts, and S3 locations. Essential for monitoring upload progress
-  and validating data before ingestion.
-
-  **Use Cases:**
-  - Monitor file upload progress
-  - Verify files are ready for ingestion
-  - Check file formats and sizes
-  - Track storage usage per table
-  - Identify failed or incomplete uploads
-  - Pre-ingestion validation
-
-  **Returned Metadata:**
-  - File ID, name, and format (parquet, csv, json)
-  - Size in bytes and row count (if available)
-  - Upload status and method
-  - Creation and upload timestamps
-  - S3 key for reference
-
-  **Upload Status Values:**
-  - `pending`: Upload URL generated, awaiting upload
-  - `uploaded`: Successfully uploaded, ready for ingestion
-  - `disabled`: Excluded from ingestion
-  - `archived`: Soft deleted
-  - `failed`: Upload failed
-
-  **Important Notes:**
-  - Only `uploaded` files are ingested
-  - Check `row_count` to estimate data volume
-  - Use `total_size_bytes` for storage monitoring
-  - Files with `failed` status should be deleted and re-uploaded
-  - File listing is included - no credit consumption
-
-  Args:
-      graph_id (str):
-      table_name (str): Table name
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]
-  """
-
-  return sync_detailed(
-    graph_id=graph_id,
-    table_name=table_name,
-    client=client,
-  ).parsed
-
-
-async def asyncio_detailed(
-  graph_id: str,
-  table_name: str,
-  *,
-  client: AuthenticatedClient,
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]:
-  """List Files in Staging Table
-
-   List all files uploaded to a staging table with comprehensive metadata.
-
-  Get a complete inventory of all files in a staging table, including upload status,
-  file sizes, row counts, and S3 locations. Essential for monitoring upload progress
-  and validating data before ingestion.
-
-  **Use Cases:**
-  - Monitor file upload progress
-  - Verify files are ready for ingestion
-  - Check file formats and sizes
-  - Track storage usage per table
-  - Identify failed or incomplete uploads
-  - Pre-ingestion validation
-
-  **Returned Metadata:**
-  - File ID, name, and format (parquet, csv, json)
-  - Size in bytes and row count (if available)
-  - Upload status and method
-  - Creation and upload timestamps
-  - S3 key for reference
-
-  **Upload Status Values:**
-  - `pending`: Upload URL generated, awaiting upload
-  - `uploaded`: Successfully uploaded, ready for ingestion
-  - `disabled`: Excluded from ingestion
-  - `archived`: Soft deleted
-  - `failed`: Upload failed
-
-  **Important Notes:**
-  - Only `uploaded` files are ingested
-  - Check `row_count` to estimate data volume
-  - Use `total_size_bytes` for storage monitoring
-  - Files with `failed` status should be deleted and re-uploaded
-  - File listing is included - no credit consumption
-
-  Args:
-      graph_id (str):
-      table_name (str): Table name
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]
-  """
-
-  kwargs = _get_kwargs(
-    graph_id=graph_id,
-    table_name=table_name,
-  )
-
-  response = await client.get_async_httpx_client().request(**kwargs)
-
-  return _build_response(client=client, response=response)
-
-
-async def asyncio(
-  graph_id: str,
-  table_name: str,
-  *,
-  client: AuthenticatedClient,
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]]:
-  """List Files in Staging Table
-
-   List all files uploaded to a staging table with comprehensive metadata.
-
-  Get a complete inventory of all files in a staging table, including upload status,
-  file sizes, row counts, and S3 locations. Essential for monitoring upload progress
-  and validating data before ingestion.
-
-  **Use Cases:**
-  - Monitor file upload progress
-  - Verify files are ready for ingestion
-  - Check file formats and sizes
-  - Track storage usage per table
-  - Identify failed or incomplete uploads
-  - Pre-ingestion validation
-
-  **Returned Metadata:**
-  - File ID, name, and format (parquet, csv, json)
-  - Size in bytes and row count (if available)
-  - Upload status and method
-  - Creation and upload timestamps
-  - S3 key for reference
-
-  **Upload Status Values:**
-  - `pending`: Upload URL generated, awaiting upload
-  - `uploaded`: Successfully uploaded, ready for ingestion
-  - `disabled`: Excluded from ingestion
-  - `archived`: Soft deleted
-  - `failed`: Upload failed
-
-  **Important Notes:**
-  - Only `uploaded` files are ingested
-  - Check `row_count` to estimate data volume
-  - Use `total_size_bytes` for storage monitoring
-  - Files with `failed` status should be deleted and re-uploaded
-  - File listing is included - no credit consumption
-
-  Args:
-      graph_id (str):
-      table_name (str): Table name
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, ListTableFilesResponse]
-  """
-
-  return (
-    await asyncio_detailed(
-      graph_id=graph_id,
-      table_name=table_name,
-      client=client,
-    )
-  ).parsed

robosystems_client/api/tables/update_file_status.py

@@ -1,395 +0,0 @@
-from http import HTTPStatus
-from typing import Any, Optional, Union, cast
-
-import httpx
-
-from ... import errors
-from ...client import AuthenticatedClient, Client
-from ...models.error_response import ErrorResponse
-from ...models.file_status_update import FileStatusUpdate
-from ...models.http_validation_error import HTTPValidationError
-from ...models.update_file_status_response_updatefilestatus import (
-  UpdateFileStatusResponseUpdatefilestatus,
-)
-from ...types import Response
-
-
-def _get_kwargs(
-  graph_id: str,
-  file_id: str,
-  *,
-  body: FileStatusUpdate,
-) -> dict[str, Any]:
-  headers: dict[str, Any] = {}
-
-  _kwargs: dict[str, Any] = {
-    "method": "patch",
-    "url": f"/v1/graphs/{graph_id}/tables/files/{file_id}",
-  }
-
-  _kwargs["json"] = body.to_dict()
-
-  headers["Content-Type"] = "application/json"
-
-  _kwargs["headers"] = headers
-  return _kwargs
-
-
-def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[
-  Union[
-    Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus
-  ]
-]:
-  if response.status_code == 200:
-    response_200 = UpdateFileStatusResponseUpdatefilestatus.from_dict(response.json())
-
-    return response_200
-
-  if response.status_code == 400:
-    response_400 = ErrorResponse.from_dict(response.json())
-
-    return response_400
-
-  if response.status_code == 401:
-    response_401 = cast(Any, None)
-    return response_401
-
-  if response.status_code == 403:
-    response_403 = ErrorResponse.from_dict(response.json())
-
-    return response_403
-
-  if response.status_code == 404:
-    response_404 = ErrorResponse.from_dict(response.json())
-
-    return response_404
-
-  if response.status_code == 413:
-    response_413 = ErrorResponse.from_dict(response.json())
-
-    return response_413
-
-  if response.status_code == 422:
-    response_422 = HTTPValidationError.from_dict(response.json())
-
-    return response_422
-
-  if response.status_code == 500:
-    response_500 = cast(Any, None)
-    return response_500
-
-  if client.raise_on_unexpected_status:
-    raise errors.UnexpectedStatus(response.status_code, response.content)
-  else:
-    return None
-
-
-def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[
-  Union[
-    Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus
-  ]
-]:
-  return Response(
-    status_code=HTTPStatus(response.status_code),
-    content=response.content,
-    headers=response.headers,
-    parsed=_parse_response(client=client, response=response),
-  )
-
-
-def sync_detailed(
-  graph_id: str,
-  file_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: FileStatusUpdate,
-) -> Response[
-  Union[
-    Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus
-  ]
-]:
-  """Update File Upload Status
-
-   Update file status after upload completes.
-
-  Marks files as uploaded after successful S3 upload. The backend validates
-  the file, calculates size and row count, enforces storage limits, and
-  registers the DuckDB table for queries.
-
-  **Status Values:**
-  - `uploaded`: File successfully uploaded to S3 (triggers validation)
-  - `disabled`: Exclude file from ingestion
-  - `archived`: Soft delete file
-
-  **What Happens on 'uploaded' Status:**
-  1. Verify file exists in S3
-  2. Calculate actual file size
-  3. Enforce tier storage limits
-  4. Calculate or estimate row count
-  5. Update table statistics
-  6. Register DuckDB external table
-  7. File ready for ingestion
-
-  **Row Count Calculation:**
-  - **Parquet**: Exact count from file metadata
-  - **CSV**: Count rows (minus header)
-  - **JSON**: Count array elements
-  - **Fallback**: Estimate from file size if reading fails
-
-  **Storage Limits:**
-  Enforced per subscription tier. Returns HTTP 413 if limit exceeded.
-  Check current usage before large uploads.
-
-  **Important Notes:**
-  - Always call this after S3 upload completes
-  - Check response for actual row count
-  - Storage limit errors (413) mean tier upgrade needed
-  - DuckDB registration failures are non-fatal (retried later)
-  - Status updates are included - no credit consumption
-
-  Args:
-      graph_id (str):
-      file_id (str): File identifier
-      body (FileStatusUpdate):
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]]
-  """
-
-  kwargs = _get_kwargs(
-    graph_id=graph_id,
-    file_id=file_id,
-    body=body,
-  )
-
-  response = client.get_httpx_client().request(
-    **kwargs,
-  )
-
-  return _build_response(client=client, response=response)
-
-
-def sync(
-  graph_id: str,
-  file_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: FileStatusUpdate,
-) -> Optional[
-  Union[
-    Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus
-  ]
-]:
-  """Update File Upload Status
-
-   Update file status after upload completes.
-
-  Marks files as uploaded after successful S3 upload. The backend validates
-  the file, calculates size and row count, enforces storage limits, and
-  registers the DuckDB table for queries.
-
-  **Status Values:**
-  - `uploaded`: File successfully uploaded to S3 (triggers validation)
-  - `disabled`: Exclude file from ingestion
-  - `archived`: Soft delete file
-
-  **What Happens on 'uploaded' Status:**
-  1. Verify file exists in S3
-  2. Calculate actual file size
-  3. Enforce tier storage limits
-  4. Calculate or estimate row count
-  5. Update table statistics
-  6. Register DuckDB external table
-  7. File ready for ingestion
-
-  **Row Count Calculation:**
-  - **Parquet**: Exact count from file metadata
-  - **CSV**: Count rows (minus header)
-  - **JSON**: Count array elements
-  - **Fallback**: Estimate from file size if reading fails
-
-  **Storage Limits:**
-  Enforced per subscription tier. Returns HTTP 413 if limit exceeded.
-  Check current usage before large uploads.
-
-  **Important Notes:**
-  - Always call this after S3 upload completes
-  - Check response for actual row count
-  - Storage limit errors (413) mean tier upgrade needed
-  - DuckDB registration failures are non-fatal (retried later)
-  - Status updates are included - no credit consumption
-
-  Args:
-      graph_id (str):
-      file_id (str): File identifier
-      body (FileStatusUpdate):
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]
-  """
-
-  return sync_detailed(
-    graph_id=graph_id,
-    file_id=file_id,
-    client=client,
-    body=body,
-  ).parsed
-
-
-async def asyncio_detailed(
-  graph_id: str,
-  file_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: FileStatusUpdate,
-) -> Response[
-  Union[
-    Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus
-  ]
-]:
-  """Update File Upload Status
-
-   Update file status after upload completes.
-
-  Marks files as uploaded after successful S3 upload. The backend validates
-  the file, calculates size and row count, enforces storage limits, and
-  registers the DuckDB table for queries.
-
-  **Status Values:**
-  - `uploaded`: File successfully uploaded to S3 (triggers validation)
-  - `disabled`: Exclude file from ingestion
-  - `archived`: Soft delete file
-
-  **What Happens on 'uploaded' Status:**
-  1. Verify file exists in S3
-  2. Calculate actual file size
-  3. Enforce tier storage limits
-  4. Calculate or estimate row count
-  5. Update table statistics
-  6. Register DuckDB external table
-  7. File ready for ingestion
-
-  **Row Count Calculation:**
-  - **Parquet**: Exact count from file metadata
-  - **CSV**: Count rows (minus header)
-  - **JSON**: Count array elements
-  - **Fallback**: Estimate from file size if reading fails
-
-  **Storage Limits:**
-  Enforced per subscription tier. Returns HTTP 413 if limit exceeded.
-  Check current usage before large uploads.
-
-  **Important Notes:**
-  - Always call this after S3 upload completes
-  - Check response for actual row count
-  - Storage limit errors (413) mean tier upgrade needed
-  - DuckDB registration failures are non-fatal (retried later)
-  - Status updates are included - no credit consumption
-
-  Args:
-      graph_id (str):
-      file_id (str): File identifier
-      body (FileStatusUpdate):
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]]
-  """
-
-  kwargs = _get_kwargs(
-    graph_id=graph_id,
-    file_id=file_id,
-    body=body,
-  )
-
-  response = await client.get_async_httpx_client().request(**kwargs)
-
-  return _build_response(client=client, response=response)
-
-
-async def asyncio(
-  graph_id: str,
-  file_id: str,
-  *,
-  client: AuthenticatedClient,
-  body: FileStatusUpdate,
-) -> Optional[
-  Union[
-    Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus
-  ]
-]:
-  """Update File Upload Status
-
-   Update file status after upload completes.
-
-  Marks files as uploaded after successful S3 upload. The backend validates
-  the file, calculates size and row count, enforces storage limits, and
-  registers the DuckDB table for queries.
-
-  **Status Values:**
-  - `uploaded`: File successfully uploaded to S3 (triggers validation)
-  - `disabled`: Exclude file from ingestion
-  - `archived`: Soft delete file
-
-  **What Happens on 'uploaded' Status:**
-  1. Verify file exists in S3
-  2. Calculate actual file size
-  3. Enforce tier storage limits
-  4. Calculate or estimate row count
-  5. Update table statistics
-  6. Register DuckDB external table
-  7. File ready for ingestion
-
-  **Row Count Calculation:**
-  - **Parquet**: Exact count from file metadata
-  - **CSV**: Count rows (minus header)
-  - **JSON**: Count array elements
-  - **Fallback**: Estimate from file size if reading fails
-
-  **Storage Limits:**
-  Enforced per subscription tier. Returns HTTP 413 if limit exceeded.
-  Check current usage before large uploads.
-
-  **Important Notes:**
-  - Always call this after S3 upload completes
-  - Check response for actual row count
-  - Storage limit errors (413) mean tier upgrade needed
-  - DuckDB registration failures are non-fatal (retried later)
-  - Status updates are included - no credit consumption
-
-  Args:
-      graph_id (str):
-      file_id (str): File identifier
-      body (FileStatusUpdate):
-
-  Raises:
-      errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
-      httpx.TimeoutException: If the request takes longer than Client.timeout.
-
-  Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, UpdateFileStatusResponseUpdatefilestatus]
-  """
-
-  return (
-    await asyncio_detailed(
-      graph_id=graph_id,
-      file_id=file_id,
-      client=client,
-      body=body,
-    )
-  ).parsed