Enhance graph limits and materialization features with improved models (#59)

## Summary
This PR significantly enhances the robosystems client by expanding graph
limits functionality and improving the materialization system. The
changes introduce new content and quota management models while
refactoring existing structures for better API consistency.

## Key Accomplishments

### New Features
- **Content Limits Management**: Added comprehensive content limits
tracking and validation
- **Download Quota System**: Implemented download quota management with
detailed tracking capabilities
- **Enhanced Materialization**: Expanded materialization
request/response handling with improved limit checking
- **Graph Limits API**: Extended graph limits endpoint with additional
functionality

### Model Improvements
- Added `ContentLimits` model for comprehensive limit management
- Introduced `DownloadQuota` model for quota tracking and enforcement
- Enhanced `MaterializeResponse` with improved limit validation and type
checking
- Expanded `ValidationError` with better context handling
- Refactored `GraphSubscriptionTier` for improved tier management

### API Enhancements
- Updated graph tiers endpoint with improved tier availability logic
- Enhanced materialization endpoint with better request/response
handling
- Improved graph limits endpoint with additional metadata

## Breaking Changes
- Removed `CreditLimits` model in favor of more granular limit models
- Refactored file structure with model renames:
- `storage_info_overage_pricing.py` →
`materialize_response_limit_check_type_0.py`
  - `storage_info_included_per_tier.py` → `validation_error_context.py`
- Updated import paths and dependencies in `__init__.py`

## Testing Notes
- Verify all graph limits functionality works with new model structure
- Test materialization workflows with enhanced request/response handling
- Validate quota management and download limit enforcement
- Ensure backward compatibility where possible with existing
integrations

## Infrastructure Considerations
- Models have been restructured to improve maintainability and API
consistency
- Enhanced error handling and validation throughout the materialization
pipeline
- Improved type safety with better model definitions and validation
contexts

---
🤖 Generated with [Claude Code](https://claude.ai/code)

**Branch Info:**
- Source: `chore/update-packages`
- Target: `main`
- Type: feature

Co-Authored-By: Claude <noreply@anthropic.com>

Author: jfrench9

robosystems_client/api/graph_limits/get_graph_limits.py

@@ -82,6 +82,7 @@ def sync_detailed(
   - **Backup Limits**: Frequency, retention, size limits
   - **Rate Limits**: Requests per minute/hour based on tier
   - **Credit Limits**: AI operation credits (if applicable)
+  - **Content Limits**: Node, relationship, and row limits (if applicable)
 
   This unified endpoint provides all limits in one place for easier client integration.
 
@@ -125,6 +126,7 @@ def sync(
   - **Backup Limits**: Frequency, retention, size limits
   - **Rate Limits**: Requests per minute/hour based on tier
   - **Credit Limits**: AI operation credits (if applicable)
+  - **Content Limits**: Node, relationship, and row limits (if applicable)
 
   This unified endpoint provides all limits in one place for easier client integration.
 
@@ -163,6 +165,7 @@ async def asyncio_detailed(
   - **Backup Limits**: Frequency, retention, size limits
   - **Rate Limits**: Requests per minute/hour based on tier
   - **Credit Limits**: AI operation credits (if applicable)
+  - **Content Limits**: Node, relationship, and row limits (if applicable)
 
   This unified endpoint provides all limits in one place for easier client integration.
 
@@ -204,6 +207,7 @@ async def asyncio(
   - **Backup Limits**: Frequency, retention, size limits
   - **Rate Limits**: Requests per minute/hour based on tier
   - **Credit Limits**: AI operation credits (if applicable)
+  - **Content Limits**: Node, relationship, and row limits (if applicable)
 
   This unified endpoint provides all limits in one place for easier client integration.
 

robosystems_client/api/graphs/get_available_graph_tiers.py

@@ -83,11 +83,9 @@ def sync_detailed(
   - Availability status
 
   **Available Tiers:**
-  - **ladybug-standard**: Multi-tenant entry-level tier
+  - **ladybug-standard**: Dedicated entry-level tier
   - **ladybug-large**: Dedicated professional tier with subgraph support
   - **ladybug-xlarge**: Enterprise tier with maximum resources
-  - **neo4j-community-large**: Neo4j Community Edition (optional, if enabled)
-  - **neo4j-enterprise-xlarge**: Neo4j Enterprise Edition (optional, if enabled)
 
   **Use Cases:**
   - Display tier options in graph creation UI
@@ -140,11 +138,9 @@ def sync(
   - Availability status
 
   **Available Tiers:**
-  - **ladybug-standard**: Multi-tenant entry-level tier
+  - **ladybug-standard**: Dedicated entry-level tier
   - **ladybug-large**: Dedicated professional tier with subgraph support
   - **ladybug-xlarge**: Enterprise tier with maximum resources
-  - **neo4j-community-large**: Neo4j Community Edition (optional, if enabled)
-  - **neo4j-enterprise-xlarge**: Neo4j Enterprise Edition (optional, if enabled)
 
   **Use Cases:**
   - Display tier options in graph creation UI
@@ -192,11 +188,9 @@ async def asyncio_detailed(
   - Availability status
 
   **Available Tiers:**
-  - **ladybug-standard**: Multi-tenant entry-level tier
+  - **ladybug-standard**: Dedicated entry-level tier
   - **ladybug-large**: Dedicated professional tier with subgraph support
   - **ladybug-xlarge**: Enterprise tier with maximum resources
-  - **neo4j-community-large**: Neo4j Community Edition (optional, if enabled)
-  - **neo4j-enterprise-xlarge**: Neo4j Enterprise Edition (optional, if enabled)
 
   **Use Cases:**
   - Display tier options in graph creation UI
@@ -247,11 +241,9 @@ async def asyncio(
   - Availability status
 
   **Available Tiers:**
-  - **ladybug-standard**: Multi-tenant entry-level tier
+  - **ladybug-standard**: Dedicated entry-level tier
   - **ladybug-large**: Dedicated professional tier with subgraph support
   - **ladybug-xlarge**: Enterprise tier with maximum resources
-  - **neo4j-community-large**: Neo4j Community Edition (optional, if enabled)
-  - **neo4j-enterprise-xlarge**: Neo4j Enterprise Edition (optional, if enabled)
 
   **Use Cases:**
   - Display tier options in graph creation UI

robosystems_client/api/materialize/materialize_graph.py

@@ -67,6 +67,11 @@ def _parse_response(
 
     return response_409
 
+  if response.status_code == 413:
+    response_413 = ErrorResponse.from_dict(response.json())
+
+    return response_413
+
   if response.status_code == 422:
     response_422 = HTTPValidationError.from_dict(response.json())
 
@@ -150,6 +155,10 @@ def sync_detailed(
   Full graph materialization can take minutes for large datasets. Consider running
   during off-peak hours for production systems.
 
+  **Dry Run:**
+  Set `dry_run=true` to validate limits without executing. Returns current usage, tier limits,
+  and any warnings or errors. No lock is acquired, no SSE operation is created.
+
   **Credits:**
   Materialization is included - no credit consumption
 
@@ -234,6 +243,10 @@ def sync(
   Full graph materialization can take minutes for large datasets. Consider running
   during off-peak hours for production systems.
 
+  **Dry Run:**
+  Set `dry_run=true` to validate limits without executing. Returns current usage, tier limits,
+  and any warnings or errors. No lock is acquired, no SSE operation is created.
+
   **Credits:**
   Materialization is included - no credit consumption
 
@@ -313,6 +326,10 @@ async def asyncio_detailed(
   Full graph materialization can take minutes for large datasets. Consider running
   during off-peak hours for production systems.
 
+  **Dry Run:**
+  Set `dry_run=true` to validate limits without executing. Returns current usage, tier limits,
+  and any warnings or errors. No lock is acquired, no SSE operation is created.
+
   **Credits:**
   Materialization is included - no credit consumption
 
@@ -395,6 +412,10 @@ async def asyncio(
   Full graph materialization can take minutes for large datasets. Consider running
   during off-peak hours for production systems.
 
+  **Dry Run:**
+  Set `dry_run=true` to validate limits without executing. Returns current usage, tier limits,
+  and any warnings or errors. No lock is acquired, no SSE operation is created.
+
   **Credits:**
   Materialization is included - no credit consumption
 

robosystems_client/models/__init__.py

@@ -55,6 +55,7 @@
 from .connection_response import ConnectionResponse
 from .connection_response_metadata import ConnectionResponseMetadata
 from .connection_response_provider import ConnectionResponseProvider
+from .content_limits import ContentLimits
 from .copy_operation_limits import CopyOperationLimits
 from .create_api_key_request import CreateAPIKeyRequest
 from .create_api_key_response import CreateAPIKeyResponse
@@ -90,6 +91,7 @@
   DetailedTransactionsResponseDateRange,
 )
 from .detailed_transactions_response_summary import DetailedTransactionsResponseSummary
+from .download_quota import DownloadQuota
 from .email_verification_request import EmailVerificationRequest
 from .enhanced_credit_transaction_response import EnhancedCreditTransactionResponse
 from .enhanced_credit_transaction_response_metadata import (
@@ -162,6 +164,7 @@
 from .logout_user_response_logoutuser import LogoutUserResponseLogoutuser
 from .materialize_request import MaterializeRequest
 from .materialize_response import MaterializeResponse
+from .materialize_response_limit_check_type_0 import MaterializeResponseLimitCheckType0
 from .materialize_status_response import MaterializeStatusResponse
 from .mcp_tool_call import MCPToolCall
 from .mcp_tool_call_arguments import MCPToolCallArguments
@@ -252,8 +255,6 @@
 from .sso_exchange_response import SSOExchangeResponse
 from .sso_token_response import SSOTokenResponse
 from .storage_info import StorageInfo
-from .storage_info_included_per_tier import StorageInfoIncludedPerTier
-from .storage_info_overage_pricing import StorageInfoOveragePricing
 from .storage_limit_response import StorageLimitResponse
 from .storage_limits import StorageLimits
 from .storage_summary import StorageSummary
@@ -289,6 +290,7 @@
 from .user_graphs_response import UserGraphsResponse
 from .user_response import UserResponse
 from .validation_error import ValidationError
+from .validation_error_context import ValidationErrorContext
 from .view_axis_config import ViewAxisConfig
 from .view_axis_config_element_labels_type_0 import ViewAxisConfigElementLabelsType0
 from .view_axis_config_member_labels_type_0 import ViewAxisConfigMemberLabelsType0
@@ -344,6 +346,7 @@
   "ConnectionResponse",
   "ConnectionResponseMetadata",
   "ConnectionResponseProvider",
+  "ContentLimits",
   "CopyOperationLimits",
   "CreateAPIKeyRequest",
   "CreateAPIKeyResponse",
@@ -375,6 +378,7 @@
   "DetailedTransactionsResponse",
   "DetailedTransactionsResponseDateRange",
   "DetailedTransactionsResponseSummary",
+  "DownloadQuota",
   "EmailVerificationRequest",
   "EnhancedCreditTransactionResponse",
   "EnhancedCreditTransactionResponseMetadata",
@@ -433,6 +437,7 @@
   "LogoutUserResponseLogoutuser",
   "MaterializeRequest",
   "MaterializeResponse",
+  "MaterializeResponseLimitCheckType0",
   "MaterializeStatusResponse",
   "MCPToolCall",
   "MCPToolCallArguments",
@@ -507,8 +512,6 @@
   "SSOExchangeResponse",
   "SSOTokenResponse",
   "StorageInfo",
-  "StorageInfoIncludedPerTier",
-  "StorageInfoOveragePricing",
   "StorageLimitResponse",
   "StorageLimits",
   "StorageSummary",
@@ -540,6 +543,7 @@
   "UserGraphsResponse",
   "UserResponse",
   "ValidationError",
+  "ValidationErrorContext",
   "ViewAxisConfig",
   "ViewAxisConfigElementLabelsType0",
   "ViewAxisConfigMemberLabelsType0",

robosystems_client/models/backup_list_response.py

@@ -1,13 +1,16 @@
 from __future__ import annotations
 
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, TypeVar
+from typing import TYPE_CHECKING, Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..types import UNSET, Unset
+
 if TYPE_CHECKING:
   from ..models.backup_response import BackupResponse
+  from ..models.download_quota import DownloadQuota
 
 
 T = TypeVar("T", bound="BackupListResponse")
@@ -21,14 +24,20 @@ class BackupListResponse:
       backups (list[BackupResponse]):
       total_count (int):
       graph_id (str):
+      is_shared_repository (bool | Unset): Whether this is a shared repository (limits apply) Default: False.
+      download_quota (DownloadQuota | None | Unset): Download quota for shared repositories
   """
 
   backups: list[BackupResponse]
   total_count: int
   graph_id: str
+  is_shared_repository: bool | Unset = False
+  download_quota: DownloadQuota | None | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
+    from ..models.download_quota import DownloadQuota
+
     backups = []
     for backups_item_data in self.backups:
       backups_item = backups_item_data.to_dict()
@@ -38,6 +47,16 @@ def to_dict(self) -> dict[str, Any]:
 
     graph_id = self.graph_id
 
+    is_shared_repository = self.is_shared_repository
+
+    download_quota: dict[str, Any] | None | Unset
+    if isinstance(self.download_quota, Unset):
+      download_quota = UNSET
+    elif isinstance(self.download_quota, DownloadQuota):
+      download_quota = self.download_quota.to_dict()
+    else:
+      download_quota = self.download_quota
+
     field_dict: dict[str, Any] = {}
     field_dict.update(self.additional_properties)
     field_dict.update(
@@ -47,12 +66,17 @@ def to_dict(self) -> dict[str, Any]:
         "graph_id": graph_id,
       }
     )
+    if is_shared_repository is not UNSET:
+      field_dict["is_shared_repository"] = is_shared_repository
+    if download_quota is not UNSET:
+      field_dict["download_quota"] = download_quota
 
     return field_dict
 
   @classmethod
   def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
     from ..models.backup_response import BackupResponse
+    from ..models.download_quota import DownloadQuota
 
     d = dict(src_dict)
     backups = []
@@ -66,10 +90,31 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
 
     graph_id = d.pop("graph_id")
 
+    is_shared_repository = d.pop("is_shared_repository", UNSET)
+
+    def _parse_download_quota(data: object) -> DownloadQuota | None | Unset:
+      if data is None:
+        return data
+      if isinstance(data, Unset):
+        return data
+      try:
+        if not isinstance(data, dict):
+          raise TypeError()
+        download_quota_type_0 = DownloadQuota.from_dict(data)
+
+        return download_quota_type_0
+      except (TypeError, ValueError, AttributeError, KeyError):
+        pass
+      return cast(DownloadQuota | None | Unset, data)
+
+    download_quota = _parse_download_quota(d.pop("download_quota", UNSET))
+
     backup_list_response = cls(
       backups=backups,
       total_count=total_count,
       graph_id=graph_id,
+      is_shared_repository=is_shared_repository,
+      download_quota=download_quota,
     )
 
     backup_list_response.additional_properties = d