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

robosystems_client/api/connections/get_connection.py

@@ -15,15 +15,28 @@ def _get_kwargs(
   graph_id: str,
   connection_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}/connections/{connection_id}",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -75,6 +88,7 @@ def sync_detailed(
   connection_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
   """Get Connection
@@ -93,6 +107,7 @@ def sync_detailed(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Unique connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -106,6 +121,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     connection_id=connection_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -121,6 +137,7 @@ def sync(
   connection_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
   """Get Connection
@@ -139,6 +156,7 @@ def sync(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Unique connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -153,6 +171,7 @@ def sync(
     graph_id=graph_id,
     connection_id=connection_id,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -162,6 +181,7 @@ async def asyncio_detailed(
   connection_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
   """Get Connection
@@ -180,6 +200,7 @@ async def asyncio_detailed(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Unique connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -193,6 +214,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     connection_id=connection_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -206,6 +228,7 @@ async def asyncio(
   connection_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ConnectionResponse, ErrorResponse, HTTPValidationError]]:
   """Get Connection
@@ -224,6 +247,7 @@ async def asyncio(
   Args:
       graph_id (str): Graph database identifier
       connection_id (str): Unique connection identifier
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -239,6 +263,7 @@ async def asyncio(
       graph_id=graph_id,
       connection_id=connection_id,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/connections/get_connection_options.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}/connections/options",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -69,6 +82,7 @@ def sync_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
   """List Connection Options
@@ -96,6 +110,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:
@@ -108,6 +123,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     graph_id=graph_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -122,6 +138,7 @@ def sync(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
   """List Connection Options
@@ -149,6 +166,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:
@@ -162,6 +180,7 @@ def sync(
   return sync_detailed(
     graph_id=graph_id,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -170,6 +189,7 @@ async def asyncio_detailed(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
   """List Connection Options
@@ -197,6 +217,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:
@@ -209,6 +230,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     graph_id=graph_id,
+    token=token,
     authorization=authorization,
   )
 
@@ -221,6 +243,7 @@ async def asyncio(
   graph_id: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[ConnectionOptionsResponse, ErrorResponse, HTTPValidationError]]:
   """List Connection Options
@@ -248,6 +271,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:
@@ -262,6 +286,7 @@ async def asyncio(
     await asyncio_detailed(
       graph_id=graph_id,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/connections/init_o_auth.py

@@ -15,15 +15,28 @@ def _get_kwargs(
   graph_id: str,
   *,
   body: OAuthInitRequest,
+  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/oauth/init",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -67,6 +80,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: OAuthInitRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[HTTPValidationError, OAuthInitResponse]]:
   """Init Oauth
@@ -78,6 +92,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 (OAuthInitRequest): Request to initiate OAuth flow.
 
@@ -92,6 +107,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -107,6 +123,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: OAuthInitRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[HTTPValidationError, OAuthInitResponse]]:
   """Init Oauth
@@ -118,6 +135,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 (OAuthInitRequest): Request to initiate OAuth flow.
 
@@ -133,6 +151,7 @@ def sync(
     graph_id=graph_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -142,6 +161,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: OAuthInitRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[HTTPValidationError, OAuthInitResponse]]:
   """Init Oauth
@@ -153,6 +173,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 (OAuthInitRequest): Request to initiate OAuth flow.
 
@@ -167,6 +188,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     graph_id=graph_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -180,6 +202,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: OAuthInitRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[HTTPValidationError, OAuthInitResponse]]:
   """Init Oauth
@@ -191,6 +214,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 (OAuthInitRequest): Request to initiate OAuth flow.
 
@@ -207,6 +231,7 @@ async def asyncio(
       graph_id=graph_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/connections/list_connections.py

@@ -17,6 +17,7 @@ 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,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
@@ -41,6 +42,13 @@ 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] = {
@@ -100,6 +108,7 @@ 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,
 ) -> Response[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
@@ -123,6 +132,7 @@ 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]):
 
   Raises:
@@ -137,6 +147,7 @@ def sync_detailed(
     graph_id=graph_id,
     entity_id=entity_id,
     provider=provider,
+    token=token,
     authorization=authorization,
   )
 
@@ -153,6 +164,7 @@ 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,
 ) -> Optional[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
@@ -176,6 +188,7 @@ 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]):
 
   Raises:
@@ -191,6 +204,7 @@ def sync(
     client=client,
     entity_id=entity_id,
     provider=provider,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -201,6 +215,7 @@ 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,
 ) -> Response[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
@@ -224,6 +239,7 @@ 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]):
 
   Raises:
@@ -238,6 +254,7 @@ async def asyncio_detailed(
     graph_id=graph_id,
     entity_id=entity_id,
     provider=provider,
+    token=token,
     authorization=authorization,
   )
 
@@ -252,6 +269,7 @@ 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,
 ) -> Optional[Union[ErrorResponse, HTTPValidationError, list["ConnectionResponse"]]]:
   """List Connections
@@ -275,6 +293,7 @@ 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]):
 
   Raises:
@@ -291,6 +310,7 @@ async def asyncio(
       client=client,
       entity_id=entity_id,
       provider=provider,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/connections/oauth_callback.py

@@ -16,15 +16,28 @@ def _get_kwargs(
   provider: str,
   *,
   body: OAuthCallbackRequest,
+  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/oauth/callback/{provider}",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -84,6 +97,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
@@ -111,6 +125,7 @@ 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]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
@@ -126,6 +141,7 @@ def sync_detailed(
     graph_id=graph_id,
     provider=provider,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -142,6 +158,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
@@ -169,6 +186,7 @@ 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]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
@@ -185,6 +203,7 @@ def sync(
     provider=provider,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -195,6 +214,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
@@ -222,6 +242,7 @@ 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]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
@@ -237,6 +258,7 @@ async def asyncio_detailed(
     graph_id=graph_id,
     provider=provider,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -251,6 +273,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: OAuthCallbackRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, ErrorResponse, HTTPValidationError]]:
   """OAuth Callback
@@ -278,6 +301,7 @@ 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]):
       body (OAuthCallbackRequest): OAuth callback parameters.
 
@@ -295,6 +319,7 @@ async def asyncio(
       provider=provider,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed