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

robosystems_client/api/user_subscriptions/get_repository_credits.py

@@ -13,15 +13,28 @@ from ...types import UNSET, Response, Unset
 def _get_kwargs(
   repository: 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/user/subscriptions/shared-repositories/credits/{repository}",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -66,6 +79,7 @@ def sync_detailed(
   repository: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
@@ -74,6 +88,7 @@ def sync_detailed(
 
   Args:
       repository (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -86,6 +101,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     repository=repository,
+    token=token,
     authorization=authorization,
   )
 
@@ -100,6 +116,7 @@ def sync(
   repository: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
@@ -108,6 +125,7 @@ def sync(
 
   Args:
       repository (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -121,6 +139,7 @@ def sync(
   return sync_detailed(
     repository=repository,
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -129,6 +148,7 @@ async def asyncio_detailed(
   repository: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
@@ -137,6 +157,7 @@ async def asyncio_detailed(
 
   Args:
       repository (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -149,6 +170,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     repository=repository,
+    token=token,
     authorization=authorization,
   )
 
@@ -161,6 +183,7 @@ async def asyncio(
   repository: str,
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, RepositoryCreditsResponse]]:
   """Get Repository Credits
@@ -169,6 +192,7 @@ async def asyncio(
 
   Args:
       repository (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -183,6 +207,7 @@ async def asyncio(
     await asyncio_detailed(
       repository=repository,
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/user_subscriptions/get_shared_repository_credits.py

@@ -12,15 +12,28 @@ from ...types import UNSET, Response, Unset
 
 def _get_kwargs(
   *,
+  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": "/v1/user/subscriptions/shared-repositories/credits",
+    "params": params,
   }
 
   _kwargs["headers"] = headers
@@ -64,6 +77,7 @@ def _build_response(
 def sync_detailed(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
@@ -71,6 +85,7 @@ def sync_detailed(
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -82,6 +97,7 @@ def sync_detailed(
   """
 
   kwargs = _get_kwargs(
+    token=token,
     authorization=authorization,
   )
 
@@ -95,6 +111,7 @@ def sync_detailed(
 def sync(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
@@ -102,6 +119,7 @@ def sync(
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -114,6 +132,7 @@ def sync(
 
   return sync_detailed(
     client=client,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -121,6 +140,7 @@ def sync(
 async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
@@ -128,6 +148,7 @@ async def asyncio_detailed(
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -139,6 +160,7 @@ async def asyncio_detailed(
   """
 
   kwargs = _get_kwargs(
+    token=token,
     authorization=authorization,
   )
 
@@ -150,6 +172,7 @@ async def asyncio_detailed(
 async def asyncio(
   *,
   client: AuthenticatedClient,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, CreditsSummaryResponse, HTTPValidationError]]:
   """Get Credit Balances
@@ -157,6 +180,7 @@ async def asyncio(
    Retrieve credit balances for all shared repository subscriptions
 
   Args:
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
 
   Raises:
@@ -170,6 +194,7 @@ async def asyncio(
   return (
     await asyncio_detailed(
       client=client,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/user_subscriptions/get_user_shared_subscriptions.py

@@ -13,6 +13,7 @@ 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,
 ) -> dict[str, Any]:
   headers: dict[str, Any] = {}
@@ -23,6 +24,13 @@ def _get_kwargs(
 
   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] = {
@@ -73,6 +81,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
@@ -81,6 +90,7 @@ 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]):
 
   Raises:
@@ -93,6 +103,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     active_only=active_only,
+    token=token,
     authorization=authorization,
   )
 
@@ -107,6 +118,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
@@ -115,6 +127,7 @@ 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]):
 
   Raises:
@@ -128,6 +141,7 @@ def sync(
   return sync_detailed(
     client=client,
     active_only=active_only,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -136,6 +150,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
@@ -144,6 +159,7 @@ 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]):
 
   Raises:
@@ -156,6 +172,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     active_only=active_only,
+    token=token,
     authorization=authorization,
   )
 
@@ -168,6 +185,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   active_only: Union[Unset, bool] = True,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, UserSubscriptionsResponse]]:
   """Get User Subscriptions
@@ -176,6 +194,7 @@ 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]):
 
   Raises:
@@ -190,6 +209,7 @@ async def asyncio(
     await asyncio_detailed(
       client=client,
       active_only=active_only,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/user_subscriptions/subscribe_to_shared_repository.py

@@ -14,15 +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,
 ) -> 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/user/subscriptions/shared-repositories/subscribe",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -74,6 +87,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
@@ -81,6 +95,7 @@ def sync_detailed(
    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]):
       body (SubscriptionRequest): Request to create a new subscription.
 
@@ -94,6 +109,7 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -108,6 +124,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
@@ -115,6 +132,7 @@ def sync(
    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]):
       body (SubscriptionRequest): Request to create a new subscription.
 
@@ -129,6 +147,7 @@ def sync(
   return sync_detailed(
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -137,6 +156,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
@@ -144,6 +164,7 @@ async def asyncio_detailed(
    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]):
       body (SubscriptionRequest): Request to create a new subscription.
 
@@ -157,6 +178,7 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -169,6 +191,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: SubscriptionRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError, SubscriptionResponse]]:
   """Subscribe to Shared Repository
@@ -176,6 +199,7 @@ async def asyncio(
    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]):
       body (SubscriptionRequest): Request to create a new subscription.
 
@@ -191,6 +215,7 @@ async def asyncio(
     await asyncio_detailed(
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).parsed

robosystems_client/api/user_subscriptions/upgrade_shared_repository_subscription.py

@@ -14,15 +14,28 @@ def _get_kwargs(
   subscription_id: str,
   *,
   body: TierUpgradeRequest,
+  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": "put",
     "url": f"/v1/user/subscriptions/shared-repositories/{subscription_id}/upgrade",
+    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -77,6 +90,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
@@ -85,6 +99,7 @@ def sync_detailed(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
@@ -99,6 +114,7 @@ def sync_detailed(
   kwargs = _get_kwargs(
     subscription_id=subscription_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -114,6 +130,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
@@ -122,6 +139,7 @@ def sync(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
@@ -137,6 +155,7 @@ def sync(
     subscription_id=subscription_id,
     client=client,
     body=body,
+    token=token,
     authorization=authorization,
   ).parsed
 
@@ -146,6 +165,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
@@ -154,6 +174,7 @@ async def asyncio_detailed(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
@@ -168,6 +189,7 @@ async def asyncio_detailed(
   kwargs = _get_kwargs(
     subscription_id=subscription_id,
     body=body,
+    token=token,
     authorization=authorization,
   )
 
@@ -181,6 +203,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: TierUpgradeRequest,
+  token: Union[None, Unset, str] = UNSET,
   authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Upgrade Subscription Tier
@@ -189,6 +212,7 @@ async def asyncio(
 
   Args:
       subscription_id (str):
+      token (Union[None, Unset, str]): JWT token for SSE authentication
       authorization (Union[None, Unset, str]):
       body (TierUpgradeRequest): Request to upgrade subscription tier.
 
@@ -205,6 +229,7 @@ async def asyncio(
       subscription_id=subscription_id,
       client=client,
       body=body,
+      token=token,
       authorization=authorization,
     )
   ).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)