robosystems-client 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

robosystems_client/api/graphs/create_graph.py

@@ -7,34 +7,18 @@ from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.create_graph_request import CreateGraphRequest
 from ...models.http_validation_error import HTTPValidationError
-from ...types import UNSET, Response, Unset
+from ...types import Response
 
 
 def _get_kwargs(
   *,
   body: CreateGraphRequest,
-  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/graphs",
-    "params": params,
   }
 
   _kwargs["json"] = body.to_dict()
@@ -78,8 +62,6 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
 
@@ -125,13 +107,12 @@ def sync_detailed(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
-          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
-          'instance_tier': 'kuzu-standard', 'metadata': {'description': 'Main production graph',
-          'graph_name': 'Production System', 'schema_extensions': ['roboledger']}, 'tags':
-          ['production', 'finance']}.
+          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Consulting LLC', 'uri':
+          'https://acmeconsulting.com'}, 'instance_tier': 'kuzu-standard', 'metadata':
+          {'description': 'Professional consulting services with full accounting integration',
+          'graph_name': 'Acme Consulting LLC', 'schema_extensions': ['roboledger']}, 'tags':
+          ['consulting', 'professional-services']}.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -143,8 +124,6 @@ def sync_detailed(
 
   kwargs = _get_kwargs(
     body=body,
-    token=token,
-    authorization=authorization,
   )
 
   response = client.get_httpx_client().request(
@@ -158,8 +137,6 @@ def sync(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
 
@@ -205,13 +182,12 @@ def sync(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
-          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
-          'instance_tier': 'kuzu-standard', 'metadata': {'description': 'Main production graph',
-          'graph_name': 'Production System', 'schema_extensions': ['roboledger']}, 'tags':
-          ['production', 'finance']}.
+          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Consulting LLC', 'uri':
+          'https://acmeconsulting.com'}, 'instance_tier': 'kuzu-standard', 'metadata':
+          {'description': 'Professional consulting services with full accounting integration',
+          'graph_name': 'Acme Consulting LLC', 'schema_extensions': ['roboledger']}, 'tags':
+          ['consulting', 'professional-services']}.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -224,8 +200,6 @@ def sync(
   return sync_detailed(
     client=client,
     body=body,
-    token=token,
-    authorization=authorization,
   ).parsed
 
 
@@ -233,8 +207,6 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Response[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
 
@@ -280,13 +252,12 @@ async def asyncio_detailed(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
-          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
-          'instance_tier': 'kuzu-standard', 'metadata': {'description': 'Main production graph',
-          'graph_name': 'Production System', 'schema_extensions': ['roboledger']}, 'tags':
-          ['production', 'finance']}.
+          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Consulting LLC', 'uri':
+          'https://acmeconsulting.com'}, 'instance_tier': 'kuzu-standard', 'metadata':
+          {'description': 'Professional consulting services with full accounting integration',
+          'graph_name': 'Acme Consulting LLC', 'schema_extensions': ['roboledger']}, 'tags':
+          ['consulting', 'professional-services']}.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -298,8 +269,6 @@ async def asyncio_detailed(
 
   kwargs = _get_kwargs(
     body=body,
-    token=token,
-    authorization=authorization,
   )
 
   response = await client.get_async_httpx_client().request(**kwargs)
@@ -311,8 +280,6 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   body: CreateGraphRequest,
-  token: Union[None, Unset, str] = UNSET,
-  authorization: Union[None, Unset, str] = UNSET,
 ) -> Optional[Union[Any, HTTPValidationError]]:
   """Create New Graph Database
 
@@ -358,13 +325,12 @@ async def asyncio(
   - `_links.status`: Point-in-time status check endpoint
 
   Args:
-      token (Union[None, Unset, str]): JWT token for SSE authentication
-      authorization (Union[None, Unset, str]):
       body (CreateGraphRequest): Request model for creating a new graph. Example:
-          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Corp', 'uri': 'https://acme.com'},
-          'instance_tier': 'kuzu-standard', 'metadata': {'description': 'Main production graph',
-          'graph_name': 'Production System', 'schema_extensions': ['roboledger']}, 'tags':
-          ['production', 'finance']}.
+          {'initial_entity': {'cik': '0001234567', 'name': 'Acme Consulting LLC', 'uri':
+          'https://acmeconsulting.com'}, 'instance_tier': 'kuzu-standard', 'metadata':
+          {'description': 'Professional consulting services with full accounting integration',
+          'graph_name': 'Acme Consulting LLC', 'schema_extensions': ['roboledger']}, 'tags':
+          ['consulting', 'professional-services']}.
 
   Raises:
       errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
@@ -378,7 +344,5 @@ async def asyncio(
     await asyncio_detailed(
       client=client,
       body=body,
-      token=token,
-      authorization=authorization,
     )
   ).parsed

robosystems_client/api/graphs/get_available_extensions.py

@@ -1,5 +1,5 @@
 from http import HTTPStatus
-from typing import Any, Optional, Union
+from typing import Any, Optional, Union, cast
 
 import httpx
 
@@ -20,12 +20,16 @@ def _get_kwargs() -> dict[str, Any]:
 
 def _parse_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[AvailableExtensionsResponse]:
+) -> Optional[Union[Any, AvailableExtensionsResponse]]:
   if response.status_code == 200:
     response_200 = AvailableExtensionsResponse.from_dict(response.json())
 
     return response_200
 
+  if response.status_code == 500:
+    response_500 = cast(Any, None)
+    return response_500
+
   if client.raise_on_unexpected_status:
     raise errors.UnexpectedStatus(response.status_code, response.content)
   else:
@@ -34,7 +38,7 @@ def _parse_response(
 
 def _build_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[AvailableExtensionsResponse]:
+) -> Response[Union[Any, AvailableExtensionsResponse]]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -46,17 +50,45 @@ def _build_response(
 def sync_detailed(
   *,
   client: AuthenticatedClient,
-) -> Response[AvailableExtensionsResponse]:
+) -> Response[Union[Any, AvailableExtensionsResponse]]:
   """Get Available Schema Extensions
 
-   List all available schema extensions for graph creation
+   List all available schema extensions for graph creation.
+
+  Schema extensions provide pre-built industry-specific data models that extend
+  the base graph schema with specialized nodes, relationships, and properties.
+
+  **Available Extensions:**
+  - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial
+  statements
+  - **RoboInvestor**: Investment portfolio management and tracking
+  - **RoboSCM**: Supply chain management and logistics
+  - **RoboFO**: Front office operations and CRM
+  - **RoboHRM**: Human resources management
+  - **RoboEPM**: Enterprise performance management
+  - **RoboReport**: Business intelligence and reporting
+
+  **Extension Information:**
+  Each extension includes:
+  - Display name and description
+  - Node and relationship counts
+  - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs)
+
+  **Use Cases:**
+  - Browse available extensions before creating a graph
+  - Understand extension capabilities and data models
+  - Plan graph schema based on business requirements
+  - Combine multiple extensions for comprehensive data modeling
+
+  **Note:**
+  Extension listing is included - no credit consumption required.
 
   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[AvailableExtensionsResponse]
+      Response[Union[Any, AvailableExtensionsResponse]]
   """
 
   kwargs = _get_kwargs()
@@ -71,17 +103,45 @@ def sync_detailed(
 def sync(
   *,
   client: AuthenticatedClient,
-) -> Optional[AvailableExtensionsResponse]:
+) -> Optional[Union[Any, AvailableExtensionsResponse]]:
   """Get Available Schema Extensions
 
-   List all available schema extensions for graph creation
+   List all available schema extensions for graph creation.
+
+  Schema extensions provide pre-built industry-specific data models that extend
+  the base graph schema with specialized nodes, relationships, and properties.
+
+  **Available Extensions:**
+  - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial
+  statements
+  - **RoboInvestor**: Investment portfolio management and tracking
+  - **RoboSCM**: Supply chain management and logistics
+  - **RoboFO**: Front office operations and CRM
+  - **RoboHRM**: Human resources management
+  - **RoboEPM**: Enterprise performance management
+  - **RoboReport**: Business intelligence and reporting
+
+  **Extension Information:**
+  Each extension includes:
+  - Display name and description
+  - Node and relationship counts
+  - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs)
+
+  **Use Cases:**
+  - Browse available extensions before creating a graph
+  - Understand extension capabilities and data models
+  - Plan graph schema based on business requirements
+  - Combine multiple extensions for comprehensive data modeling
+
+  **Note:**
+  Extension listing is included - no credit consumption required.
 
   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:
-      AvailableExtensionsResponse
+      Union[Any, AvailableExtensionsResponse]
   """
 
   return sync_detailed(
@@ -92,17 +152,45 @@ def sync(
 async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
-) -> Response[AvailableExtensionsResponse]:
+) -> Response[Union[Any, AvailableExtensionsResponse]]:
   """Get Available Schema Extensions
 
-   List all available schema extensions for graph creation
+   List all available schema extensions for graph creation.
+
+  Schema extensions provide pre-built industry-specific data models that extend
+  the base graph schema with specialized nodes, relationships, and properties.
+
+  **Available Extensions:**
+  - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial
+  statements
+  - **RoboInvestor**: Investment portfolio management and tracking
+  - **RoboSCM**: Supply chain management and logistics
+  - **RoboFO**: Front office operations and CRM
+  - **RoboHRM**: Human resources management
+  - **RoboEPM**: Enterprise performance management
+  - **RoboReport**: Business intelligence and reporting
+
+  **Extension Information:**
+  Each extension includes:
+  - Display name and description
+  - Node and relationship counts
+  - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs)
+
+  **Use Cases:**
+  - Browse available extensions before creating a graph
+  - Understand extension capabilities and data models
+  - Plan graph schema based on business requirements
+  - Combine multiple extensions for comprehensive data modeling
+
+  **Note:**
+  Extension listing is included - no credit consumption required.
 
   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[AvailableExtensionsResponse]
+      Response[Union[Any, AvailableExtensionsResponse]]
   """
 
   kwargs = _get_kwargs()
@@ -115,17 +203,45 @@ async def asyncio_detailed(
 async def asyncio(
   *,
   client: AuthenticatedClient,
-) -> Optional[AvailableExtensionsResponse]:
+) -> Optional[Union[Any, AvailableExtensionsResponse]]:
   """Get Available Schema Extensions
 
-   List all available schema extensions for graph creation
+   List all available schema extensions for graph creation.
+
+  Schema extensions provide pre-built industry-specific data models that extend
+  the base graph schema with specialized nodes, relationships, and properties.
+
+  **Available Extensions:**
+  - **RoboLedger**: Complete accounting system with XBRL reporting, general ledger, and financial
+  statements
+  - **RoboInvestor**: Investment portfolio management and tracking
+  - **RoboSCM**: Supply chain management and logistics
+  - **RoboFO**: Front office operations and CRM
+  - **RoboHRM**: Human resources management
+  - **RoboEPM**: Enterprise performance management
+  - **RoboReport**: Business intelligence and reporting
+
+  **Extension Information:**
+  Each extension includes:
+  - Display name and description
+  - Node and relationship counts
+  - Context-aware capabilities (e.g., SEC repositories get different features than entity graphs)
+
+  **Use Cases:**
+  - Browse available extensions before creating a graph
+  - Understand extension capabilities and data models
+  - Plan graph schema based on business requirements
+  - Combine multiple extensions for comprehensive data modeling
+
+  **Note:**
+  Extension listing is included - no credit consumption required.
 
   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:
-      AvailableExtensionsResponse
+      Union[Any, AvailableExtensionsResponse]
   """
 
   return (