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

robosystems_client/api/billing/{update_org_payment_method.py → 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/api/credits_/list_credit_transactions.py

@@ -141,7 +141,7 @@ def sync_detailed(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,
@@ -206,7 +206,7 @@ def sync(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,
@@ -266,7 +266,7 @@ async def asyncio_detailed(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,
@@ -329,7 +329,7 @@ async def asyncio(
   No credits are consumed for viewing transaction history.
 
   Args:
-      graph_id (str):
+      graph_id (str): Graph database identifier
       transaction_type (Union[None, Unset, str]): Filter by transaction type (allocation,
           consumption, bonus, refund)
       operation_type (Union[None, Unset, str]): Filter by operation type (e.g., entity_lookup,

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/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/extensions/__init__.py

@@ -14,6 +14,14 @@ from .query_client import (
   QueryOptions,
   QueuedQueryError,
 )
+from .agent_client import (
+  AgentClient,
+  AgentResult,
+  QueuedAgentResponse,
+  AgentQueryRequest,
+  AgentOptions,
+  QueuedAgentError,
+)
 from .operation_client import (
   OperationClient,
   OperationStatus,
@@ -118,6 +126,13 @@ __all__ = [
   "QueryRequest",
   "QueryOptions",
   "QueuedQueryError",
+  # Agent Client
+  "AgentClient",
+  "AgentResult",
+  "QueuedAgentResponse",
+  "AgentQueryRequest",
+  "AgentOptions",
+  "QueuedAgentError",
   # Operation Client
   "OperationClient",
   "OperationStatus",
@@ -189,6 +204,16 @@ def stream_query(graph_id: str, query: str, parameters=None, chunk_size=None):
   return extensions.query.stream_query(graph_id, query, parameters, chunk_size)
 
 
+def agent_query(graph_id: str, message: str, context=None):
+  """Execute an agent query using the default extensions instance"""
+  return extensions.agent.query(graph_id, message, context)
+
+
+def analyze_financials(graph_id: str, message: str, on_progress=None):
+  """Execute financial agent using the default extensions instance"""
+  return extensions.agent.analyze_financials(graph_id, message, on_progress)
+
+
 # DataFrame convenience functions (if pandas is available)
 if (
   HAS_PANDAS