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

robosystems_client/api/connections/list_connections.py

@@ -17,17 +17,13 @@ def _get_kwargs(
   *,
   entity_id: Union[None, Unset, str] = UNSET,
   provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
-  cookies = {}
-  if auth_token is not UNSET:
-    cookies["auth-token"] = auth_token
-
   params: dict[str, Any] = {}
 
   json_entity_id: Union[None, Unset, str]
@@ -46,13 +42,19 @@ def _get_kwargs(
     json_provider = provider
   params["provider"] = json_provider
 
+  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}/connections",
     "params": params,
-    "cookies": cookies,
   }
 
   _kwargs["headers"] = headers
@@ -106,8 +108,8 @@ def sync_detailed(
   client: AuthenticatedClient,
   entity_id: Union[None, Unset, str] = UNSET,
   provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
 
@@ -130,8 +132,8 @@ def sync_detailed(
       graph_id (str): Graph database identifier
       entity_id (Union[None, Unset, str]): Filter by entity ID
       provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -145,8 +147,8 @@ def sync_detailed(
     graph_id=graph_id,
     entity_id=entity_id,
     provider=provider,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -162,8 +164,8 @@ def sync(
   client: AuthenticatedClient,
   entity_id: Union[None, Unset, str] = UNSET,
   provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
 
@@ -186,8 +188,8 @@ def sync(
       graph_id (str): Graph database identifier
       entity_id (Union[None, Unset, str]): Filter by entity ID
       provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -202,8 +204,8 @@ def sync(
     client=client,
     entity_id=entity_id,
     provider=provider,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -213,8 +215,8 @@ async def asyncio_detailed(
   client: AuthenticatedClient,
   entity_id: Union[None, Unset, str] = UNSET,
   provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
 
@@ -237,8 +239,8 @@ async def asyncio_detailed(
       graph_id (str): Graph database identifier
       entity_id (Union[None, Unset, str]): Filter by entity ID
       provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -252,8 +254,8 @@ async def asyncio_detailed(
     graph_id=graph_id,
     entity_id=entity_id,
     provider=provider,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -267,8 +269,8 @@ async def asyncio(
   client: AuthenticatedClient,
   entity_id: Union[None, Unset, str] = UNSET,
   provider: Union[ListConnectionsProviderType0, None, Unset] = UNSET,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
 
@@ -291,8 +293,8 @@ async def asyncio(
       graph_id (str): Graph database identifier
       entity_id (Union[None, Unset, str]): Filter by entity ID
       provider (Union[ListConnectionsProviderType0, None, Unset]): Filter by provider type
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -308,7 +310,7 @@ async def asyncio(
       client=client,
       entity_id=entity_id,
       provider=provider,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/api/connections/oauth_callback.py

@@ -16,21 +16,28 @@ def _get_kwargs(
   provider: str,
   *,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
-  cookies = {}
-  if auth_token is not UNSET:
-    cookies["auth-token"] = auth_token
+  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/oauth/callback/{provider}",
-    "cookies": cookies,
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -90,8 +97,8 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
 
@@ -108,7 +115,7 @@ def sync_detailed(
   - **QuickBooks**: Accounting data integration
 
   Security measures:
-  - State validation prevents CSRF attacks
+  - State validation prevents session hijacking
   - User context is verified
   - Tokens are encrypted before storage
   - Full audit trail is maintained
@@ -118,8 +125,8 @@ def sync_detailed(
   Args:
       graph_id (str): Graph database identifier
       provider (str): OAuth provider name
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
   Raises:
@@ -134,8 +141,8 @@ def sync_detailed(
     graph_id=graph_id,
     provider=provider,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -151,8 +158,8 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
 
@@ -169,7 +176,7 @@ def sync(
   - **QuickBooks**: Accounting data integration
 
   Security measures:
-  - State validation prevents CSRF attacks
+  - State validation prevents session hijacking
   - User context is verified
   - Tokens are encrypted before storage
   - Full audit trail is maintained
@@ -179,8 +186,8 @@ def sync(
   Args:
       graph_id (str): Graph database identifier
       provider (str): OAuth provider name
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
   Raises:
@@ -196,8 +203,8 @@ def sync(
     provider=provider,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -207,8 +214,8 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
 
@@ -225,7 +232,7 @@ async def asyncio_detailed(
   - **QuickBooks**: Accounting data integration
 
   Security measures:
-  - State validation prevents CSRF attacks
+  - State validation prevents session hijacking
   - User context is verified
   - Tokens are encrypted before storage
   - Full audit trail is maintained
@@ -235,8 +242,8 @@ async def asyncio_detailed(
   Args:
       graph_id (str): Graph database identifier
       provider (str): OAuth provider name
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
   Raises:
@@ -251,8 +258,8 @@ async def asyncio_detailed(
     graph_id=graph_id,
     provider=provider,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -266,8 +273,8 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
 
@@ -284,7 +291,7 @@ async def asyncio(
   - **QuickBooks**: Accounting data integration
 
   Security measures:
-  - State validation prevents CSRF attacks
+  - State validation prevents session hijacking
   - User context is verified
   - Tokens are encrypted before storage
   - Full audit trail is maintained
@@ -294,8 +301,8 @@ async def asyncio(
   Args:
       graph_id (str): Graph database identifier
       provider (str): OAuth provider name
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
   Raises:
@@ -312,7 +319,7 @@ async def asyncio(
       provider=provider,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/api/connections/sync_connection.py

@@ -19,21 +19,28 @@ def _get_kwargs(
   connection_id: str,
   *,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
-  cookies = {}
-  if auth_token is not UNSET:
-    cookies["auth-token"] = auth_token
+  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",
-    "cookies": cookies,
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -94,8 +101,8 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
 ]:
@@ -129,8 +136,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
   Raises:
@@ -145,8 +152,8 @@ def sync_detailed(
     graph_id=graph_id,
     connection_id=connection_id,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -162,8 +169,8 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
 ]:
@@ -197,8 +204,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
   Raises:
@@ -214,8 +221,8 @@ def sync(
     connection_id=connection_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -225,8 +232,8 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
 ]:
@@ -260,8 +267,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
   Raises:
@@ -276,8 +283,8 @@ async def asyncio_detailed(
     graph_id=graph_id,
     connection_id=connection_id,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -291,8 +298,8 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: SyncConnectionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[
   Union[ErrorResponse, HTTPValidationError, SyncConnectionResponseSyncconnection]
 ]:
@@ -326,8 +333,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (SyncConnectionRequest): Request to sync a connection.
 
   Raises:
@@ -344,7 +351,7 @@ async def asyncio(
       connection_id=connection_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/api/copy/copy_data_to_graph.py

@@ -17,21 +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,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
   if not isinstance(authorization, Unset):
     headers["authorization"] = authorization
 
-  cookies = {}
-  if auth_token is not UNSET:
-    cookies["auth-token"] = auth_token
+  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",
-    "cookies": cookies,
+    "params": params,
   }
 
   _kwargs["json"]: dict[str, Any]
@@ -102,8 +109,8 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
 
@@ -169,8 +176,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
   Raises:
@@ -184,8 +191,8 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -200,8 +207,8 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
 
@@ -267,8 +274,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
   Raises:
@@ -283,8 +290,8 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -293,8 +300,8 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
 
@@ -360,8 +367,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
   Raises:
@@ -375,8 +382,8 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -389,8 +396,8 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: Union["DataFrameCopyRequest", "S3CopyRequest", "URLCopyRequest"],
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CopyResponse, HTTPValidationError]]:
   """Copy Data to Graph
 
@@ -456,8 +463,8 @@ 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]):
-      auth_token (Union[None, Unset, str]):
       body (Union['DataFrameCopyRequest', 'S3CopyRequest', 'URLCopyRequest']):
 
   Raises:
@@ -473,7 +480,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed