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

robosystems_client/api/agent/recommend_agent.py

@@ -15,15 +15,28 @@ def _get_kwargs(
   graph_id: str,
   *,
   body: AgentRecommendationRequest,
+  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}/agent/recommend",
+    "url": f"/v1/graphs/{graph_id}/agent/recommend",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -70,6 +83,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: AgentRecommendationRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[AgentRecommendationResponse, Any, HTTPValidationError]]:
   """Get agent recommendations
@@ -91,6 +105,7 @@ def sync_detailed(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (AgentRecommendationRequest): Request for agent recommendations.
 
@@ -105,6 +120,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -120,6 +136,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: AgentRecommendationRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[AgentRecommendationResponse, Any, HTTPValidationError]]:
   """Get agent recommendations
@@ -141,6 +158,7 @@ def sync(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (AgentRecommendationRequest): Request for agent recommendations.
 
@@ -156,6 +174,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -165,6 +184,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: AgentRecommendationRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[AgentRecommendationResponse, Any, HTTPValidationError]]:
   """Get agent recommendations
@@ -186,6 +206,7 @@ async def asyncio_detailed(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (AgentRecommendationRequest): Request for agent recommendations.
 
@@ -200,6 +221,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -213,6 +235,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: AgentRecommendationRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[AgentRecommendationResponse, Any, HTTPValidationError]]:
   """Get agent recommendations
@@ -234,6 +257,7 @@ async def asyncio(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (AgentRecommendationRequest): Request for agent recommendations.
 
@@ -250,6 +274,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/backup/create_backup.py

@@ -15,15 +15,28 @@ def _get_kwargs(
   graph_id: str,
   *,
   body: BackupCreateRequest,
+  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}/backups",
+    "url": f"/v1/graphs/{graph_id}/backups",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -82,6 +95,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: BackupCreateRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
   """Create Backup
@@ -131,6 +145,7 @@ def sync_detailed(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (BackupCreateRequest): Request model for creating a backup.
 
@@ -145,6 +160,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -160,6 +176,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: BackupCreateRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
   """Create Backup
@@ -209,6 +226,7 @@ def sync(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (BackupCreateRequest): Request model for creating a backup.
 
@@ -224,6 +242,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -233,6 +252,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: BackupCreateRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
   """Create Backup
@@ -282,6 +302,7 @@ async def asyncio_detailed(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (BackupCreateRequest): Request model for creating a backup.
 
@@ -296,6 +317,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -309,6 +331,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: BackupCreateRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
   """Create Backup
@@ -358,6 +381,7 @@ async def asyncio(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (BackupCreateRequest): Request model for creating a backup.
 
@@ -374,6 +398,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/backup/export_backup.py

@@ -13,15 +13,28 @@ def _get_kwargs(
   graph_id: str,
   backup_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": "post",
-    "url": f"/v1/{graph_id}/backups/{backup_id}/export",
+    "url": f"/v1/graphs/{graph_id}/backups/{backup_id}/export",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -66,6 +79,7 @@ def sync_detailed(
   backup_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Export Kuzu backup for download
@@ -75,6 +89,7 @@ def sync_detailed(
   Args:
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -88,6 +103,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     backup_id=backup_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -103,6 +119,7 @@ def sync(
   backup_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Export Kuzu backup for download
@@ -112,6 +129,7 @@ def sync(
   Args:
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -126,6 +144,7 @@ def sync(
     graph_id=graph_id,
     backup_id=backup_id,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -135,6 +154,7 @@ async def asyncio_detailed(
   backup_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Export Kuzu backup for download
@@ -144,6 +164,7 @@ async def asyncio_detailed(
   Args:
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -157,6 +178,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     backup_id=backup_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -170,6 +192,7 @@ async def asyncio(
   backup_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Export Kuzu backup for download
@@ -179,6 +202,7 @@ async def asyncio(
   Args:
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -194,6 +218,7 @@ async def asyncio(
       graph_id=graph_id,
       backup_id=backup_id,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/backup/get_backup_download_url.py

@@ -17,6 +17,7 @@ def _get_kwargs(
   backup_id: str,
   *,
   expires_in: Union[Unset, int] = 3600,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
@@ -27,11 +28,18 @@ def _get_kwargs(
 
   params["expires_in"] = expires_in
 
+  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}/backups/{backup_id}/download",
+    "url": f"/v1/graphs/{graph_id}/backups/{backup_id}/download",
     "params": params,
   }
 
@@ -88,6 +96,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   expires_in: Union[Unset, int] = 3600,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[
   Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
@@ -100,6 +109,7 @@ def sync_detailed(
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
       expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -114,6 +124,7 @@ def sync_detailed(
     graph_id=graph_id,
     backup_id=backup_id,
     expires_in=expires_in,
+    token=token,
     authorization=authorization,
   )
 
@@ -130,6 +141,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   expires_in: Union[Unset, int] = 3600,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[
   Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
@@ -142,6 +154,7 @@ def sync(
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
       expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -157,6 +170,7 @@ def sync(
     backup_id=backup_id,
     client=client,
     expires_in=expires_in,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -167,6 +181,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   expires_in: Union[Unset, int] = 3600,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[
   Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
@@ -179,6 +194,7 @@ async def asyncio_detailed(
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
       expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -193,6 +209,7 @@ async def asyncio_detailed(
     graph_id=graph_id,
     backup_id=backup_id,
     expires_in=expires_in,
+    token=token,
     authorization=authorization,
   )
 
@@ -207,6 +224,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   expires_in: Union[Unset, int] = 3600,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[
   Union[Any, GetBackupDownloadUrlResponseGetbackupdownloadurl, HTTPValidationError]
@@ -219,6 +237,7 @@ async def asyncio(
       graph_id (str): Graph database identifier
       backup_id (str): Backup identifier
       expires_in (Union[Unset, int]): URL expiration time in seconds Default: 3600.
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -235,6 +254,7 @@ async def asyncio(
       backup_id=backup_id,
       client=client,
       expires_in=expires_in,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/backup/get_backup_stats.py

@@ -13,15 +13,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}/backups/stats",
+    "url": f"/v1/graphs/{graph_id}/backups/stats",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -60,6 +73,7 @@ def sync_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[BackupStatsResponse, HTTPValidationError]]:
   """Get backup statistics
@@ -68,6 +82,7 @@ def sync_detailed(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -80,6 +95,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     graph_id=graph_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -94,6 +110,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[BackupStatsResponse, HTTPValidationError]]:
   """Get backup statistics
@@ -102,6 +119,7 @@ def sync(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -115,6 +133,7 @@ def sync(
   return sync_detailed(
     graph_id=graph_id,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -123,6 +142,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[BackupStatsResponse, HTTPValidationError]]:
   """Get backup statistics
@@ -131,6 +151,7 @@ async def asyncio_detailed(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -143,6 +164,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     graph_id=graph_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -155,6 +177,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[BackupStatsResponse, HTTPValidationError]]:
   """Get backup statistics
@@ -163,6 +186,7 @@ async def asyncio(
 
   Args:
       graph_id (str): Graph database identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -177,6 +201,7 @@ async def asyncio(
     await asyncio_detailed(
       graph_id=graph_id,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed