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

robosystems_client/api/user_subscriptions/get_shared_repository_credits.py

@@ -12,21 +12,28 @@ from ...types import UNSET, Response, Unset
 
 def _get_kwargs(
   *,
+  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": "get",
     "url": "/v1/user/subscriptions/shared-repositories/credits",
-    "cookies": cookies,
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -70,16 +77,16 @@ def _build_response(
 def sync_detailed(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
 
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      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.
@@ -90,8 +97,8 @@ def sync_detailed(
   """
 
   kwargs = _get_kwargs(
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -104,16 +111,16 @@ def sync_detailed(
 def sync(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
 
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      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.
@@ -125,24 +132,24 @@ def sync(
 
   return sync_detailed(
     client=client,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
 async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
 
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      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.
@@ -153,8 +160,8 @@ async def asyncio_detailed(
   """
 
   kwargs = _get_kwargs(
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -165,16 +172,16 @@ async def asyncio_detailed(
 async def asyncio(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
 
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      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.
@@ -187,7 +194,7 @@ async def asyncio(
   return (
     await asyncio_detailed(
       client=client,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py

@@ -13,28 +13,30 @@ from ...types import UNSET, Response, Unset
 def _get_kwargs(
   *,
   active_only: Union[Unset, bool] = True,
+  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] = {}
 
   params["active_only"] = active_only
 
+  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": "/v1/user/subscriptions/shared-repositories",
     "params": params,
-    "cookies": cookies,
   }
 
   _kwargs["headers"] = headers
@@ -79,8 +81,8 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
 
@@ -88,8 +90,8 @@ def sync_detailed(
 
   Args:
       active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
+      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.
@@ -101,8 +103,8 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     active_only=active_only,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -116,8 +118,8 @@ def sync(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
 
@@ -125,8 +127,8 @@ def sync(
 
   Args:
       active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
+      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.
@@ -139,8 +141,8 @@ def sync(
   return sync_detailed(
     client=client,
     active_only=active_only,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -148,8 +150,8 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
 
@@ -157,8 +159,8 @@ async def asyncio_detailed(
 
   Args:
       active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
+      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.
@@ -170,8 +172,8 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     active_only=active_only,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -183,8 +185,8 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
 
@@ -192,8 +194,8 @@ async def asyncio(
 
   Args:
       active_only (Union[Unset, bool]): Only return active subscriptions Default: True.
+      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.
@@ -207,7 +209,7 @@ async def asyncio(
     await asyncio_detailed(
       client=client,
       active_only=active_only,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py

@@ -14,21 +14,28 @@ from ...types import UNSET, Response, Unset
 def _get_kwargs(
   *,
   body: SubscriptionRequest,
+  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": "/v1/user/subscriptions/shared-repositories/subscribe",
-    "cookies": cookies,
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -80,16 +87,16 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
 
    Create a new subscription to a shared repository add-on with specified tier
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (SubscriptionRequest): Request to create a new subscription.
 
   Raises:
@@ -102,8 +109,8 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -117,16 +124,16 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
 
    Create a new subscription to a shared repository add-on with specified tier
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (SubscriptionRequest): Request to create a new subscription.
 
   Raises:
@@ -140,8 +147,8 @@ def sync(
   return sync_detailed(
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -149,16 +156,16 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
 
    Create a new subscription to a shared repository add-on with specified tier
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (SubscriptionRequest): Request to create a new subscription.
 
   Raises:
@@ -171,8 +178,8 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -184,16 +191,16 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
 
    Create a new subscription to a shared repository add-on with specified tier
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (SubscriptionRequest): Request to create a new subscription.
 
   Raises:
@@ -208,7 +215,7 @@ async def asyncio(
     await asyncio_detailed(
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py

@@ -14,21 +14,28 @@ def _get_kwargs(
   subscription_id: str,
   *,
   body: TierUpgradeRequest,
+  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": "put",
     "url": f"/v1/user/subscriptions/shared-repositories/{subscription_id}/upgrade",
-    "cookies": cookies,
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -83,8 +90,8 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -92,8 +99,8 @@ def sync_detailed(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -107,8 +114,8 @@ def sync_detailed(
   kwargs = _get_kwargs(
     subscription_id=subscription_id,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = client.get_httpx_client().request(
@@ -123,8 +130,8 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -132,8 +139,8 @@ def sync(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -148,8 +155,8 @@ def sync(
     subscription_id=subscription_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   ).parsed
 
 
@@ -158,8 +165,8 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -167,8 +174,8 @@ async def asyncio_detailed(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -182,8 +189,8 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     subscription_id=subscription_id,
     body=body,
+    token=token,
     authorization=authorization,
-    auth_token=auth_token,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -196,8 +203,8 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
-  auth_token: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
 
@@ -205,8 +212,8 @@ async def asyncio(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
-      auth_token (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
   Raises:
@@ -222,7 +229,7 @@ async def asyncio(
       subscription_id=subscription_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
-      auth_token=auth_token,
     )
   ).parsed

robosystems_client/extensions/__init__.py

@@ -54,6 +54,47 @@ from .auth_integration import (
   create_development_extensions,
 )
 
+# JWT Token utilities
+from .token_utils import (
+  validate_jwt_format,
+  extract_jwt_from_header,
+  decode_jwt_payload,
+  is_jwt_expired,
+  get_jwt_claims,
+  get_jwt_expiration,
+  extract_token_from_environment,
+  extract_token_from_cookie,
+  find_valid_token,
+  TokenManager,
+  TokenSource,
+)
+
+# DataFrame utilities (optional - requires pandas)
+try:
+  from .dataframe_utils import (
+    query_result_to_dataframe,
+    DataFrameQueryClient,
+    HAS_PANDAS,
+  )
+
+  # Re-export the imported functions for module API
+  from .dataframe_utils import (
+    parse_datetime_columns,
+    stream_to_dataframe as _stream_to_dataframe,
+    dataframe_to_cypher_params,
+    export_query_to_csv,
+    compare_dataframes,
+  )
+except ImportError:
+  HAS_PANDAS = False
+  DataFrameQueryClient = None
+  # Set placeholders for optional functions
+  parse_datetime_columns = None
+  _stream_to_dataframe = None
+  dataframe_to_cypher_params = None
+  export_query_to_csv = None
+  compare_dataframes = None
+
 __all__ = [
   # Core extension classes
   "RoboSystemsExtensions",
@@ -101,6 +142,21 @@ __all__ = [
   "create_extensions",
   "create_production_extensions",
   "create_development_extensions",
+  # JWT Token utilities
+  "validate_jwt_format",
+  "extract_jwt_from_header",
+  "decode_jwt_payload",
+  "is_jwt_expired",
+  "get_jwt_claims",
+  "get_jwt_expiration",
+  "extract_token_from_environment",
+  "extract_token_from_cookie",
+  "find_valid_token",
+  "TokenManager",
+  "TokenSource",
+  # DataFrame utilities (optional)
+  "HAS_PANDAS",
+  "DataFrameQueryClient",
 ]
 
 # Create a default extensions instance
@@ -135,3 +191,17 @@ def copy_from_s3(
   return extensions.copy_from_s3(
     graph_id, table_name, s3_path, access_key_id, secret_access_key, **kwargs
   )
+
+
+# DataFrame convenience functions (if pandas is available)
+if HAS_PANDAS:
+
+  def query_to_dataframe(graph_id: str, query: str, parameters=None, **kwargs):
+    """Execute query and return results as pandas DataFrame"""
+    result = execute_query(graph_id, query, parameters)
+    return query_result_to_dataframe(result, **kwargs)
+
+  def stream_to_dataframe(graph_id: str, query: str, parameters=None, chunk_size=10000):
+    """Stream query results and return as pandas DataFrame"""
+    stream = stream_query(graph_id, query, parameters, chunk_size)
+    return _stream_to_dataframe(stream, chunk_size)