robosystems-client 0.1.9__py3-none-any.whl

robosystems_client/models/__init__.py

@@ -0,0 +1,379 @@
+"""Contains all the data models used in inputs/outputs"""
+
+from .account_info import AccountInfo
+from .add_on_credit_info import AddOnCreditInfo
+from .agent_message import AgentMessage
+from .agent_request import AgentRequest
+from .agent_request_context_type_0 import AgentRequestContextType0
+from .agent_response import AgentResponse
+from .agent_response_metadata_type_0 import AgentResponseMetadataType0
+from .api_key_info import APIKeyInfo
+from .api_keys_response import APIKeysResponse
+from .auth_response import AuthResponse
+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_export_request import BackupExportRequest
+from .backup_list_response import BackupListResponse
+from .backup_response import BackupResponse
+from .backup_restore_request import BackupRestoreRequest
+from .backup_stats_response import BackupStatsResponse
+from .backup_stats_response_backup_formats import BackupStatsResponseBackupFormats
+from .cancel_operation_response_canceloperation import (
+  CancelOperationResponseCanceloperation,
+)
+from .cancellation_response import CancellationResponse
+from .check_credit_balance_response_checkcreditbalance import (
+  CheckCreditBalanceResponseCheckcreditbalance,
+)
+from .connection_options_response import ConnectionOptionsResponse
+from .connection_provider_info import ConnectionProviderInfo
+from .connection_provider_info_auth_type import ConnectionProviderInfoAuthType
+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 .create_api_key_request import CreateAPIKeyRequest
+from .create_api_key_response import CreateAPIKeyResponse
+from .create_connection_request import CreateConnectionRequest
+from .create_connection_request_provider import CreateConnectionRequestProvider
+from .create_graph_request import CreateGraphRequest
+from .credit_check_request import CreditCheckRequest
+from .credit_summary import CreditSummary
+from .credit_summary_response import CreditSummaryResponse
+from .credits_summary_response import CreditsSummaryResponse
+from .credits_summary_response_credits_by_addon_item import (
+  CreditsSummaryResponseCreditsByAddonItem,
+)
+from .custom_schema_definition import CustomSchemaDefinition
+from .custom_schema_definition_metadata import CustomSchemaDefinitionMetadata
+from .custom_schema_definition_nodes_item import CustomSchemaDefinitionNodesItem
+from .custom_schema_definition_relationships_item import (
+  CustomSchemaDefinitionRelationshipsItem,
+)
+from .cypher_query_request import CypherQueryRequest
+from .cypher_query_request_parameters_type_0 import CypherQueryRequestParametersType0
+from .database_health_response import DatabaseHealthResponse
+from .database_info_response import DatabaseInfoResponse
+from .detailed_transactions_response import DetailedTransactionsResponse
+from .detailed_transactions_response_date_range import (
+  DetailedTransactionsResponseDateRange,
+)
+from .detailed_transactions_response_summary import DetailedTransactionsResponseSummary
+from .enhanced_credit_transaction_response import EnhancedCreditTransactionResponse
+from .enhanced_credit_transaction_response_metadata import (
+  EnhancedCreditTransactionResponseMetadata,
+)
+from .error_response import ErrorResponse
+from .exchange_token_request import ExchangeTokenRequest
+from .exchange_token_request_metadata_type_0 import ExchangeTokenRequestMetadataType0
+from .get_all_credit_summaries_response_getallcreditsummaries import (
+  GetAllCreditSummariesResponseGetallcreditsummaries,
+)
+from .get_backup_download_url_response_getbackupdownloadurl import (
+  GetBackupDownloadUrlResponseGetbackupdownloadurl,
+)
+from .get_current_auth_user_response_getcurrentauthuser import (
+  GetCurrentAuthUserResponseGetcurrentauthuser,
+)
+from .get_current_graph_bill_response_getcurrentgraphbill import (
+  GetCurrentGraphBillResponseGetcurrentgraphbill,
+)
+from .get_graph_billing_history_response_getgraphbillinghistory import (
+  GetGraphBillingHistoryResponseGetgraphbillinghistory,
+)
+from .get_graph_monthly_bill_response_getgraphmonthlybill import (
+  GetGraphMonthlyBillResponseGetgraphmonthlybill,
+)
+from .get_graph_schema_info_response_getgraphschemainfo import (
+  GetGraphSchemaInfoResponseGetgraphschemainfo,
+)
+from .get_graph_usage_details_response_getgraphusagedetails import (
+  GetGraphUsageDetailsResponseGetgraphusagedetails,
+)
+from .get_mcp_health_response_getmcphealth import GetMCPHealthResponseGetmcphealth
+from .get_operation_status_response_getoperationstatus import (
+  GetOperationStatusResponseGetoperationstatus,
+)
+from .get_storage_usage_response_getstorageusage import (
+  GetStorageUsageResponseGetstorageusage,
+)
+from .graph_info import GraphInfo
+from .graph_metadata import GraphMetadata
+from .graph_metrics_response import GraphMetricsResponse
+from .graph_metrics_response_estimated_size import GraphMetricsResponseEstimatedSize
+from .graph_metrics_response_health_status import GraphMetricsResponseHealthStatus
+from .graph_metrics_response_node_counts import GraphMetricsResponseNodeCounts
+from .graph_metrics_response_relationship_counts import (
+  GraphMetricsResponseRelationshipCounts,
+)
+from .graph_usage_response import GraphUsageResponse
+from .graph_usage_response_query_statistics import GraphUsageResponseQueryStatistics
+from .graph_usage_response_recent_activity import GraphUsageResponseRecentActivity
+from .graph_usage_response_storage_usage import GraphUsageResponseStorageUsage
+from .health_status import HealthStatus
+from .health_status_details_type_0 import HealthStatusDetailsType0
+from .http_validation_error import HTTPValidationError
+from .initial_entity_data import InitialEntityData
+from .kuzu_backup_health_response_kuzubackuphealth import (
+  KuzuBackupHealthResponseKuzubackuphealth,
+)
+from .link_token_request import LinkTokenRequest
+from .link_token_request_options_type_0 import LinkTokenRequestOptionsType0
+from .link_token_request_provider_type_0 import LinkTokenRequestProviderType0
+from .list_connections_provider_type_0 import ListConnectionsProviderType0
+from .list_schema_extensions_response_listschemaextensions import (
+  ListSchemaExtensionsResponseListschemaextensions,
+)
+from .login_request import LoginRequest
+from .logout_user_response_logoutuser import LogoutUserResponseLogoutuser
+from .mcp_tool_call import MCPToolCall
+from .mcp_tool_call_arguments import MCPToolCallArguments
+from .mcp_tools_response import MCPToolsResponse
+from .mcp_tools_response_tools_item import MCPToolsResponseToolsItem
+from .o_auth_callback_request import OAuthCallbackRequest
+from .o_auth_init_request import OAuthInitRequest
+from .o_auth_init_request_additional_params_type_0 import (
+  OAuthInitRequestAdditionalParamsType0,
+)
+from .o_auth_init_response import OAuthInitResponse
+from .password_check_request import PasswordCheckRequest
+from .password_check_response import PasswordCheckResponse
+from .password_check_response_character_types import PasswordCheckResponseCharacterTypes
+from .password_policy_response import PasswordPolicyResponse
+from .password_policy_response_policy import PasswordPolicyResponsePolicy
+from .plaid_connection_config import PlaidConnectionConfig
+from .plaid_connection_config_accounts_type_0_item import (
+  PlaidConnectionConfigAccountsType0Item,
+)
+from .plaid_connection_config_institution_type_0 import (
+  PlaidConnectionConfigInstitutionType0,
+)
+from .quick_books_connection_config import QuickBooksConnectionConfig
+from .register_request import RegisterRequest
+from .repository_credits_response import RepositoryCreditsResponse
+from .repository_plan import RepositoryPlan
+from .repository_type import RepositoryType
+from .response_mode import ResponseMode
+from .schema_export_response import SchemaExportResponse
+from .schema_export_response_data_stats_type_0 import SchemaExportResponseDataStatsType0
+from .schema_export_response_schema_definition_type_0 import (
+  SchemaExportResponseSchemaDefinitionType0,
+)
+from .schema_validation_request import SchemaValidationRequest
+from .schema_validation_request_schema_definition_type_0 import (
+  SchemaValidationRequestSchemaDefinitionType0,
+)
+from .schema_validation_response import SchemaValidationResponse
+from .schema_validation_response_compatibility_type_0 import (
+  SchemaValidationResponseCompatibilityType0,
+)
+from .schema_validation_response_stats_type_0 import SchemaValidationResponseStatsType0
+from .sec_connection_config import SECConnectionConfig
+from .sso_complete_request import SSOCompleteRequest
+from .sso_exchange_request import SSOExchangeRequest
+from .sso_exchange_response import SSOExchangeResponse
+from .sso_login_request import SSOLoginRequest
+from .sso_token_response import SSOTokenResponse
+from .storage_limit_response import StorageLimitResponse
+from .subscription_info import SubscriptionInfo
+from .subscription_info_metadata import SubscriptionInfoMetadata
+from .subscription_request import SubscriptionRequest
+from .subscription_response import SubscriptionResponse
+from .success_response import SuccessResponse
+from .success_response_data_type_0 import SuccessResponseDataType0
+from .sync_connection_request import SyncConnectionRequest
+from .sync_connection_request_sync_options_type_0 import (
+  SyncConnectionRequestSyncOptionsType0,
+)
+from .sync_connection_response_syncconnection import (
+  SyncConnectionResponseSyncconnection,
+)
+from .tier_upgrade_request import TierUpgradeRequest
+from .transaction_summary_response import TransactionSummaryResponse
+from .update_api_key_request import UpdateAPIKeyRequest
+from .update_password_request import UpdatePasswordRequest
+from .update_user_request import UpdateUserRequest
+from .upgrade_subscription_request import UpgradeSubscriptionRequest
+from .user_analytics_response import UserAnalyticsResponse
+from .user_analytics_response_api_usage import UserAnalyticsResponseApiUsage
+from .user_analytics_response_graph_usage import UserAnalyticsResponseGraphUsage
+from .user_analytics_response_limits import UserAnalyticsResponseLimits
+from .user_analytics_response_recent_activity_item import (
+  UserAnalyticsResponseRecentActivityItem,
+)
+from .user_analytics_response_user_info import UserAnalyticsResponseUserInfo
+from .user_graph_summary import UserGraphSummary
+from .user_graphs_response import UserGraphsResponse
+from .user_limits_response import UserLimitsResponse
+from .user_response import UserResponse
+from .user_subscriptions_response import UserSubscriptionsResponse
+from .user_usage_response import UserUsageResponse
+from .user_usage_response_graphs import UserUsageResponseGraphs
+from .user_usage_summary_response import UserUsageSummaryResponse
+from .user_usage_summary_response_usage_vs_limits import (
+  UserUsageSummaryResponseUsageVsLimits,
+)
+from .validation_error import ValidationError
+
+__all__ = (
+  "AccountInfo",
+  "AddOnCreditInfo",
+  "AgentMessage",
+  "AgentRequest",
+  "AgentRequestContextType0",
+  "AgentResponse",
+  "AgentResponseMetadataType0",
+  "APIKeyInfo",
+  "APIKeysResponse",
+  "AuthResponse",
+  "AuthResponseUser",
+  "AvailableExtension",
+  "AvailableExtensionsResponse",
+  "BackupCreateRequest",
+  "BackupExportRequest",
+  "BackupListResponse",
+  "BackupResponse",
+  "BackupRestoreRequest",
+  "BackupStatsResponse",
+  "BackupStatsResponseBackupFormats",
+  "CancellationResponse",
+  "CancelOperationResponseCanceloperation",
+  "CheckCreditBalanceResponseCheckcreditbalance",
+  "ConnectionOptionsResponse",
+  "ConnectionProviderInfo",
+  "ConnectionProviderInfoAuthType",
+  "ConnectionProviderInfoProvider",
+  "ConnectionResponse",
+  "ConnectionResponseMetadata",
+  "ConnectionResponseProvider",
+  "CreateAPIKeyRequest",
+  "CreateAPIKeyResponse",
+  "CreateConnectionRequest",
+  "CreateConnectionRequestProvider",
+  "CreateGraphRequest",
+  "CreditCheckRequest",
+  "CreditsSummaryResponse",
+  "CreditsSummaryResponseCreditsByAddonItem",
+  "CreditSummary",
+  "CreditSummaryResponse",
+  "CustomSchemaDefinition",
+  "CustomSchemaDefinitionMetadata",
+  "CustomSchemaDefinitionNodesItem",
+  "CustomSchemaDefinitionRelationshipsItem",
+  "CypherQueryRequest",
+  "CypherQueryRequestParametersType0",
+  "DatabaseHealthResponse",
+  "DatabaseInfoResponse",
+  "DetailedTransactionsResponse",
+  "DetailedTransactionsResponseDateRange",
+  "DetailedTransactionsResponseSummary",
+  "EnhancedCreditTransactionResponse",
+  "EnhancedCreditTransactionResponseMetadata",
+  "ErrorResponse",
+  "ExchangeTokenRequest",
+  "ExchangeTokenRequestMetadataType0",
+  "GetAllCreditSummariesResponseGetallcreditsummaries",
+  "GetBackupDownloadUrlResponseGetbackupdownloadurl",
+  "GetCurrentAuthUserResponseGetcurrentauthuser",
+  "GetCurrentGraphBillResponseGetcurrentgraphbill",
+  "GetGraphBillingHistoryResponseGetgraphbillinghistory",
+  "GetGraphMonthlyBillResponseGetgraphmonthlybill",
+  "GetGraphSchemaInfoResponseGetgraphschemainfo",
+  "GetGraphUsageDetailsResponseGetgraphusagedetails",
+  "GetMCPHealthResponseGetmcphealth",
+  "GetOperationStatusResponseGetoperationstatus",
+  "GetStorageUsageResponseGetstorageusage",
+  "GraphInfo",
+  "GraphMetadata",
+  "GraphMetricsResponse",
+  "GraphMetricsResponseEstimatedSize",
+  "GraphMetricsResponseHealthStatus",
+  "GraphMetricsResponseNodeCounts",
+  "GraphMetricsResponseRelationshipCounts",
+  "GraphUsageResponse",
+  "GraphUsageResponseQueryStatistics",
+  "GraphUsageResponseRecentActivity",
+  "GraphUsageResponseStorageUsage",
+  "HealthStatus",
+  "HealthStatusDetailsType0",
+  "HTTPValidationError",
+  "InitialEntityData",
+  "KuzuBackupHealthResponseKuzubackuphealth",
+  "LinkTokenRequest",
+  "LinkTokenRequestOptionsType0",
+  "LinkTokenRequestProviderType0",
+  "ListConnectionsProviderType0",
+  "ListSchemaExtensionsResponseListschemaextensions",
+  "LoginRequest",
+  "LogoutUserResponseLogoutuser",
+  "MCPToolCall",
+  "MCPToolCallArguments",
+  "MCPToolsResponse",
+  "MCPToolsResponseToolsItem",
+  "OAuthCallbackRequest",
+  "OAuthInitRequest",
+  "OAuthInitRequestAdditionalParamsType0",
+  "OAuthInitResponse",
+  "PasswordCheckRequest",
+  "PasswordCheckResponse",
+  "PasswordCheckResponseCharacterTypes",
+  "PasswordPolicyResponse",
+  "PasswordPolicyResponsePolicy",
+  "PlaidConnectionConfig",
+  "PlaidConnectionConfigAccountsType0Item",
+  "PlaidConnectionConfigInstitutionType0",
+  "QuickBooksConnectionConfig",
+  "RegisterRequest",
+  "RepositoryCreditsResponse",
+  "RepositoryPlan",
+  "RepositoryType",
+  "ResponseMode",
+  "SchemaExportResponse",
+  "SchemaExportResponseDataStatsType0",
+  "SchemaExportResponseSchemaDefinitionType0",
+  "SchemaValidationRequest",
+  "SchemaValidationRequestSchemaDefinitionType0",
+  "SchemaValidationResponse",
+  "SchemaValidationResponseCompatibilityType0",
+  "SchemaValidationResponseStatsType0",
+  "SECConnectionConfig",
+  "SSOCompleteRequest",
+  "SSOExchangeRequest",
+  "SSOExchangeResponse",
+  "SSOLoginRequest",
+  "SSOTokenResponse",
+  "StorageLimitResponse",
+  "SubscriptionInfo",
+  "SubscriptionInfoMetadata",
+  "SubscriptionRequest",
+  "SubscriptionResponse",
+  "SuccessResponse",
+  "SuccessResponseDataType0",
+  "SyncConnectionRequest",
+  "SyncConnectionRequestSyncOptionsType0",
+  "SyncConnectionResponseSyncconnection",
+  "TierUpgradeRequest",
+  "TransactionSummaryResponse",
+  "UpdateAPIKeyRequest",
+  "UpdatePasswordRequest",
+  "UpdateUserRequest",
+  "UpgradeSubscriptionRequest",
+  "UserAnalyticsResponse",
+  "UserAnalyticsResponseApiUsage",
+  "UserAnalyticsResponseGraphUsage",
+  "UserAnalyticsResponseLimits",
+  "UserAnalyticsResponseRecentActivityItem",
+  "UserAnalyticsResponseUserInfo",
+  "UserGraphsResponse",
+  "UserGraphSummary",
+  "UserLimitsResponse",
+  "UserResponse",
+  "UserSubscriptionsResponse",
+  "UserUsageResponse",
+  "UserUsageResponseGraphs",
+  "UserUsageSummaryResponse",
+  "UserUsageSummaryResponseUsageVsLimits",
+  "ValidationError",
+)

robosystems_client/models/account_info.py

@@ -0,0 +1,79 @@
+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="AccountInfo")
+
+
+@_attrs_define
+class AccountInfo:
+  """Provider account information.
+
+  Example:
+      {'provider': 'github', 'provider_account_id': '12345', 'provider_type': 'oauth'}
+
+  Attributes:
+      provider (str): Authentication provider ID (e.g., 'github', 'google')
+      provider_type (str): Type of provider
+      provider_account_id (str): Account ID at the provider
+  """
+
+  provider: str
+  provider_type: str
+  provider_account_id: str
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    provider = self.provider
+
+    provider_type = self.provider_type
+
+    provider_account_id = self.provider_account_id
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "provider": provider,
+        "provider_type": provider_type,
+        "provider_account_id": provider_account_id,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    provider = d.pop("provider")
+
+    provider_type = d.pop("provider_type")
+
+    provider_account_id = d.pop("provider_account_id")
+
+    account_info = cls(
+      provider=provider,
+      provider_type=provider_type,
+      provider_account_id=provider_account_id,
+    )
+
+    account_info.additional_properties = d
+    return account_info
+
+  @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/add_on_credit_info.py

@@ -0,0 +1,119 @@
+from collections.abc import Mapping
+from typing import Any, TypeVar, Union
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="AddOnCreditInfo")
+
+
+@_attrs_define
+class AddOnCreditInfo:
+  """Credit information for a specific add-on.
+
+  Attributes:
+      subscription_id (str): Subscription ID
+      addon_type (str): Add-on type (e.g., sec_data)
+      name (str): Display name of the add-on
+      tier (str): Subscription tier
+      credits_remaining (float): Credits remaining
+      credits_allocated (float): Monthly credit allocation
+      credits_consumed (float): Credits consumed this month
+      rollover_amount (Union[Unset, float]): Credits rolled over from previous month Default: 0.0.
+  """
+
+  subscription_id: str
+  addon_type: str
+  name: str
+  tier: str
+  credits_remaining: float
+  credits_allocated: float
+  credits_consumed: float
+  rollover_amount: Union[Unset, float] = 0.0
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    subscription_id = self.subscription_id
+
+    addon_type = self.addon_type
+
+    name = self.name
+
+    tier = self.tier
+
+    credits_remaining = self.credits_remaining
+
+    credits_allocated = self.credits_allocated
+
+    credits_consumed = self.credits_consumed
+
+    rollover_amount = self.rollover_amount
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "subscription_id": subscription_id,
+        "addon_type": addon_type,
+        "name": name,
+        "tier": tier,
+        "credits_remaining": credits_remaining,
+        "credits_allocated": credits_allocated,
+        "credits_consumed": credits_consumed,
+      }
+    )
+    if rollover_amount is not UNSET:
+      field_dict["rollover_amount"] = rollover_amount
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    subscription_id = d.pop("subscription_id")
+
+    addon_type = d.pop("addon_type")
+
+    name = d.pop("name")
+
+    tier = d.pop("tier")
+
+    credits_remaining = d.pop("credits_remaining")
+
+    credits_allocated = d.pop("credits_allocated")
+
+    credits_consumed = d.pop("credits_consumed")
+
+    rollover_amount = d.pop("rollover_amount", UNSET)
+
+    add_on_credit_info = cls(
+      subscription_id=subscription_id,
+      addon_type=addon_type,
+      name=name,
+      tier=tier,
+      credits_remaining=credits_remaining,
+      credits_allocated=credits_allocated,
+      credits_consumed=credits_consumed,
+      rollover_amount=rollover_amount,
+    )
+
+    add_on_credit_info.additional_properties = d
+    return add_on_credit_info
+
+  @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/agent_message.py

@@ -0,0 +1,68 @@
+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="AgentMessage")
+
+
+@_attrs_define
+class AgentMessage:
+  """Message in conversation history.
+
+  Attributes:
+      role (str): Message role (user/assistant)
+      content (str): Message content
+  """
+
+  role: str
+  content: str
+  additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+  def to_dict(self) -> dict[str, Any]:
+    role = self.role
+
+    content = self.content
+
+    field_dict: dict[str, Any] = {}
+    field_dict.update(self.additional_properties)
+    field_dict.update(
+      {
+        "role": role,
+        "content": content,
+      }
+    )
+
+    return field_dict
+
+  @classmethod
+  def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+    d = dict(src_dict)
+    role = d.pop("role")
+
+    content = d.pop("content")
+
+    agent_message = cls(
+      role=role,
+      content=content,
+    )
+
+    agent_message.additional_properties = d
+    return agent_message
+
+  @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