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

robosystems_client/api/query/execute_cypher_query.py

@@ -6,6 +6,7 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.cypher_query_request import CypherQueryRequest
+from ...models.execute_cypher_query_response_200 import ExecuteCypherQueryResponse200
 from ...models.http_validation_error import HTTPValidationError
 from ...models.response_mode import ResponseMode
 from ...types import UNSET, Response, Unset
@@ -59,7 +60,7 @@ def _get_kwargs(
 
 def _parse_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, HTTPValidationError]]:
+) -> Optional[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]:
   if response.status_code == 200:
     content_type = response.headers.get("content-type", "")
     if (
@@ -67,7 +68,8 @@ def _parse_response(
       or response.headers.get("x-stream-format") == "ndjson"
     ):
       return None
-    response_200 = response.json()
+    response_200 = ExecuteCypherQueryResponse200.from_dict(response.json())
+
     return response_200
 
   if response.status_code == 202:
@@ -111,7 +113,7 @@ def _parse_response(
 
 def _build_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, HTTPValidationError]]:
+) -> Response[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -128,7 +130,7 @@ def sync_detailed(
   mode: Union[None, ResponseMode, Unset] = UNSET,
   chunk_size: Union[None, Unset, int] = UNSET,
   test_mode: Union[Unset, bool] = False,
-) -> Response[Union[Any, HTTPValidationError]]:
+) -> Response[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]:
   r"""Execute Cypher Query (Read-Only)
 
    Execute a read-only Cypher query with intelligent response optimization.
@@ -207,7 +209,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError]]
+      Response[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]
   """
 
   kwargs = _get_kwargs(
@@ -233,7 +235,7 @@ def sync(
   mode: Union[None, ResponseMode, Unset] = UNSET,
   chunk_size: Union[None, Unset, int] = UNSET,
   test_mode: Union[Unset, bool] = False,
-) -> Optional[Union[Any, HTTPValidationError]]:
+) -> Optional[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]:
   r"""Execute Cypher Query (Read-Only)
 
    Execute a read-only Cypher query with intelligent response optimization.
@@ -312,7 +314,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError]
+      Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]
   """
 
   return sync_detailed(
@@ -333,7 +335,7 @@ async def asyncio_detailed(
   mode: Union[None, ResponseMode, Unset] = UNSET,
   chunk_size: Union[None, Unset, int] = UNSET,
   test_mode: Union[Unset, bool] = False,
-) -> Response[Union[Any, HTTPValidationError]]:
+) -> Response[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]:
   r"""Execute Cypher Query (Read-Only)
 
    Execute a read-only Cypher query with intelligent response optimization.
@@ -412,7 +414,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError]]
+      Response[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]
   """
 
   kwargs = _get_kwargs(
@@ -436,7 +438,7 @@ async def asyncio(
   mode: Union[None, ResponseMode, Unset] = UNSET,
   chunk_size: Union[None, Unset, int] = UNSET,
   test_mode: Union[Unset, bool] = False,
-) -> Optional[Union[Any, HTTPValidationError]]:
+) -> Optional[Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]]:
   r"""Execute Cypher Query (Read-Only)
 
    Execute a read-only Cypher query with intelligent response optimization.
@@ -515,7 +517,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError]
+      Union[Any, ExecuteCypherQueryResponse200, HTTPValidationError]
   """
 
   return (

robosystems_client/models/__init__.py

@@ -29,6 +29,8 @@ from .auth_response_user import AuthResponseUser
 from .available_extension import AvailableExtension
 from .available_extensions_response import AvailableExtensionsResponse
 from .backup_create_request import BackupCreateRequest
+from .backup_download_url_response import BackupDownloadUrlResponse
+from .backup_limits import BackupLimits
 from .backup_list_response import BackupListResponse
 from .backup_response import BackupResponse
 from .backup_restore_request import BackupRestoreRequest
@@ -52,6 +54,7 @@ from .connection_provider_info_provider import ConnectionProviderInfoProvider
 from .connection_response import ConnectionResponse
 from .connection_response_metadata import ConnectionResponseMetadata
 from .connection_response_provider import ConnectionResponseProvider
+from .copy_operation_limits import CopyOperationLimits
 from .create_api_key_request import CreateAPIKeyRequest
 from .create_api_key_response import CreateAPIKeyResponse
 from .create_connection_request import CreateConnectionRequest
@@ -59,6 +62,7 @@ from .create_connection_request_provider import CreateConnectionRequestProvider
 from .create_graph_request import CreateGraphRequest
 from .create_subgraph_request import CreateSubgraphRequest
 from .create_subgraph_request_metadata_type_0 import CreateSubgraphRequestMetadataType0
+from .credit_limits import CreditLimits
 from .credit_summary import CreditSummary
 from .credit_summary_response import CreditSummaryResponse
 from .credits_summary_response import CreditsSummaryResponse
@@ -91,6 +95,10 @@ from .enhanced_credit_transaction_response_metadata import (
 from .error_response import ErrorResponse
 from .exchange_token_request import ExchangeTokenRequest
 from .exchange_token_request_metadata_type_0 import ExchangeTokenRequestMetadataType0
+from .execute_cypher_query_response_200 import ExecuteCypherQueryResponse200
+from .execute_cypher_query_response_200_data_item import (
+  ExecuteCypherQueryResponse200DataItem,
+)
 from .file_info import FileInfo
 from .file_status_update import FileStatusUpdate
 from .file_upload_request import FileUploadRequest
@@ -105,9 +113,6 @@ from .get_all_credit_summaries_response_getallcreditsummaries import (
 from .get_all_shared_repository_limits_response_getallsharedrepositorylimits import (
   GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits,
 )
-from .get_backup_download_url_response_getbackupdownloadurl import (
-  GetBackupDownloadUrlResponseGetbackupdownloadurl,
-)
 from .get_current_auth_user_response_getcurrentauthuser import (
   GetCurrentAuthUserResponseGetcurrentauthuser,
 )
@@ -118,9 +123,6 @@ from .get_file_info_response import GetFileInfoResponse
 from .get_graph_billing_history_response_getgraphbillinghistory import (
   GetGraphBillingHistoryResponseGetgraphbillinghistory,
 )
-from .get_graph_limits_response_getgraphlimits import (
-  GetGraphLimitsResponseGetgraphlimits,
-)
 from .get_graph_monthly_bill_response_getgraphmonthlybill import (
   GetGraphMonthlyBillResponseGetgraphmonthlybill,
 )
@@ -137,6 +139,7 @@ from .get_storage_usage_response_getstorageusage import (
   GetStorageUsageResponseGetstorageusage,
 )
 from .graph_info import GraphInfo
+from .graph_limits_response import GraphLimitsResponse
 from .graph_metadata import GraphMetadata
 from .graph_metrics_response import GraphMetricsResponse
 from .graph_metrics_response_estimated_size import GraphMetricsResponseEstimatedSize
@@ -183,7 +186,9 @@ from .plaid_connection_config_accounts_type_0_item import (
 from .plaid_connection_config_institution_type_0 import (
   PlaidConnectionConfigInstitutionType0,
 )
+from .query_limits import QueryLimits
 from .quick_books_connection_config import QuickBooksConnectionConfig
+from .rate_limits import RateLimits
 from .register_request import RegisterRequest
 from .repository_credits_response import RepositoryCreditsResponse
 from .repository_plan import RepositoryPlan
@@ -217,6 +222,7 @@ from .sso_exchange_request import SSOExchangeRequest
 from .sso_exchange_response import SSOExchangeResponse
 from .sso_token_response import SSOTokenResponse
 from .storage_limit_response import StorageLimitResponse
+from .storage_limits import StorageLimits
 from .subgraph_quota_response import SubgraphQuotaResponse
 from .subgraph_response import SubgraphResponse
 from .subgraph_response_metadata_type_0 import SubgraphResponseMetadataType0
@@ -295,6 +301,8 @@ __all__ = (
   "AvailableExtension",
   "AvailableExtensionsResponse",
   "BackupCreateRequest",
+  "BackupDownloadUrlResponse",
+  "BackupLimits",
   "BackupListResponse",
   "BackupResponse",
   "BackupRestoreRequest",
@@ -314,6 +322,7 @@ __all__ = (
   "ConnectionResponse",
   "ConnectionResponseMetadata",
   "ConnectionResponseProvider",
+  "CopyOperationLimits",
   "CreateAPIKeyRequest",
   "CreateAPIKeyResponse",
   "CreateConnectionRequest",
@@ -321,6 +330,7 @@ __all__ = (
   "CreateGraphRequest",
   "CreateSubgraphRequest",
   "CreateSubgraphRequestMetadataType0",
+  "CreditLimits",
   "CreditsSummaryResponse",
   "CreditsSummaryResponseCreditsByAddonType0Item",
   "CreditSummary",
@@ -345,6 +355,8 @@ __all__ = (
   "ErrorResponse",
   "ExchangeTokenRequest",
   "ExchangeTokenRequestMetadataType0",
+  "ExecuteCypherQueryResponse200",
+  "ExecuteCypherQueryResponse200DataItem",
   "FileInfo",
   "FileStatusUpdate",
   "FileUploadRequest",
@@ -353,18 +365,17 @@ __all__ = (
   "ForgotPasswordResponseForgotpassword",
   "GetAllCreditSummariesResponseGetallcreditsummaries",
   "GetAllSharedRepositoryLimitsResponseGetallsharedrepositorylimits",
-  "GetBackupDownloadUrlResponseGetbackupdownloadurl",
   "GetCurrentAuthUserResponseGetcurrentauthuser",
   "GetCurrentGraphBillResponseGetcurrentgraphbill",
   "GetFileInfoResponse",
   "GetGraphBillingHistoryResponseGetgraphbillinghistory",
-  "GetGraphLimitsResponseGetgraphlimits",
   "GetGraphMonthlyBillResponseGetgraphmonthlybill",
   "GetGraphUsageDetailsResponseGetgraphusagedetails",
   "GetOperationStatusResponseGetoperationstatus",
   "GetSharedRepositoryLimitsResponseGetsharedrepositorylimits",
   "GetStorageUsageResponseGetstorageusage",
   "GraphInfo",
+  "GraphLimitsResponse",
   "GraphMetadata",
   "GraphMetricsResponse",
   "GraphMetricsResponseEstimatedSize",
@@ -403,7 +414,9 @@ __all__ = (
   "PlaidConnectionConfig",
   "PlaidConnectionConfigAccountsType0Item",
   "PlaidConnectionConfigInstitutionType0",
+  "QueryLimits",
   "QuickBooksConnectionConfig",
+  "RateLimits",
   "RegisterRequest",
   "RepositoryCreditsResponse",
   "RepositoryPlan",
@@ -429,6 +442,7 @@ __all__ = (
   "SSOExchangeResponse",
   "SSOTokenResponse",
   "StorageLimitResponse",
+  "StorageLimits",
   "SubgraphQuotaResponse",
   "SubgraphResponse",
   "SubgraphResponseMetadataType0",

robosystems_client/models/agent_response.py

@@ -30,7 +30,7 @@ class AgentResponse:
       mode_used (AgentMode): Agent execution modes.
       metadata (Union['AgentResponseMetadataType0', None, Unset]): Response metadata including routing info
       tokens_used (Union['AgentResponseTokensUsedType0', None, Unset]): Token usage statistics
-      confidence_score (Union[None, Unset, float]): Confidence score of the response
+      confidence_score (Union[None, Unset, float]): Confidence score of the response (0.0-1.0 scale)
       operation_id (Union[None, Unset, str]): Operation ID for SSE monitoring
       is_partial (Union[Unset, bool]): Whether this is a partial response Default: False.
       error_details (Union['AgentResponseErrorDetailsType0', None, Unset]): Error details if any

robosystems_client/models/auth_response.py

@@ -21,11 +21,15 @@ class AuthResponse:
       user (AuthResponseUser): User information
       message (str): Success message
       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
   """
 
   user: "AuthResponseUser"
   message: str
   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]:
@@ -39,6 +43,18 @@ class AuthResponse:
     else:
       token = self.token
 
+    expires_in: Union[None, Unset, int]
+    if isinstance(self.expires_in, Unset):
+      expires_in = UNSET
+    else:
+      expires_in = self.expires_in
+
+    refresh_threshold: Union[None, Unset, int]
+    if isinstance(self.refresh_threshold, Unset):
+      refresh_threshold = UNSET
+    else:
+      refresh_threshold = self.refresh_threshold
+
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
     field_dict.update(
@@ -49,6 +65,10 @@ class AuthResponse:
     )
     if token is not UNSET:
       field_dict["token"] = token
+    if expires_in is not UNSET:
+      field_dict["expires_in"] = expires_in
+    if refresh_threshold is not UNSET:
+      field_dict["refresh_threshold"] = refresh_threshold
 
     return field_dict
 
@@ -70,10 +90,30 @@ class AuthResponse:
 
     token = _parse_token(d.pop("token", UNSET))
 
+    def _parse_expires_in(data: object) -> Union[None, Unset, int]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, int], data)
+
+    expires_in = _parse_expires_in(d.pop("expires_in", UNSET))
+
+    def _parse_refresh_threshold(data: object) -> Union[None, Unset, int]:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      return cast(Union[None, Unset, int], data)
+
+    refresh_threshold = _parse_refresh_threshold(d.pop("refresh_threshold", UNSET))
+
     auth_response = cls(
       user=user,
       message=message,
       token=token,
+      expires_in=expires_in,
+      refresh_threshold=refresh_threshold,
     )
 
     auth_response.additional_properties = d

robosystems_client/models/backup_download_url_response.py

@@ -0,0 +1,92 @@
+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="BackupDownloadUrlResponse")
+
+
+@_attrs_define
+class BackupDownloadUrlResponse:
+  """Response model for backup download URL generation.
+
+  Attributes:
+      download_url (str): Pre-signed S3 URL for downloading the backup file
+      expires_in (int): URL expiration time in seconds from now
+      expires_at (float): Unix timestamp when the URL expires
+      backup_id (str): Backup identifier
+      graph_id (str): Graph database identifier
+  """
+
+  download_url: str
+  expires_in: int
+  expires_at: float
+  backup_id: str
+  graph_id: str
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    download_url = self.download_url
+
+    expires_in = self.expires_in
+
+    expires_at = self.expires_at
+
+    backup_id = self.backup_id
+
+    graph_id = self.graph_id
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "download_url": download_url,
+        "expires_in": expires_in,
+        "expires_at": expires_at,
+        "backup_id": backup_id,
+        "graph_id": graph_id,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    download_url = d.pop("download_url")
+
+    expires_in = d.pop("expires_in")
+
+    expires_at = d.pop("expires_at")
+
+    backup_id = d.pop("backup_id")
+
+    graph_id = d.pop("graph_id")
+
+    backup_download_url_response = cls(
+      download_url=download_url,
+      expires_in=expires_in,
+      expires_at=expires_at,
+      backup_id=backup_id,
+      graph_id=graph_id,
+    )
+
+    backup_download_url_response.additional_properties = d
+    return backup_download_url_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

robosystems_client/models/backup_limits.py

@@ -0,0 +1,76 @@
+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="BackupLimits")
+
+
+@_attrs_define
+class BackupLimits:
+  """Backup operation limits.
+
+  Attributes:
+      max_backup_size_gb (float): Maximum backup size in GB
+      backup_retention_days (int): Backup retention period in days
+      max_backups_per_day (int): Maximum backups per day
+  """
+
+  max_backup_size_gb: float
+  backup_retention_days: int
+  max_backups_per_day: int
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    max_backup_size_gb = self.max_backup_size_gb
+
+    backup_retention_days = self.backup_retention_days
+
+    max_backups_per_day = self.max_backups_per_day
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "max_backup_size_gb": max_backup_size_gb,
+        "backup_retention_days": backup_retention_days,
+        "max_backups_per_day": max_backups_per_day,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    max_backup_size_gb = d.pop("max_backup_size_gb")
+
+    backup_retention_days = d.pop("backup_retention_days")
+
+    max_backups_per_day = d.pop("max_backups_per_day")
+
+    backup_limits = cls(
+      max_backup_size_gb=max_backup_size_gb,
+      backup_retention_days=backup_retention_days,
+      max_backups_per_day=max_backups_per_day,
+    )
+
+    backup_limits.additional_properties = d
+    return backup_limits
+
+  @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

robosystems_client/models/batch_agent_request.py

@@ -18,7 +18,7 @@ class BatchAgentRequest:
   """Request for batch processing multiple queries.
 
   Attributes:
-      queries (list['AgentRequest']): List of queries to process
+      queries (list['AgentRequest']): List of queries to process (max 10)
       parallel (Union[Unset, bool]): Process queries in parallel Default: False.
   """
 

robosystems_client/models/batch_agent_response.py

@@ -16,8 +16,8 @@ class BatchAgentResponse:
   """Response for batch processing.
 
   Attributes:
-      results (list['AgentResponse']): List of agent responses
-      total_execution_time (float): Total execution time
+      results (list['AgentResponse']): List of agent responses (includes successes and failures)
+      total_execution_time (float): Total execution time in seconds
       parallel_processed (bool): Whether queries were processed in parallel
   """
 

robosystems_client/models/copy_operation_limits.py

@@ -0,0 +1,100 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="CopyOperationLimits")
+
+
+@_attrs_define
+class CopyOperationLimits:
+  """Copy/ingestion operation limits.
+
+  Attributes:
+      max_file_size_gb (float): Maximum file size in GB
+      timeout_seconds (int): Operation timeout in seconds
+      concurrent_operations (int): Maximum concurrent operations
+      max_files_per_operation (int): Maximum files per operation
+      daily_copy_operations (int): Daily operation limit
+      supported_formats (list[str]): Supported file formats
+  """
+
+  max_file_size_gb: float
+  timeout_seconds: int
+  concurrent_operations: int
+  max_files_per_operation: int
+  daily_copy_operations: int
+  supported_formats: list[str]
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    max_file_size_gb = self.max_file_size_gb
+
+    timeout_seconds = self.timeout_seconds
+
+    concurrent_operations = self.concurrent_operations
+
+    max_files_per_operation = self.max_files_per_operation
+
+    daily_copy_operations = self.daily_copy_operations
+
+    supported_formats = self.supported_formats
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "max_file_size_gb": max_file_size_gb,
+        "timeout_seconds": timeout_seconds,
+        "concurrent_operations": concurrent_operations,
+        "max_files_per_operation": max_files_per_operation,
+        "daily_copy_operations": daily_copy_operations,
+        "supported_formats": supported_formats,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    max_file_size_gb = d.pop("max_file_size_gb")
+
+    timeout_seconds = d.pop("timeout_seconds")
+
+    concurrent_operations = d.pop("concurrent_operations")
+
+    max_files_per_operation = d.pop("max_files_per_operation")
+
+    daily_copy_operations = d.pop("daily_copy_operations")
+
+    supported_formats = cast(list[str], d.pop("supported_formats"))
+
+    copy_operation_limits = cls(
+      max_file_size_gb=max_file_size_gb,
+      timeout_seconds=timeout_seconds,
+      concurrent_operations=concurrent_operations,
+      max_files_per_operation=max_files_per_operation,
+      daily_copy_operations=daily_copy_operations,
+      supported_formats=supported_formats,
+    )
+
+    copy_operation_limits.additional_properties = d
+    return copy_operation_limits
+
+  @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

robosystems_client/models/create_graph_request.py

@@ -19,23 +19,22 @@ T = TypeVar("T", bound="CreateGraphRequest")
 class CreateGraphRequest:
   """Request model for creating a new graph.
 
-  Example:
-      {'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']}
-
-  Attributes:
-      metadata (GraphMetadata): Metadata for graph creation.
-      instance_tier (Union[Unset, str]): Instance tier: kuzu-standard, kuzu-large, kuzu-xlarge, neo4j-community-large,
-          neo4j-enterprise-xlarge Default: 'kuzu-standard'.
-      custom_schema (Union['CustomSchemaDefinition', None, Unset]): Custom schema definition to apply
-      initial_entity (Union['InitialEntityData', None, Unset]): Optional initial entity to create in the graph. If
-          provided, creates a entity-focused graph.
-      create_entity (Union[Unset, bool]): Whether to create the entity node and upload initial data. Only applies when
-          initial_entity is provided. Set to False to create graph without populating entity data (useful for file-based
-          ingestion workflows). Default: True.
-      tags (Union[Unset, list[str]]): Optional tags for organization
+  Use this to create either:
+  - **Entity graphs**: Standard graphs with entity schema and optional extensions
+  - **Custom graphs**: Generic graphs with fully custom schema definitions
+
+      Attributes:
+          metadata (GraphMetadata): Metadata for graph creation.
+          instance_tier (Union[Unset, str]): Instance tier: kuzu-standard, kuzu-large, kuzu-xlarge, neo4j-community-large,
+              neo4j-enterprise-xlarge Default: 'kuzu-standard'.
+          custom_schema (Union['CustomSchemaDefinition', None, Unset]): Custom schema definition to apply. If provided,
+              creates a generic custom graph. If omitted, creates an entity graph using schema_extensions.
+          initial_entity (Union['InitialEntityData', None, Unset]): Optional initial entity to create in the graph. If
+              provided with entity graph, populates the first entity node.
+          create_entity (Union[Unset, bool]): Whether to create the entity node and upload initial data. Only applies when
+              initial_entity is provided. Set to False to create graph without populating entity data (useful for file-based
+              ingestion workflows). Default: True.
+          tags (Union[Unset, list[str]]): Optional tags for organization
   """
 
   metadata: "GraphMetadata"