robosystems-client 0.1.17__py3-none-any.whl → 0.1.18__py3-none-any.whl

robosystems_client/api/connections/sync_connection.py

@@ -19,15 +19,28 @@ def _get_kwargs(
   connection_id: str,
   *,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
+  params: dict[str, Any] = {}
+
+  json_token: Union[None, Unset, str]
+  if isinstance(token, Unset):
+    json_token = UNSET
+  else:
+    json_token = token
+  params["token"] = json_token
+
+  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
   _kwargs: dict[str, Any] = {
     "method": "post",
     "url": f"/v1/{graph_id}/connections/{connection_id}/sync",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -88,6 +101,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
@@ -122,6 +136,7 @@ def sync_detailed(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
@@ -137,6 +152,7 @@ def sync_detailed(
     graph_id=graph_id,
     connection_id=connection_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -153,6 +169,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
@@ -187,6 +204,7 @@ def sync(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
@@ -203,6 +221,7 @@ def sync(
     connection_id=connection_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -213,6 +232,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
@@ -247,6 +267,7 @@ async def asyncio_detailed(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
@@ -262,6 +283,7 @@ async def asyncio_detailed(
     graph_id=graph_id,
     connection_id=connection_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -276,6 +298,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
@@ -310,6 +333,7 @@ async def asyncio(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
@@ -327,6 +351,7 @@ async def asyncio(
       connection_id=connection_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/copy/copy_data_to_graph.py

@@ -17,15 +17,28 @@ def _get_kwargs(
   graph_id: str,
   *,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
+  params: dict[str, Any] = {}
+
+  json_token: Union[None, Unset, str]
+  if isinstance(token, Unset):
+    json_token = UNSET
+  else:
+    json_token = token
+  params["token"] = json_token
+
+  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
   _kwargs: dict[str, Any] = {
     "method": "post",
     "url": f"/v1/{graph_id}/copy",
+    "params": params,
   }
 
   _kwargs["json"]: dict[str, Any]
@@ -96,6 +109,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
@@ -162,6 +176,7 @@ def sync_detailed(
   Args:
       graph_id (str): Target graph identifier (user graphs only - shared repositories not
           allowed)
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
@@ -176,6 +191,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -191,6 +207,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
@@ -257,6 +274,7 @@ def sync(
   Args:
       graph_id (str): Target graph identifier (user graphs only - shared repositories not
           allowed)
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
@@ -272,6 +290,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -281,6 +300,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
@@ -347,6 +367,7 @@ async def asyncio_detailed(
   Args:
       graph_id (str): Target graph identifier (user graphs only - shared repositories not
           allowed)
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
@@ -361,6 +382,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -374,6 +396,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
@@ -440,6 +463,7 @@ async def asyncio(
   Args:
       graph_id (str): Target graph identifier (user graphs only - shared repositories not
           allowed)
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
@@ -456,6 +480,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/create/create_graph.py

@@ -13,15 +13,28 @@ from ...types import UNSET, Response, Unset
 def _get_kwargs(
   *,
   body: CreateGraphRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
+  params: dict[str, Any] = {}
+
+  json_token: Union[None, Unset, str]
+  if isinstance(token, Unset):
+    json_token = UNSET
+  else:
+    json_token = token
+  params["token"] = json_token
+
+  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
   _kwargs: dict[str, Any] = {
     "method": "post",
     "url": "/v1/create/graph",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -63,6 +76,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
@@ -109,6 +123,7 @@ def sync_detailed(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
           {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
@@ -126,6 +141,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -140,6 +156,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
@@ -186,6 +203,7 @@ def sync(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
           {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
@@ -204,6 +222,7 @@ def sync(
   return sync_detailed(
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -212,6 +231,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
@@ -258,6 +278,7 @@ async def asyncio_detailed(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
           {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
@@ -275,6 +296,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -287,6 +309,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
@@ -333,6 +356,7 @@ async def asyncio(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
           {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
@@ -352,6 +376,7 @@ async def asyncio(
     await asyncio_detailed(
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/graph_analytics/get_graph_metrics.py

@@ -14,15 +14,28 @@ from ...types import UNSET, Response, Unset
 def _get_kwargs(
   graph_id: str,
   *,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
+  params: dict[str, Any] = {}
+
+  json_token: Union[None, Unset, str]
+  if isinstance(token, Unset):
+    json_token = UNSET
+  else:
+    json_token = token
+  params["token"] = json_token
+
+  params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
   _kwargs: dict[str, Any] = {
     "method": "get",
     "url": f"/v1/{graph_id}/analytics",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -73,6 +86,7 @@ def sync_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
   """Get Graph Metrics
@@ -97,6 +111,7 @@ def sync_detailed(
 
   Args:
       graph_id (str): The graph ID to get metrics for
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -109,6 +124,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     graph_id=graph_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -123,6 +139,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
   """Get Graph Metrics
@@ -147,6 +164,7 @@ def sync(
 
   Args:
       graph_id (str): The graph ID to get metrics for
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -160,6 +178,7 @@ def sync(
   return sync_detailed(
     graph_id=graph_id,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -168,6 +187,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
   """Get Graph Metrics
@@ -192,6 +212,7 @@ async def asyncio_detailed(
 
   Args:
       graph_id (str): The graph ID to get metrics for
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -204,6 +225,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     graph_id=graph_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -216,6 +238,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, GraphMetricsResponse, HTTPValidationError]]:
   """Get Graph Metrics
@@ -240,6 +263,7 @@ async def asyncio(
 
   Args:
       graph_id (str): The graph ID to get metrics for
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -254,6 +278,7 @@ async def asyncio(
     await asyncio_detailed(
       graph_id=graph_id,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/graph_analytics/get_graph_usage_stats.py

@@ -15,6 +15,7 @@ def _get_kwargs(
   graph_id: str,
   *,
   include_details: Union[Unset, bool] = False,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
@@ -25,6 +26,13 @@ def _get_kwargs(
 
   params["include_details"] = include_details
 
+  json_token: Union[None, Unset, str]
+  if isinstance(token, Unset):
+    json_token = UNSET
+  else:
+    json_token = token
+  params["token"] = json_token
+
   params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
 
   _kwargs: dict[str, Any] = {
@@ -78,6 +86,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   include_details: Union[Unset, bool] = False,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Usage Statistics
@@ -110,6 +119,7 @@ def sync_detailed(
       graph_id (str): The graph ID to get usage stats for
       include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default:
           False.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -123,6 +133,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     include_details=include_details,
+    token=token,
     authorization=authorization,
   )
 
@@ -138,6 +149,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   include_details: Union[Unset, bool] = False,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Usage Statistics
@@ -170,6 +182,7 @@ def sync(
       graph_id (str): The graph ID to get usage stats for
       include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default:
           False.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -184,6 +197,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     include_details=include_details,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -193,6 +207,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   include_details: Union[Unset, bool] = False,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Usage Statistics
@@ -225,6 +240,7 @@ async def asyncio_detailed(
       graph_id (str): The graph ID to get usage stats for
       include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default:
           False.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -238,6 +254,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     include_details=include_details,
+    token=token,
     authorization=authorization,
   )
 
@@ -251,6 +268,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   include_details: Union[Unset, bool] = False,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Usage Statistics
@@ -283,6 +301,7 @@ async def asyncio(
       graph_id (str): The graph ID to get usage stats for
       include_details (Union[Unset, bool]): Include detailed metrics (may be slower) Default:
           False.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -298,6 +317,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       include_details=include_details,
+      token=token,
       authorization=authorization,
     )
   ).parsed