asteroid-odyssey 1.3.5__py3-none-any.whl → 1.3.7__py3-none-any.whl

asteroid_odyssey/agents_v1_gen/models/agent_profile.py

@@ -43,7 +43,9 @@ class AgentProfile(BaseModel):
     cookies: List[Cookie] = Field(description="List of cookies associated with this agent profile")
     created_at: datetime = Field(description="The date and time the agent profile was created")
     updated_at: datetime = Field(description="The last update time of the agent profile")
-    __properties: ClassVar[List[str]] = ["id", "name", "description", "organization_id", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials", "cookies", "created_at", "updated_at"]
+    tracing_snapshots: StrictBool = Field(description="Whether to enable tracing snapshots for the profile")
+    extra_stealth: StrictBool = Field(description="Whether to enable extra stealth for the profile")
+    __properties: ClassVar[List[str]] = ["id", "name", "description", "organization_id", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials", "cookies", "created_at", "updated_at", "tracing_snapshots", "extra_stealth"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -121,7 +123,9 @@ class AgentProfile(BaseModel):
             "credentials": [Credential.from_dict(_item) for _item in obj["credentials"]] if obj.get("credentials") is not None else None,
             "cookies": [Cookie.from_dict(_item) for _item in obj["cookies"]] if obj.get("cookies") is not None else None,
             "created_at": obj.get("created_at"),
-            "updated_at": obj.get("updated_at")
+            "updated_at": obj.get("updated_at"),
+            "tracing_snapshots": obj.get("tracing_snapshots"),
+            "extra_stealth": obj.get("extra_stealth")
         })
         return _obj
 

asteroid_odyssey/agents_v1_gen/models/create_agent_profile_request.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from typing_extensions import Annotated
 from asteroid_odyssey.agents_v1_gen.models.cookie import Cookie
 from asteroid_odyssey.agents_v1_gen.models.country_code import CountryCode
@@ -40,7 +40,9 @@ class CreateAgentProfileRequest(BaseModel):
     sticky_ip: StrictBool = Field(description="Whether the same IP address should be used for all executions of this profile")
     credentials: List[Credential] = Field(description="Optional list of credentials to create with the profile")
     cookies: List[Cookie] = Field(description="Optional list of cookies to create with the profile")
-    __properties: ClassVar[List[str]] = ["name", "description", "organization_id", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials", "cookies"]
+    tracing_snapshots: Optional[StrictBool] = Field(default=True, description="Whether to enable tracing snapshots for the profile")
+    extra_stealth: Optional[StrictBool] = Field(default=False, description="Whether to enable extra stealth for the profile")
+    __properties: ClassVar[List[str]] = ["name", "description", "organization_id", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials", "cookies", "tracing_snapshots", "extra_stealth"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -115,7 +117,9 @@ class CreateAgentProfileRequest(BaseModel):
             "captcha_solver_active": obj.get("captcha_solver_active") if obj.get("captcha_solver_active") is not None else False,
             "sticky_ip": obj.get("sticky_ip") if obj.get("sticky_ip") is not None else False,
             "credentials": [Credential.from_dict(_item) for _item in obj["credentials"]] if obj.get("credentials") is not None else None,
-            "cookies": [Cookie.from_dict(_item) for _item in obj["cookies"]] if obj.get("cookies") is not None else None
+            "cookies": [Cookie.from_dict(_item) for _item in obj["cookies"]] if obj.get("cookies") is not None else None,
+            "tracing_snapshots": obj.get("tracing_snapshots") if obj.get("tracing_snapshots") is not None else True,
+            "extra_stealth": obj.get("extra_stealth") if obj.get("extra_stealth") is not None else False
         })
         return _obj
 

asteroid_odyssey/agents_v1_gen/models/update_agent_profile_request.py

@@ -41,7 +41,9 @@ class UpdateAgentProfileRequest(BaseModel):
     credentials_to_delete: Optional[List[StrictStr]] = Field(default=None, description="List of credential IDs to delete from the profile")
     cookies_to_add: Optional[List[Cookie]] = Field(default=None, description="List of cookies to add to the profile")
     cookies_to_delete: Optional[List[StrictStr]] = Field(default=None, description="List of cookie IDs to delete from the profile")
-    __properties: ClassVar[List[str]] = ["name", "description", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials_to_add", "credentials_to_delete", "cookies_to_add", "cookies_to_delete"]
+    tracing_snapshots: Optional[StrictBool] = Field(default=True, description="Whether to enable tracing snapshots for the profile")
+    extra_stealth: Optional[StrictBool] = Field(default=False, description="Whether to enable extra stealth for the profile")
+    __properties: ClassVar[List[str]] = ["name", "description", "proxy_cc", "proxy_type", "captcha_solver_active", "sticky_ip", "credentials_to_add", "credentials_to_delete", "cookies_to_add", "cookies_to_delete", "tracing_snapshots", "extra_stealth"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -117,7 +119,9 @@ class UpdateAgentProfileRequest(BaseModel):
             "credentials_to_add": [Credential.from_dict(_item) for _item in obj["credentials_to_add"]] if obj.get("credentials_to_add") is not None else None,
             "credentials_to_delete": obj.get("credentials_to_delete"),
             "cookies_to_add": [Cookie.from_dict(_item) for _item in obj["cookies_to_add"]] if obj.get("cookies_to_add") is not None else None,
-            "cookies_to_delete": obj.get("cookies_to_delete")
+            "cookies_to_delete": obj.get("cookies_to_delete"),
+            "tracing_snapshots": obj.get("tracing_snapshots") if obj.get("tracing_snapshots") is not None else True,
+            "extra_stealth": obj.get("extra_stealth") if obj.get("extra_stealth") is not None else False
         })
         return _obj
 

asteroid_odyssey/agents_v2_gen/__init__.py

@@ -36,18 +36,23 @@ from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_action_failed
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_action_started import ActivityPayloadUnionActionStarted
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_file_added import ActivityPayloadUnionFileAdded
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_generic import ActivityPayloadUnionGeneric
+from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_graph_updated import ActivityPayloadUnionGraphUpdated
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_status_changed import ActivityPayloadUnionStatusChanged
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_step_completed import ActivityPayloadUnionStepCompleted
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_step_started import ActivityPayloadUnionStepStarted
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_terminal import ActivityPayloadUnionTerminal
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_transitioned_node import ActivityPayloadUnionTransitionedNode
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_user_message_received import ActivityPayloadUnionUserMessageReceived
+from asteroid_odyssey.agents_v2_gen.models.agent_list200_response import AgentList200Response
+from asteroid_odyssey.agents_v2_gen.models.agents_agent_base import AgentsAgentBase
+from asteroid_odyssey.agents_v2_gen.models.agents_agent_sort_field import AgentsAgentSortField
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity import AgentsExecutionActivity
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_completed_payload import AgentsExecutionActivityActionCompletedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_failed_payload import AgentsExecutionActivityActionFailedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_started_payload import AgentsExecutionActivityActionStartedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_file_added_payload import AgentsExecutionActivityFileAddedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_generic_payload import AgentsExecutionActivityGenericPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_graph_updated_payload import AgentsExecutionActivityGraphUpdatedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_payload_union import AgentsExecutionActivityPayloadUnion
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_status_changed_payload import AgentsExecutionActivityStatusChangedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_step_completed_payload import AgentsExecutionActivityStepCompletedPayload
@@ -59,12 +64,18 @@ from asteroid_odyssey.agents_v2_gen.models.agents_execution_cancel_reason import
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_cancelled_payload import AgentsExecutionCancelledPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_completed_payload import AgentsExecutionCompletedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_failed_payload import AgentsExecutionFailedPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_graph_update import AgentsExecutionGraphUpdate
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_node_details import AgentsExecutionNodeDetails
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_paused_payload import AgentsExecutionPausedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_status import AgentsExecutionStatus
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_terminal_payload import AgentsExecutionTerminalPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_transition_details import AgentsExecutionTransitionDetails
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_update_type import AgentsExecutionUpdateType
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_user_messages_add_text_body import AgentsExecutionUserMessagesAddTextBody
 from asteroid_odyssey.agents_v2_gen.models.agents_files_file import AgentsFilesFile
 from asteroid_odyssey.agents_v2_gen.models.common_bad_request_error_body import CommonBadRequestErrorBody
+from asteroid_odyssey.agents_v2_gen.models.common_error import CommonError
 from asteroid_odyssey.agents_v2_gen.models.common_forbidden_error_body import CommonForbiddenErrorBody
 from asteroid_odyssey.agents_v2_gen.models.common_not_found_error_body import CommonNotFoundErrorBody
+from asteroid_odyssey.agents_v2_gen.models.common_sort_direction import CommonSortDirection
 from asteroid_odyssey.agents_v2_gen.models.version import Version

asteroid_odyssey/agents_v2_gen/api/default_api.py

@@ -19,9 +19,12 @@ from typing_extensions import Annotated
 from pydantic import Field, StrictBytes, StrictInt, StrictStr, field_validator
 from typing import List, Optional, Tuple, Union
 from typing_extensions import Annotated
+from asteroid_odyssey.agents_v2_gen.models.agent_list200_response import AgentList200Response
+from asteroid_odyssey.agents_v2_gen.models.agents_agent_sort_field import AgentsAgentSortField
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity import AgentsExecutionActivity
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_user_messages_add_text_body import AgentsExecutionUserMessagesAddTextBody
 from asteroid_odyssey.agents_v2_gen.models.agents_files_file import AgentsFilesFile
+from asteroid_odyssey.agents_v2_gen.models.common_sort_direction import CommonSortDirection
 
 from asteroid_odyssey.agents_v2_gen.api_client import ApiClient, RequestSerialized
 from asteroid_odyssey.agents_v2_gen.api_response import ApiResponse
@@ -41,6 +44,351 @@ class DefaultApi:
         self.api_client = api_client
 
 
+    @validate_call
+    def agent_list(
+        self,
+        page_size: StrictInt,
+        page: StrictInt,
+        organization_id: Optional[StrictStr] = None,
+        search_name: Optional[StrictStr] = None,
+        sort_field: Optional[AgentsAgentSortField] = None,
+        sort_direction: Optional[CommonSortDirection] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> AgentList200Response:
+        """agent_list
+
+
+        :param page_size: (required)
+        :type page_size: int
+        :param page: (required)
+        :type page: int
+        :param organization_id:
+        :type organization_id: str
+        :param search_name:
+        :type search_name: str
+        :param sort_field:
+        :type sort_field: AgentsAgentSortField
+        :param sort_direction:
+        :type sort_direction: CommonSortDirection
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._agent_list_serialize(
+            page_size=page_size,
+            page=page,
+            organization_id=organization_id,
+            search_name=search_name,
+            sort_field=sort_field,
+            sort_direction=sort_direction,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AgentList200Response",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def agent_list_with_http_info(
+        self,
+        page_size: StrictInt,
+        page: StrictInt,
+        organization_id: Optional[StrictStr] = None,
+        search_name: Optional[StrictStr] = None,
+        sort_field: Optional[AgentsAgentSortField] = None,
+        sort_direction: Optional[CommonSortDirection] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[AgentList200Response]:
+        """agent_list
+
+
+        :param page_size: (required)
+        :type page_size: int
+        :param page: (required)
+        :type page: int
+        :param organization_id:
+        :type organization_id: str
+        :param search_name:
+        :type search_name: str
+        :param sort_field:
+        :type sort_field: AgentsAgentSortField
+        :param sort_direction:
+        :type sort_direction: CommonSortDirection
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._agent_list_serialize(
+            page_size=page_size,
+            page=page,
+            organization_id=organization_id,
+            search_name=search_name,
+            sort_field=sort_field,
+            sort_direction=sort_direction,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AgentList200Response",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def agent_list_without_preload_content(
+        self,
+        page_size: StrictInt,
+        page: StrictInt,
+        organization_id: Optional[StrictStr] = None,
+        search_name: Optional[StrictStr] = None,
+        sort_field: Optional[AgentsAgentSortField] = None,
+        sort_direction: Optional[CommonSortDirection] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """agent_list
+
+
+        :param page_size: (required)
+        :type page_size: int
+        :param page: (required)
+        :type page: int
+        :param organization_id:
+        :type organization_id: str
+        :param search_name:
+        :type search_name: str
+        :param sort_field:
+        :type sort_field: AgentsAgentSortField
+        :param sort_direction:
+        :type sort_direction: CommonSortDirection
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._agent_list_serialize(
+            page_size=page_size,
+            page=page,
+            organization_id=organization_id,
+            search_name=search_name,
+            sort_field=sort_field,
+            sort_direction=sort_direction,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AgentList200Response",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _agent_list_serialize(
+        self,
+        page_size,
+        page,
+        organization_id,
+        search_name,
+        sort_field,
+        sort_direction,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        if organization_id is not None:
+            
+            _query_params.append(('organizationId', organization_id))
+            
+        if page_size is not None:
+            
+            _query_params.append(('pageSize', page_size))
+            
+        if page is not None:
+            
+            _query_params.append(('page', page))
+            
+        if search_name is not None:
+            
+            _query_params.append(('searchName', search_name))
+            
+        if sort_field is not None:
+            
+            _query_params.append(('sort_field', sort_field.value))
+            
+        if sort_direction is not None:
+            
+            _query_params.append(('sort_direction', sort_direction.value))
+            
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'application/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'ApiKeyAuth'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/agents',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def execution_activities_get(
         self,

asteroid_odyssey/agents_v2_gen/models/__init__.py

@@ -19,18 +19,23 @@ from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_action_failed
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_action_started import ActivityPayloadUnionActionStarted
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_file_added import ActivityPayloadUnionFileAdded
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_generic import ActivityPayloadUnionGeneric
+from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_graph_updated import ActivityPayloadUnionGraphUpdated
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_status_changed import ActivityPayloadUnionStatusChanged
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_step_completed import ActivityPayloadUnionStepCompleted
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_step_started import ActivityPayloadUnionStepStarted
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_terminal import ActivityPayloadUnionTerminal
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_transitioned_node import ActivityPayloadUnionTransitionedNode
 from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_user_message_received import ActivityPayloadUnionUserMessageReceived
+from asteroid_odyssey.agents_v2_gen.models.agent_list200_response import AgentList200Response
+from asteroid_odyssey.agents_v2_gen.models.agents_agent_base import AgentsAgentBase
+from asteroid_odyssey.agents_v2_gen.models.agents_agent_sort_field import AgentsAgentSortField
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity import AgentsExecutionActivity
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_completed_payload import AgentsExecutionActivityActionCompletedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_failed_payload import AgentsExecutionActivityActionFailedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_started_payload import AgentsExecutionActivityActionStartedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_file_added_payload import AgentsExecutionActivityFileAddedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_generic_payload import AgentsExecutionActivityGenericPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_graph_updated_payload import AgentsExecutionActivityGraphUpdatedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_payload_union import AgentsExecutionActivityPayloadUnion
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_status_changed_payload import AgentsExecutionActivityStatusChangedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_step_completed_payload import AgentsExecutionActivityStepCompletedPayload
@@ -42,12 +47,18 @@ from asteroid_odyssey.agents_v2_gen.models.agents_execution_cancel_reason import
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_cancelled_payload import AgentsExecutionCancelledPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_completed_payload import AgentsExecutionCompletedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_failed_payload import AgentsExecutionFailedPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_graph_update import AgentsExecutionGraphUpdate
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_node_details import AgentsExecutionNodeDetails
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_paused_payload import AgentsExecutionPausedPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_status import AgentsExecutionStatus
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_terminal_payload import AgentsExecutionTerminalPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_transition_details import AgentsExecutionTransitionDetails
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_update_type import AgentsExecutionUpdateType
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_user_messages_add_text_body import AgentsExecutionUserMessagesAddTextBody
 from asteroid_odyssey.agents_v2_gen.models.agents_files_file import AgentsFilesFile
 from asteroid_odyssey.agents_v2_gen.models.common_bad_request_error_body import CommonBadRequestErrorBody
+from asteroid_odyssey.agents_v2_gen.models.common_error import CommonError
 from asteroid_odyssey.agents_v2_gen.models.common_forbidden_error_body import CommonForbiddenErrorBody
 from asteroid_odyssey.agents_v2_gen.models.common_not_found_error_body import CommonNotFoundErrorBody
+from asteroid_odyssey.agents_v2_gen.models.common_sort_direction import CommonSortDirection
 from asteroid_odyssey.agents_v2_gen.models.version import Version

asteroid_odyssey/agents_v2_gen/models/activity_payload_union_graph_updated.py

@@ -0,0 +1,100 @@
+# coding: utf-8
+
+"""
+    Agent Service
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_graph_updated_payload import AgentsExecutionActivityGraphUpdatedPayload
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ActivityPayloadUnionGraphUpdated(BaseModel):
+    """
+    ActivityPayloadUnionGraphUpdated
+    """ # noqa: E501
+    activity_type: StrictStr = Field(alias="activityType")
+    data: AgentsExecutionActivityGraphUpdatedPayload
+    __properties: ClassVar[List[str]] = ["activityType", "data"]
+
+    @field_validator('activity_type')
+    def activity_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['graph_updated']):
+            raise ValueError("must be one of enum values ('graph_updated')")
+        return value
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ActivityPayloadUnionGraphUpdated from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of data
+        if self.data:
+            _dict['data'] = self.data.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ActivityPayloadUnionGraphUpdated from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "activityType": obj.get("activityType"),
+            "data": AgentsExecutionActivityGraphUpdatedPayload.from_dict(obj["data"]) if obj.get("data") is not None else None
+        })
+        return _obj
+
+