robosystems-client 0.2.6__tar.gz → 0.2.8__tar.gz

{robosystems_client-0.2.6 → robosystems_client-0.2.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robosystems-client
-Version: 0.2.6
+Version: 0.2.8
 Summary: Python Client for RoboSystems financial graph database API
 Author: RFS LLC
 License: MIT

{robosystems_client-0.2.6 → robosystems_client-0.2.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "robosystems-client"
-version = "0.2.6"
+version = "0.2.8"
 description = "Python Client for RoboSystems financial graph database API"
 readme = "README.md"
 requires-python = ">=3.10"

{robosystems_client-0.2.6 → robosystems_client-0.2.8}/robosystems_client/api/graphs/get_available_graph_tiers.py

@@ -5,6 +5,7 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
+from ...models.available_graph_tiers_response import AvailableGraphTiersResponse
 from ...models.http_validation_error import HTTPValidationError
 from ...types import UNSET, Response, Unset
 
@@ -30,9 +31,10 @@ def _get_kwargs(
 
 def _parse_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, HTTPValidationError]]:
+) -> Optional[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]:
   if response.status_code == 200:
-    response_200 = response.json()
+    response_200 = AvailableGraphTiersResponse.from_dict(response.json())
+
     return response_200
 
   if response.status_code == 422:
@@ -52,7 +54,7 @@ def _parse_response(
 
 def _build_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, HTTPValidationError]]:
+) -> Response[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -65,7 +67,7 @@ def sync_detailed(
   *,
   client: AuthenticatedClient,
   include_disabled: Union[Unset, bool] = False,
-) -> Response[Union[Any, HTTPValidationError]]:
+) -> Response[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]:
   """Get Available Graph Tiers
 
    List all available graph database tier configurations.
@@ -104,7 +106,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError]]
+      Response[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]
   """
 
   kwargs = _get_kwargs(
@@ -122,7 +124,7 @@ def sync(
   *,
   client: AuthenticatedClient,
   include_disabled: Union[Unset, bool] = False,
-) -> Optional[Union[Any, HTTPValidationError]]:
+) -> Optional[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]:
   """Get Available Graph Tiers
 
    List all available graph database tier configurations.
@@ -161,7 +163,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError]
+      Union[Any, AvailableGraphTiersResponse, HTTPValidationError]
   """
 
   return sync_detailed(
@@ -174,7 +176,7 @@ async def asyncio_detailed(
   *,
   client: AuthenticatedClient,
   include_disabled: Union[Unset, bool] = False,
-) -> Response[Union[Any, HTTPValidationError]]:
+) -> Response[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]:
   """Get Available Graph Tiers
 
    List all available graph database tier configurations.
@@ -213,7 +215,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, HTTPValidationError]]
+      Response[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]
   """
 
   kwargs = _get_kwargs(
@@ -229,7 +231,7 @@ async def asyncio(
   *,
   client: AuthenticatedClient,
   include_disabled: Union[Unset, bool] = False,
-) -> Optional[Union[Any, HTTPValidationError]]:
+) -> Optional[Union[Any, AvailableGraphTiersResponse, HTTPValidationError]]:
   """Get Available Graph Tiers
 
    List all available graph database tier configurations.
@@ -268,7 +270,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, HTTPValidationError]
+      Union[Any, AvailableGraphTiersResponse, HTTPValidationError]
   """
 
   return (

{robosystems_client-0.2.6 → robosystems_client-0.2.8}/robosystems_client/api/service_offerings/get_service_offerings.py

@@ -6,6 +6,7 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.error_response import ErrorResponse
+from ...models.service_offerings_response import ServiceOfferingsResponse
 from ...types import Response
 
 
@@ -20,9 +21,10 @@ def _get_kwargs() -> dict[str, Any]:
 
 def _parse_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Optional[Union[Any, ErrorResponse]]:
+) -> Optional[Union[ErrorResponse, ServiceOfferingsResponse]]:
   if response.status_code == 200:
-    response_200 = response.json()
+    response_200 = ServiceOfferingsResponse.from_dict(response.json())
+
     return response_200
 
   if response.status_code == 500:
@@ -38,7 +40,7 @@ def _parse_response(
 
 def _build_response(
   *, client: Union[AuthenticatedClient, Client], response: httpx.Response
-) -> Response[Union[Any, ErrorResponse]]:
+) -> Response[Union[ErrorResponse, ServiceOfferingsResponse]]:
   return Response(
     status_code=HTTPStatus(response.status_code),
     content=response.content,
@@ -50,7 +52,7 @@ def _build_response(
 def sync_detailed(
   *,
   client: Union[AuthenticatedClient, Client],
-) -> Response[Union[Any, ErrorResponse]]:
+) -> Response[Union[ErrorResponse, ServiceOfferingsResponse]]:
   """Get Service Offerings
 
    Get comprehensive information about all subscription offerings.
@@ -75,7 +77,7 @@ def sync_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, ErrorResponse]]
+      Response[Union[ErrorResponse, ServiceOfferingsResponse]]
   """
 
   kwargs = _get_kwargs()
@@ -90,7 +92,7 @@ def sync_detailed(
 def sync(
   *,
   client: Union[AuthenticatedClient, Client],
-) -> Optional[Union[Any, ErrorResponse]]:
+) -> Optional[Union[ErrorResponse, ServiceOfferingsResponse]]:
   """Get Service Offerings
 
    Get comprehensive information about all subscription offerings.
@@ -115,7 +117,7 @@ def sync(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, ErrorResponse]
+      Union[ErrorResponse, ServiceOfferingsResponse]
   """
 
   return sync_detailed(
@@ -126,7 +128,7 @@ def sync(
 async def asyncio_detailed(
   *,
   client: Union[AuthenticatedClient, Client],
-) -> Response[Union[Any, ErrorResponse]]:
+) -> Response[Union[ErrorResponse, ServiceOfferingsResponse]]:
   """Get Service Offerings
 
    Get comprehensive information about all subscription offerings.
@@ -151,7 +153,7 @@ async def asyncio_detailed(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Response[Union[Any, ErrorResponse]]
+      Response[Union[ErrorResponse, ServiceOfferingsResponse]]
   """
 
   kwargs = _get_kwargs()
@@ -164,7 +166,7 @@ async def asyncio_detailed(
 async def asyncio(
   *,
   client: Union[AuthenticatedClient, Client],
-) -> Optional[Union[Any, ErrorResponse]]:
+) -> Optional[Union[ErrorResponse, ServiceOfferingsResponse]]:
   """Get Service Offerings
 
    Get comprehensive information about all subscription offerings.
@@ -189,7 +191,7 @@ async def asyncio(
       httpx.TimeoutException: If the request takes longer than Client.timeout.
 
   Returns:
-      Union[Any, ErrorResponse]
+      Union[ErrorResponse, ServiceOfferingsResponse]
   """
 
   return (

{robosystems_client-0.2.6 → robosystems_client-0.2.8}/robosystems_client/extensions/query_client.py

@@ -141,39 +141,100 @@ class QueryClient:
       if hasattr(response, "parsed") and response.parsed:
         response_data = response.parsed
 
-        # Handle both dict and object responses
+        # Handle both dict and attrs object responses
         if isinstance(response_data, dict):
           # Response is a plain dict
           data = response_data
+          has_data = "data" in data and "columns" in data
         else:
-          # Response is an object with additional_properties
-          data = (
-            response_data.additional_properties
-            if hasattr(response_data, "additional_properties")
-            else response_data
-          )
+          # Response is an attrs object - check for attributes directly
+          data = response_data
+          has_data = hasattr(data, "data") and hasattr(data, "columns")
 
         # Check if this is an immediate response
-        if "data" in data and "columns" in data:
+        if has_data:
+          # Extract data from either dict or attrs object
+          if isinstance(data, dict):
+            result_data = data["data"]
+            result_columns = data["columns"]
+            result_row_count = data.get("row_count", len(data["data"]))
+            result_execution_time = data.get("execution_time_ms", 0)
+            result_timestamp = data.get("timestamp", datetime.now().isoformat())
+          else:
+            # attrs object - access attributes directly
+            from ..types import UNSET
+
+            raw_data = data.data if data.data is not UNSET else []
+            # Convert data items to dicts if they're objects
+            result_data = []
+            for item in raw_data:
+              if hasattr(item, "to_dict"):
+                result_data.append(item.to_dict())
+              elif hasattr(item, "additional_properties"):
+                result_data.append(item.additional_properties)
+              else:
+                result_data.append(item)
+            result_columns = data.columns if data.columns is not UNSET else []
+            result_row_count = (
+              data.row_count if data.row_count is not UNSET else len(result_data)
+            )
+            result_execution_time = (
+              data.execution_time_ms if data.execution_time_ms is not UNSET else 0
+            )
+            result_timestamp = (
+              data.timestamp
+              if data.timestamp is not UNSET
+              else datetime.now().isoformat()
+            )
+
           return QueryResult(
-            data=data["data"],
-            columns=data["columns"],
-            row_count=data.get("row_count", len(data["data"])),
-            execution_time_ms=data.get("execution_time_ms", 0),
+            data=result_data,
+            columns=result_columns,
+            row_count=result_row_count,
+            execution_time_ms=result_execution_time,
             graph_id=graph_id,
-            timestamp=data.get("timestamp", datetime.now().isoformat()),
+            timestamp=result_timestamp,
           )
 
         # Check if this is a queued response
-        if data.get("status") == "queued" and "operation_id" in data:
-          queued_response = QueuedQueryResponse(
-            status=data["status"],
-            operation_id=data["operation_id"],
-            queue_position=data.get("queue_position", 0),
-            estimated_wait_seconds=data.get("estimated_wait_seconds", 0),
-            message=data.get("message", "Query queued"),
+        is_queued = False
+        queued_response = None
+
+        if isinstance(data, dict):
+          is_queued = data.get("status") == "queued" and "operation_id" in data
+          if is_queued:
+            queued_response = QueuedQueryResponse(
+              status=data["status"],
+              operation_id=data["operation_id"],
+              queue_position=data.get("queue_position", 0),
+              estimated_wait_seconds=data.get("estimated_wait_seconds", 0),
+              message=data.get("message", "Query queued"),
+            )
+        else:
+          is_queued = (
+            hasattr(data, "status")
+            and hasattr(data, "operation_id")
+            and getattr(data, "status", None) == "queued"
           )
+          if is_queued:
+            from ..types import UNSET
+
+            queued_response = QueuedQueryResponse(
+              status=data.status,
+              operation_id=data.operation_id,
+              queue_position=data.queue_position
+              if hasattr(data, "queue_position") and data.queue_position is not UNSET
+              else 0,
+              estimated_wait_seconds=data.estimated_wait_seconds
+              if hasattr(data, "estimated_wait_seconds")
+              and data.estimated_wait_seconds is not UNSET
+              else 0,
+              message=data.message
+              if hasattr(data, "message") and data.message is not UNSET
+              else "Query queued",
+            )
 
+        if is_queued and queued_response:
           # Notify about queue status
           if options.on_queue_update:
             options.on_queue_update(

{robosystems_client-0.2.6 → robosystems_client-0.2.8}/robosystems_client/models/__init__.py

@@ -28,6 +28,7 @@ from .auth_response import AuthResponse
 from .auth_response_user import AuthResponseUser
 from .available_extension import AvailableExtension
 from .available_extensions_response import AvailableExtensionsResponse
+from .available_graph_tiers_response import AvailableGraphTiersResponse
 from .backup_create_request import BackupCreateRequest
 from .backup_download_url_response import BackupDownloadUrlResponse
 from .backup_limits import BackupLimits
@@ -148,6 +149,13 @@ from .graph_metrics_response_node_counts import GraphMetricsResponseNodeCounts
 from .graph_metrics_response_relationship_counts import (
   GraphMetricsResponseRelationshipCounts,
 )
+from .graph_subscription_tier import GraphSubscriptionTier
+from .graph_subscriptions import GraphSubscriptions
+from .graph_tier_backup import GraphTierBackup
+from .graph_tier_copy_operations import GraphTierCopyOperations
+from .graph_tier_info import GraphTierInfo
+from .graph_tier_instance import GraphTierInstance
+from .graph_tier_limits import GraphTierLimits
 from .graph_usage_response import GraphUsageResponse
 from .graph_usage_response_query_statistics import GraphUsageResponseQueryStatistics
 from .graph_usage_response_recent_activity import GraphUsageResponseRecentActivity
@@ -174,6 +182,13 @@ from .o_auth_init_request_additional_params_type_0 import (
   OAuthInitRequestAdditionalParamsType0,
 )
 from .o_auth_init_response import OAuthInitResponse
+from .offering_repository_plan import OfferingRepositoryPlan
+from .offering_repository_plan_rate_limits_type_0 import (
+  OfferingRepositoryPlanRateLimitsType0,
+)
+from .operation_costs import OperationCosts
+from .operation_costs_ai_operations import OperationCostsAiOperations
+from .operation_costs_token_pricing import OperationCostsTokenPricing
 from .password_check_request import PasswordCheckRequest
 from .password_check_response import PasswordCheckResponse
 from .password_check_response_character_types import PasswordCheckResponseCharacterTypes
@@ -191,7 +206,9 @@ 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_info import RepositoryInfo
 from .repository_plan import RepositoryPlan
+from .repository_subscriptions import RepositorySubscriptions
 from .repository_type import RepositoryType
 from .resend_verification_email_response_resendverificationemail import (
   ResendVerificationEmailResponseResendverificationemail,
@@ -217,10 +234,15 @@ from .schema_validation_response_compatibility_type_0 import (
 from .schema_validation_response_stats_type_0 import SchemaValidationResponseStatsType0
 from .sec_connection_config import SECConnectionConfig
 from .selection_criteria import SelectionCriteria
+from .service_offering_summary import ServiceOfferingSummary
+from .service_offerings_response import ServiceOfferingsResponse
 from .sso_complete_request import SSOCompleteRequest
 from .sso_exchange_request import SSOExchangeRequest
 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 .subgraph_quota_response import SubgraphQuotaResponse
@@ -247,6 +269,7 @@ from .table_list_response import TableListResponse
 from .table_query_request import TableQueryRequest
 from .table_query_response import TableQueryResponse
 from .tier_upgrade_request import TierUpgradeRequest
+from .token_pricing import TokenPricing
 from .transaction_summary_response import TransactionSummaryResponse
 from .update_api_key_request import UpdateAPIKeyRequest
 from .update_file_status_response_updatefilestatus import (
@@ -300,6 +323,7 @@ __all__ = (
   "AuthResponseUser",
   "AvailableExtension",
   "AvailableExtensionsResponse",
+  "AvailableGraphTiersResponse",
   "BackupCreateRequest",
   "BackupDownloadUrlResponse",
   "BackupLimits",
@@ -382,6 +406,13 @@ __all__ = (
   "GraphMetricsResponseHealthStatus",
   "GraphMetricsResponseNodeCounts",
   "GraphMetricsResponseRelationshipCounts",
+  "GraphSubscriptions",
+  "GraphSubscriptionTier",
+  "GraphTierBackup",
+  "GraphTierCopyOperations",
+  "GraphTierInfo",
+  "GraphTierInstance",
+  "GraphTierLimits",
   "GraphUsageResponse",
   "GraphUsageResponseQueryStatistics",
   "GraphUsageResponseRecentActivity",
@@ -406,6 +437,11 @@ __all__ = (
   "OAuthInitRequest",
   "OAuthInitRequestAdditionalParamsType0",
   "OAuthInitResponse",
+  "OfferingRepositoryPlan",
+  "OfferingRepositoryPlanRateLimitsType0",
+  "OperationCosts",
+  "OperationCostsAiOperations",
+  "OperationCostsTokenPricing",
   "PasswordCheckRequest",
   "PasswordCheckResponse",
   "PasswordCheckResponseCharacterTypes",
@@ -419,7 +455,9 @@ __all__ = (
   "RateLimits",
   "RegisterRequest",
   "RepositoryCreditsResponse",
+  "RepositoryInfo",
   "RepositoryPlan",
+  "RepositorySubscriptions",
   "RepositoryType",
   "ResendVerificationEmailResponseResendverificationemail",
   "ResetPasswordRequest",
@@ -437,10 +475,15 @@ __all__ = (
   "SchemaValidationResponseStatsType0",
   "SECConnectionConfig",
   "SelectionCriteria",
+  "ServiceOfferingsResponse",
+  "ServiceOfferingSummary",
   "SSOCompleteRequest",
   "SSOExchangeRequest",
   "SSOExchangeResponse",
   "SSOTokenResponse",
+  "StorageInfo",
+  "StorageInfoIncludedPerTier",
+  "StorageInfoOveragePricing",
   "StorageLimitResponse",
   "StorageLimits",
   "SubgraphQuotaResponse",
@@ -463,6 +506,7 @@ __all__ = (
   "TableQueryRequest",
   "TableQueryResponse",
   "TierUpgradeRequest",
+  "TokenPricing",
   "TransactionSummaryResponse",
   "UpdateAPIKeyRequest",
   "UpdateFileStatusResponseUpdatefilestatus",

robosystems_client-0.2.8/robosystems_client/models/available_graph_tiers_response.py

@@ -0,0 +1,74 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+  from ..models.graph_tier_info import GraphTierInfo
+
+
+T = TypeVar("T", bound="AvailableGraphTiersResponse")
+
+
+@_attrs_define
+class AvailableGraphTiersResponse:
+  """Response containing available graph tiers.
+
+  Attributes:
+      tiers (list['GraphTierInfo']): List of available tiers
+  """
+
+  tiers: list["GraphTierInfo"]
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    tiers = []
+    for tiers_item_data in self.tiers:
+      tiers_item = tiers_item_data.to_dict()
+      tiers.append(tiers_item)
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "tiers": tiers,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    from ..models.graph_tier_info import GraphTierInfo
+
+    d = dict(src_dict)
+    tiers = []
+    _tiers = d.pop("tiers")
+    for tiers_item_data in _tiers:
+      tiers_item = GraphTierInfo.from_dict(tiers_item_data)
+
+      tiers.append(tiers_item)
+
+    available_graph_tiers_response = cls(
+      tiers=tiers,
+    )
+
+    available_graph_tiers_response.additional_properties = d
+    return available_graph_tiers_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