robosystems-client 0.2.15__py3-none-any.whl → 0.2.16__py3-none-any.whl

robosystems_client/models/__init__.py

@@ -40,8 +40,6 @@ from .backup_stats_response_backup_formats import BackupStatsResponseBackupForma
 from .batch_agent_request import BatchAgentRequest
 from .batch_agent_response import BatchAgentResponse
 from .billing_customer import BillingCustomer
-from .bulk_ingest_request import BulkIngestRequest
-from .bulk_ingest_response import BulkIngestResponse
 from .cancel_operation_response_canceloperation import (
   CancelOperationResponseCanceloperation,
 )
@@ -69,6 +67,7 @@ from .create_org_request import CreateOrgRequest
 from .create_repository_subscription_request import CreateRepositorySubscriptionRequest
 from .create_subgraph_request import CreateSubgraphRequest
 from .create_subgraph_request_metadata_type_0 import CreateSubgraphRequestMetadataType0
+from .create_view_request import CreateViewRequest
 from .credit_limits import CreditLimits
 from .credit_summary import CreditSummary
 from .credit_summary_operation_breakdown import CreditSummaryOperationBreakdown
@@ -96,6 +95,7 @@ from .enhanced_credit_transaction_response import EnhancedCreditTransactionRespo
 from .enhanced_credit_transaction_response_metadata import (
   EnhancedCreditTransactionResponseMetadata,
 )
+from .enhanced_file_status_layers import EnhancedFileStatusLayers
 from .error_response import ErrorResponse
 from .exchange_token_request import ExchangeTokenRequest
 from .exchange_token_request_metadata_type_0 import ExchangeTokenRequestMetadataType0
@@ -103,7 +103,9 @@ from .execute_cypher_query_response_200 import ExecuteCypherQueryResponse200
 from .execute_cypher_query_response_200_data_item import (
   ExecuteCypherQueryResponse200DataItem,
 )
+from .fact_detail import FactDetail
 from .file_info import FileInfo
+from .file_layer_status import FileLayerStatus
 from .file_status_update import FileStatusUpdate
 from .file_upload_request import FileUploadRequest
 from .file_upload_response import FileUploadResponse
@@ -158,6 +160,9 @@ from .list_subgraphs_response import ListSubgraphsResponse
 from .list_table_files_response import ListTableFilesResponse
 from .login_request import LoginRequest
 from .logout_user_response_logoutuser import LogoutUserResponseLogoutuser
+from .materialize_request import MaterializeRequest
+from .materialize_response import MaterializeResponse
+from .materialize_status_response import MaterializeStatusResponse
 from .mcp_tool_call import MCPToolCall
 from .mcp_tool_call_arguments import MCPToolCallArguments
 from .mcp_tools_response import MCPToolsResponse
@@ -220,6 +225,8 @@ from .resend_verification_email_response_resendverificationemail import (
 from .reset_password_request import ResetPasswordRequest
 from .reset_password_validate_response import ResetPasswordValidateResponse
 from .response_mode import ResponseMode
+from .save_view_request import SaveViewRequest
+from .save_view_response import SaveViewResponse
 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 (
@@ -250,6 +257,7 @@ from .storage_info_overage_pricing import StorageInfoOveragePricing
 from .storage_limit_response import StorageLimitResponse
 from .storage_limits import StorageLimits
 from .storage_summary import StorageSummary
+from .structure_detail import StructureDetail
 from .subgraph_quota_response import SubgraphQuotaResponse
 from .subgraph_response import SubgraphResponse
 from .subgraph_response_metadata_type_0 import SubgraphResponseMetadataType0
@@ -265,7 +273,6 @@ from .sync_connection_response_syncconnection import (
   SyncConnectionResponseSyncconnection,
 )
 from .table_info import TableInfo
-from .table_ingest_result import TableIngestResult
 from .table_list_response import TableListResponse
 from .table_query_request import TableQueryRequest
 from .table_query_response import TableQueryResponse
@@ -273,9 +280,7 @@ from .token_pricing import TokenPricing
 from .transaction_summary_response import TransactionSummaryResponse
 from .upcoming_invoice import UpcomingInvoice
 from .update_api_key_request import UpdateAPIKeyRequest
-from .update_file_status_response_updatefilestatus import (
-  UpdateFileStatusResponseUpdatefilestatus,
-)
+from .update_file_response_updatefile import UpdateFileResponseUpdatefile
 from .update_member_role_request import UpdateMemberRoleRequest
 from .update_org_request import UpdateOrgRequest
 from .update_password_request import UpdatePasswordRequest
@@ -284,6 +289,12 @@ from .upgrade_subscription_request import UpgradeSubscriptionRequest
 from .user_graphs_response import UserGraphsResponse
 from .user_response import UserResponse
 from .validation_error import ValidationError
+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
+from .view_config import ViewConfig
+from .view_source import ViewSource
+from .view_source_type import ViewSourceType
 
 __all__ = (
   "AccountInfo",
@@ -322,8 +333,6 @@ __all__ = (
   "BatchAgentRequest",
   "BatchAgentResponse",
   "BillingCustomer",
-  "BulkIngestRequest",
-  "BulkIngestResponse",
   "CancelOperationResponseCanceloperation",
   "CheckCreditBalanceResponseCheckcreditbalance",
   "CheckoutResponse",
@@ -347,6 +356,7 @@ __all__ = (
   "CreateRepositorySubscriptionRequest",
   "CreateSubgraphRequest",
   "CreateSubgraphRequestMetadataType0",
+  "CreateViewRequest",
   "CreditLimits",
   "CreditSummary",
   "CreditSummaryOperationBreakdown",
@@ -368,12 +378,15 @@ __all__ = (
   "EmailVerificationRequest",
   "EnhancedCreditTransactionResponse",
   "EnhancedCreditTransactionResponseMetadata",
+  "EnhancedFileStatusLayers",
   "ErrorResponse",
   "ExchangeTokenRequest",
   "ExchangeTokenRequestMetadataType0",
   "ExecuteCypherQueryResponse200",
   "ExecuteCypherQueryResponse200DataItem",
+  "FactDetail",
   "FileInfo",
+  "FileLayerStatus",
   "FileStatusUpdate",
   "FileUploadRequest",
   "FileUploadResponse",
@@ -418,6 +431,9 @@ __all__ = (
   "ListTableFilesResponse",
   "LoginRequest",
   "LogoutUserResponseLogoutuser",
+  "MaterializeRequest",
+  "MaterializeResponse",
+  "MaterializeStatusResponse",
   "MCPToolCall",
   "MCPToolCallArguments",
   "MCPToolsResponse",
@@ -470,6 +486,8 @@ __all__ = (
   "ResetPasswordRequest",
   "ResetPasswordValidateResponse",
   "ResponseMode",
+  "SaveViewRequest",
+  "SaveViewResponse",
   "SchemaExportResponse",
   "SchemaExportResponseDataStatsType0",
   "SchemaExportResponseSchemaDefinitionType0",
@@ -494,6 +512,7 @@ __all__ = (
   "StorageLimitResponse",
   "StorageLimits",
   "StorageSummary",
+  "StructureDetail",
   "SubgraphQuotaResponse",
   "SubgraphResponse",
   "SubgraphResponseMetadataType0",
@@ -505,7 +524,6 @@ __all__ = (
   "SyncConnectionRequestSyncOptionsType0",
   "SyncConnectionResponseSyncconnection",
   "TableInfo",
-  "TableIngestResult",
   "TableListResponse",
   "TableQueryRequest",
   "TableQueryResponse",
@@ -513,7 +531,7 @@ __all__ = (
   "TransactionSummaryResponse",
   "UpcomingInvoice",
   "UpdateAPIKeyRequest",
-  "UpdateFileStatusResponseUpdatefilestatus",
+  "UpdateFileResponseUpdatefile",
   "UpdateMemberRoleRequest",
   "UpdateOrgRequest",
   "UpdatePasswordRequest",
@@ -522,4 +540,10 @@ __all__ = (
   "UserGraphsResponse",
   "UserResponse",
   "ValidationError",
+  "ViewAxisConfig",
+  "ViewAxisConfigElementLabelsType0",
+  "ViewAxisConfigMemberLabelsType0",
+  "ViewConfig",
+  "ViewSource",
+  "ViewSourceType",
 )

robosystems_client/models/account_info.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/agent_list_response.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import TYPE_CHECKING, Any, TypeVar
 
@@ -20,7 +22,7 @@ class AgentListResponse:
       total (int): Total number of agents
   """
 
-  agents: "AgentListResponseAgents"
+  agents: AgentListResponseAgents
   total: int
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 

robosystems_client/models/agent_list_response_agents.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import TYPE_CHECKING, Any, TypeVar
 
@@ -17,7 +19,7 @@ T = TypeVar("T", bound="AgentListResponseAgents")
 class AgentListResponseAgents:
   """Dictionary of available agents with metadata"""
 
-  additional_properties: dict[str, "AgentListResponseAgentsAdditionalProperty"] = (
+  additional_properties: dict[str, AgentListResponseAgentsAdditionalProperty] = (
     _attrs_field(init=False, factory=dict)
   )
 
@@ -52,11 +54,11 @@ class AgentListResponseAgents:
   def additional_keys(self) -> list[str]:
     return list(self.additional_properties.keys())
 
-  def __getitem__(self, key: str) -> "AgentListResponseAgentsAdditionalProperty":
+  def __getitem__(self, key: str) -> AgentListResponseAgentsAdditionalProperty:
     return self.additional_properties[key]
 
   def __setitem__(
-    self, key: str, value: "AgentListResponseAgentsAdditionalProperty"
+    self, key: str, value: AgentListResponseAgentsAdditionalProperty
   ) -> None:
     self.additional_properties[key] = value
 

robosystems_client/models/agent_list_response_agents_additional_property.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/agent_message.py

@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 import datetime
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -18,12 +20,12 @@ class AgentMessage:
   Attributes:
       role (str): Message role (user/assistant)
       content (str): Message content
-      timestamp (Union[None, Unset, datetime.datetime]): Message timestamp
+      timestamp (datetime.datetime | None | Unset): Message timestamp
   """
 
   role: str
   content: str
-  timestamp: Union[None, Unset, datetime.datetime] = UNSET
+  timestamp: datetime.datetime | None | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -31,7 +33,7 @@ class AgentMessage:
 
     content = self.content
 
-    timestamp: Union[None, Unset, str]
+    timestamp: None | str | Unset
     if isinstance(self.timestamp, Unset):
       timestamp = UNSET
     elif isinstance(self.timestamp, datetime.datetime):
@@ -59,7 +61,7 @@ class AgentMessage:
 
     content = d.pop("content")
 
-    def _parse_timestamp(data: object) -> Union[None, Unset, datetime.datetime]:
+    def _parse_timestamp(data: object) -> datetime.datetime | None | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
@@ -70,9 +72,9 @@ class AgentMessage:
         timestamp_type_0 = isoparse(data)
 
         return timestamp_type_0
-      except:  # noqa: E722
+      except (TypeError, ValueError, AttributeError, KeyError):
         pass
-      return cast(Union[None, Unset, datetime.datetime], data)
+      return cast(datetime.datetime | None | Unset, data)
 
     timestamp = _parse_timestamp(d.pop("timestamp", UNSET))
 

robosystems_client/models/agent_metadata_response.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -20,8 +22,8 @@ class AgentMetadataResponse:
       capabilities (list[str]): Agent capabilities
       supported_modes (list[str]): Supported execution modes
       requires_credits (bool): Whether agent requires credits
-      author (Union[None, Unset, str]): Agent author
-      tags (Union[Unset, list[str]]): Agent tags
+      author (None | str | Unset): Agent author
+      tags (list[str] | Unset): Agent tags
   """
 
   name: str
@@ -30,8 +32,8 @@ class AgentMetadataResponse:
   capabilities: list[str]
   supported_modes: list[str]
   requires_credits: bool
-  author: Union[None, Unset, str] = UNSET
-  tags: Union[Unset, list[str]] = UNSET
+  author: None | str | Unset = UNSET
+  tags: list[str] | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -47,13 +49,13 @@ class AgentMetadataResponse:
 
     requires_credits = self.requires_credits
 
-    author: Union[None, Unset, str]
+    author: None | str | Unset
     if isinstance(self.author, Unset):
       author = UNSET
     else:
       author = self.author
 
-    tags: Union[Unset, list[str]] = UNSET
+    tags: list[str] | Unset = UNSET
     if not isinstance(self.tags, Unset):
       tags = self.tags
 
@@ -91,12 +93,12 @@ class AgentMetadataResponse:
 
     requires_credits = d.pop("requires_credits")
 
-    def _parse_author(data: object) -> Union[None, Unset, str]:
+    def _parse_author(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     author = _parse_author(d.pop("author", UNSET))
 

robosystems_client/models/agent_recommendation.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import Any, TypeVar, Union, cast
+from typing import Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -18,14 +20,14 @@ class AgentRecommendation:
       agent_name (str): Agent display name
       confidence (float): Confidence score (0-1)
       capabilities (list[str]): Agent capabilities
-      reason (Union[None, Unset, str]): Reason for recommendation
+      reason (None | str | Unset): Reason for recommendation
   """
 
   agent_type: str
   agent_name: str
   confidence: float
   capabilities: list[str]
-  reason: Union[None, Unset, str] = UNSET
+  reason: None | str | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -37,7 +39,7 @@ class AgentRecommendation:
 
     capabilities = self.capabilities
 
-    reason: Union[None, Unset, str]
+    reason: None | str | Unset
     if isinstance(self.reason, Unset):
       reason = UNSET
     else:
@@ -69,12 +71,12 @@ class AgentRecommendation:
 
     capabilities = cast(list[str], d.pop("capabilities"))
 
-    def _parse_reason(data: object) -> Union[None, Unset, str]:
+    def _parse_reason(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     reason = _parse_reason(d.pop("reason", UNSET))
 

robosystems_client/models/agent_recommendation_request.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -21,11 +23,11 @@ class AgentRecommendationRequest:
 
   Attributes:
       query (str): Query to analyze
-      context (Union['AgentRecommendationRequestContextType0', None, Unset]): Additional context
+      context (AgentRecommendationRequestContextType0 | None | Unset): Additional context
   """
 
   query: str
-  context: Union["AgentRecommendationRequestContextType0", None, Unset] = UNSET
+  context: AgentRecommendationRequestContextType0 | None | Unset = UNSET
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -35,7 +37,7 @@ class AgentRecommendationRequest:
 
     query = self.query
 
-    context: Union[None, Unset, dict[str, Any]]
+    context: dict[str, Any] | None | Unset
     if isinstance(self.context, Unset):
       context = UNSET
     elif isinstance(self.context, AgentRecommendationRequestContextType0):
@@ -66,7 +68,7 @@ class AgentRecommendationRequest:
 
     def _parse_context(
       data: object,
-    ) -> Union["AgentRecommendationRequestContextType0", None, Unset]:
+    ) -> AgentRecommendationRequestContextType0 | None | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
@@ -77,9 +79,9 @@ class AgentRecommendationRequest:
         context_type_0 = AgentRecommendationRequestContextType0.from_dict(data)
 
         return context_type_0
-      except:  # noqa: E722
+      except (TypeError, ValueError, AttributeError, KeyError):
         pass
-      return cast(Union["AgentRecommendationRequestContextType0", None, Unset], data)
+      return cast(AgentRecommendationRequestContextType0 | None | Unset, data)
 
     context = _parse_context(d.pop("context", UNSET))
 

robosystems_client/models/agent_recommendation_request_context_type_0.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar
 

robosystems_client/models/agent_recommendation_response.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import TYPE_CHECKING, Any, TypeVar
 
@@ -16,11 +18,11 @@ class AgentRecommendationResponse:
   """Response for agent recommendations.
 
   Attributes:
-      recommendations (list['AgentRecommendation']): List of agent recommendations sorted by confidence
+      recommendations (list[AgentRecommendation]): List of agent recommendations sorted by confidence
       query (str): The analyzed query
   """
 
-  recommendations: list["AgentRecommendation"]
+  recommendations: list[AgentRecommendation]
   query: str
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 

robosystems_client/models/agent_request.py

@@ -1,5 +1,7 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from typing import TYPE_CHECKING, Any, TypeVar, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -22,27 +24,26 @@ class AgentRequest:
 
   Attributes:
       message (str): The query or message to process
-      history (Union[Unset, list['AgentMessage']]): Conversation history
-      context (Union['AgentRequestContextType0', None, Unset]): Additional context for analysis (e.g., enable_rag,
+      history (list[AgentMessage] | Unset): Conversation history
+      context (AgentRequestContextType0 | None | Unset): Additional context for analysis (e.g., enable_rag,
           include_schema)
-      mode (Union[AgentMode, None, Unset]): Execution mode Default: AgentMode.STANDARD.
-      agent_type (Union[None, Unset, str]): Specific agent type to use (optional)
-      selection_criteria (Union['SelectionCriteria', None, Unset]): Criteria for agent selection
-      force_extended_analysis (Union[Unset, bool]): Force extended analysis mode with comprehensive research Default:
-          False.
-      enable_rag (Union[Unset, bool]): Enable RAG context enrichment Default: True.
-      stream (Union[Unset, bool]): Enable streaming response Default: False.
+      mode (AgentMode | None | Unset): Execution mode Default: AgentMode.STANDARD.
+      agent_type (None | str | Unset): Specific agent type to use (optional)
+      selection_criteria (None | SelectionCriteria | Unset): Criteria for agent selection
+      force_extended_analysis (bool | Unset): Force extended analysis mode with comprehensive research Default: False.
+      enable_rag (bool | Unset): Enable RAG context enrichment Default: True.
+      stream (bool | Unset): Enable streaming response Default: False.
   """
 
   message: str
-  history: Union[Unset, list["AgentMessage"]] = UNSET
-  context: Union["AgentRequestContextType0", None, Unset] = UNSET
-  mode: Union[AgentMode, None, Unset] = AgentMode.STANDARD
-  agent_type: Union[None, Unset, str] = UNSET
-  selection_criteria: Union["SelectionCriteria", None, Unset] = UNSET
-  force_extended_analysis: Union[Unset, bool] = False
-  enable_rag: Union[Unset, bool] = True
-  stream: Union[Unset, bool] = False
+  history: list[AgentMessage] | Unset = UNSET
+  context: AgentRequestContextType0 | None | Unset = UNSET
+  mode: AgentMode | None | Unset = AgentMode.STANDARD
+  agent_type: None | str | Unset = UNSET
+  selection_criteria: None | SelectionCriteria | Unset = UNSET
+  force_extended_analysis: bool | Unset = False
+  enable_rag: bool | Unset = True
+  stream: bool | Unset = False
   additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
   def to_dict(self) -> dict[str, Any]:
@@ -51,14 +52,14 @@ class AgentRequest:
 
     message = self.message
 
-    history: Union[Unset, list[dict[str, Any]]] = UNSET
+    history: list[dict[str, Any]] | Unset = UNSET
     if not isinstance(self.history, Unset):
       history = []
       for history_item_data in self.history:
         history_item = history_item_data.to_dict()
         history.append(history_item)
 
-    context: Union[None, Unset, dict[str, Any]]
+    context: dict[str, Any] | None | Unset
     if isinstance(self.context, Unset):
       context = UNSET
     elif isinstance(self.context, AgentRequestContextType0):
@@ -66,7 +67,7 @@ class AgentRequest:
     else:
       context = self.context
 
-    mode: Union[None, Unset, str]
+    mode: None | str | Unset
     if isinstance(self.mode, Unset):
       mode = UNSET
     elif isinstance(self.mode, AgentMode):
@@ -74,13 +75,13 @@ class AgentRequest:
     else:
       mode = self.mode
 
-    agent_type: Union[None, Unset, str]
+    agent_type: None | str | Unset
     if isinstance(self.agent_type, Unset):
       agent_type = UNSET
     else:
       agent_type = self.agent_type
 
-    selection_criteria: Union[None, Unset, dict[str, Any]]
+    selection_criteria: dict[str, Any] | None | Unset
     if isinstance(self.selection_criteria, Unset):
       selection_criteria = UNSET
     elif isinstance(self.selection_criteria, SelectionCriteria):
@@ -129,14 +130,16 @@ class AgentRequest:
     d = dict(src_dict)
     message = d.pop("message")
 
-    history = []
     _history = d.pop("history", UNSET)
-    for history_item_data in _history or []:
-      history_item = AgentMessage.from_dict(history_item_data)
+    history: list[AgentMessage] | Unset = UNSET
+    if _history is not UNSET:
+      history = []
+      for history_item_data in _history:
+        history_item = AgentMessage.from_dict(history_item_data)
 
-      history.append(history_item)
+        history.append(history_item)
 
-    def _parse_context(data: object) -> Union["AgentRequestContextType0", None, Unset]:
+    def _parse_context(data: object) -> AgentRequestContextType0 | None | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
@@ -147,13 +150,13 @@ class AgentRequest:
         context_type_0 = AgentRequestContextType0.from_dict(data)
 
         return context_type_0
-      except:  # noqa: E722
+      except (TypeError, ValueError, AttributeError, KeyError):
         pass
-      return cast(Union["AgentRequestContextType0", None, Unset], data)
+      return cast(AgentRequestContextType0 | None | Unset, data)
 
     context = _parse_context(d.pop("context", UNSET))
 
-    def _parse_mode(data: object) -> Union[AgentMode, None, Unset]:
+    def _parse_mode(data: object) -> AgentMode | None | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
@@ -164,24 +167,22 @@ class AgentRequest:
         mode_type_0 = AgentMode(data)
 
         return mode_type_0
-      except:  # noqa: E722
+      except (TypeError, ValueError, AttributeError, KeyError):
         pass
-      return cast(Union[AgentMode, None, Unset], data)
+      return cast(AgentMode | None | Unset, data)
 
     mode = _parse_mode(d.pop("mode", UNSET))
 
-    def _parse_agent_type(data: object) -> Union[None, Unset, str]:
+    def _parse_agent_type(data: object) -> None | str | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
         return data
-      return cast(Union[None, Unset, str], data)
+      return cast(None | str | Unset, data)
 
     agent_type = _parse_agent_type(d.pop("agent_type", UNSET))
 
-    def _parse_selection_criteria(
-      data: object,
-    ) -> Union["SelectionCriteria", None, Unset]:
+    def _parse_selection_criteria(data: object) -> None | SelectionCriteria | Unset:
       if data is None:
         return data
       if isinstance(data, Unset):
@@ -192,9 +193,9 @@ class AgentRequest:
         selection_criteria_type_0 = SelectionCriteria.from_dict(data)
 
         return selection_criteria_type_0
-      except:  # noqa: E722
+      except (TypeError, ValueError, AttributeError, KeyError):
         pass
-      return cast(Union["SelectionCriteria", None, Unset], data)
+      return cast(None | SelectionCriteria | Unset, data)
 
     selection_criteria = _parse_selection_criteria(d.pop("selection_criteria", UNSET))
 

robosystems_client/models/agent_request_context_type_0.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 from typing import Any, TypeVar