asteroid-odyssey 1.3.13__py3-none-any.whl → 1.6.1__py3-none-any.whl

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_action_started_info.py

@@ -17,21 +17,23 @@ 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.agents_execution_llm_call_started_details import AgentsExecutionLLMCallStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_nav_to_started_details import AgentsExecutionNavToStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_scratchpad_read_started_details import AgentsExecutionScratchpadReadStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_scratchpad_write_started_details import AgentsExecutionScratchpadWriteStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_script_hybrid_playwright_started_details import AgentsExecutionScriptHybridPlaywrightStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_script_playwright_started_details import AgentsExecutionScriptPlaywrightStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_scriptpad_read_started_details import AgentsExecutionScriptpadReadStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_scriptpad_run_function_started_details import AgentsExecutionScriptpadRunFunctionStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_scriptpad_search_replace_started_details import AgentsExecutionScriptpadSearchReplaceStartedDetails
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_util_get_datetime_started_details import AgentsExecutionUtilGetDatetimeStartedDetails
+from asteroid_odyssey.models.agents_execution_agent_query_context_started_details import AgentsExecutionAgentQueryContextStartedDetails
+from asteroid_odyssey.models.agents_execution_llm_call_started_details import AgentsExecutionLLMCallStartedDetails
+from asteroid_odyssey.models.agents_execution_nav_to_started_details import AgentsExecutionNavToStartedDetails
+from asteroid_odyssey.models.agents_execution_scratchpad_read_started_details import AgentsExecutionScratchpadReadStartedDetails
+from asteroid_odyssey.models.agents_execution_scratchpad_write_started_details import AgentsExecutionScratchpadWriteStartedDetails
+from asteroid_odyssey.models.agents_execution_script_eval_started_details import AgentsExecutionScriptEvalStartedDetails
+from asteroid_odyssey.models.agents_execution_script_hybrid_playwright_started_details import AgentsExecutionScriptHybridPlaywrightStartedDetails
+from asteroid_odyssey.models.agents_execution_script_playwright_started_details import AgentsExecutionScriptPlaywrightStartedDetails
+from asteroid_odyssey.models.agents_execution_scriptpad_read_started_details import AgentsExecutionScriptpadReadStartedDetails
+from asteroid_odyssey.models.agents_execution_scriptpad_run_function_started_details import AgentsExecutionScriptpadRunFunctionStartedDetails
+from asteroid_odyssey.models.agents_execution_scriptpad_search_replace_started_details import AgentsExecutionScriptpadSearchReplaceStartedDetails
+from asteroid_odyssey.models.agents_execution_util_get_datetime_started_details import AgentsExecutionUtilGetDatetimeStartedDetails
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-AGENTSEXECUTIONACTIVITYACTIONSTARTEDINFO_ONE_OF_SCHEMAS = ["AgentsExecutionLLMCallStartedDetails", "AgentsExecutionNavToStartedDetails", "AgentsExecutionScratchpadReadStartedDetails", "AgentsExecutionScratchpadWriteStartedDetails", "AgentsExecutionScriptHybridPlaywrightStartedDetails", "AgentsExecutionScriptPlaywrightStartedDetails", "AgentsExecutionScriptpadReadStartedDetails", "AgentsExecutionScriptpadRunFunctionStartedDetails", "AgentsExecutionScriptpadSearchReplaceStartedDetails", "AgentsExecutionUtilGetDatetimeStartedDetails"]
+AGENTSEXECUTIONACTIVITYACTIONSTARTEDINFO_ONE_OF_SCHEMAS = ["AgentsExecutionAgentQueryContextStartedDetails", "AgentsExecutionLLMCallStartedDetails", "AgentsExecutionNavToStartedDetails", "AgentsExecutionScratchpadReadStartedDetails", "AgentsExecutionScratchpadWriteStartedDetails", "AgentsExecutionScriptEvalStartedDetails", "AgentsExecutionScriptHybridPlaywrightStartedDetails", "AgentsExecutionScriptPlaywrightStartedDetails", "AgentsExecutionScriptpadReadStartedDetails", "AgentsExecutionScriptpadRunFunctionStartedDetails", "AgentsExecutionScriptpadSearchReplaceStartedDetails", "AgentsExecutionUtilGetDatetimeStartedDetails"]
 
 class AgentsExecutionActivityActionStartedInfo(BaseModel):
     """
@@ -49,16 +51,20 @@ class AgentsExecutionActivityActionStartedInfo(BaseModel):
     oneof_schema_5_validator: Optional[AgentsExecutionScriptPlaywrightStartedDetails] = None
     # data type: AgentsExecutionScriptHybridPlaywrightStartedDetails
     oneof_schema_6_validator: Optional[AgentsExecutionScriptHybridPlaywrightStartedDetails] = None
+    # data type: AgentsExecutionScriptEvalStartedDetails
+    oneof_schema_7_validator: Optional[AgentsExecutionScriptEvalStartedDetails] = None
     # data type: AgentsExecutionScriptpadSearchReplaceStartedDetails
-    oneof_schema_7_validator: Optional[AgentsExecutionScriptpadSearchReplaceStartedDetails] = None
+    oneof_schema_8_validator: Optional[AgentsExecutionScriptpadSearchReplaceStartedDetails] = None
     # data type: AgentsExecutionUtilGetDatetimeStartedDetails
-    oneof_schema_8_validator: Optional[AgentsExecutionUtilGetDatetimeStartedDetails] = None
+    oneof_schema_9_validator: Optional[AgentsExecutionUtilGetDatetimeStartedDetails] = None
     # data type: AgentsExecutionScriptpadReadStartedDetails
-    oneof_schema_9_validator: Optional[AgentsExecutionScriptpadReadStartedDetails] = None
+    oneof_schema_10_validator: Optional[AgentsExecutionScriptpadReadStartedDetails] = None
     # data type: AgentsExecutionLLMCallStartedDetails
-    oneof_schema_10_validator: Optional[AgentsExecutionLLMCallStartedDetails] = None
-    actual_instance: Optional[Union[AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails]] = None
-    one_of_schemas: Set[str] = { "AgentsExecutionLLMCallStartedDetails", "AgentsExecutionNavToStartedDetails", "AgentsExecutionScratchpadReadStartedDetails", "AgentsExecutionScratchpadWriteStartedDetails", "AgentsExecutionScriptHybridPlaywrightStartedDetails", "AgentsExecutionScriptPlaywrightStartedDetails", "AgentsExecutionScriptpadReadStartedDetails", "AgentsExecutionScriptpadRunFunctionStartedDetails", "AgentsExecutionScriptpadSearchReplaceStartedDetails", "AgentsExecutionUtilGetDatetimeStartedDetails" }
+    oneof_schema_11_validator: Optional[AgentsExecutionLLMCallStartedDetails] = None
+    # data type: AgentsExecutionAgentQueryContextStartedDetails
+    oneof_schema_12_validator: Optional[AgentsExecutionAgentQueryContextStartedDetails] = None
+    actual_instance: Optional[Union[AgentsExecutionAgentQueryContextStartedDetails, AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptEvalStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails]] = None
+    one_of_schemas: Set[str] = { "AgentsExecutionAgentQueryContextStartedDetails", "AgentsExecutionLLMCallStartedDetails", "AgentsExecutionNavToStartedDetails", "AgentsExecutionScratchpadReadStartedDetails", "AgentsExecutionScratchpadWriteStartedDetails", "AgentsExecutionScriptEvalStartedDetails", "AgentsExecutionScriptHybridPlaywrightStartedDetails", "AgentsExecutionScriptPlaywrightStartedDetails", "AgentsExecutionScriptpadReadStartedDetails", "AgentsExecutionScriptpadRunFunctionStartedDetails", "AgentsExecutionScriptpadSearchReplaceStartedDetails", "AgentsExecutionUtilGetDatetimeStartedDetails" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -114,6 +120,11 @@ class AgentsExecutionActivityActionStartedInfo(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionScriptHybridPlaywrightStartedDetails`")
         else:
             match += 1
+        # validate data type: AgentsExecutionScriptEvalStartedDetails
+        if not isinstance(v, AgentsExecutionScriptEvalStartedDetails):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionScriptEvalStartedDetails`")
+        else:
+            match += 1
         # validate data type: AgentsExecutionScriptpadSearchReplaceStartedDetails
         if not isinstance(v, AgentsExecutionScriptpadSearchReplaceStartedDetails):
             error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionScriptpadSearchReplaceStartedDetails`")
@@ -134,12 +145,17 @@ class AgentsExecutionActivityActionStartedInfo(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionLLMCallStartedDetails`")
         else:
             match += 1
+        # validate data type: AgentsExecutionAgentQueryContextStartedDetails
+        if not isinstance(v, AgentsExecutionAgentQueryContextStartedDetails):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgentsExecutionAgentQueryContextStartedDetails`")
+        else:
+            match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionAgentQueryContextStartedDetails, AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptEvalStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionAgentQueryContextStartedDetails, AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptEvalStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -190,6 +206,12 @@ class AgentsExecutionActivityActionStartedInfo(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into AgentsExecutionScriptEvalStartedDetails
+        try:
+            instance.actual_instance = AgentsExecutionScriptEvalStartedDetails.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into AgentsExecutionScriptpadSearchReplaceStartedDetails
         try:
             instance.actual_instance = AgentsExecutionScriptpadSearchReplaceStartedDetails.from_json(json_str)
@@ -214,13 +236,19 @@ class AgentsExecutionActivityActionStartedInfo(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into AgentsExecutionAgentQueryContextStartedDetails
+        try:
+            instance.actual_instance = AgentsExecutionAgentQueryContextStartedDetails.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 AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionAgentQueryContextStartedDetails, AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptEvalStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into AgentsExecutionActivityActionStartedInfo with oneOf schemas: AgentsExecutionAgentQueryContextStartedDetails, AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptEvalStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -234,7 +262,7 @@ class AgentsExecutionActivityActionStartedInfo(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AgentsExecutionAgentQueryContextStartedDetails, AgentsExecutionLLMCallStartedDetails, AgentsExecutionNavToStartedDetails, AgentsExecutionScratchpadReadStartedDetails, AgentsExecutionScratchpadWriteStartedDetails, AgentsExecutionScriptEvalStartedDetails, AgentsExecutionScriptHybridPlaywrightStartedDetails, AgentsExecutionScriptPlaywrightStartedDetails, AgentsExecutionScriptpadReadStartedDetails, AgentsExecutionScriptpadRunFunctionStartedDetails, AgentsExecutionScriptpadSearchReplaceStartedDetails, AgentsExecutionUtilGetDatetimeStartedDetails]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_action_started_payload.py

@@ -19,8 +19,8 @@ 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_execution_action_name import AgentsExecutionActionName
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_activity_action_started_info import AgentsExecutionActivityActionStartedInfo
+from asteroid_odyssey.models.agents_execution_action_name import AgentsExecutionActionName
+from asteroid_odyssey.models.agents_execution_activity_action_started_info import AgentsExecutionActivityActionStartedInfo
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,12 +28,12 @@ class AgentsExecutionActivityActionStartedPayload(BaseModel):
     """
     AgentsExecutionActivityActionStartedPayload
     """ # noqa: E501
-    action_id: StrictStr = Field(alias="actionId")
-    action_name: AgentsExecutionActionName = Field(alias="actionName")
     activity_type: StrictStr = Field(alias="activityType")
-    info: Optional[AgentsExecutionActivityActionStartedInfo] = None
     message: StrictStr
-    __properties: ClassVar[List[str]] = ["actionId", "actionName", "activityType", "info", "message"]
+    action_name: AgentsExecutionActionName = Field(alias="actionName")
+    action_id: StrictStr = Field(alias="actionId")
+    info: Optional[AgentsExecutionActivityActionStartedInfo] = None
+    __properties: ClassVar[List[str]] = ["activityType", "message", "actionName", "actionId", "info"]
 
     @field_validator('activity_type')
     def activity_type_validate_enum(cls, value):
@@ -96,11 +96,11 @@ class AgentsExecutionActivityActionStartedPayload(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "actionId": obj.get("actionId"),
-            "actionName": obj.get("actionName"),
             "activityType": obj.get("activityType"),
-            "info": AgentsExecutionActivityActionStartedInfo.from_dict(obj["info"]) if obj.get("info") is not None else None,
-            "message": obj.get("message")
+            "message": obj.get("message"),
+            "actionName": obj.get("actionName"),
+            "actionId": obj.get("actionId"),
+            "info": AgentsExecutionActivityActionStartedInfo.from_dict(obj["info"]) if obj.get("info") is not None else None
         })
         return _obj
 

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_file_added_payload.py

@@ -29,11 +29,11 @@ class AgentsExecutionActivityFileAddedPayload(BaseModel):
     activity_type: StrictStr = Field(alias="activityType")
     file_id: StrictStr = Field(alias="fileId")
     file_name: StrictStr = Field(alias="fileName")
-    file_size: StrictInt = Field(alias="fileSize")
     mime_type: StrictStr = Field(alias="mimeType")
-    presigned_url: StrictStr = Field(alias="presignedUrl")
+    file_size: StrictInt = Field(alias="fileSize")
     source: StrictStr
-    __properties: ClassVar[List[str]] = ["activityType", "fileId", "fileName", "fileSize", "mimeType", "presignedUrl", "source"]
+    presigned_url: StrictStr = Field(alias="presignedUrl")
+    __properties: ClassVar[List[str]] = ["activityType", "fileId", "fileName", "mimeType", "fileSize", "source", "presignedUrl"]
 
     @field_validator('activity_type')
     def activity_type_validate_enum(cls, value):
@@ -103,10 +103,10 @@ class AgentsExecutionActivityFileAddedPayload(BaseModel):
             "activityType": obj.get("activityType"),
             "fileId": obj.get("fileId"),
             "fileName": obj.get("fileName"),
-            "fileSize": obj.get("fileSize"),
             "mimeType": obj.get("mimeType"),
-            "presignedUrl": obj.get("presignedUrl"),
-            "source": obj.get("source")
+            "fileSize": obj.get("fileSize"),
+            "source": obj.get("source"),
+            "presignedUrl": obj.get("presignedUrl")
         })
         return _obj
 

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_payload_union.py

@@ -17,20 +17,20 @@ 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.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 asteroid_odyssey.models.agents_execution_activity_action_completed_payload import AgentsExecutionActivityActionCompletedPayload
+from asteroid_odyssey.models.agents_execution_activity_action_failed_payload import AgentsExecutionActivityActionFailedPayload
+from asteroid_odyssey.models.agents_execution_activity_action_started_payload import AgentsExecutionActivityActionStartedPayload
+from asteroid_odyssey.models.agents_execution_activity_file_added_payload import AgentsExecutionActivityFileAddedPayload
+from asteroid_odyssey.models.agents_execution_activity_generic_payload import AgentsExecutionActivityGenericPayload
+from asteroid_odyssey.models.agents_execution_activity_playwright_script_generated_payload import AgentsExecutionActivityPlaywrightScriptGeneratedPayload
+from asteroid_odyssey.models.agents_execution_activity_reasoning_payload import AgentsExecutionActivityReasoningPayload
+from asteroid_odyssey.models.agents_execution_activity_status_changed_payload import AgentsExecutionActivityStatusChangedPayload
+from asteroid_odyssey.models.agents_execution_activity_step_completed_payload import AgentsExecutionActivityStepCompletedPayload
+from asteroid_odyssey.models.agents_execution_activity_step_started_payload import AgentsExecutionActivityStepStartedPayload
+from asteroid_odyssey.models.agents_execution_activity_transitioned_node_payload import AgentsExecutionActivityTransitionedNodePayload
+from asteroid_odyssey.models.agents_execution_activity_user_message_received_payload import AgentsExecutionActivityUserMessageReceivedPayload
+from asteroid_odyssey.models.agents_execution_activity_workflow_updated_payload import AgentsExecutionActivityWorkflowUpdatedPayload
+from asteroid_odyssey.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

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_playwright_script_generated_payload.py

@@ -19,7 +19,7 @@ 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 asteroid_odyssey.models.agents_graph_models_nodes_properties_playwright_script_llm_var import AgentsGraphModelsNodesPropertiesPlaywrightScriptLLMVar
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,13 +28,13 @@ 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"]
+    llm_vars: Optional[List[AgentsGraphModelsNodesPropertiesPlaywrightScriptLLMVar]] = Field(default=None, alias="llmVars")
+    context: StrictStr
+    old_script: Optional[StrictStr] = Field(default=None, alias="oldScript")
+    __properties: ClassVar[List[str]] = ["activityType", "nodeId", "nodeName", "script", "llmVars", "context", "oldScript"]
 
     @field_validator('activity_type')
     def activity_type_validate_enum(cls, value):
@@ -102,12 +102,12 @@ class AgentsExecutionActivityPlaywrightScriptGeneratedPayload(BaseModel):
 
         _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")
+            "script": obj.get("script"),
+            "llmVars": [AgentsGraphModelsNodesPropertiesPlaywrightScriptLLMVar.from_dict(_item) for _item in obj["llmVars"]] if obj.get("llmVars") is not None else None,
+            "context": obj.get("context"),
+            "oldScript": obj.get("oldScript")
         })
         return _obj
 

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_status_changed_payload.py

@@ -19,12 +19,12 @@ 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_execution_awaiting_confirmation_payload import AgentsExecutionAwaitingConfirmationPayload
-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_paused_payload import AgentsExecutionPausedPayload
-from asteroid_odyssey.agents_v2_gen.models.agents_execution_status import AgentsExecutionStatus
+from asteroid_odyssey.models.agents_execution_awaiting_confirmation_payload import AgentsExecutionAwaitingConfirmationPayload
+from asteroid_odyssey.models.agents_execution_cancelled_payload import AgentsExecutionCancelledPayload
+from asteroid_odyssey.models.agents_execution_completed_payload import AgentsExecutionCompletedPayload
+from asteroid_odyssey.models.agents_execution_failed_payload import AgentsExecutionFailedPayload
+from asteroid_odyssey.models.agents_execution_paused_payload import AgentsExecutionPausedPayload
+from asteroid_odyssey.models.agents_execution_status import AgentsExecutionStatus
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -33,13 +33,13 @@ 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")
+    status: AgentsExecutionStatus
     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]] = ["activityType", "awaitingConfirmationPayload", "cancelledPayload", "completedPayload", "failedPayload", "pausedPayload", "status"]
+    awaiting_confirmation_payload: Optional[AgentsExecutionAwaitingConfirmationPayload] = Field(default=None, alias="awaitingConfirmationPayload")
+    cancelled_payload: Optional[AgentsExecutionCancelledPayload] = Field(default=None, alias="cancelledPayload")
+    __properties: ClassVar[List[str]] = ["activityType", "status", "completedPayload", "failedPayload", "pausedPayload", "awaitingConfirmationPayload", "cancelledPayload"]
 
     @field_validator('activity_type')
     def activity_type_validate_enum(cls, value):
@@ -87,12 +87,6 @@ class AgentsExecutionActivityStatusChangedPayload(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of awaiting_confirmation_payload
-        if self.awaiting_confirmation_payload:
-            _dict['awaitingConfirmationPayload'] = self.awaiting_confirmation_payload.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of cancelled_payload
-        if self.cancelled_payload:
-            _dict['cancelledPayload'] = self.cancelled_payload.to_dict()
         # override the default output from pydantic by calling `to_dict()` of completed_payload
         if self.completed_payload:
             _dict['completedPayload'] = self.completed_payload.to_dict()
@@ -102,6 +96,12 @@ class AgentsExecutionActivityStatusChangedPayload(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of paused_payload
         if self.paused_payload:
             _dict['pausedPayload'] = self.paused_payload.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of awaiting_confirmation_payload
+        if self.awaiting_confirmation_payload:
+            _dict['awaitingConfirmationPayload'] = self.awaiting_confirmation_payload.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of cancelled_payload
+        if self.cancelled_payload:
+            _dict['cancelledPayload'] = self.cancelled_payload.to_dict()
         return _dict
 
     @classmethod
@@ -115,12 +115,12 @@ class AgentsExecutionActivityStatusChangedPayload(BaseModel):
 
         _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,
+            "status": obj.get("status"),
             "completedPayload": AgentsExecutionCompletedPayload.from_dict(obj["completedPayload"]) if obj.get("completedPayload") is not None else None,
             "failedPayload": AgentsExecutionFailedPayload.from_dict(obj["failedPayload"]) if obj.get("failedPayload") is not None else None,
             "pausedPayload": AgentsExecutionPausedPayload.from_dict(obj["pausedPayload"]) if obj.get("pausedPayload") is not None else None,
-            "status": obj.get("status")
+            "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
         })
         return _obj
 

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_transitioned_node_payload.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from asteroid_odyssey.agents_v2_gen.models.agents_graph_models_transitions_transition_type import AgentsGraphModelsTransitionsTransitionType
+from asteroid_odyssey.models.agents_graph_models_transitions_transition_type import AgentsGraphModelsTransitionsTransitionType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,12 +28,12 @@ class AgentsExecutionActivityTransitionedNodePayload(BaseModel):
     AgentsExecutionActivityTransitionedNodePayload
     """ # noqa: E501
     activity_type: StrictStr = Field(alias="activityType")
-    from_node_duration: Optional[StrictInt] = Field(default=None, alias="fromNodeDuration")
+    new_node_uuid: StrictStr = Field(alias="newNodeUUID")
     new_node_name: StrictStr = Field(alias="newNodeName")
     new_node_type: StrictStr = Field(alias="newNodeType")
-    new_node_uuid: StrictStr = Field(alias="newNodeUUID")
+    from_node_duration: Optional[StrictInt] = Field(default=None, alias="fromNodeDuration")
     transition_type: Optional[AgentsGraphModelsTransitionsTransitionType] = Field(default=None, alias="transitionType")
-    __properties: ClassVar[List[str]] = ["activityType", "fromNodeDuration", "newNodeName", "newNodeType", "newNodeUUID", "transitionType"]
+    __properties: ClassVar[List[str]] = ["activityType", "newNodeUUID", "newNodeName", "newNodeType", "fromNodeDuration", "transitionType"]
 
     @field_validator('activity_type')
     def activity_type_validate_enum(cls, value):
@@ -94,10 +94,10 @@ class AgentsExecutionActivityTransitionedNodePayload(BaseModel):
 
         _obj = cls.model_validate({
             "activityType": obj.get("activityType"),
-            "fromNodeDuration": obj.get("fromNodeDuration"),
+            "newNodeUUID": obj.get("newNodeUUID"),
             "newNodeName": obj.get("newNodeName"),
             "newNodeType": obj.get("newNodeType"),
-            "newNodeUUID": obj.get("newNodeUUID"),
+            "fromNodeDuration": obj.get("fromNodeDuration"),
             "transitionType": obj.get("transitionType")
         })
         return _obj

asteroid_odyssey/{agents_v2_gen/models → models}/agents_execution_activity_workflow_updated_payload.py

@@ -19,7 +19,7 @@ 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_workflow_update import AgentsExecutionWorkflowUpdate
+from asteroid_odyssey.models.agents_execution_workflow_update import AgentsExecutionWorkflowUpdate
 from typing import Optional, Set
 from typing_extensions import Self
 

asteroid_odyssey/models/agents_execution_agent_query_context_completed_details.py

@@ -0,0 +1,98 @@
+# 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 typing import Optional, Set
+from typing_extensions import Self
+
+class AgentsExecutionAgentQueryContextCompletedDetails(BaseModel):
+    """
+    AgentsExecutionAgentQueryContextCompletedDetails
+    """ # noqa: E501
+    action_name: StrictStr = Field(alias="actionName")
+    query: StrictStr
+    answer: StrictStr
+    __properties: ClassVar[List[str]] = ["actionName", "query", "answer"]
+
+    @field_validator('action_name')
+    def action_name_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['agent_query_context']):
+            raise ValueError("must be one of enum values ('agent_query_context')")
+        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 AgentsExecutionAgentQueryContextCompletedDetails 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,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentsExecutionAgentQueryContextCompletedDetails from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "actionName": obj.get("actionName"),
+            "query": obj.get("query"),
+            "answer": obj.get("answer")
+        })
+        return _obj
+
+

asteroid_odyssey/models/agents_execution_agent_query_context_started_details.py

@@ -0,0 +1,96 @@
+# 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 typing import Optional, Set
+from typing_extensions import Self
+
+class AgentsExecutionAgentQueryContextStartedDetails(BaseModel):
+    """
+    AgentsExecutionAgentQueryContextStartedDetails
+    """ # noqa: E501
+    action_name: StrictStr = Field(alias="actionName")
+    query: StrictStr
+    __properties: ClassVar[List[str]] = ["actionName", "query"]
+
+    @field_validator('action_name')
+    def action_name_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['agent_query_context']):
+            raise ValueError("must be one of enum values ('agent_query_context')")
+        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 AgentsExecutionAgentQueryContextStartedDetails 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,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AgentsExecutionAgentQueryContextStartedDetails from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "actionName": obj.get("actionName"),
+            "query": obj.get("query")
+        })
+        return _obj
+
+