robosystems-client 0.2.11__py3-none-any.whl → 0.2.12__py3-none-any.whl

robosystems_client/api/org_usage/get_org_usage.py

@@ -0,0 +1,186 @@
+from http import HTTPStatus
+from typing import Any, Optional, Union
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.http_validation_error import HTTPValidationError
+from ...models.org_usage_response import OrgUsageResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+  org_id: str,
+  *,
+  days: Union[Unset, int] = 30,
+) -> dict[str, Any]:
+  params: dict[str, Any] = {}
+
+  params["days"] = days
+
+  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/orgs/{org_id}/usage",
+    "params": params,
+  }
+
+  return _kwargs
+
+
+def _parse_response(
+  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Optional[Union[HTTPValidationError, OrgUsageResponse]]:
+  if response.status_code == 200:
+    response_200 = OrgUsageResponse.from_dict(response.json())
+
+    return response_200
+
+  if response.status_code == 422:
+    response_422 = HTTPValidationError.from_dict(response.json())
+
+    return response_422
+
+  if client.raise_on_unexpected_status:
+    raise errors.UnexpectedStatus(response.status_code, response.content)
+  else:
+    return None
+
+
+def _build_response(
+  *, client: Union[AuthenticatedClient, Client], response: httpx.Response
+) -> Response[Union[HTTPValidationError, OrgUsageResponse]]:
+  return Response(
+    status_code=HTTPStatus(response.status_code),
+    content=response.content,
+    headers=response.headers,
+    parsed=_parse_response(client=client, response=response),
+  )
+
+
+def sync_detailed(
+  org_id: str,
+  *,
+  client: AuthenticatedClient,
+  days: Union[Unset, int] = 30,
+) -> Response[Union[HTTPValidationError, OrgUsageResponse]]:
+  """Get Organization Usage
+
+   Get detailed usage statistics for an organization aggregated across all graphs.
+
+  Args:
+      org_id (str):
+      days (Union[Unset, int]):  Default: 30.
+
+  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, OrgUsageResponse]]
+  """
+
+  kwargs = _get_kwargs(
+    org_id=org_id,
+    days=days,
+  )
+
+  response = client.get_httpx_client().request(
+    **kwargs,
+  )
+
+  return _build_response(client=client, response=response)
+
+
+def sync(
+  org_id: str,
+  *,
+  client: AuthenticatedClient,
+  days: Union[Unset, int] = 30,
+) -> Optional[Union[HTTPValidationError, OrgUsageResponse]]:
+  """Get Organization Usage
+
+   Get detailed usage statistics for an organization aggregated across all graphs.
+
+  Args:
+      org_id (str):
+      days (Union[Unset, int]):  Default: 30.
+
+  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, OrgUsageResponse]
+  """
+
+  return sync_detailed(
+    org_id=org_id,
+    client=client,
+    days=days,
+  ).parsed
+
+
+async def asyncio_detailed(
+  org_id: str,
+  *,
+  client: AuthenticatedClient,
+  days: Union[Unset, int] = 30,
+) -> Response[Union[HTTPValidationError, OrgUsageResponse]]:
+  """Get Organization Usage
+
+   Get detailed usage statistics for an organization aggregated across all graphs.
+
+  Args:
+      org_id (str):
+      days (Union[Unset, int]):  Default: 30.
+
+  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, OrgUsageResponse]]
+  """
+
+  kwargs = _get_kwargs(
+    org_id=org_id,
+    days=days,
+  )
+
+  response = await client.get_async_httpx_client().request(**kwargs)
+
+  return _build_response(client=client, response=response)
+
+
+async def asyncio(
+  org_id: str,
+  *,
+  client: AuthenticatedClient,
+  days: Union[Unset, int] = 30,
+) -> Optional[Union[HTTPValidationError, OrgUsageResponse]]:
+  """Get Organization Usage
+
+   Get detailed usage statistics for an organization aggregated across all graphs.
+
+  Args:
+      org_id (str):
+      days (Union[Unset, int]):  Default: 30.
+
+  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, OrgUsageResponse]
+  """
+
+  return (
+    await asyncio_detailed(
+      org_id=org_id,
+      client=client,
+      days=days,
+    )
+  ).parsed

robosystems_client/api/service_offerings/get_service_offerings.py

@@ -61,9 +61,15 @@ def sync_detailed(
   and shared repository subscriptions. This is the primary endpoint for frontend
   applications to display subscription options.
 
+  **Pricing Model:**
+  - Graph subscriptions are **per-graph** with infrastructure-based pricing
+  - Each graph you create has its own monthly subscription
+  - Organizations can have multiple graphs with different infrastructure tiers
+  - Credits are allocated per-graph, not shared across organization
+
   Includes:
-  - Graph subscription tiers (standard, enterprise, premium)
-  - Shared repository subscriptions (SEC, industry, economic data)
+  - Graph infrastructure tiers (kuzu-standard, kuzu-large, kuzu-xlarge) - per-graph pricing
+  - Shared repository subscriptions (SEC, industry, economic data) - org-level
   - Operation costs and credit information
   - Features and capabilities for each tier
   - Enabled/disabled status for repositories
@@ -101,9 +107,15 @@ def sync(
   and shared repository subscriptions. This is the primary endpoint for frontend
   applications to display subscription options.
 
+  **Pricing Model:**
+  - Graph subscriptions are **per-graph** with infrastructure-based pricing
+  - Each graph you create has its own monthly subscription
+  - Organizations can have multiple graphs with different infrastructure tiers
+  - Credits are allocated per-graph, not shared across organization
+
   Includes:
-  - Graph subscription tiers (standard, enterprise, premium)
-  - Shared repository subscriptions (SEC, industry, economic data)
+  - Graph infrastructure tiers (kuzu-standard, kuzu-large, kuzu-xlarge) - per-graph pricing
+  - Shared repository subscriptions (SEC, industry, economic data) - org-level
   - Operation costs and credit information
   - Features and capabilities for each tier
   - Enabled/disabled status for repositories
@@ -137,9 +149,15 @@ async def asyncio_detailed(
   and shared repository subscriptions. This is the primary endpoint for frontend
   applications to display subscription options.
 
+  **Pricing Model:**
+  - Graph subscriptions are **per-graph** with infrastructure-based pricing
+  - Each graph you create has its own monthly subscription
+  - Organizations can have multiple graphs with different infrastructure tiers
+  - Credits are allocated per-graph, not shared across organization
+
   Includes:
-  - Graph subscription tiers (standard, enterprise, premium)
-  - Shared repository subscriptions (SEC, industry, economic data)
+  - Graph infrastructure tiers (kuzu-standard, kuzu-large, kuzu-xlarge) - per-graph pricing
+  - Shared repository subscriptions (SEC, industry, economic data) - org-level
   - Operation costs and credit information
   - Features and capabilities for each tier
   - Enabled/disabled status for repositories
@@ -175,9 +193,15 @@ async def asyncio(
   and shared repository subscriptions. This is the primary endpoint for frontend
   applications to display subscription options.
 
+  **Pricing Model:**
+  - Graph subscriptions are **per-graph** with infrastructure-based pricing
+  - Each graph you create has its own monthly subscription
+  - Organizations can have multiple graphs with different infrastructure tiers
+  - Credits are allocated per-graph, not shared across organization
+
   Includes:
-  - Graph subscription tiers (standard, enterprise, premium)
-  - Shared repository subscriptions (SEC, industry, economic data)
+  - Graph infrastructure tiers (kuzu-standard, kuzu-large, kuzu-xlarge) - per-graph pricing
+  - Shared repository subscriptions (SEC, industry, economic data) - org-level
   - Operation costs and credit information
   - Features and capabilities for each tier
   - Enabled/disabled status for repositories

robosystems_client/api/usage/get_graph_usage_analytics.py

@@ -95,7 +95,7 @@ def sync_detailed(
 ) -> Response[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Graph Usage Analytics
 
-   Get comprehensive usage analytics tracked by the GraphUsageTracking model.
+   Get comprehensive usage analytics tracked by the GraphUsage model.
 
   Provides temporal usage patterns including:
   - **Storage Analytics**: GB-hours for billing, breakdown by type (files, tables, graphs, subgraphs)
@@ -171,7 +171,7 @@ def sync(
 ) -> Optional[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Graph Usage Analytics
 
-   Get comprehensive usage analytics tracked by the GraphUsageTracking model.
+   Get comprehensive usage analytics tracked by the GraphUsage model.
 
   Provides temporal usage patterns including:
   - **Storage Analytics**: GB-hours for billing, breakdown by type (files, tables, graphs, subgraphs)
@@ -242,7 +242,7 @@ async def asyncio_detailed(
 ) -> Response[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Graph Usage Analytics
 
-   Get comprehensive usage analytics tracked by the GraphUsageTracking model.
+   Get comprehensive usage analytics tracked by the GraphUsage model.
 
   Provides temporal usage patterns including:
   - **Storage Analytics**: GB-hours for billing, breakdown by type (files, tables, graphs, subgraphs)
@@ -316,7 +316,7 @@ async def asyncio(
 ) -> Optional[Union[ErrorResponse, GraphUsageResponse, HTTPValidationError]]:
   """Get Graph Usage Analytics
 
-   Get comprehensive usage analytics tracked by the GraphUsageTracking model.
+   Get comprehensive usage analytics tracked by the GraphUsage model.
 
   Provides temporal usage patterns including:
   - **Storage Analytics**: GB-hours for billing, breakdown by type (files, tables, graphs, subgraphs)

robosystems_client/models/__init__.py

@@ -24,6 +24,7 @@ from .agent_response_tokens_used_type_0 import AgentResponseTokensUsedType0
 from .api_key_info import APIKeyInfo
 from .api_keys_response import APIKeysResponse
 from .auth_response import AuthResponse
+from .auth_response_org_type_0 import AuthResponseOrgType0
 from .auth_response_user import AuthResponseUser
 from .available_extension import AvailableExtension
 from .available_extensions_response import AvailableExtensionsResponse
@@ -65,6 +66,7 @@ from .create_checkout_request_resource_config import CreateCheckoutRequestResour
 from .create_connection_request import CreateConnectionRequest
 from .create_connection_request_provider import CreateConnectionRequestProvider
 from .create_graph_request import CreateGraphRequest
+from .create_org_request import CreateOrgRequest
 from .create_repository_subscription_request import CreateRepositorySubscriptionRequest
 from .create_subgraph_request import CreateSubgraphRequest
 from .create_subgraph_request_metadata_type_0 import CreateSubgraphRequestMetadataType0
@@ -144,6 +146,7 @@ from .health_status import HealthStatus
 from .health_status_details_type_0 import HealthStatusDetailsType0
 from .http_validation_error import HTTPValidationError
 from .initial_entity_data import InitialEntityData
+from .invite_member_request import InviteMemberRequest
 from .invoice import Invoice
 from .invoice_line_item import InvoiceLineItem
 from .invoices_response import InvoicesResponse
@@ -151,6 +154,7 @@ from .link_token_request import LinkTokenRequest
 from .link_token_request_options_type_0 import LinkTokenRequestOptionsType0
 from .link_token_request_provider_type_0 import LinkTokenRequestProviderType0
 from .list_connections_provider_type_0 import ListConnectionsProviderType0
+from .list_org_graphs_response_200_item import ListOrgGraphsResponse200Item
 from .list_subgraphs_response import ListSubgraphsResponse
 from .list_table_files_response import ListTableFilesResponse
 from .login_request import LoginRequest
@@ -172,6 +176,22 @@ from .offering_repository_plan_rate_limits_type_0 import (
 from .operation_costs import OperationCosts
 from .operation_costs_ai_operations import OperationCostsAiOperations
 from .operation_costs_token_pricing import OperationCostsTokenPricing
+from .org_detail_response import OrgDetailResponse
+from .org_detail_response_graphs_item import OrgDetailResponseGraphsItem
+from .org_detail_response_limits_type_0 import OrgDetailResponseLimitsType0
+from .org_detail_response_members_item import OrgDetailResponseMembersItem
+from .org_limits_response import OrgLimitsResponse
+from .org_limits_response_current_usage import OrgLimitsResponseCurrentUsage
+from .org_list_response import OrgListResponse
+from .org_member_list_response import OrgMemberListResponse
+from .org_member_response import OrgMemberResponse
+from .org_response import OrgResponse
+from .org_role import OrgRole
+from .org_type import OrgType
+from .org_usage_response import OrgUsageResponse
+from .org_usage_response_daily_trend_item import OrgUsageResponseDailyTrendItem
+from .org_usage_response_graph_details_item import OrgUsageResponseGraphDetailsItem
+from .org_usage_summary import OrgUsageSummary
 from .password_check_request import PasswordCheckRequest
 from .password_check_response import PasswordCheckResponse
 from .password_check_response_character_types import PasswordCheckResponseCharacterTypes
@@ -256,16 +276,15 @@ from .update_api_key_request import UpdateAPIKeyRequest
 from .update_file_status_response_updatefilestatus import (
   UpdateFileStatusResponseUpdatefilestatus,
 )
+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
-from .user_limits_response import UserLimitsResponse
 from .user_response import UserResponse
-from .user_usage_response import UserUsageResponse
-from .user_usage_response_graphs import UserUsageResponseGraphs
 from .validation_error import ValidationError
 
 __all__ = (
@@ -289,6 +308,7 @@ __all__ = (
   "APIKeyInfo",
   "APIKeysResponse",
   "AuthResponse",
+  "AuthResponseOrgType0",
   "AuthResponseUser",
   "AvailableExtension",
   "AvailableExtensionsResponse",
@@ -326,6 +346,7 @@ __all__ = (
   "CreateConnectionRequest",
   "CreateConnectionRequestProvider",
   "CreateGraphRequest",
+  "CreateOrgRequest",
   "CreateRepositorySubscriptionRequest",
   "CreateSubgraphRequest",
   "CreateSubgraphRequestMetadataType0",
@@ -387,6 +408,7 @@ __all__ = (
   "HealthStatusDetailsType0",
   "HTTPValidationError",
   "InitialEntityData",
+  "InviteMemberRequest",
   "Invoice",
   "InvoiceLineItem",
   "InvoicesResponse",
@@ -394,6 +416,7 @@ __all__ = (
   "LinkTokenRequestOptionsType0",
   "LinkTokenRequestProviderType0",
   "ListConnectionsProviderType0",
+  "ListOrgGraphsResponse200Item",
   "ListSubgraphsResponse",
   "ListTableFilesResponse",
   "LoginRequest",
@@ -411,6 +434,22 @@ __all__ = (
   "OperationCosts",
   "OperationCostsAiOperations",
   "OperationCostsTokenPricing",
+  "OrgDetailResponse",
+  "OrgDetailResponseGraphsItem",
+  "OrgDetailResponseLimitsType0",
+  "OrgDetailResponseMembersItem",
+  "OrgLimitsResponse",
+  "OrgLimitsResponseCurrentUsage",
+  "OrgListResponse",
+  "OrgMemberListResponse",
+  "OrgMemberResponse",
+  "OrgResponse",
+  "OrgRole",
+  "OrgType",
+  "OrgUsageResponse",
+  "OrgUsageResponseDailyTrendItem",
+  "OrgUsageResponseGraphDetailsItem",
+  "OrgUsageSummary",
   "PasswordCheckRequest",
   "PasswordCheckResponse",
   "PasswordCheckResponseCharacterTypes",
@@ -477,15 +516,14 @@ __all__ = (
   "UpcomingInvoice",
   "UpdateAPIKeyRequest",
   "UpdateFileStatusResponseUpdatefilestatus",
+  "UpdateMemberRoleRequest",
+  "UpdateOrgRequest",
   "UpdatePasswordRequest",
   "UpdatePaymentMethodRequest",
   "UpdatePaymentMethodResponse",
   "UpdateUserRequest",
   "UpgradeSubscriptionRequest",
   "UserGraphsResponse",
-  "UserLimitsResponse",
   "UserResponse",
-  "UserUsageResponse",
-  "UserUsageResponseGraphs",
   "ValidationError",
 )

robosystems_client/models/auth_response.py

@@ -7,6 +7,7 @@ from attrs import field as _attrs_field
 from ..types import UNSET, Unset
 
 if TYPE_CHECKING:
+  from ..models.auth_response_org_type_0 import AuthResponseOrgType0
   from ..models.auth_response_user import AuthResponseUser
 
 
@@ -20,6 +21,8 @@ class AuthResponse:
   Attributes:
       user (AuthResponseUser): User information
       message (str): Success message
+      org (Union['AuthResponseOrgType0', None, Unset]): Organization information (personal org created automatically
+          on registration)
       token (Union[None, Unset, str]): JWT authentication token (optional for cookie-based auth)
       expires_in (Union[None, Unset, int]): Token expiry time in seconds from now
       refresh_threshold (Union[None, Unset, int]): Recommended refresh threshold in seconds before expiry
@@ -27,16 +30,27 @@ class AuthResponse:
 
   user: "AuthResponseUser"
   message: str
+  org: Union["AuthResponseOrgType0", None, Unset] = UNSET
   token: Union[None, Unset, str] = UNSET
   expires_in: Union[None, Unset, int] = UNSET
   refresh_threshold: Union[None, Unset, int] = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
+    from ..models.auth_response_org_type_0 import AuthResponseOrgType0
+
     user = self.user.to_dict()
 
     message = self.message
 
+    org: Union[None, Unset, dict[str, Any]]
+    if isinstance(self.org, Unset):
+      org = UNSET
+    elif isinstance(self.org, AuthResponseOrgType0):
+      org = self.org.to_dict()
+    else:
+      org = self.org
+
     token: Union[None, Unset, str]
     if isinstance(self.token, Unset):
       token = UNSET
@@ -63,6 +77,8 @@ class AuthResponse:
         "message": message,
       }
     )
+    if org is not UNSET:
+      field_dict["org"] = org
     if token is not UNSET:
       field_dict["token"] = token
     if expires_in is not UNSET:
@@ -74,6 +90,7 @@ class AuthResponse:
 
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.auth_response_org_type_0 import AuthResponseOrgType0
     from ..models.auth_response_user import AuthResponseUser
 
     d = dict(src_dict)
@@ -81,6 +98,23 @@ class AuthResponse:
 
     message = d.pop("message")
 
+    def _parse_org(data: object) -> Union["AuthResponseOrgType0", None, Unset]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      try:
+        if not isinstance(data, dict):
+          raise TypeError()
+        org_type_0 = AuthResponseOrgType0.from_dict(data)
+
+        return org_type_0
+      except:  # noqa: E722
+        pass
+      return cast(Union["AuthResponseOrgType0", None, Unset], data)
+
+    org = _parse_org(d.pop("org", UNSET))
+
     def _parse_token(data: object) -> Union[None, Unset, str]:
       if data is None:
         return data
@@ -111,6 +145,7 @@ class AuthResponse:
     auth_response = cls(
       user=user,
       message=message,
+      org=org,
       token=token,
       expires_in=expires_in,
       refresh_threshold=refresh_threshold,

robosystems_client/models/{user_usage_response_graphs.py → auth_response_org_type_0.py}

@@ -4,12 +4,12 @@ from typing import Any, TypeVar
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
-T = TypeVar("T", bound="UserUsageResponseGraphs")
+T = TypeVar("T", bound="AuthResponseOrgType0")
 
 
 @_attrs_define
-class UserUsageResponseGraphs:
-  """Graph usage statistics (current/limit/remaining)"""
+class AuthResponseOrgType0:
+  """ """
 
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -22,10 +22,10 @@ class UserUsageResponseGraphs:
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     d = dict(src_dict)
-    user_usage_response_graphs = cls()
+    auth_response_org_type_0 = cls()
 
-    user_usage_response_graphs.additional_properties = d
-    return user_usage_response_graphs
+    auth_response_org_type_0.additional_properties = d
+    return auth_response_org_type_0
 
   @property
   def additional_keys(self) -> list[str]:

robosystems_client/models/billing_customer.py

@@ -15,18 +15,18 @@ T = TypeVar("T", bound="BillingCustomer")
 
 @_attrs_define
 class BillingCustomer:
-  """Billing customer information.
+  """Billing customer information for an organization.
 
   Attributes:
-      user_id (str): User ID
-      has_payment_method (bool): Whether customer has a payment method on file
+      org_id (str): Organization ID
+      has_payment_method (bool): Whether organization has a payment method on file
       invoice_billing_enabled (bool): Whether invoice billing is enabled (enterprise customers)
       payment_methods (list['PaymentMethod']): List of payment methods on file
       created_at (str): Customer creation timestamp (ISO format)
       stripe_customer_id (Union[None, Unset, str]): Stripe customer ID if applicable
   """
 
-  user_id: str
+  org_id: str
   has_payment_method: bool
   invoice_billing_enabled: bool
   payment_methods: list["PaymentMethod"]
@@ -35,7 +35,7 @@ class BillingCustomer:
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
-    user_id = self.user_id
+    org_id = self.org_id
 
     has_payment_method = self.has_payment_method
 
@@ -58,7 +58,7 @@ class BillingCustomer:
     field_dict.update(self.additional_properties)
     field_dict.update(
       {
-        "user_id": user_id,
+        "org_id": org_id,
         "has_payment_method": has_payment_method,
         "invoice_billing_enabled": invoice_billing_enabled,
         "payment_methods": payment_methods,
@@ -75,7 +75,7 @@ class BillingCustomer:
     from ..models.payment_method import PaymentMethod
 
     d = dict(src_dict)
-    user_id = d.pop("user_id")
+    org_id = d.pop("org_id")
 
     has_payment_method = d.pop("has_payment_method")
 
@@ -100,7 +100,7 @@ class BillingCustomer:
     stripe_customer_id = _parse_stripe_customer_id(d.pop("stripe_customer_id", UNSET))
 
     billing_customer = cls(
-      user_id=user_id,
+      org_id=org_id,
       has_payment_method=has_payment_method,
       invoice_billing_enabled=invoice_billing_enabled,
       payment_methods=payment_methods,