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

robosystems_client/api/subgraphs/create_subgraph.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union
+from typing import Any
 
 import httpx
 
@@ -32,12 +32,12 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[HTTPValidationError, SubgraphResponse]]:
-  if response.status_code == 201:
-    response_201 = SubgraphResponse.from_dict(response.json())
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> HTTPValidationError | SubgraphResponse | None:
+  if response.status_code == 200:
+    response_200 = SubgraphResponse.from_dict(response.json())
 
-    return response_201
+    return response_200
 
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
@@ -51,8 +51,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[HTTPValidationError, SubgraphResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[HTTPValidationError | SubgraphResponse]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -66,10 +66,10 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: CreateSubgraphRequest,
-) -> Response[Union[HTTPValidationError, SubgraphResponse]]:
+) -> Response[HTTPValidationError | SubgraphResponse]:
   """Create Subgraph
 
-   Create a new subgraph within a parent graph.
+   Create a new subgraph within a parent graph, with optional data forking.
 
   **Requirements:**
   - Valid authentication
@@ -79,9 +79,18 @@ def sync_detailed(
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
+  **Fork Mode:**
+  When `fork_parent=true`, the operation:
+  - Returns immediately with an operation_id for SSE monitoring
+  - Copies data from parent graph to the new subgraph
+  - Supports selective forking via metadata.fork_options
+  - Tracks progress in real-time via SSE
+
   **Returns:**
-  - Created subgraph details including its unique ID
-  - Subgraph ID format: `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
+  - Without fork: Immediate SubgraphResponse with created subgraph details
+  - With fork: Operation response with SSE monitoring endpoint
+
+  **Subgraph ID format:** `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
 
   **Usage:**
   - Subgraphs share parent's credit pool
@@ -97,7 +106,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[HTTPValidationError, SubgraphResponse]]
+      Response[HTTPValidationError | SubgraphResponse]
   """
 
   kwargs = _get_kwargs(
@@ -117,10 +126,10 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: CreateSubgraphRequest,
-) -> Optional[Union[HTTPValidationError, SubgraphResponse]]:
+) -> HTTPValidationError | SubgraphResponse | None:
   """Create Subgraph
 
-   Create a new subgraph within a parent graph.
+   Create a new subgraph within a parent graph, with optional data forking.
 
   **Requirements:**
   - Valid authentication
@@ -130,9 +139,18 @@ def sync(
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
+  **Fork Mode:**
+  When `fork_parent=true`, the operation:
+  - Returns immediately with an operation_id for SSE monitoring
+  - Copies data from parent graph to the new subgraph
+  - Supports selective forking via metadata.fork_options
+  - Tracks progress in real-time via SSE
+
   **Returns:**
-  - Created subgraph details including its unique ID
-  - Subgraph ID format: `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
+  - Without fork: Immediate SubgraphResponse with created subgraph details
+  - With fork: Operation response with SSE monitoring endpoint
+
+  **Subgraph ID format:** `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
 
   **Usage:**
   - Subgraphs share parent's credit pool
@@ -148,7 +166,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[HTTPValidationError, SubgraphResponse]
+      HTTPValidationError | SubgraphResponse
   """
 
   return sync_detailed(
@@ -163,10 +181,10 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: CreateSubgraphRequest,
-) -> Response[Union[HTTPValidationError, SubgraphResponse]]:
+) -> Response[HTTPValidationError | SubgraphResponse]:
   """Create Subgraph
 
-   Create a new subgraph within a parent graph.
+   Create a new subgraph within a parent graph, with optional data forking.
 
   **Requirements:**
   - Valid authentication
@@ -176,9 +194,18 @@ async def asyncio_detailed(
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
+  **Fork Mode:**
+  When `fork_parent=true`, the operation:
+  - Returns immediately with an operation_id for SSE monitoring
+  - Copies data from parent graph to the new subgraph
+  - Supports selective forking via metadata.fork_options
+  - Tracks progress in real-time via SSE
+
   **Returns:**
-  - Created subgraph details including its unique ID
-  - Subgraph ID format: `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
+  - Without fork: Immediate SubgraphResponse with created subgraph details
+  - With fork: Operation response with SSE monitoring endpoint
+
+  **Subgraph ID format:** `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
 
   **Usage:**
   - Subgraphs share parent's credit pool
@@ -194,7 +221,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[HTTPValidationError, SubgraphResponse]]
+      Response[HTTPValidationError | SubgraphResponse]
   """
 
   kwargs = _get_kwargs(
@@ -212,10 +239,10 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: CreateSubgraphRequest,
-) -> Optional[Union[HTTPValidationError, SubgraphResponse]]:
+) -> HTTPValidationError | SubgraphResponse | None:
   """Create Subgraph
 
-   Create a new subgraph within a parent graph.
+   Create a new subgraph within a parent graph, with optional data forking.
 
   **Requirements:**
   - Valid authentication
@@ -225,9 +252,18 @@ async def asyncio(
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
+  **Fork Mode:**
+  When `fork_parent=true`, the operation:
+  - Returns immediately with an operation_id for SSE monitoring
+  - Copies data from parent graph to the new subgraph
+  - Supports selective forking via metadata.fork_options
+  - Tracks progress in real-time via SSE
+
   **Returns:**
-  - Created subgraph details including its unique ID
-  - Subgraph ID format: `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
+  - Without fork: Immediate SubgraphResponse with created subgraph details
+  - With fork: Operation response with SSE monitoring endpoint
+
+  **Subgraph ID format:** `{parent_id}_{subgraph_name}` (e.g., kg1234567890abcdef_dev)
 
   **Usage:**
   - Subgraphs share parent's credit pool
@@ -243,7 +279,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[HTTPValidationError, SubgraphResponse]
+      HTTPValidationError | SubgraphResponse
   """
 
   return (

robosystems_client/api/subgraphs/delete_subgraph.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, DeleteSubgraphResponse, HTTPValidationError]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | DeleteSubgraphResponse | HTTPValidationError | None:
   if response.status_code == 200:
     response_200 = DeleteSubgraphResponse.from_dict(response.json())
 
@@ -76,8 +76,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | DeleteSubgraphResponse | HTTPValidationError]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -92,7 +92,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: DeleteSubgraphRequest,
-) -> Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]:
+) -> Response[Any | DeleteSubgraphResponse | HTTPValidationError]:
   """Delete Subgraph
 
    Delete a subgraph database.
@@ -130,7 +130,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]
+      Response[Any | DeleteSubgraphResponse | HTTPValidationError]
   """
 
   kwargs = _get_kwargs(
@@ -152,7 +152,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: DeleteSubgraphRequest,
-) -> Optional[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]:
+) -> Any | DeleteSubgraphResponse | HTTPValidationError | None:
   """Delete Subgraph
 
    Delete a subgraph database.
@@ -190,7 +190,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, DeleteSubgraphResponse, HTTPValidationError]
+      Any | DeleteSubgraphResponse | HTTPValidationError
   """
 
   return sync_detailed(
@@ -207,7 +207,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: DeleteSubgraphRequest,
-) -> Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]:
+) -> Response[Any | DeleteSubgraphResponse | HTTPValidationError]:
   """Delete Subgraph
 
    Delete a subgraph database.
@@ -245,7 +245,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]
+      Response[Any | DeleteSubgraphResponse | HTTPValidationError]
   """
 
   kwargs = _get_kwargs(
@@ -265,7 +265,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: DeleteSubgraphRequest,
-) -> Optional[Union[Any, DeleteSubgraphResponse, HTTPValidationError]]:
+) -> Any | DeleteSubgraphResponse | HTTPValidationError | None:
   """Delete Subgraph
 
    Delete a subgraph database.
@@ -303,7 +303,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, DeleteSubgraphResponse, HTTPValidationError]
+      Any | DeleteSubgraphResponse | HTTPValidationError
   """
 
   return (

robosystems_client/api/subgraphs/get_subgraph_info.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, HTTPValidationError, SubgraphResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | HTTPValidationError | SubgraphResponse | None:
   if response.status_code == 200:
     response_200 = SubgraphResponse.from_dict(response.json())
 
@@ -62,8 +62,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, HTTPValidationError, SubgraphResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | HTTPValidationError | SubgraphResponse]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -77,7 +77,7 @@ def sync_detailed(
   subgraph_name: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, HTTPValidationError, SubgraphResponse]]:
+) -> Response[Any | HTTPValidationError | SubgraphResponse]:
   """Get Subgraph Details
 
    Get detailed information about a specific subgraph.
@@ -114,7 +114,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError, SubgraphResponse]]
+      Response[Any | HTTPValidationError | SubgraphResponse]
   """
 
   kwargs = _get_kwargs(
@@ -134,7 +134,7 @@ def sync(
   subgraph_name: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, HTTPValidationError, SubgraphResponse]]:
+) -> Any | HTTPValidationError | SubgraphResponse | None:
   """Get Subgraph Details
 
    Get detailed information about a specific subgraph.
@@ -171,7 +171,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError, SubgraphResponse]
+      Any | HTTPValidationError | SubgraphResponse
   """
 
   return sync_detailed(
@@ -186,7 +186,7 @@ async def asyncio_detailed(
   subgraph_name: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, HTTPValidationError, SubgraphResponse]]:
+) -> Response[Any | HTTPValidationError | SubgraphResponse]:
   """Get Subgraph Details
 
    Get detailed information about a specific subgraph.
@@ -223,7 +223,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError, SubgraphResponse]]
+      Response[Any | HTTPValidationError | SubgraphResponse]
   """
 
   kwargs = _get_kwargs(
@@ -241,7 +241,7 @@ async def asyncio(
   subgraph_name: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, HTTPValidationError, SubgraphResponse]]:
+) -> Any | HTTPValidationError | SubgraphResponse | None:
   """Get Subgraph Details
 
    Get detailed information about a specific subgraph.
@@ -278,7 +278,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError, SubgraphResponse]
+      Any | HTTPValidationError | SubgraphResponse
   """
 
   return (

robosystems_client/api/subgraphs/get_subgraph_quota.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union, cast
+from typing import Any, cast
 
 import httpx
 
@@ -22,8 +22,8 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | HTTPValidationError | SubgraphQuotaResponse | None:
   if response.status_code == 200:
     response_200 = SubgraphQuotaResponse.from_dict(response.json())
 
@@ -57,8 +57,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | HTTPValidationError | SubgraphQuotaResponse]:
   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[Any, HTTPValidationError, SubgraphQuotaResponse]]:
+) -> Response[Any | HTTPValidationError | SubgraphQuotaResponse]:
   """Get Subgraph Quota
 
    Get subgraph quota and usage information for a parent graph.
@@ -100,7 +100,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]
+      Response[Any | HTTPValidationError | SubgraphQuotaResponse]
   """
 
   kwargs = _get_kwargs(
@@ -118,7 +118,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]:
+) -> Any | HTTPValidationError | SubgraphQuotaResponse | None:
   """Get Subgraph Quota
 
    Get subgraph quota and usage information for a parent graph.
@@ -147,7 +147,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError, SubgraphQuotaResponse]
+      Any | HTTPValidationError | SubgraphQuotaResponse
   """
 
   return sync_detailed(
@@ -160,7 +160,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]:
+) -> Response[Any | HTTPValidationError | SubgraphQuotaResponse]:
   """Get Subgraph Quota
 
    Get subgraph quota and usage information for a parent graph.
@@ -189,7 +189,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]
+      Response[Any | HTTPValidationError | SubgraphQuotaResponse]
   """
 
   kwargs = _get_kwargs(
@@ -205,7 +205,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, HTTPValidationError, SubgraphQuotaResponse]]:
+) -> Any | HTTPValidationError | SubgraphQuotaResponse | None:
   """Get Subgraph Quota
 
    Get subgraph quota and usage information for a parent graph.
@@ -234,7 +234,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError, SubgraphQuotaResponse]
+      Any | HTTPValidationError | SubgraphQuotaResponse
   """
 
   return (

robosystems_client/api/subgraphs/list_subgraphs.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union
+from typing import Any
 
 import httpx
 
@@ -22,8 +22,8 @@ def _get_kwargs(
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[HTTPValidationError, ListSubgraphsResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> HTTPValidationError | ListSubgraphsResponse | None:
   if response.status_code == 200:
     response_200 = ListSubgraphsResponse.from_dict(response.json())
 
@@ -41,8 +41,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[HTTPValidationError, ListSubgraphsResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[HTTPValidationError | ListSubgraphsResponse]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -55,7 +55,7 @@ def sync_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[HTTPValidationError, ListSubgraphsResponse]]:
+) -> Response[HTTPValidationError | ListSubgraphsResponse]:
   """List Subgraphs
 
    List all subgraphs for a parent graph.
@@ -77,7 +77,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[HTTPValidationError, ListSubgraphsResponse]]
+      Response[HTTPValidationError | ListSubgraphsResponse]
   """
 
   kwargs = _get_kwargs(
@@ -95,7 +95,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[HTTPValidationError, ListSubgraphsResponse]]:
+) -> HTTPValidationError | ListSubgraphsResponse | None:
   """List Subgraphs
 
    List all subgraphs for a parent graph.
@@ -117,7 +117,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[HTTPValidationError, ListSubgraphsResponse]
+      HTTPValidationError | ListSubgraphsResponse
   """
 
   return sync_detailed(
@@ -130,7 +130,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[HTTPValidationError, ListSubgraphsResponse]]:
+) -> Response[HTTPValidationError | ListSubgraphsResponse]:
   """List Subgraphs
 
    List all subgraphs for a parent graph.
@@ -152,7 +152,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[HTTPValidationError, ListSubgraphsResponse]]
+      Response[HTTPValidationError | ListSubgraphsResponse]
   """
 
   kwargs = _get_kwargs(
@@ -168,7 +168,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[HTTPValidationError, ListSubgraphsResponse]]:
+) -> HTTPValidationError | ListSubgraphsResponse | None:
   """List Subgraphs
 
    List all subgraphs for a parent graph.
@@ -190,7 +190,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[HTTPValidationError, ListSubgraphsResponse]
+      HTTPValidationError | ListSubgraphsResponse
   """
 
   return (