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

robosystems_client/api/graphs/get_graphs.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union, cast
+from typing import Any, cast
 
 import httpx
 
@@ -19,8 +19,8 @@ def _get_kwargs() -> dict[str, Any]:
 
 
 def _parse_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, UserGraphsResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | UserGraphsResponse | None:
   if response.status_code == 200:
     response_200 = UserGraphsResponse.from_dict(response.json())
 
@@ -37,8 +37,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, UserGraphsResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | UserGraphsResponse]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -50,7 +50,7 @@ def _build_response(
 def sync_detailed(
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, UserGraphsResponse]]:
+) -> Response[Any | UserGraphsResponse]:
   r"""Get User Graphs and Repositories
 
    List all graph databases and shared repositories accessible to the current user.
@@ -100,7 +100,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, UserGraphsResponse]]
+      Response[Any | UserGraphsResponse]
   """
 
   kwargs = _get_kwargs()
@@ -115,7 +115,7 @@ def sync_detailed(
 def sync(
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, UserGraphsResponse]]:
+) -> Any | UserGraphsResponse | None:
   r"""Get User Graphs and Repositories
 
    List all graph databases and shared repositories accessible to the current user.
@@ -165,7 +165,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, UserGraphsResponse]
+      Any | UserGraphsResponse
   """
 
   return sync_detailed(
@@ -176,7 +176,7 @@ def sync(
 async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
-) -> Response[Union[Any, UserGraphsResponse]]:
+) -> Response[Any | UserGraphsResponse]:
   r"""Get User Graphs and Repositories
 
    List all graph databases and shared repositories accessible to the current user.
@@ -226,7 +226,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, UserGraphsResponse]]
+      Response[Any | UserGraphsResponse]
   """
 
   kwargs = _get_kwargs()
@@ -239,7 +239,7 @@ async def asyncio_detailed(
 async def asyncio(
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[Any, UserGraphsResponse]]:
+) -> Any | UserGraphsResponse | None:
   r"""Get User Graphs and Repositories
 
    List all graph databases and shared repositories accessible to the current user.
@@ -289,7 +289,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, UserGraphsResponse]
+      Any | UserGraphsResponse
   """
 
   return (

robosystems_client/api/graphs/select_graph.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, HTTPValidationError, SuccessResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> ErrorResponse | HTTPValidationError | SuccessResponse | None:
   if response.status_code == 200:
     response_200 = SuccessResponse.from_dict(response.json())
 
@@ -57,8 +57,8 @@ def _parse_response(
 
 
 def _build_response(
-  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]:
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[ErrorResponse | HTTPValidationError | SuccessResponse]:
   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, HTTPValidationError, SuccessResponse]]:
+) -> Response[ErrorResponse | HTTPValidationError | SuccessResponse]:
   """Select Graph
 
    Select a specific graph as the active workspace for the user.
@@ -112,7 +112,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]
+      Response[ErrorResponse | HTTPValidationError | SuccessResponse]
   """
 
   kwargs = _get_kwargs(
@@ -130,7 +130,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]:
+) -> ErrorResponse | HTTPValidationError | SuccessResponse | None:
   """Select Graph
 
    Select a specific graph as the active workspace for the user.
@@ -171,7 +171,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[ErrorResponse, HTTPValidationError, SuccessResponse]
+      ErrorResponse | HTTPValidationError | SuccessResponse
   """
 
   return sync_detailed(
@@ -184,7 +184,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]:
+) -> Response[ErrorResponse | HTTPValidationError | SuccessResponse]:
   """Select Graph
 
    Select a specific graph as the active workspace for the user.
@@ -225,7 +225,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]
+      Response[ErrorResponse | HTTPValidationError | SuccessResponse]
   """
 
   kwargs = _get_kwargs(
@@ -241,7 +241,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
-) -> Optional[Union[ErrorResponse, HTTPValidationError, SuccessResponse]]:
+) -> ErrorResponse | HTTPValidationError | SuccessResponse | None:
   """Select Graph
 
    Select a specific graph as the active workspace for the user.
@@ -282,7 +282,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[ErrorResponse, HTTPValidationError, SuccessResponse]
+      ErrorResponse | HTTPValidationError | SuccessResponse
   """
 
   return (

robosystems_client/api/materialization/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""

robosystems_client/api/materialization/get_materialization_status.py

@@ -0,0 +1,272 @@
+from http import HTTPStatus
+from typing import Any, 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.materialize_status_response import MaterializeStatusResponse
+from ...types import Response
+
+
+def _get_kwargs(
+  graph_id: str,
+) -> dict[str, Any]:
+  _kwargs: dict[str, Any] = {
+    "method": "get",
+    "url": f"/v1/graphs/{graph_id}/materialize/status",
+  }
+
+  return _kwargs
+
+
+def _parse_response(
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse | None:
+  if response.status_code == 200:
+    response_200 = MaterializeStatusResponse.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 client.raise_on_unexpected_status:
+    raise errors.UnexpectedStatus(response.status_code, response.content)
+  else:
+    return None
+
+
+def _build_response(
+  *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse]:
+  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,
+  *,
+  client: AuthenticatedClient,
+) -> Response[Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse]:
+  """Get Materialization Status
+
+   Get current materialization status for the graph.
+
+  Shows whether the graph is stale (DuckDB has changes not yet in graph database),
+  when it was last materialized, and how long since last materialization.
+
+  **Status Information:**
+  - Whether graph is currently stale
+  - Reason for staleness if applicable
+  - When graph became stale
+  - When graph was last materialized
+  - Total materialization count
+  - Hours since last materialization
+
+  **Use Cases:**
+  - Decide if materialization is needed
+  - Monitor graph freshness
+  - Track materialization history
+  - Understand data pipeline state
+
+  **Important Notes:**
+  - Stale graph means DuckDB has changes not in graph
+  - Graph becomes stale after file deletions
+  - Materialization clears staleness
+  - Status retrieval is included - no credit consumption
+
+  Args:
+      graph_id (str):
+
+  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[Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse]
+  """
+
+  kwargs = _get_kwargs(
+    graph_id=graph_id,
+  )
+
+  response = client.get_httpx_client().request(
+    **kwargs,
+  )
+
+  return _build_response(client=client, response=response)
+
+
+def sync(
+  graph_id: str,
+  *,
+  client: AuthenticatedClient,
+) -> Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse | None:
+  """Get Materialization Status
+
+   Get current materialization status for the graph.
+
+  Shows whether the graph is stale (DuckDB has changes not yet in graph database),
+  when it was last materialized, and how long since last materialization.
+
+  **Status Information:**
+  - Whether graph is currently stale
+  - Reason for staleness if applicable
+  - When graph became stale
+  - When graph was last materialized
+  - Total materialization count
+  - Hours since last materialization
+
+  **Use Cases:**
+  - Decide if materialization is needed
+  - Monitor graph freshness
+  - Track materialization history
+  - Understand data pipeline state
+
+  **Important Notes:**
+  - Stale graph means DuckDB has changes not in graph
+  - Graph becomes stale after file deletions
+  - Materialization clears staleness
+  - Status retrieval is included - no credit consumption
+
+  Args:
+      graph_id (str):
+
+  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:
+      Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse
+  """
+
+  return sync_detailed(
+    graph_id=graph_id,
+    client=client,
+  ).parsed
+
+
+async def asyncio_detailed(
+  graph_id: str,
+  *,
+  client: AuthenticatedClient,
+) -> Response[Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse]:
+  """Get Materialization Status
+
+   Get current materialization status for the graph.
+
+  Shows whether the graph is stale (DuckDB has changes not yet in graph database),
+  when it was last materialized, and how long since last materialization.
+
+  **Status Information:**
+  - Whether graph is currently stale
+  - Reason for staleness if applicable
+  - When graph became stale
+  - When graph was last materialized
+  - Total materialization count
+  - Hours since last materialization
+
+  **Use Cases:**
+  - Decide if materialization is needed
+  - Monitor graph freshness
+  - Track materialization history
+  - Understand data pipeline state
+
+  **Important Notes:**
+  - Stale graph means DuckDB has changes not in graph
+  - Graph becomes stale after file deletions
+  - Materialization clears staleness
+  - Status retrieval is included - no credit consumption
+
+  Args:
+      graph_id (str):
+
+  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[Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse]
+  """
+
+  kwargs = _get_kwargs(
+    graph_id=graph_id,
+  )
+
+  response = await client.get_async_httpx_client().request(**kwargs)
+
+  return _build_response(client=client, response=response)
+
+
+async def asyncio(
+  graph_id: str,
+  *,
+  client: AuthenticatedClient,
+) -> Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse | None:
+  """Get Materialization Status
+
+   Get current materialization status for the graph.
+
+  Shows whether the graph is stale (DuckDB has changes not yet in graph database),
+  when it was last materialized, and how long since last materialization.
+
+  **Status Information:**
+  - Whether graph is currently stale
+  - Reason for staleness if applicable
+  - When graph became stale
+  - When graph was last materialized
+  - Total materialization count
+  - Hours since last materialization
+
+  **Use Cases:**
+  - Decide if materialization is needed
+  - Monitor graph freshness
+  - Track materialization history
+  - Understand data pipeline state
+
+  **Important Notes:**
+  - Stale graph means DuckDB has changes not in graph
+  - Graph becomes stale after file deletions
+  - Materialization clears staleness
+  - Status retrieval is included - no credit consumption
+
+  Args:
+      graph_id (str):
+
+  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:
+      Any | ErrorResponse | HTTPValidationError | MaterializeStatusResponse
+  """
+
+  return (
+    await asyncio_detailed(
+      graph_id=graph_id,
+      client=client,
+    )
+  ).parsed