robosystems-client 0.2.13__tar.gz → 0.2.14__tar.gz

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.13
+Version: 0.2.14
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "robosystems-client"
-version = "0.2.13"
+version = "0.2.14"
 description = "Python Client for RoboSystems financial graph database API"
 readme = "README.md"
 requires-python = ">=3.10"

robosystems_client-0.2.13/robosystems_client/api/billing/update_org_payment_method.py → robosystems_client-0.2.14/robosystems_client/api/billing/create_portal_session.py

@@ -6,36 +6,26 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.http_validation_error import HTTPValidationError
-from ...models.update_payment_method_request import UpdatePaymentMethodRequest
-from ...models.update_payment_method_response import UpdatePaymentMethodResponse
+from ...models.portal_session_response import PortalSessionResponse
 from ...types import Response
 
 
 def _get_kwargs(
   org_id: str,
-  *,
-  body: UpdatePaymentMethodRequest,
 ) -> dict[str, Any]:
-  headers: dict[str, Any] = {}
-
   _kwargs: dict[str, Any] = {
     "method": "post",
-    "url": f"/v1/billing/customer/{org_id}/payment-method",
+    "url": f"/v1/billing/customer/{org_id}/portal",
   }
 
-  _kwargs["json"] = body.to_dict()
-
-  headers["Content-Type"] = "application/json"
-
-  _kwargs["headers"] = headers
   return _kwargs
 
 
 def _parse_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[HTTPValidationError, UpdatePaymentMethodResponse]]:
+) -> Optional[Union[HTTPValidationError, PortalSessionResponse]]:
   if response.status_code == 200:
-    response_200 = UpdatePaymentMethodResponse.from_dict(response.json())
+    response_200 = PortalSessionResponse.from_dict(response.json())
 
     return response_200
 
@@ -52,7 +42,7 @@ def _parse_response(
 
 def _build_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[HTTPValidationError, UpdatePaymentMethodResponse]]:
+) -> Response[Union[HTTPValidationError, PortalSessionResponse]]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -65,32 +55,37 @@ def sync_detailed(
   org_id: str,
   *,
   client: AuthenticatedClient,
-  body: UpdatePaymentMethodRequest,
-) -> Response[Union[HTTPValidationError, UpdatePaymentMethodResponse]]:
-  """Update Organization Default Payment Method
+) -> Response[Union[HTTPValidationError, PortalSessionResponse]]:
+  """Create Customer Portal Session
 
-   Update the default payment method for the organization.
+   Create a Stripe Customer Portal session for managing payment methods.
 
-  This changes which payment method will be used for future subscription charges.
+  The portal allows users to:
+  - Add new payment methods
+  - Remove existing payment methods
+  - Update default payment method
+  - View billing history
+
+  The user will be redirected to Stripe's hosted portal page and returned to the billing page when
+  done.
 
   **Requirements:**
   - User must be an OWNER of the organization
+  - Organization must have a Stripe customer ID (i.e., has gone through checkout at least once)
 
   Args:
       org_id (str):
-      body (UpdatePaymentMethodRequest): Request to update default payment method.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[HTTPValidationError, UpdatePaymentMethodResponse]]
+      Response[Union[HTTPValidationError, PortalSessionResponse]]
   """
 
   kwargs = _get_kwargs(
     org_id=org_id,
-    body=body,
   )
 
   response = client.get_httpx_client().request(
@@ -104,33 +99,38 @@ def sync(
   org_id: str,
   *,
   client: AuthenticatedClient,
-  body: UpdatePaymentMethodRequest,
-) -> Optional[Union[HTTPValidationError, UpdatePaymentMethodResponse]]:
-  """Update Organization Default Payment Method
+) -> Optional[Union[HTTPValidationError, PortalSessionResponse]]:
+  """Create Customer Portal Session
+
+   Create a Stripe Customer Portal session for managing payment methods.
 
-   Update the default payment method for the organization.
+  The portal allows users to:
+  - Add new payment methods
+  - Remove existing payment methods
+  - Update default payment method
+  - View billing history
 
-  This changes which payment method will be used for future subscription charges.
+  The user will be redirected to Stripe's hosted portal page and returned to the billing page when
+  done.
 
   **Requirements:**
   - User must be an OWNER of the organization
+  - Organization must have a Stripe customer ID (i.e., has gone through checkout at least once)
 
   Args:
       org_id (str):
-      body (UpdatePaymentMethodRequest): Request to update default payment method.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[HTTPValidationError, UpdatePaymentMethodResponse]
+      Union[HTTPValidationError, PortalSessionResponse]
   """
 
   return sync_detailed(
     org_id=org_id,
     client=client,
-    body=body,
   ).parsed
 
 
@@ -138,32 +138,37 @@ async def asyncio_detailed(
   org_id: str,
   *,
   client: AuthenticatedClient,
-  body: UpdatePaymentMethodRequest,
-) -> Response[Union[HTTPValidationError, UpdatePaymentMethodResponse]]:
-  """Update Organization Default Payment Method
+) -> Response[Union[HTTPValidationError, PortalSessionResponse]]:
+  """Create Customer Portal Session
 
-   Update the default payment method for the organization.
+   Create a Stripe Customer Portal session for managing payment methods.
 
-  This changes which payment method will be used for future subscription charges.
+  The portal allows users to:
+  - Add new payment methods
+  - Remove existing payment methods
+  - Update default payment method
+  - View billing history
+
+  The user will be redirected to Stripe's hosted portal page and returned to the billing page when
+  done.
 
   **Requirements:**
   - User must be an OWNER of the organization
+  - Organization must have a Stripe customer ID (i.e., has gone through checkout at least once)
 
   Args:
       org_id (str):
-      body (UpdatePaymentMethodRequest): Request to update default payment method.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[HTTPValidationError, UpdatePaymentMethodResponse]]
+      Response[Union[HTTPValidationError, PortalSessionResponse]]
   """
 
   kwargs = _get_kwargs(
     org_id=org_id,
-    body=body,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -175,33 +180,38 @@ async def asyncio(
   org_id: str,
   *,
   client: AuthenticatedClient,
-  body: UpdatePaymentMethodRequest,
-) -> Optional[Union[HTTPValidationError, UpdatePaymentMethodResponse]]:
-  """Update Organization Default Payment Method
+) -> Optional[Union[HTTPValidationError, PortalSessionResponse]]:
+  """Create Customer Portal Session
+
+   Create a Stripe Customer Portal session for managing payment methods.
 
-   Update the default payment method for the organization.
+  The portal allows users to:
+  - Add new payment methods
+  - Remove existing payment methods
+  - Update default payment method
+  - View billing history
 
-  This changes which payment method will be used for future subscription charges.
+  The user will be redirected to Stripe's hosted portal page and returned to the billing page when
+  done.
 
   **Requirements:**
   - User must be an OWNER of the organization
+  - Organization must have a Stripe customer ID (i.e., has gone through checkout at least once)
 
   Args:
       org_id (str):
-      body (UpdatePaymentMethodRequest): Request to update default payment method.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[HTTPValidationError, UpdatePaymentMethodResponse]
+      Union[HTTPValidationError, PortalSessionResponse]
   """
 
   return (
     await asyncio_detailed(
       org_id=org_id,
       client=client,
-      body=body,
     )
   ).parsed

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/robosystems_client/api/subgraphs/create_subgraph.py

@@ -75,7 +75,7 @@ def sync_detailed(
   - Valid authentication
   - Parent graph must exist and be accessible to the user
   - User must have 'admin' permission on the parent graph
-  - Parent graph tier must support subgraphs (Enterprise or Premium only)
+  - Parent graph tier must support subgraphs (Kuzu Large/XLarge or Neo4j Enterprise XLarge)
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
@@ -120,7 +120,7 @@ def sync(
   - Valid authentication
   - Parent graph must exist and be accessible to the user
   - User must have 'admin' permission on the parent graph
-  - Parent graph tier must support subgraphs (Enterprise or Premium only)
+  - Parent graph tier must support subgraphs (Kuzu Large/XLarge or Neo4j Enterprise XLarge)
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
@@ -160,7 +160,7 @@ async def asyncio_detailed(
   - Valid authentication
   - Parent graph must exist and be accessible to the user
   - User must have 'admin' permission on the parent graph
-  - Parent graph tier must support subgraphs (Enterprise or Premium only)
+  - Parent graph tier must support subgraphs (Kuzu Large/XLarge or Neo4j Enterprise XLarge)
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 
@@ -203,7 +203,7 @@ async def asyncio(
   - Valid authentication
   - Parent graph must exist and be accessible to the user
   - User must have 'admin' permission on the parent graph
-  - Parent graph tier must support subgraphs (Enterprise or Premium only)
+  - Parent graph tier must support subgraphs (Kuzu Large/XLarge or Neo4j Enterprise XLarge)
   - Must be within subgraph quota limits
   - Subgraph name must be unique within the parent graph
 

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/robosystems_client/api/subgraphs/delete_subgraph.py

@@ -111,8 +111,8 @@ def sync_detailed(
   All data in the subgraph will be lost.
 
   **Backup Location:**
-  If backup requested, stored in S3 at:
-  `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup`
+  If backup requested, stored in S3 Kuzu database bucket at:
+  `s3://{kuzu_s3_bucket}/{instance_id}/{database_name}_{timestamp}.backup`
 
   Args:
       graph_id (str):
@@ -165,8 +165,8 @@ def sync(
   All data in the subgraph will be lost.
 
   **Backup Location:**
-  If backup requested, stored in S3 at:
-  `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup`
+  If backup requested, stored in S3 Kuzu database bucket at:
+  `s3://{kuzu_s3_bucket}/{instance_id}/{database_name}_{timestamp}.backup`
 
   Args:
       graph_id (str):
@@ -214,8 +214,8 @@ async def asyncio_detailed(
   All data in the subgraph will be lost.
 
   **Backup Location:**
-  If backup requested, stored in S3 at:
-  `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup`
+  If backup requested, stored in S3 Kuzu database bucket at:
+  `s3://{kuzu_s3_bucket}/{instance_id}/{database_name}_{timestamp}.backup`
 
   Args:
       graph_id (str):
@@ -266,8 +266,8 @@ async def asyncio(
   All data in the subgraph will be lost.
 
   **Backup Location:**
-  If backup requested, stored in S3 at:
-  `s3://robosystems-backups/{instance_id}/{database_name}_{timestamp}.backup`
+  If backup requested, stored in S3 Kuzu database bucket at:
+  `s3://{kuzu_s3_bucket}/{instance_id}/{database_name}_{timestamp}.backup`
 
   Args:
       graph_id (str):

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/robosystems_client/models/__init__.py

@@ -45,7 +45,6 @@ from .bulk_ingest_response import BulkIngestResponse
 from .cancel_operation_response_canceloperation import (
   CancelOperationResponseCanceloperation,
 )
-from .cancellation_response import CancellationResponse
 from .check_credit_balance_response_checkcreditbalance import (
   CheckCreditBalanceResponseCheckcreditbalance,
 )
@@ -208,6 +207,7 @@ from .plaid_connection_config_accounts_type_0_item import (
 from .plaid_connection_config_institution_type_0 import (
   PlaidConnectionConfigInstitutionType0,
 )
+from .portal_session_response import PortalSessionResponse
 from .query_limits import QueryLimits
 from .quick_books_connection_config import QuickBooksConnectionConfig
 from .rate_limits import RateLimits
@@ -279,8 +279,6 @@ from .update_file_status_response_updatefilestatus import (
 from .update_member_role_request import UpdateMemberRoleRequest
 from .update_org_request import UpdateOrgRequest
 from .update_password_request import UpdatePasswordRequest
-from .update_payment_method_request import UpdatePaymentMethodRequest
-from .update_payment_method_response import UpdatePaymentMethodResponse
 from .update_user_request import UpdateUserRequest
 from .upgrade_subscription_request import UpgradeSubscriptionRequest
 from .user_graphs_response import UserGraphsResponse
@@ -326,7 +324,6 @@ __all__ = (
   "BillingCustomer",
   "BulkIngestRequest",
   "BulkIngestResponse",
-  "CancellationResponse",
   "CancelOperationResponseCanceloperation",
   "CheckCreditBalanceResponseCheckcreditbalance",
   "CheckoutResponse",
@@ -462,6 +459,7 @@ __all__ = (
   "PlaidConnectionConfig",
   "PlaidConnectionConfigAccountsType0Item",
   "PlaidConnectionConfigInstitutionType0",
+  "PortalSessionResponse",
   "QueryLimits",
   "QuickBooksConnectionConfig",
   "RateLimits",
@@ -519,8 +517,6 @@ __all__ = (
   "UpdateMemberRoleRequest",
   "UpdateOrgRequest",
   "UpdatePasswordRequest",
-  "UpdatePaymentMethodRequest",
-  "UpdatePaymentMethodResponse",
   "UpdateUserRequest",
   "UpgradeSubscriptionRequest",
   "UserGraphsResponse",

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/robosystems_client/models/create_subgraph_request.py

@@ -24,8 +24,7 @@ class CreateSubgraphRequest:
       name (str): Alphanumeric name for the subgraph (e.g., dev, staging, prod1)
       display_name (str): Human-readable display name for the subgraph
       description (Union[None, Unset, str]): Optional description of the subgraph's purpose
-      schema_extensions (Union[None, Unset, list[str]]): Schema extensions to include (inherits from parent by
-          default)
+      schema_extensions (Union[Unset, list[str]]): Schema extensions to include (inherits from parent by default)
       subgraph_type (Union[Unset, SubgraphType]): Types of subgraphs.
       metadata (Union['CreateSubgraphRequestMetadataType0', None, Unset]): Additional metadata for the subgraph
   """
@@ -33,7 +32,7 @@ class CreateSubgraphRequest:
   name: str
   display_name: str
   description: Union[None, Unset, str] = UNSET
-  schema_extensions: Union[None, Unset, list[str]] = UNSET
+  schema_extensions: Union[Unset, list[str]] = UNSET
   subgraph_type: Union[Unset, SubgraphType] = UNSET
   metadata: Union["CreateSubgraphRequestMetadataType0", None, Unset] = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
@@ -53,13 +52,8 @@ class CreateSubgraphRequest:
     else:
       description = self.description
 
-    schema_extensions: Union[None, Unset, list[str]]
-    if isinstance(self.schema_extensions, Unset):
-      schema_extensions = UNSET
-    elif isinstance(self.schema_extensions, list):
-      schema_extensions = self.schema_extensions
-
-    else:
+    schema_extensions: Union[Unset, list[str]] = UNSET
+    if not isinstance(self.schema_extensions, Unset):
       schema_extensions = self.schema_extensions
 
     subgraph_type: Union[Unset, str] = UNSET
@@ -113,22 +107,7 @@ class CreateSubgraphRequest:
 
     description = _parse_description(d.pop("description", UNSET))
 
-    def _parse_schema_extensions(data: object) -> Union[None, Unset, list[str]]:
-      if data is None:
-        return data
-      if isinstance(data, Unset):
-        return data
-      try:
-        if not isinstance(data, list):
-          raise TypeError()
-        schema_extensions_type_0 = cast(list[str], data)
-
-        return schema_extensions_type_0
-      except:  # noqa: E722
-        pass
-      return cast(Union[None, Unset, list[str]], data)
-
-    schema_extensions = _parse_schema_extensions(d.pop("schema_extensions", UNSET))
+    schema_extensions = cast(list[str], d.pop("schema_extensions", UNSET))
 
     _subgraph_type = d.pop("subgraph_type", UNSET)
     subgraph_type: Union[Unset, SubgraphType]

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/robosystems_client/models/graph_subscription_response.py

@@ -26,6 +26,8 @@ class GraphSubscriptionResponse:
       current_period_end (Union[None, Unset, str]): Current billing period end
       started_at (Union[None, Unset, str]): Subscription start date
       canceled_at (Union[None, Unset, str]): Cancellation date
+      ends_at (Union[None, Unset, str]): Subscription end date (when access will be revoked, especially relevant for
+          cancelled subscriptions)
   """
 
   id: str
@@ -40,6 +42,7 @@ class GraphSubscriptionResponse:
   current_period_end: Union[None, Unset, str] = UNSET
   started_at: Union[None, Unset, str] = UNSET
   canceled_at: Union[None, Unset, str] = UNSET
+  ends_at: Union[None, Unset, str] = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -83,6 +86,12 @@ class GraphSubscriptionResponse:
     else:
       canceled_at = self.canceled_at
 
+    ends_at: Union[None, Unset, str]
+    if isinstance(self.ends_at, Unset):
+      ends_at = UNSET
+    else:
+      ends_at = self.ends_at
+
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
     field_dict.update(
@@ -105,6 +114,8 @@ class GraphSubscriptionResponse:
       field_dict["started_at"] = started_at
     if canceled_at is not UNSET:
       field_dict["canceled_at"] = canceled_at
+    if ends_at is not UNSET:
+      field_dict["ends_at"] = ends_at
 
     return field_dict
 
@@ -165,6 +176,15 @@ class GraphSubscriptionResponse:
 
     canceled_at = _parse_canceled_at(d.pop("canceled_at", UNSET))
 
+    def _parse_ends_at(data: object) -> Union[None, Unset, str]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, str], data)
+
+    ends_at = _parse_ends_at(d.pop("ends_at", UNSET))
+
     graph_subscription_response = cls(
       id=id,
       resource_type=resource_type,
@@ -178,6 +198,7 @@ class GraphSubscriptionResponse:
       current_period_end=current_period_end,
       started_at=started_at,
       canceled_at=canceled_at,
+      ends_at=ends_at,
     )
 
     graph_subscription_response.additional_properties = d

{robosystems_client-0.2.13 → robosystems_client-0.2.14}/robosystems_client/models/list_subgraphs_response.py

@@ -21,6 +21,8 @@ class ListSubgraphsResponse:
       parent_graph_id (str): Parent graph identifier
       parent_graph_name (str): Parent graph name
       parent_graph_tier (str): Parent graph tier
+      subgraphs_enabled (bool): Whether subgraphs are enabled for this tier (requires Kuzu Large/XLarge or Neo4j
+          Enterprise XLarge)
       subgraph_count (int): Total number of subgraphs
       subgraphs (list['SubgraphSummary']): List of subgraphs
       max_subgraphs (Union[None, Unset, int]): Maximum allowed subgraphs for this tier (None = unlimited)
@@ -30,6 +32,7 @@ class ListSubgraphsResponse:
   parent_graph_id: str
   parent_graph_name: str
   parent_graph_tier: str
+  subgraphs_enabled: bool
   subgraph_count: int
   subgraphs: list["SubgraphSummary"]
   max_subgraphs: Union[None, Unset, int] = UNSET
@@ -43,6 +46,8 @@ class ListSubgraphsResponse:
 
     parent_graph_tier = self.parent_graph_tier
 
+    subgraphs_enabled = self.subgraphs_enabled
+
     subgraph_count = self.subgraph_count
 
     subgraphs = []
@@ -69,6 +74,7 @@ class ListSubgraphsResponse:
         "parent_graph_id": parent_graph_id,
         "parent_graph_name": parent_graph_name,
         "parent_graph_tier": parent_graph_tier,
+        "subgraphs_enabled": subgraphs_enabled,
         "subgraph_count": subgraph_count,
         "subgraphs": subgraphs,
       }
@@ -91,6 +97,8 @@ class ListSubgraphsResponse:
 
     parent_graph_tier = d.pop("parent_graph_tier")
 
+    subgraphs_enabled = d.pop("subgraphs_enabled")
+
     subgraph_count = d.pop("subgraph_count")
 
     subgraphs = []
@@ -122,6 +130,7 @@ class ListSubgraphsResponse:
       parent_graph_id=parent_graph_id,
       parent_graph_name=parent_graph_name,
       parent_graph_tier=parent_graph_tier,
+      subgraphs_enabled=subgraphs_enabled,
       subgraph_count=subgraph_count,
       subgraphs=subgraphs,
       max_subgraphs=max_subgraphs,

robosystems_client-0.2.14/robosystems_client/models/portal_session_response.py

@@ -0,0 +1,60 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="PortalSessionResponse")
+
+
+@_attrs_define
+class PortalSessionResponse:
+  """Response for customer portal session creation.
+
+  Attributes:
+      portal_url (str): Stripe Customer Portal URL where user can manage payment methods
+  """
+
+  portal_url: str
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    portal_url = self.portal_url
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "portal_url": portal_url,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    portal_url = d.pop("portal_url")
+
+    portal_session_response = cls(
+      portal_url=portal_url,
+    )
+
+    portal_session_response.additional_properties = d
+    return portal_session_response
+
+  @property
+  def additional_keys(self) -> list[str]:
+    return list(self.additional_properties.keys())
+
+  def __getitem__(self, key: str) -> Any:
+    return self.additional_properties[key]
+
+  def __setitem__(self, key: str, value: Any) -> None:
+    self.additional_properties[key] = value
+
+  def __delitem__(self, key: str) -> None:
+    del self.additional_properties[key]
+
+  def __contains__(self, key: str) -> bool:
+    return key in self.additional_properties