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

robosystems_client/api/subscriptions/upgrade_subscription.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union, cast
+from typing import Any, cast
 
 import httpx
 
@@ -32,8 +32,8 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | GraphSubscriptionResponse | HTTPValidationError | None:
   if response.status_code == 200:
     response_200 = GraphSubscriptionResponse.from_dict(response.json())
 
@@ -55,8 +55,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | GraphSubscriptionResponse | HTTPValidationError]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -70,7 +70,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: UpgradeSubscriptionRequest,
-) -> Response[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]:
+) -> Response[Any | GraphSubscriptionResponse | HTTPValidationError]:
   """Upgrade Subscription
 
    Upgrade a subscription to a different plan.
@@ -87,7 +87,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]
+      Response[Any | GraphSubscriptionResponse | HTTPValidationError]
   """
 
   kwargs = _get_kwargs(
@@ -107,7 +107,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: UpgradeSubscriptionRequest,
-) -> Optional[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]:
+) -> Any | GraphSubscriptionResponse | HTTPValidationError | None:
   """Upgrade Subscription
 
    Upgrade a subscription to a different plan.
@@ -124,7 +124,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, GraphSubscriptionResponse, HTTPValidationError]
+      Any | GraphSubscriptionResponse | HTTPValidationError
   """
 
   return sync_detailed(
@@ -139,7 +139,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: UpgradeSubscriptionRequest,
-) -> Response[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]:
+) -> Response[Any | GraphSubscriptionResponse | HTTPValidationError]:
   """Upgrade Subscription
 
    Upgrade a subscription to a different plan.
@@ -156,7 +156,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]
+      Response[Any | GraphSubscriptionResponse | HTTPValidationError]
   """
 
   kwargs = _get_kwargs(
@@ -174,7 +174,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: UpgradeSubscriptionRequest,
-) -> Optional[Union[Any, GraphSubscriptionResponse, HTTPValidationError]]:
+) -> Any | GraphSubscriptionResponse | HTTPValidationError | None:
   """Upgrade Subscription
 
    Upgrade a subscription to a different plan.
@@ -191,7 +191,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, GraphSubscriptionResponse, HTTPValidationError]
+      Any | GraphSubscriptionResponse | HTTPValidationError
   """
 
   return (

robosystems_client/api/tables/list_tables.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union, cast
+from typing import Any, cast
 
 import httpx
 
@@ -23,8 +23,8 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | ErrorResponse | HTTPValidationError | TableListResponse | None:
   if response.status_code == 200:
     response_200 = TableListResponse.from_dict(response.json())
 
@@ -60,8 +60,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | ErrorResponse | HTTPValidationError | TableListResponse]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -74,7 +74,7 @@ def sync_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
+) -> Response[Any | ErrorResponse | HTTPValidationError | TableListResponse]:
   """List Staging Tables
 
    List all DuckDB staging tables with comprehensive metrics and status.
@@ -120,7 +120,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]
+      Response[Any | ErrorResponse | HTTPValidationError | TableListResponse]
   """
 
   kwargs = _get_kwargs(
@@ -138,7 +138,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
+) -> Any | ErrorResponse | HTTPValidationError | TableListResponse | None:
   """List Staging Tables
 
    List all DuckDB staging tables with comprehensive metrics and status.
@@ -184,7 +184,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]
+      Any | ErrorResponse | HTTPValidationError | TableListResponse
   """
 
   return sync_detailed(
@@ -197,7 +197,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
+) -> Response[Any | ErrorResponse | HTTPValidationError | TableListResponse]:
   """List Staging Tables
 
    List all DuckDB staging tables with comprehensive metrics and status.
@@ -243,7 +243,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]
+      Response[Any | ErrorResponse | HTTPValidationError | TableListResponse]
   """
 
   kwargs = _get_kwargs(
@@ -259,7 +259,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]]:
+) -> Any | ErrorResponse | HTTPValidationError | TableListResponse | None:
   """List Staging Tables
 
    List all DuckDB staging tables with comprehensive metrics and status.
@@ -305,7 +305,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, TableListResponse]
+      Any | ErrorResponse | HTTPValidationError | TableListResponse
   """
 
   return (

robosystems_client/api/tables/query_tables.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union, cast
+from typing import Any, cast
 
 import httpx
 
@@ -33,8 +33,8 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | ErrorResponse | HTTPValidationError | TableQueryResponse | None:
   if response.status_code == 200:
     response_200 = TableQueryResponse.from_dict(response.json())
 
@@ -79,8 +79,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | ErrorResponse | HTTPValidationError | TableQueryResponse]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -94,7 +94,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: TableQueryRequest,
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]:
+) -> Response[Any | ErrorResponse | HTTPValidationError | TableQueryResponse]:
   r"""Query Staging Tables with SQL
 
    Execute SQL queries on DuckDB staging tables for data inspection and validation.
@@ -146,6 +146,12 @@ def sync_detailed(
   - Read-only: No INSERT, UPDATE, DELETE
   - User's tables only: Cannot query other users' data
 
+  **Subgraph Support:**
+  This endpoint accepts both parent graph IDs and subgraph IDs.
+  - Parent graph: Use `graph_id` like `kg0123456789abcdef`
+  - Subgraph: Use full subgraph ID like `kg0123456789abcdef_dev`
+  Each subgraph has its own independent staging tables.
+
   **Shared Repositories:**
   Shared repositories (SEC, etc.) do not allow direct SQL queries.
   Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query`
@@ -162,7 +168,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]
+      Response[Any | ErrorResponse | HTTPValidationError | TableQueryResponse]
   """
 
   kwargs = _get_kwargs(
@@ -182,7 +188,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: TableQueryRequest,
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]:
+) -> Any | ErrorResponse | HTTPValidationError | TableQueryResponse | None:
   r"""Query Staging Tables with SQL
 
    Execute SQL queries on DuckDB staging tables for data inspection and validation.
@@ -234,6 +240,12 @@ def sync(
   - Read-only: No INSERT, UPDATE, DELETE
   - User's tables only: Cannot query other users' data
 
+  **Subgraph Support:**
+  This endpoint accepts both parent graph IDs and subgraph IDs.
+  - Parent graph: Use `graph_id` like `kg0123456789abcdef`
+  - Subgraph: Use full subgraph ID like `kg0123456789abcdef_dev`
+  Each subgraph has its own independent staging tables.
+
   **Shared Repositories:**
   Shared repositories (SEC, etc.) do not allow direct SQL queries.
   Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query`
@@ -250,7 +262,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]
+      Any | ErrorResponse | HTTPValidationError | TableQueryResponse
   """
 
   return sync_detailed(
@@ -265,7 +277,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: TableQueryRequest,
-) -> Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]:
+) -> Response[Any | ErrorResponse | HTTPValidationError | TableQueryResponse]:
   r"""Query Staging Tables with SQL
 
    Execute SQL queries on DuckDB staging tables for data inspection and validation.
@@ -317,6 +329,12 @@ async def asyncio_detailed(
   - Read-only: No INSERT, UPDATE, DELETE
   - User's tables only: Cannot query other users' data
 
+  **Subgraph Support:**
+  This endpoint accepts both parent graph IDs and subgraph IDs.
+  - Parent graph: Use `graph_id` like `kg0123456789abcdef`
+  - Subgraph: Use full subgraph ID like `kg0123456789abcdef_dev`
+  Each subgraph has its own independent staging tables.
+
   **Shared Repositories:**
   Shared repositories (SEC, etc.) do not allow direct SQL queries.
   Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query`
@@ -333,7 +351,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]
+      Response[Any | ErrorResponse | HTTPValidationError | TableQueryResponse]
   """
 
   kwargs = _get_kwargs(
@@ -351,7 +369,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: TableQueryRequest,
-) -> Optional[Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]]:
+) -> Any | ErrorResponse | HTTPValidationError | TableQueryResponse | None:
   r"""Query Staging Tables with SQL
 
    Execute SQL queries on DuckDB staging tables for data inspection and validation.
@@ -403,6 +421,12 @@ async def asyncio(
   - Read-only: No INSERT, UPDATE, DELETE
   - User's tables only: Cannot query other users' data
 
+  **Subgraph Support:**
+  This endpoint accepts both parent graph IDs and subgraph IDs.
+  - Parent graph: Use `graph_id` like `kg0123456789abcdef`
+  - Subgraph: Use full subgraph ID like `kg0123456789abcdef_dev`
+  Each subgraph has its own independent staging tables.
+
   **Shared Repositories:**
   Shared repositories (SEC, etc.) do not allow direct SQL queries.
   Use the graph query endpoint instead: `POST /v1/graphs/{graph_id}/query`
@@ -419,7 +443,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, ErrorResponse, HTTPValidationError, TableQueryResponse]
+      Any | ErrorResponse | HTTPValidationError | TableQueryResponse
   """
 
   return (

robosystems_client/api/usage/get_graph_metrics.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union
+from typing import Any
 
 import httpx
 
@@ -23,8 +23,8 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> ErrorResponse | GraphMetricsResponse | HTTPValidationError | None:
   if response.status_code == 200:
     response_200 = GraphMetricsResponse.from_dict(response.json())
 
@@ -57,8 +57,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[ErrorResponse | GraphMetricsResponse | HTTPValidationError]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -71,7 +71,7 @@ def sync_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
+) -> Response[ErrorResponse | GraphMetricsResponse | HTTPValidationError]:
   """Get Graph Metrics
 
    Get comprehensive metrics for the graph database.
@@ -100,7 +100,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]
+      Response[ErrorResponse | GraphMetricsResponse | HTTPValidationError]
   """
 
   kwargs = _get_kwargs(
@@ -118,7 +118,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
+) -> ErrorResponse | GraphMetricsResponse | HTTPValidationError | None:
   """Get Graph Metrics
 
    Get comprehensive metrics for the graph database.
@@ -147,7 +147,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]
+      ErrorResponse | GraphMetricsResponse | HTTPValidationError
   """
 
   return sync_detailed(
@@ -160,7 +160,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
+) -> Response[ErrorResponse | GraphMetricsResponse | HTTPValidationError]:
   """Get Graph Metrics
 
    Get comprehensive metrics for the graph database.
@@ -189,7 +189,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]
+      Response[ErrorResponse | GraphMetricsResponse | HTTPValidationError]
   """
 
   kwargs = _get_kwargs(
@@ -205,7 +205,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
+) -> ErrorResponse | GraphMetricsResponse | HTTPValidationError | None:
   """Get Graph Metrics
 
    Get comprehensive metrics for the graph database.
@@ -234,7 +234,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]
+      ErrorResponse | GraphMetricsResponse | HTTPValidationError
   """
 
   return (