asteroid-odyssey 1.3.6__py3-none-any.whl → 1.3.8__py3-none-any.whl

asteroid_odyssey/agents_v2_gen/models/agents_execution_activity_payload_union.py

@@ -17,54 +17,60 @@ import json
 import pprint
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
 from typing import Any, List, Optional
-from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_action_completed import ActivityPayloadUnionActionCompleted
-from asteroid_odyssey.agents_v2_gen.models.activity_payload_union_action_failed import ActivityPayloadUnionActionFailed
-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.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_playwright_script_generated_payload import AgentsExecutionActivityPlaywrightScriptGeneratedPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_reasoning_payload import AgentsExecutionActivityReasoningPayload
+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
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_step_started_payload import AgentsExecutionActivityStepStartedPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_transitioned_node_payload import AgentsExecutionActivityTransitionedNodePayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_user_message_received_payload import AgentsExecutionActivityUserMessageReceivedPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_workflow_updated_payload import AgentsExecutionActivityWorkflowUpdatedPayload
+from asteroid_odyssey.agents_v2_gen.models.agents_execution_terminal_payload import AgentsExecutionTerminalPayload
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-AGENTSEXECUTIONACTIVITYPAYLOADUNION_ONE_OF_SCHEMAS = ["ActivityPayloadUnionActionCompleted", "ActivityPayloadUnionActionFailed", "ActivityPayloadUnionActionStarted", "ActivityPayloadUnionFileAdded", "ActivityPayloadUnionGeneric", "ActivityPayloadUnionGraphUpdated", "ActivityPayloadUnionStatusChanged", "ActivityPayloadUnionStepCompleted", "ActivityPayloadUnionStepStarted", "ActivityPayloadUnionTerminal", "ActivityPayloadUnionTransitionedNode", "ActivityPayloadUnionUserMessageReceived"]
+AGENTSEXECUTIONACTIVITYPAYLOADUNION_ONE_OF_SCHEMAS = ["AgentsExecutionActivityActionCompletedPayload", "AgentsExecutionActivityActionFailedPayload", "AgentsExecutionActivityActionStartedPayload", "AgentsExecutionActivityFileAddedPayload", "AgentsExecutionActivityGenericPayload", "AgentsExecutionActivityPlaywrightScriptGeneratedPayload", "AgentsExecutionActivityReasoningPayload", "AgentsExecutionActivityStatusChangedPayload", "AgentsExecutionActivityStepCompletedPayload", "AgentsExecutionActivityStepStartedPayload", "AgentsExecutionActivityTransitionedNodePayload", "AgentsExecutionActivityUserMessageReceivedPayload", "AgentsExecutionActivityWorkflowUpdatedPayload", "AgentsExecutionTerminalPayload"]
 
 class AgentsExecutionActivityPayloadUnion(BaseModel):
     """
     AgentsExecutionActivityPayloadUnion
     """
-    # data type: ActivityPayloadUnionTerminal
-    oneof_schema_1_validator: Optional[ActivityPayloadUnionTerminal] = None
-    # data type: ActivityPayloadUnionGeneric
-    oneof_schema_2_validator: Optional[ActivityPayloadUnionGeneric] = None
-    # data type: ActivityPayloadUnionStepStarted
-    oneof_schema_3_validator: Optional[ActivityPayloadUnionStepStarted] = None
-    # data type: ActivityPayloadUnionStepCompleted
-    oneof_schema_4_validator: Optional[ActivityPayloadUnionStepCompleted] = None
-    # data type: ActivityPayloadUnionTransitionedNode
-    oneof_schema_5_validator: Optional[ActivityPayloadUnionTransitionedNode] = None
-    # data type: ActivityPayloadUnionStatusChanged
-    oneof_schema_6_validator: Optional[ActivityPayloadUnionStatusChanged] = None
-    # data type: ActivityPayloadUnionActionStarted
-    oneof_schema_7_validator: Optional[ActivityPayloadUnionActionStarted] = None
-    # data type: ActivityPayloadUnionActionCompleted
-    oneof_schema_8_validator: Optional[ActivityPayloadUnionActionCompleted] = None
-    # data type: ActivityPayloadUnionActionFailed
-    oneof_schema_9_validator: Optional[ActivityPayloadUnionActionFailed] = None
-    # data type: ActivityPayloadUnionUserMessageReceived
-    oneof_schema_10_validator: Optional[ActivityPayloadUnionUserMessageReceived] = None
-    # data type: ActivityPayloadUnionFileAdded
-    oneof_schema_11_validator: Optional[ActivityPayloadUnionFileAdded] = None
-    # data type: ActivityPayloadUnionGraphUpdated
-    oneof_schema_12_validator: Optional[ActivityPayloadUnionGraphUpdated] = None
-    actual_instance: Optional[Union[ActivityPayloadUnionActionCompleted, ActivityPayloadUnionActionFailed, ActivityPayloadUnionActionStarted, ActivityPayloadUnionFileAdded, ActivityPayloadUnionGeneric, ActivityPayloadUnionGraphUpdated, ActivityPayloadUnionStatusChanged, ActivityPayloadUnionStepCompleted, ActivityPayloadUnionStepStarted, ActivityPayloadUnionTerminal, ActivityPayloadUnionTransitionedNode, ActivityPayloadUnionUserMessageReceived]] = None
-    one_of_schemas: Set[str] = { "ActivityPayloadUnionActionCompleted", "ActivityPayloadUnionActionFailed", "ActivityPayloadUnionActionStarted", "ActivityPayloadUnionFileAdded", "ActivityPayloadUnionGeneric", "ActivityPayloadUnionGraphUpdated", "ActivityPayloadUnionStatusChanged", "ActivityPayloadUnionStepCompleted", "ActivityPayloadUnionStepStarted", "ActivityPayloadUnionTerminal", "ActivityPayloadUnionTransitionedNode", "ActivityPayloadUnionUserMessageReceived" }
+    # data type: AgentsExecutionTerminalPayload
+    oneof_schema_1_validator: Optional[AgentsExecutionTerminalPayload] = None
+    # data type: AgentsExecutionActivityGenericPayload
+    oneof_schema_2_validator: Optional[AgentsExecutionActivityGenericPayload] = None
+    # data type: AgentsExecutionActivityReasoningPayload
+    oneof_schema_3_validator: Optional[AgentsExecutionActivityReasoningPayload] = None
+    # data type: AgentsExecutionActivityStepStartedPayload
+    oneof_schema_4_validator: Optional[AgentsExecutionActivityStepStartedPayload] = None
+    # data type: AgentsExecutionActivityStepCompletedPayload
+    oneof_schema_5_validator: Optional[AgentsExecutionActivityStepCompletedPayload] = None
+    # data type: AgentsExecutionActivityTransitionedNodePayload
+    oneof_schema_6_validator: Optional[AgentsExecutionActivityTransitionedNodePayload] = None
+    # data type: AgentsExecutionActivityStatusChangedPayload
+    oneof_schema_7_validator: Optional[AgentsExecutionActivityStatusChangedPayload] = None
+    # data type: AgentsExecutionActivityActionStartedPayload
+    oneof_schema_8_validator: Optional[AgentsExecutionActivityActionStartedPayload] = None
+    # data type: AgentsExecutionActivityActionCompletedPayload
+    oneof_schema_9_validator: Optional[AgentsExecutionActivityActionCompletedPayload] = None
+    # data type: AgentsExecutionActivityActionFailedPayload
+    oneof_schema_10_validator: Optional[AgentsExecutionActivityActionFailedPayload] = None
+    # data type: AgentsExecutionActivityUserMessageReceivedPayload
+    oneof_schema_11_validator: Optional[AgentsExecutionActivityUserMessageReceivedPayload] = None
+    # data type: AgentsExecutionActivityFileAddedPayload
+    oneof_schema_12_validator: Optional[AgentsExecutionActivityFileAddedPayload] = None
+    # data type: AgentsExecutionActivityWorkflowUpdatedPayload
+    oneof_schema_13_validator: Optional[AgentsExecutionActivityWorkflowUpdatedPayload] = None
+    # data type: AgentsExecutionActivityPlaywrightScriptGeneratedPayload
+    oneof_schema_14_validator: Optional[AgentsExecutionActivityPlaywrightScriptGeneratedPayload] = None
+    actual_instance: Optional[Union[AgentsExecutionActivityActionCompletedPayload, AgentsExecutionActivityActionFailedPayload, AgentsExecutionActivityActionStartedPayload, AgentsExecutionActivityFileAddedPayload, AgentsExecutionActivityGenericPayload, AgentsExecutionActivityPlaywrightScriptGeneratedPayload, AgentsExecutionActivityReasoningPayload, AgentsExecutionActivityStatusChangedPayload, AgentsExecutionActivityStepCompletedPayload, AgentsExecutionActivityStepStartedPayload, AgentsExecutionActivityTransitionedNodePayload, AgentsExecutionActivityUserMessageReceivedPayload, AgentsExecutionActivityWorkflowUpdatedPayload, AgentsExecutionTerminalPayload]] = None
+    one_of_schemas: Set[str] = { "AgentsExecutionActivityActionCompletedPayload", "AgentsExecutionActivityActionFailedPayload", "AgentsExecutionActivityActionStartedPayload", "AgentsExecutionActivityFileAddedPayload", "AgentsExecutionActivityGenericPayload", "AgentsExecutionActivityPlaywrightScriptGeneratedPayload", "AgentsExecutionActivityReasoningPayload", "AgentsExecutionActivityStatusChangedPayload", "AgentsExecutionActivityStepCompletedPayload", "AgentsExecutionActivityStepStartedPayload", "AgentsExecutionActivityTransitionedNodePayload", "AgentsExecutionActivityUserMessageReceivedPayload", "AgentsExecutionActivityWorkflowUpdatedPayload", "AgentsExecutionTerminalPayload" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -90,72 +96,82 @@ class AgentsExecutionActivityPayloadUnion(BaseModel):
         instance = AgentsExecutionActivityPayloadUnion.model_construct()
         error_messages = []
         match = 0
-        # validate data type: ActivityPayloadUnionTerminal
-        if not isinstance(v, ActivityPayloadUnionTerminal):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionTerminal`")
+        # validate data type: AgentsExecutionTerminalPayload
+        if not isinstance(v, AgentsExecutionTerminalPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionTerminalPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionGeneric
-        if not isinstance(v, ActivityPayloadUnionGeneric):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionGeneric`")
+        # validate data type: AgentsExecutionActivityGenericPayload
+        if not isinstance(v, AgentsExecutionActivityGenericPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityGenericPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionStepStarted
-        if not isinstance(v, ActivityPayloadUnionStepStarted):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionStepStarted`")
+        # validate data type: AgentsExecutionActivityReasoningPayload
+        if not isinstance(v, AgentsExecutionActivityReasoningPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityReasoningPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionStepCompleted
-        if not isinstance(v, ActivityPayloadUnionStepCompleted):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionStepCompleted`")
+        # validate data type: AgentsExecutionActivityStepStartedPayload
+        if not isinstance(v, AgentsExecutionActivityStepStartedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityStepStartedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionTransitionedNode
-        if not isinstance(v, ActivityPayloadUnionTransitionedNode):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionTransitionedNode`")
+        # validate data type: AgentsExecutionActivityStepCompletedPayload
+        if not isinstance(v, AgentsExecutionActivityStepCompletedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityStepCompletedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionStatusChanged
-        if not isinstance(v, ActivityPayloadUnionStatusChanged):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionStatusChanged`")
+        # validate data type: AgentsExecutionActivityTransitionedNodePayload
+        if not isinstance(v, AgentsExecutionActivityTransitionedNodePayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityTransitionedNodePayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionActionStarted
-        if not isinstance(v, ActivityPayloadUnionActionStarted):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionActionStarted`")
+        # validate data type: AgentsExecutionActivityStatusChangedPayload
+        if not isinstance(v, AgentsExecutionActivityStatusChangedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityStatusChangedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionActionCompleted
-        if not isinstance(v, ActivityPayloadUnionActionCompleted):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionActionCompleted`")
+        # validate data type: AgentsExecutionActivityActionStartedPayload
+        if not isinstance(v, AgentsExecutionActivityActionStartedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityActionStartedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionActionFailed
-        if not isinstance(v, ActivityPayloadUnionActionFailed):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionActionFailed`")
+        # validate data type: AgentsExecutionActivityActionCompletedPayload
+        if not isinstance(v, AgentsExecutionActivityActionCompletedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityActionCompletedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionUserMessageReceived
-        if not isinstance(v, ActivityPayloadUnionUserMessageReceived):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionUserMessageReceived`")
+        # validate data type: AgentsExecutionActivityActionFailedPayload
+        if not isinstance(v, AgentsExecutionActivityActionFailedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityActionFailedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionFileAdded
-        if not isinstance(v, ActivityPayloadUnionFileAdded):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionFileAdded`")
+        # validate data type: AgentsExecutionActivityUserMessageReceivedPayload
+        if not isinstance(v, AgentsExecutionActivityUserMessageReceivedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityUserMessageReceivedPayload`")
         else:
             match += 1
-        # validate data type: ActivityPayloadUnionGraphUpdated
-        if not isinstance(v, ActivityPayloadUnionGraphUpdated):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ActivityPayloadUnionGraphUpdated`")
+        # validate data type: AgentsExecutionActivityFileAddedPayload
+        if not isinstance(v, AgentsExecutionActivityFileAddedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityFileAddedPayload`")
+        else:
+            match += 1
+        # validate data type: AgentsExecutionActivityWorkflowUpdatedPayload
+        if not isinstance(v, AgentsExecutionActivityWorkflowUpdatedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityWorkflowUpdatedPayload`")
+        else:
+            match += 1
+        # validate data type: AgentsExecutionActivityPlaywrightScriptGeneratedPayload
+        if not isinstance(v, AgentsExecutionActivityPlaywrightScriptGeneratedPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionActivityPlaywrightScriptGeneratedPayload`")
         else:
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in AgentsExecutionActivityPayloadUnion with oneOf schemas: ActivityPayloadUnionActionCompleted, ActivityPayloadUnionActionFailed, ActivityPayloadUnionActionStarted, ActivityPayloadUnionFileAdded, ActivityPayloadUnionGeneric, ActivityPayloadUnionGraphUpdated, ActivityPayloadUnionStatusChanged, ActivityPayloadUnionStepCompleted, ActivityPayloadUnionStepStarted, ActivityPayloadUnionTerminal, ActivityPayloadUnionTransitionedNode, ActivityPayloadUnionUserMessageReceived. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in AgentsExecutionActivityPayloadUnion with oneOf schemas: AgentsExecutionActivityActionCompletedPayload, AgentsExecutionActivityActionFailedPayload, AgentsExecutionActivityActionStartedPayload, AgentsExecutionActivityFileAddedPayload, AgentsExecutionActivityGenericPayload, AgentsExecutionActivityPlaywrightScriptGeneratedPayload, AgentsExecutionActivityReasoningPayload, AgentsExecutionActivityStatusChangedPayload, AgentsExecutionActivityStepCompletedPayload, AgentsExecutionActivityStepStartedPayload, AgentsExecutionActivityTransitionedNodePayload, AgentsExecutionActivityUserMessageReceivedPayload, AgentsExecutionActivityWorkflowUpdatedPayload, AgentsExecutionTerminalPayload. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in AgentsExecutionActivityPayloadUnion with oneOf schemas: ActivityPayloadUnionActionCompleted, ActivityPayloadUnionActionFailed, ActivityPayloadUnionActionStarted, ActivityPayloadUnionFileAdded, ActivityPayloadUnionGeneric, ActivityPayloadUnionGraphUpdated, ActivityPayloadUnionStatusChanged, ActivityPayloadUnionStepCompleted, ActivityPayloadUnionStepStarted, ActivityPayloadUnionTerminal, ActivityPayloadUnionTransitionedNode, ActivityPayloadUnionUserMessageReceived. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in AgentsExecutionActivityPayloadUnion with oneOf schemas: AgentsExecutionActivityActionCompletedPayload, AgentsExecutionActivityActionFailedPayload, AgentsExecutionActivityActionStartedPayload, AgentsExecutionActivityFileAddedPayload, AgentsExecutionActivityGenericPayload, AgentsExecutionActivityPlaywrightScriptGeneratedPayload, AgentsExecutionActivityReasoningPayload, AgentsExecutionActivityStatusChangedPayload, AgentsExecutionActivityStepCompletedPayload, AgentsExecutionActivityStepStartedPayload, AgentsExecutionActivityTransitionedNodePayload, AgentsExecutionActivityUserMessageReceivedPayload, AgentsExecutionActivityWorkflowUpdatedPayload, AgentsExecutionTerminalPayload. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -170,85 +186,97 @@ class AgentsExecutionActivityPayloadUnion(BaseModel):
         error_messages = []
         match = 0
 
-        # deserialize data into ActivityPayloadUnionTerminal
+        # deserialize data into AgentsExecutionTerminalPayload
+        try:
+            instance.actual_instance = AgentsExecutionTerminalPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into AgentsExecutionActivityGenericPayload
+        try:
+            instance.actual_instance = AgentsExecutionActivityGenericPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into AgentsExecutionActivityReasoningPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionTerminal.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityReasoningPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionGeneric
+        # deserialize data into AgentsExecutionActivityStepStartedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionGeneric.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityStepStartedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionStepStarted
+        # deserialize data into AgentsExecutionActivityStepCompletedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionStepStarted.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityStepCompletedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionStepCompleted
+        # deserialize data into AgentsExecutionActivityTransitionedNodePayload
         try:
-            instance.actual_instance = ActivityPayloadUnionStepCompleted.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityTransitionedNodePayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionTransitionedNode
+        # deserialize data into AgentsExecutionActivityStatusChangedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionTransitionedNode.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityStatusChangedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionStatusChanged
+        # deserialize data into AgentsExecutionActivityActionStartedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionStatusChanged.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityActionStartedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionActionStarted
+        # deserialize data into AgentsExecutionActivityActionCompletedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionActionStarted.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityActionCompletedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionActionCompleted
+        # deserialize data into AgentsExecutionActivityActionFailedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionActionCompleted.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityActionFailedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionActionFailed
+        # deserialize data into AgentsExecutionActivityUserMessageReceivedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionActionFailed.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityUserMessageReceivedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionUserMessageReceived
+        # deserialize data into AgentsExecutionActivityFileAddedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionUserMessageReceived.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityFileAddedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionFileAdded
+        # deserialize data into AgentsExecutionActivityWorkflowUpdatedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionFileAdded.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityWorkflowUpdatedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ActivityPayloadUnionGraphUpdated
+        # deserialize data into AgentsExecutionActivityPlaywrightScriptGeneratedPayload
         try:
-            instance.actual_instance = ActivityPayloadUnionGraphUpdated.from_json(json_str)
+            instance.actual_instance = AgentsExecutionActivityPlaywrightScriptGeneratedPayload.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into AgentsExecutionActivityPayloadUnion with oneOf schemas: ActivityPayloadUnionActionCompleted, ActivityPayloadUnionActionFailed, ActivityPayloadUnionActionStarted, ActivityPayloadUnionFileAdded, ActivityPayloadUnionGeneric, ActivityPayloadUnionGraphUpdated, ActivityPayloadUnionStatusChanged, ActivityPayloadUnionStepCompleted, ActivityPayloadUnionStepStarted, ActivityPayloadUnionTerminal, ActivityPayloadUnionTransitionedNode, ActivityPayloadUnionUserMessageReceived. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into AgentsExecutionActivityPayloadUnion with oneOf schemas: AgentsExecutionActivityActionCompletedPayload, AgentsExecutionActivityActionFailedPayload, AgentsExecutionActivityActionStartedPayload, AgentsExecutionActivityFileAddedPayload, AgentsExecutionActivityGenericPayload, AgentsExecutionActivityPlaywrightScriptGeneratedPayload, AgentsExecutionActivityReasoningPayload, AgentsExecutionActivityStatusChangedPayload, AgentsExecutionActivityStepCompletedPayload, AgentsExecutionActivityStepStartedPayload, AgentsExecutionActivityTransitionedNodePayload, AgentsExecutionActivityUserMessageReceivedPayload, AgentsExecutionActivityWorkflowUpdatedPayload, AgentsExecutionTerminalPayload. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into AgentsExecutionActivityPayloadUnion with oneOf schemas: ActivityPayloadUnionActionCompleted, ActivityPayloadUnionActionFailed, ActivityPayloadUnionActionStarted, ActivityPayloadUnionFileAdded, ActivityPayloadUnionGeneric, ActivityPayloadUnionGraphUpdated, ActivityPayloadUnionStatusChanged, ActivityPayloadUnionStepCompleted, ActivityPayloadUnionStepStarted, ActivityPayloadUnionTerminal, ActivityPayloadUnionTransitionedNode, ActivityPayloadUnionUserMessageReceived. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into AgentsExecutionActivityPayloadUnion with oneOf schemas: AgentsExecutionActivityActionCompletedPayload, AgentsExecutionActivityActionFailedPayload, AgentsExecutionActivityActionStartedPayload, AgentsExecutionActivityFileAddedPayload, AgentsExecutionActivityGenericPayload, AgentsExecutionActivityPlaywrightScriptGeneratedPayload, AgentsExecutionActivityReasoningPayload, AgentsExecutionActivityStatusChangedPayload, AgentsExecutionActivityStepCompletedPayload, AgentsExecutionActivityStepStartedPayload, AgentsExecutionActivityTransitionedNodePayload, AgentsExecutionActivityUserMessageReceivedPayload, AgentsExecutionActivityWorkflowUpdatedPayload, AgentsExecutionTerminalPayload. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -262,7 +290,7 @@ class AgentsExecutionActivityPayloadUnion(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], ActivityPayloadUnionActionCompleted, ActivityPayloadUnionActionFailed, ActivityPayloadUnionActionStarted, ActivityPayloadUnionFileAdded, ActivityPayloadUnionGeneric, ActivityPayloadUnionGraphUpdated, ActivityPayloadUnionStatusChanged, ActivityPayloadUnionStepCompleted, ActivityPayloadUnionStepStarted, ActivityPayloadUnionTerminal, ActivityPayloadUnionTransitionedNode, ActivityPayloadUnionUserMessageReceived]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AgentsExecutionActivityActionCompletedPayload, AgentsExecutionActivityActionFailedPayload, AgentsExecutionActivityActionStartedPayload, AgentsExecutionActivityFileAddedPayload, AgentsExecutionActivityGenericPayload, AgentsExecutionActivityPlaywrightScriptGeneratedPayload, AgentsExecutionActivityReasoningPayload, AgentsExecutionActivityStatusChangedPayload, AgentsExecutionActivityStepCompletedPayload, AgentsExecutionActivityStepStartedPayload, AgentsExecutionActivityTransitionedNodePayload, AgentsExecutionActivityUserMessageReceivedPayload, AgentsExecutionActivityWorkflowUpdatedPayload, AgentsExecutionTerminalPayload]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

asteroid_odyssey/agents_v2_gen/models/agents_execution_activity_playwright_script_generated_payload.py

@@ -0,0 +1,114 @@
+# 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, Optional
+from asteroid_odyssey.agents_v2_gen.models.agents_graph_models_nodes_properties_playwright_script_llm_var import AgentsGraphModelsNodesPropertiesPlaywrightScriptLLMVar
+from typing import Optional, Set
+from typing_extensions import Self
+
+class AgentsExecutionActivityPlaywrightScriptGeneratedPayload(BaseModel):
+    """
+    AgentsExecutionActivityPlaywrightScriptGeneratedPayload
+    """ # noqa: E501
+    activity_type: StrictStr = Field(alias="activityType")
+    context: StrictStr
+    llm_vars: Optional[List[AgentsGraphModelsNodesPropertiesPlaywrightScriptLLMVar]] = Field(default=None, alias="llmVars")
+    node_id: StrictStr = Field(alias="nodeId")
+    node_name: StrictStr = Field(alias="nodeName")
+    old_script: Optional[StrictStr] = Field(default=None, alias="oldScript")
+    script: StrictStr
+    __properties: ClassVar[List[str]] = ["activityType", "context", "llmVars", "nodeId", "nodeName", "oldScript", "script"]
+
+    @field_validator('activity_type')
+    def activity_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['playwright_script_generated']):
+            raise ValueError("must be one of enum values ('playwright_script_generated')")
+        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 AgentsExecutionActivityPlaywrightScriptGeneratedPayload 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 each item in llm_vars (list)
+        _items = []
+        if self.llm_vars:
+            for _item_llm_vars in self.llm_vars:
+                if _item_llm_vars:
+                    _items.append(_item_llm_vars.to_dict())
+            _dict['llmVars'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentsExecutionActivityPlaywrightScriptGeneratedPayload 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"),
+            "context": obj.get("context"),
+            "llmVars": [AgentsGraphModelsNodesPropertiesPlaywrightScriptLLMVar.from_dict(_item) for _item in obj["llmVars"]] if obj.get("llmVars") is not None else None,
+            "nodeId": obj.get("nodeId"),
+            "nodeName": obj.get("nodeName"),
+            "oldScript": obj.get("oldScript"),
+            "script": obj.get("script")
+        })
+        return _obj
+
+

asteroid_odyssey/agents_v2_gen/models/{activity_payload_union_terminal.py → agents_execution_activity_reasoning_payload.py}

@@ -19,23 +19,22 @@ 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_terminal_payload import AgentsExecutionTerminalPayload
 from typing import Optional, Set
 from typing_extensions import Self
 
-class ActivityPayloadUnionTerminal(BaseModel):
+class AgentsExecutionActivityReasoningPayload(BaseModel):
     """
-    ActivityPayloadUnionTerminal
+    AgentsExecutionActivityReasoningPayload
     """ # noqa: E501
     activity_type: StrictStr = Field(alias="activityType")
-    data: AgentsExecutionTerminalPayload
-    __properties: ClassVar[List[str]] = ["activityType", "data"]
+    reasoning: StrictStr
+    __properties: ClassVar[List[str]] = ["activityType", "reasoning"]
 
     @field_validator('activity_type')
     def activity_type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['terminal']):
-            raise ValueError("must be one of enum values ('terminal')")
+        if value not in set(['reasoning']):
+            raise ValueError("must be one of enum values ('reasoning')")
         return value
 
     model_config = ConfigDict(
@@ -56,7 +55,7 @@ class ActivityPayloadUnionTerminal(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of ActivityPayloadUnionTerminal from a JSON string"""
+        """Create an instance of AgentsExecutionActivityReasoningPayload from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -77,14 +76,11 @@ class ActivityPayloadUnionTerminal(BaseModel):
             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 ActivityPayloadUnionTerminal from a dict"""
+        """Create an instance of AgentsExecutionActivityReasoningPayload from a dict"""
         if obj is None:
             return None
 
@@ -93,7 +89,7 @@ class ActivityPayloadUnionTerminal(BaseModel):
 
         _obj = cls.model_validate({
             "activityType": obj.get("activityType"),
-            "data": AgentsExecutionTerminalPayload.from_dict(obj["data"]) if obj.get("data") is not None else None
+            "reasoning": obj.get("reasoning")
         })
         return _obj
 

asteroid_odyssey/agents_v2_gen/models/agents_execution_activity_status_changed_payload.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_awaiting_confirmation_payload import AgentsExecutionAwaitingConfirmationPayload
 from asteroid_odyssey.agents_v2_gen.models.agents_execution_cancelled_payload import AgentsExecutionCancelledPayload
@@ -32,13 +32,21 @@ class AgentsExecutionActivityStatusChangedPayload(BaseModel):
     """
     AgentsExecutionActivityStatusChangedPayload
     """ # noqa: E501
+    activity_type: StrictStr = Field(alias="activityType")
     awaiting_confirmation_payload: Optional[AgentsExecutionAwaitingConfirmationPayload] = Field(default=None, alias="awaitingConfirmationPayload")
     cancelled_payload: Optional[AgentsExecutionCancelledPayload] = Field(default=None, alias="cancelledPayload")
     completed_payload: Optional[AgentsExecutionCompletedPayload] = Field(default=None, alias="completedPayload")
     failed_payload: Optional[AgentsExecutionFailedPayload] = Field(default=None, alias="failedPayload")
     paused_payload: Optional[AgentsExecutionPausedPayload] = Field(default=None, alias="pausedPayload")
     status: AgentsExecutionStatus
-    __properties: ClassVar[List[str]] = ["awaitingConfirmationPayload", "cancelledPayload", "completedPayload", "failedPayload", "pausedPayload", "status"]
+    __properties: ClassVar[List[str]] = ["activityType", "awaitingConfirmationPayload", "cancelledPayload", "completedPayload", "failedPayload", "pausedPayload", "status"]
+
+    @field_validator('activity_type')
+    def activity_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['status_changed']):
+            raise ValueError("must be one of enum values ('status_changed')")
+        return value
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -106,6 +114,7 @@ class AgentsExecutionActivityStatusChangedPayload(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "activityType": obj.get("activityType"),
             "awaitingConfirmationPayload": AgentsExecutionAwaitingConfirmationPayload.from_dict(obj["awaitingConfirmationPayload"]) if obj.get("awaitingConfirmationPayload") is not None else None,
             "cancelledPayload": AgentsExecutionCancelledPayload.from_dict(obj["cancelledPayload"]) if obj.get("cancelledPayload") is not None else None,
             "completedPayload": AgentsExecutionCompletedPayload.from_dict(obj["completedPayload"]) if obj.get("completedPayload") is not None else None,

asteroid_odyssey/agents_v2_gen/models/agents_execution_activity_step_completed_payload.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictInt
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
@@ -26,8 +26,16 @@ class AgentsExecutionActivityStepCompletedPayload(BaseModel):
     """
     AgentsExecutionActivityStepCompletedPayload
     """ # noqa: E501
+    activity_type: StrictStr = Field(alias="activityType")
     step_number: StrictInt = Field(alias="stepNumber")
-    __properties: ClassVar[List[str]] = ["stepNumber"]
+    __properties: ClassVar[List[str]] = ["activityType", "stepNumber"]
+
+    @field_validator('activity_type')
+    def activity_type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['step_completed']):
+            raise ValueError("must be one of enum values ('step_completed')")
+        return value
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -80,6 +88,7 @@ class AgentsExecutionActivityStepCompletedPayload(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "activityType": obj.get("activityType"),
             "stepNumber": obj.get("stepNumber")
         })
         return _obj