amigo_sdk 0.11.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.11.0"
+__version__ = "0.12.0"
 from .sdk_client import AmigoClient, AsyncAmigoClient
 
 __all__ = ["__version__", "AmigoClient", "AsyncAmigoClient"]

amigo_sdk/generated/model.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  <stdin>
-#   timestamp: 2025-08-27T15:09:46+00:00
+#   timestamp: 2025-08-27T16:38:45+00:00
 
 from __future__ import annotations
 
@@ -8847,12 +8847,6 @@ class SendEmailSideEffect(BaseModel):
     )
 
 
-class SideEffect(RootModel[Union[SendEmailSideEffect, EmitEventSideEffect]]):
-    root: Union[SendEmailSideEffect, EmitEventSideEffect] = Field(
-        ..., discriminator='type'
-    )
-
-
 class SideEffectStateNextState(AnnotationStateInputNextState):
     pass
 
@@ -8860,7 +8854,9 @@ class SideEffectStateNextState(AnnotationStateInputNextState):
 class SideEffectState(BaseModel):
     type: Literal['side-effect'] = Field(..., title='Type')
     next_state: Union[str, SideEffectStateNextState] = Field(..., title='Next State')
-    side_effect: SideEffect
+    side_effect: Union[SendEmailSideEffect, EmitEventSideEffect] = Field(
+        ..., discriminator='type'
+    )
 
 
 class SideEffectStateTransitionLog(BaseModel):
@@ -8958,18 +8954,10 @@ class StateOrRefName(RootModel[str]):
     root: str = Field(..., pattern='^[A-Za-z0-9_]+$')
 
 
-class StrippedNonemptyStringAZ(RootModel[str]):
-    root: str = Field(..., min_length=1, pattern='[A-Z_]+')
-
-
 class StrippedNonemptyStringWS(RootModel[str]):
     root: str = Field(..., min_length=1, pattern='^[\\w\\s]+$')
 
 
-class StrippedNonemptyStringAZ09(RootModel[str]):
-    root: str = Field(..., min_length=1, pattern='^[a-z0-9_]+$')
-
-
 class Tag(BaseModel):
     key: str = Field(..., title='Key')
     value: Optional[str] = Field(..., title='Value')
@@ -9195,8 +9183,11 @@ class UnitTestRunSpecPersonaVersionInfo(RootModel[List]):
 
 
 class UpdateRequest(BaseModel):
-    name: StrippedNonemptyStringAZ = Field(
-        ..., description='The name of the environment variable to update.'
+    name: str = Field(
+        ...,
+        description='The name of the environment variable to update.',
+        min_length=1,
+        pattern='[A-Z_]+',
     )
     value: str = Field(
         ...,
@@ -9210,14 +9201,6 @@ class UserDimensionOutput(BaseModel):
     tags: List[str] = Field(..., title='Tags')
 
 
-class UserMessageAudioConfig(
-    RootModel[Union[MP3UserMessageAudioConfig, PCMUserMessageAudioConfig]]
-):
-    root: Union[MP3UserMessageAudioConfig, PCMUserMessageAudioConfig] = Field(
-        ..., discriminator='type'
-    )
-
-
 class UserMessageAvailableEvent(BaseModel):
     type: Literal['user-message-available'] = Field(
         'user-message-available', title='Type'
@@ -9334,10 +9317,6 @@ class WebhookDestination(BaseModel):
     )
 
 
-class WebhookType(RootModel[Literal['conversation-post-processing-complete']]):
-    root: Literal['conversation-post-processing-complete']
-
-
 class WorkingMemory(BaseModel):
     content: str = Field(..., description='The content of the memory.', title='Content')
     context: str = Field(..., description='The context of the memory.', title='Context')
@@ -9464,10 +9443,6 @@ class AmigoLibPydanticBaseModelStrippedNonemptyString2(
     pass
 
 
-class AmigoLibPydanticBaseModelStrippedNonemptyString3(RootModel[str]):
-    root: str = Field(..., max_length=256, min_length=1)
-
-
 class AmigoLibPydanticBaseModelStrippedNonemptyString4(RootModel[str]):
     root: str = Field(..., max_length=3, min_length=1)
 
@@ -9541,8 +9516,10 @@ class ConversationGenerateConversationStarterRequest(BaseModel):
         le=10,
         title='Max Count',
     )
-    generation_instructions: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The instruction for the conversation starter generation.'
+    generation_instructions: str = Field(
+        ...,
+        description='The instruction for the conversation starter generation.',
+        min_length=1,
     )
 
 
@@ -9580,7 +9557,7 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionInjectInstr
     BaseModel
 ):
     type: Literal['inject-instruction'] = Field(..., title='Type')
-    instruction: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    instruction: str = Field(..., min_length=1)
     overrides_instructions: bool = Field(
         ...,
         description='During injection, whether the original instruction of the state is overriden with this instruction.',
@@ -9600,7 +9577,7 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestInjectInstructionA
     BaseModel
 ):
     type: Literal['inject-instruction'] = Field(..., title='Type')
-    instruction: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    instruction: str = Field(..., min_length=1)
     overrides_instructions: bool = Field(..., title='Overrides Instructions')
 
 
@@ -9796,8 +9773,8 @@ class MetricUpdateMetricRequest(BaseModel):
 
 
 class OrganizationCreateAgentRequest(BaseModel):
-    agent_name: AmigoLibPydanticBaseModelStrippedNonemptyString3 = Field(
-        ..., description='The name of the new agent.'
+    agent_name: str = Field(
+        ..., description='The name of the new agent.', max_length=256, min_length=1
     )
 
 
@@ -10483,8 +10460,11 @@ class OrganizationCreateOrganizationResponse(BaseModel):
 
 
 class OrganizationCreateServiceHierarchicalStateMachineRequest(BaseModel):
-    state_machine_name: AmigoLibPydanticBaseModelStrippedNonemptyString3 = Field(
-        ..., description='The name of the new state machine.'
+    state_machine_name: str = Field(
+        ...,
+        description='The name of the new state machine.',
+        max_length=256,
+        min_length=1,
     )
 
 
@@ -10703,8 +10683,8 @@ class ServiceUpdateServiceRequest(BaseModel):
 
 
 class SimulationCreateSimulationPersonaRequestInitialVersion(BaseModel):
-    background: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The background of the simulation persona.'
+    background: str = Field(
+        ..., description='The background of the simulation persona.', min_length=1
     )
     preferred_language: Optional[
         Union[IdentityInputDefaultSpokenLanguage, FieldNotSet]
@@ -10731,8 +10711,8 @@ class SimulationCreateSimulationPersonaResponse(BaseModel):
 
 
 class SimulationCreateSimulationPersonaVersionRequest(BaseModel):
-    background: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The background of the simulation persona.'
+    background: str = Field(
+        ..., description='The background of the simulation persona.', min_length=1
     )
     preferred_language: Optional[
         Union[IdentityInputDefaultSpokenLanguage, FieldNotSet]
@@ -10783,8 +10763,8 @@ class SimulationCreateSimulationUnitTestResponse(BaseModel):
 
 
 class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The name of the simulation unit test set.'
+    name: str = Field(
+        ..., description='The name of the simulation unit test set.', min_length=1
     )
     description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         ..., description='The description of the simulation unit test set.'
@@ -11052,12 +11032,14 @@ class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
 
 
 class ToolCreateToolRequest(BaseModel):
-    name: StrippedNonemptyStringAZ09 = Field(
+    name: str = Field(
         ...,
         description='The name of the tool. It must be unique among all non-deprecated tools in the organization.',
+        min_length=1,
+        pattern='^[a-z0-9_]+$',
     )
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='A description of the tool.'
+    description: str = Field(
+        ..., description='A description of the tool.', min_length=1
     )
     tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
         ..., description='The tags of this tool.', title='Tags'
@@ -11288,7 +11270,7 @@ class WebhookDestinationCreateWebhookDestinationRequest(BaseModel):
         description='The URL to which the webhook will be sent. The URL must be in HTTPS.',
         title='Url',
     )
-    accepted_types: List[WebhookType] = Field(
+    accepted_types: List[Literal['conversation-post-processing-complete']] = Field(
         ..., description='A list of webhook types to receive.', title='Accepted Types'
     )
     retry_attempts: Optional[int] = Field(
@@ -11352,10 +11334,12 @@ class WebhookDestinationUpdateWebhookDestinationRequest(BaseModel):
         description='The number of attempts to retry sending the webhook event in case of failure. If not specified, this field is not updated.',
         title='Retry Attempts',
     )
-    accepted_types: Optional[List[WebhookType]] = Field(
-        None,
-        description='A list of webhook types to receive. If not specified, this field is not updated.',
-        title='Accepted Types',
+    accepted_types: Optional[List[Literal['conversation-post-processing-complete']]] = (
+        Field(
+            None,
+            description='A list of webhook types to receive. If not specified, this field is not updated.',
+            title='Accepted Types',
+        )
     )
 
 
@@ -11648,7 +11632,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         [], description='The IDs of the messages to retrieve.', title='Id'
     )
     message_type: Optional[List[MessageType]] = Field(
-        ['external-event', 'user-message', 'agent-message'],
+        ['user-message', 'agent-message', 'external-event'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -11689,7 +11673,9 @@ class InteractWithConversationParametersQuery(BaseModel):
         description='The format of the response that will be sent to the user.',
         title='Response Format',
     )
-    request_audio_config: Optional[UserMessageAudioConfig] = Field(
+    request_audio_config: Optional[
+        Union[MP3UserMessageAudioConfig, PCMUserMessageAudioConfig]
+    ] = Field(
         None,
         description='Configuration for the user message audio. This is only required if `request_format` is set to `voice`.',
         title='Request Audio Config',
@@ -11992,9 +11978,10 @@ class SearchDynamicBehaviorSetsParametersQueryCreator(RootModel[List[str]]):
 
 
 class SearchDynamicBehaviorSetsParametersQuery(BaseModel):
-    query: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    query: str = Field(
         ...,
         description='The query to search for. Any sets containing the terms in its name and the triggers of its latest version would be returned.',
+        min_length=1,
     )
     is_active: Optional[bool] = Field(
         None,
@@ -12135,9 +12122,10 @@ class SearchMetricsParametersQueryCreator(RootModel[List[str]]):
 
 
 class SearchMetricsParametersQuery(BaseModel):
-    query: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    query: str = Field(
         ...,
         description='The query to search for. Any metrics containing the terms in its name and description would be returned.',
+        min_length=1,
     )
     applied_to_service: Optional[SearchMetricsParametersQueryAppliedToService] = Field(
         None,
@@ -12401,9 +12389,10 @@ class GetSimulationScenariosParametersQuery(BaseModel):
 
 
 class SearchSimulationPersonasParametersQuery(BaseModel):
-    query: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    query: str = Field(
         ...,
         description='The query to search for. Any simulation personas containing the terms in its name and the background of its latest version would be returned.',
+        min_length=1,
     )
     role: Optional[List[str]] = Field(
         [], description='The roles of the simulation personas.', title='Role'
@@ -12421,9 +12410,10 @@ class SearchSimulationPersonasParametersQuery(BaseModel):
 
 
 class SearchSimulationScenariosParametersQuery(BaseModel):
-    query: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    query: str = Field(
         ...,
         description='The query to search for. Any simulation scenarios containing the terms in its name would be returned.',
+        min_length=1,
     )
     creator: Optional[List[str]] = Field(
         [],
@@ -12558,9 +12548,10 @@ class SearchSimulationUnitTestsParametersQueryMetricIdItem(
 
 
 class SearchSimulationUnitTestsParametersQuery(BaseModel):
-    query: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    query: str = Field(
         ...,
         description='The query to search for. Any simulation unit tests containing the terms in its name would be returned.',
+        min_length=1,
     )
     creator: Optional[List[str]] = Field(
         [],
@@ -12668,9 +12659,10 @@ class SearchSimulationUnitTestSetsParametersQueryUnitTestIdItem(
 
 
 class SearchSimulationUnitTestSetsParametersQuery(BaseModel):
-    query: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    query: str = Field(
         ...,
         description='The query to search for. Any simulation unit test sets containing the terms in its name or description would be returned.',
+        min_length=1,
     )
     creator: Optional[List[str]] = Field(
         [],
@@ -12960,7 +12952,7 @@ class ActionStateTransitionLog(BaseModel):
 
 class AnnotationStateInput(BaseModel):
     type: Literal['annotation'] = Field(..., title='Type')
-    inner_thought: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    inner_thought: str = Field(..., min_length=1)
     next_state: Union[StateOrRefName, AnnotationStateInputNextState] = Field(
         ..., title='Next State'
     )
@@ -12980,12 +12972,6 @@ class CategoricalMetricValueInput(BaseModel):
     )
 
 
-class Condition(RootModel[Union[EqualCondition, InCondition, NotEqualCondition]]):
-    root: Union[EqualCondition, InCondition, NotEqualCondition] = Field(
-        ..., discriminator='type'
-    )
-
-
 class DecisionStateOutput(BaseModel):
     type: Literal['decision'] = Field(..., title='Type')
     exit_conditions: List[ExitConditionOutput] = Field(
@@ -13051,8 +13037,11 @@ class DecisionStateTransitionLog(BaseModel):
 
 
 class DeleteRequest(BaseModel):
-    name: StrippedNonemptyStringAZ = Field(
-        ..., description='The name of the environment variable to delete.'
+    name: str = Field(
+        ...,
+        description='The name of the environment variable to delete.',
+        min_length=1,
+        pattern='[A-Z_]+',
     )
 
 
@@ -13081,26 +13070,8 @@ class DynamicBehaviorSetVersion(BaseModel):
     action: AmigoLibMongoCollectionsDynamicBehaviorSetVersionAction
 
 
-class EngageUserAgentActionEvent(
-    RootModel[
-        Union[
-            EngageUserDynamicBehaviorOverrideEvent,
-            EngageUserMessageFragmentGeneratedEvent,
-            ToolCallStartedEvent,
-            ToolCallEndedEvent,
-        ]
-    ]
-):
-    root: Union[
-        EngageUserDynamicBehaviorOverrideEvent,
-        EngageUserMessageFragmentGeneratedEvent,
-        ToolCallStartedEvent,
-        ToolCallEndedEvent,
-    ] = Field(..., discriminator='type')
-
-
 class ExitConditionInput(BaseModel):
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    description: str = Field(..., min_length=1)
     next_state: Union[StateOrRefName, ExitConditionInputNextState] = Field(
         ..., title='Next State'
     )
@@ -13130,8 +13101,8 @@ class IdentityOutput(BaseModel):
 
 class InitialExternalEventConfigInput(BaseModel):
     type: Literal['external-event'] = Field(..., title='Type')
-    message: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The message of the initial external event.'
+    message: str = Field(
+        ..., description='The message of the initial external event.', min_length=1
     )
 
 
@@ -13143,17 +13114,12 @@ class InitialMessageConfigInput(
     )
 
 
-class InitialMessageConfigOutput(
-    RootModel[Union[InitialUserMessageConfig, InitialExternalEventConfigOutput]]
-):
-    root: Union[InitialUserMessageConfig, InitialExternalEventConfigOutput] = Field(
-        ..., discriminator='type'
-    )
-
-
 class InsertRequest(BaseModel):
-    name: StrippedNonemptyStringAZ = Field(
-        ..., description='The name of the environment variable to insert.'
+    name: str = Field(
+        ...,
+        description='The name of the environment variable to insert.',
+        min_length=1,
+        pattern='[A-Z_]+',
     )
     value: str = Field(
         ...,
@@ -13246,100 +13212,6 @@ class MessageInstance(BaseModel):
     )
 
 
-class MetricTypeInput(
-    RootModel[
-        Union[
-            TotalActiveUserMetricInput,
-            TotalConversationCountMetricInput,
-            TotalMessageCountMetricInput,
-            ConversationByTimeMetricInput,
-            TotalARRMetricInput,
-        ]
-    ]
-):
-    root: Union[
-        TotalActiveUserMetricInput,
-        TotalConversationCountMetricInput,
-        TotalMessageCountMetricInput,
-        ConversationByTimeMetricInput,
-        TotalARRMetricInput,
-    ] = Field(..., discriminator='type')
-
-
-class MetricTypeOutput(
-    RootModel[
-        Union[
-            TotalActiveUserMetricOutput,
-            TotalConversationCountMetricOutput,
-            TotalMessageCountMetricOutput,
-            ConversationByTimeMetricOutput,
-            TotalARRMetricOutput,
-        ]
-    ]
-):
-    root: Union[
-        TotalActiveUserMetricOutput,
-        TotalConversationCountMetricOutput,
-        TotalMessageCountMetricOutput,
-        ConversationByTimeMetricOutput,
-        TotalARRMetricOutput,
-    ] = Field(..., discriminator='type')
-
-
-class MetricValueInput(
-    RootModel[
-        Union[NumericalMetricValue, CategoricalMetricValueInput, BooleanMetricValue]
-    ]
-):
-    root: Union[
-        NumericalMetricValue, CategoricalMetricValueInput, BooleanMetricValue
-    ] = Field(..., discriminator='type')
-
-
-class MetricValueOutput(
-    RootModel[
-        Union[BooleanMetricValue, NumericalMetricValue, CategoricalMetricValueOutput]
-    ]
-):
-    root: Union[
-        BooleanMetricValue, NumericalMetricValue, CategoricalMetricValueOutput
-    ] = Field(..., discriminator='type')
-
-
-class NavigateStateMachineAgentActionEvent(
-    RootModel[
-        Union[
-            NavigateStateMachineDynamicBehaviorOverrideEvent,
-            NavigateStateMachineActionStateActionSelectedEvent,
-            NavigateStateMachineExitConditionSelectedEvent,
-            NavigateStateMachineRecallStateDynamicQueriesGeneratedEvent,
-            NavigateStateMachineRecallStateStaticQueriesRetrievedEvent,
-            NavigateStateMachineRecallStateActiveMemoryRetrievedEvent,
-            NavigateStateMachineReflectionStatePromptRetrievedEvent,
-            NavigateStateMachineReflectionStateReflectionGeneratedEvent,
-            NavigateStateMachineAnnotationStateAnnotationRetrievedEvent,
-            NavigateStateMachineStateTransitionedEvent,
-            ToolCallStartedEvent,
-            ToolCallEndedEvent,
-        ]
-    ]
-):
-    root: Union[
-        NavigateStateMachineDynamicBehaviorOverrideEvent,
-        NavigateStateMachineActionStateActionSelectedEvent,
-        NavigateStateMachineExitConditionSelectedEvent,
-        NavigateStateMachineRecallStateDynamicQueriesGeneratedEvent,
-        NavigateStateMachineRecallStateStaticQueriesRetrievedEvent,
-        NavigateStateMachineRecallStateActiveMemoryRetrievedEvent,
-        NavigateStateMachineReflectionStatePromptRetrievedEvent,
-        NavigateStateMachineReflectionStateReflectionGeneratedEvent,
-        NavigateStateMachineAnnotationStateAnnotationRetrievedEvent,
-        NavigateStateMachineStateTransitionedEvent,
-        ToolCallStartedEvent,
-        ToolCallEndedEvent,
-    ] = Field(..., discriminator='type')
-
-
 class NumericalMetricSuccessCriterion(BaseModel):
     type: Literal['numerical'] = Field(..., title='Type')
     verb: Verb
@@ -13353,10 +13225,12 @@ class PermissionGrantInput(BaseModel):
         title='Action',
     )
     permission_name: str = Field(..., title='Permission Name')
-    conditions: Dict[str, Condition] = Field(
-        ...,
-        description='A dictionary of attribute name to condition that must be met for this grant to be applicable.',
-        title='Conditions',
+    conditions: Dict[str, Union[EqualCondition, InCondition, NotEqualCondition]] = (
+        Field(
+            ...,
+            description='A dictionary of attribute name to condition that must be met for this grant to be applicable.',
+            title='Conditions',
+        )
     )
     description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = None
 
@@ -13370,10 +13244,12 @@ class PermissionGrantOutput(BaseModel):
     permission_name: str = Field(
         ..., description='The name of the permission.', title='Permission Name'
     )
-    conditions: Dict[str, Condition] = Field(
-        ...,
-        description='A dictionary of attribute name to condition that must be met for this grant to be applicable.',
-        title='Conditions',
+    conditions: Dict[str, Union[EqualCondition, InCondition, NotEqualCondition]] = (
+        Field(
+            ...,
+            description='A dictionary of attribute name to condition that must be met for this grant to be applicable.',
+            title='Conditions',
+        )
     )
     description: Optional[str] = Field(
         None, description='An optional explanation of this grant.', title='Description'
@@ -13454,10 +13330,10 @@ class ReflectionStateTransitionLog(BaseModel):
 
 
 class RelationshipToDeveloperInput(BaseModel):
-    ownership: AmigoLibPydanticBaseModelStrippedNonemptyString1
-    type: AmigoLibPydanticBaseModelStrippedNonemptyString1
-    conversation_visibility: AmigoLibPydanticBaseModelStrippedNonemptyString1
-    thought_visibility: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    ownership: str = Field(..., min_length=1)
+    type: str = Field(..., min_length=1)
+    conversation_visibility: str = Field(..., min_length=1)
+    thought_visibility: str = Field(..., min_length=1)
 
 
 class RoleInstance(BaseModel):
@@ -13487,22 +13363,6 @@ class RoleInstance(BaseModel):
     )
 
 
-class SelectDynamicBehaviorAgentActionEvent(
-    RootModel[
-        Union[
-            SelectDynamicBehaviorVectorGeneratedEvent,
-            SelectDynamicBehaviorRelevanceRetrievalCompletedEvent,
-            SelectDynamicBehaviorCompletedEvent,
-        ]
-    ]
-):
-    root: Union[
-        SelectDynamicBehaviorVectorGeneratedEvent,
-        SelectDynamicBehaviorRelevanceRetrievalCompletedEvent,
-        SelectDynamicBehaviorCompletedEvent,
-    ] = Field(..., discriminator='type')
-
-
 class SimulationPersonaInstance(BaseModel):
     id: str = Field(..., description='The ID of the simulation persona.', title='Id')
     name: str = Field(
@@ -13593,7 +13453,9 @@ class SimulationScenarioVersion(BaseModel):
         description='The instructions for the simulation scenario.',
         title='Instructions',
     )
-    initial_message_config: Optional[InitialMessageConfigOutput] = None
+    initial_message_config: Optional[
+        Union[InitialUserMessageConfig, InitialExternalEventConfigOutput]
+    ] = None
 
 
 class SimulationUnitTestSet(BaseModel):
@@ -13672,52 +13534,6 @@ class SimulationUnitTestSetInstance(BaseModel):
     )
 
 
-class State(
-    RootModel[
-        Union[
-            ActionStateOutput,
-            DecisionStateOutput,
-            RecallStateOutput,
-            AnnotationStateOutput,
-            ReflectionStateOutput,
-            SideEffectState,
-        ]
-    ]
-):
-    root: Union[
-        ActionStateOutput,
-        DecisionStateOutput,
-        RecallStateOutput,
-        AnnotationStateOutput,
-        ReflectionStateOutput,
-        SideEffectState,
-    ] = Field(..., discriminator='type')
-
-
-class StateTransitionLog(
-    RootModel[
-        Union[
-            ActionStateTransitionLog,
-            DecisionStateTransitionLog,
-            RecallStateTransitionLog,
-            JumpbackStateTransitionLog,
-            AnnotationStateTransitionLog,
-            ReflectionStateTransitionLog,
-            SideEffectStateTransitionLog,
-        ]
-    ]
-):
-    root: Union[
-        ActionStateTransitionLog,
-        DecisionStateTransitionLog,
-        RecallStateTransitionLog,
-        JumpbackStateTransitionLog,
-        AnnotationStateTransitionLog,
-        ReflectionStateTransitionLog,
-        SideEffectStateTransitionLog,
-    ] = Field(..., discriminator='type')
-
-
 class SuccessCriterionDescriptionInput(
     RootModel[
         Union[
@@ -13734,22 +13550,6 @@ class SuccessCriterionDescriptionInput(
     ] = Field(..., discriminator='type')
 
 
-class SuccessCriterionDescriptionOutput(
-    RootModel[
-        Union[
-            BooleanMetricSuccessCriterion,
-            NumericalMetricSuccessCriterion,
-            CategoricalMetricSuccessCriterionOutput,
-        ]
-    ]
-):
-    root: Union[
-        BooleanMetricSuccessCriterion,
-        NumericalMetricSuccessCriterion,
-        CategoricalMetricSuccessCriterionOutput,
-    ] = Field(..., discriminator='type')
-
-
 class ToolCallSpecInput(BaseModel):
     tool_id: str = Field(
         ...,
@@ -13762,9 +13562,10 @@ class ToolCallSpecInput(BaseModel):
         description='A [Python packaging version constraint](https://packaging.python.org/en/latest/specifications/version-specifiers/#id5) of the tool.',
         title='Version Constraint',
     )
-    additional_instruction: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    additional_instruction: str = Field(
         ...,
         description="Additional instruction to be supplied to the LLM in addition to the tool's description.",
+        min_length=1,
     )
 
 
@@ -13844,7 +13645,7 @@ class UnitTestRunResult(BaseModel):
 
 
 class UserDimensionInput(BaseModel):
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    description: str = Field(..., min_length=1)
     tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         ..., title='Tags'
     )
@@ -13966,26 +13767,6 @@ class WebhookDeliveryInstance(BaseModel):
     )
 
 
-class SrcAppAmigoServiceConversationAction(
-    RootModel[
-        Union[
-            SelectDynamicBehaviorAgentActionEvent,
-            CheckIfActiveMemoryNeededAgentActionEvent,
-            ExtractActiveMemoriesAgentActionEvent,
-            NavigateStateMachineAgentActionEvent,
-            EngageUserAgentActionEvent,
-        ]
-    ]
-):
-    root: Union[
-        SelectDynamicBehaviorAgentActionEvent,
-        CheckIfActiveMemoryNeededAgentActionEvent,
-        ExtractActiveMemoriesAgentActionEvent,
-        NavigateStateMachineAgentActionEvent,
-        EngageUserAgentActionEvent,
-    ]
-
-
 class ConversationGetConversationMessagesResponse(BaseModel):
     messages: List[MessageInstance] = Field(
         ..., description='The retrieved messages.', title='Messages'
@@ -14016,7 +13797,17 @@ class ConversationGetInteractionInsightsResponse(BaseModel):
         description='The objective of the current state.',
         title='Current State Objective',
     )
-    state_transition_logs: List[StateTransitionLog] = Field(
+    state_transition_logs: List[
+        Union[
+            ActionStateTransitionLog,
+            DecisionStateTransitionLog,
+            RecallStateTransitionLog,
+            JumpbackStateTransitionLog,
+            AnnotationStateTransitionLog,
+            ReflectionStateTransitionLog,
+            SideEffectStateTransitionLog,
+        ]
+    ] = Field(
         ...,
         description='The state transition log of the interaction.',
         title='State Transition Logs',
@@ -14130,12 +13921,13 @@ class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
 
 
 class MetricCreateMetricRequest(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    name: str = Field(
         ...,
         description='The name of the metric. Must be unique within the organization.',
+        min_length=1,
     )
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The description of the metric.'
+    description: str = Field(
+        ..., description='The description of the metric.', min_length=1
     )
     applied_to_services: List[MetricCreateMetricRequestAppliedToService] = Field(
         ...,
@@ -14148,24 +13940,29 @@ class MetricCreateMetricRequest(BaseModel):
     tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
-    metric_value: MetricValueInput
+    metric_value: Union[
+        NumericalMetricValue, CategoricalMetricValueInput, BooleanMetricValue
+    ] = Field(..., discriminator='type')
 
 
 class OrganizationCreateOrganizationRequest(BaseModel):
-    org_name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='A human friendly name of the organization.'
+    org_name: str = Field(
+        ..., description='A human friendly name of the organization.', min_length=1
     )
-    title: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    title: str = Field(
         ...,
         description='An advertising tagline for the services offered by this organization.',
+        min_length=1,
     )
-    main_description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    main_description: str = Field(
         ...,
         description='A description of the services offered by this organization that is displayed on the login page for the Amigo frontend for this organization.',
+        min_length=1,
     )
-    sub_description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    sub_description: str = Field(
         ...,
         description='Additional descriptions of the services offered by this organization that is displayed below `main_description` in a smaller font.',
+        min_length=1,
     )
     user_dimensions: List[UserDimensionInput] = Field(
         ...,
@@ -14211,9 +14008,10 @@ class OrganizationCreateOrganizationRequest(BaseModel):
             title='Onboarding Instructions',
         )
     )
-    azure_devops_team_name: AmigoLibPydanticBaseModelStrippedNonemptyString2 = Field(
+    azure_devops_team_name: str = Field(
         ...,
         description="The name of the Azure DevOps team to create (if it doesn't exist) for this organization. All orgs with the same Azure DevOps team name will share access to the same Azure DevOps resources, namely\nthe source code for tools.",
+        min_length=1,
     )
 
 
@@ -14274,7 +14072,15 @@ class OrganizationGetApiKeysResponse(BaseModel):
 
 
 class OrganizationGetOrganizationMetricsRequest(BaseModel):
-    metrics: List[MetricTypeInput] = Field(
+    metrics: List[
+        Union[
+            TotalActiveUserMetricInput,
+            TotalConversationCountMetricInput,
+            TotalMessageCountMetricInput,
+            ConversationByTimeMetricInput,
+            TotalARRMetricInput,
+        ]
+    ] = Field(
         ...,
         description='A list of metrics to compute. Each type of metric can only appear once.',
         min_length=1,
@@ -14283,7 +14089,15 @@ class OrganizationGetOrganizationMetricsRequest(BaseModel):
 
 
 class OrganizationGetOrganizationMetricsResponse(BaseModel):
-    metrics: List[MetricTypeOutput] = Field(
+    metrics: List[
+        Union[
+            TotalActiveUserMetricOutput,
+            TotalConversationCountMetricOutput,
+            TotalMessageCountMetricOutput,
+            ConversationByTimeMetricOutput,
+            TotalARRMetricOutput,
+        ]
+    ] = Field(
         ...,
         description='The metrics that were requested, in the same order as they were requested.',
         title='Metrics',
@@ -14359,12 +14173,14 @@ class OrganizationModifyOrganizationRequest(BaseModel):
 
 
 class RoleCreateRoleRequest(BaseModel):
-    role_name: AmigoLibPydanticBaseModelStrippedNonemptyString3 = Field(
+    role_name: str = Field(
         ...,
         description='The name of the role to create. The role must have a max length of 256 characters.',
+        max_length=256,
+        min_length=1,
     )
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='A description about the role.'
+    description: str = Field(
+        ..., description='A description about the role.', min_length=1
     )
     permission_grants: List[PermissionGrantInput] = Field(
         ...,
@@ -14428,11 +14244,9 @@ class ServiceCreateServiceRequest(BaseModel):
         pattern='^[a-f0-9]{24}$',
         title='Agent Id',
     )
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The name of this service.'
-    )
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='A description of this service.'
+    name: str = Field(..., description='The name of this service.', min_length=1)
+    description: str = Field(
+        ..., description='A description of this service.', min_length=1
     )
     is_active: bool = Field(
         ...,
@@ -14453,11 +14267,11 @@ class ServiceUpsertServiceVersionSetRequest(BaseModel):
 
 
 class SimulationCreateSimulationPersonaRequest(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The name of the simulation persona.'
+    name: str = Field(
+        ..., description='The name of the simulation persona.', min_length=1
     )
-    role: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The role of the simulation persona.'
+    role: str = Field(
+        ..., description='The role of the simulation persona.', min_length=1
     )
     tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
         ...,
@@ -14470,13 +14284,15 @@ class SimulationCreateSimulationPersonaRequest(BaseModel):
 
 
 class SimulationCreateSimulationScenarioRequestInitialVersion(BaseModel):
-    objective: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The objective of the simulation scenario.'
+    objective: str = Field(
+        ..., description='The objective of the simulation scenario.', min_length=1
     )
-    instructions: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The instructions of the simulation scenario.'
+    instructions: str = Field(
+        ..., description='The instructions of the simulation scenario.', min_length=1
     )
-    initial_message_config: Optional[InitialMessageConfigInput] = Field(
+    initial_message_config: Optional[
+        Union[InitialUserMessageConfig, InitialExternalEventConfigInput]
+    ] = Field(
         ...,
         description='The initial message configuration for the simulation scenario version. If not specified, an agent message opens the simulated conversation.',
     )
@@ -14488,9 +14304,13 @@ SimulationCreateSimulationScenarioVersionRequest = (
 
 
 class SimulationCreateSimulationUnitTestRequestSuccessCriterion(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    name: str = Field(..., min_length=1)
     metric_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Metric Id')
-    criterion: SuccessCriterionDescriptionInput
+    criterion: Union[
+        NumericalMetricSuccessCriterion,
+        BooleanMetricSuccessCriterion,
+        CategoricalMetricSuccessCriterionInput,
+    ] = Field(..., discriminator='type')
 
 
 class SimulationGetSimulationPersonasResponse(BaseModel):
@@ -14655,12 +14475,10 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
 
 
 class UserCreateInvitedUserRequest(BaseModel):
-    first_name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The first name of the user.'
-    )
-    last_name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The last name of the user.'
+    first_name: str = Field(
+        ..., description='The first name of the user.', min_length=1
     )
+    last_name: str = Field(..., description='The last name of the user.', min_length=1)
     email: EmailStr = Field(
         ...,
         description='The email of the user. This email uniquely identifies the user in the organization.',
@@ -14720,7 +14538,7 @@ class WebhookDestinationGetWebhookDeliveriesResponse(BaseModel):
 
 class ActionStateInput(BaseModel):
     type: Literal['action'] = Field(..., title='Type')
-    objective: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    objective: str = Field(..., min_length=1)
     actions: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         ..., min_length=1, title='Actions'
     )
@@ -14798,9 +14616,35 @@ class Conversation(BaseModel):
 
 class CurrentAgentActionEvent(BaseModel):
     type: Literal['current-agent-action'] = Field('current-agent-action', title='Type')
-    action: SrcAppAmigoServiceConversationAction = Field(
-        ..., description='The type of action the agent is performing right now.'
-    )
+    action: Union[
+        CheckIfActiveMemoryNeededAgentActionEvent,
+        ExtractActiveMemoriesAgentActionEvent,
+        Union[
+            SelectDynamicBehaviorVectorGeneratedEvent,
+            SelectDynamicBehaviorRelevanceRetrievalCompletedEvent,
+            SelectDynamicBehaviorCompletedEvent,
+        ],
+        Union[
+            NavigateStateMachineDynamicBehaviorOverrideEvent,
+            NavigateStateMachineActionStateActionSelectedEvent,
+            NavigateStateMachineExitConditionSelectedEvent,
+            NavigateStateMachineRecallStateDynamicQueriesGeneratedEvent,
+            NavigateStateMachineRecallStateStaticQueriesRetrievedEvent,
+            NavigateStateMachineRecallStateActiveMemoryRetrievedEvent,
+            NavigateStateMachineReflectionStatePromptRetrievedEvent,
+            NavigateStateMachineReflectionStateReflectionGeneratedEvent,
+            NavigateStateMachineAnnotationStateAnnotationRetrievedEvent,
+            NavigateStateMachineStateTransitionedEvent,
+            ToolCallStartedEvent,
+            ToolCallEndedEvent,
+        ],
+        Union[
+            EngageUserDynamicBehaviorOverrideEvent,
+            EngageUserMessageFragmentGeneratedEvent,
+            ToolCallStartedEvent,
+            ToolCallEndedEvent,
+        ],
+    ] = Field(..., description='The type of action the agent is performing right now.')
 
 
 class DecisionStateInput(BaseModel):
@@ -14809,28 +14653,14 @@ class DecisionStateInput(BaseModel):
     decision_guidelines: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         ..., title='Decision Guidelines'
     )
-    objective: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    objective: str = Field(..., min_length=1)
     tool_call_specs: List[ToolCallSpecInput] = Field(..., title='Tool Call Specs')
 
 
-class EvaluationSource(
-    RootModel[
-        Union[
-            PostSessionEvaluationSource,
-            ManualEvaluationSource,
-            SimulationEvaluationSource,
-        ]
-    ]
-):
-    root: Union[
-        PostSessionEvaluationSource, ManualEvaluationSource, SimulationEvaluationSource
-    ] = Field(..., discriminator='type')
-
-
 class IdentityInput(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1
-    role: AmigoLibPydanticBaseModelStrippedNonemptyString1
-    developed_by: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    name: str = Field(..., min_length=1)
+    role: str = Field(..., min_length=1)
+    developed_by: str = Field(..., min_length=1)
     default_spoken_language: IdentityInputDefaultSpokenLanguage = Field(
         ..., title='Default Spoken Language'
     )
@@ -14863,8 +14693,12 @@ class Metric(BaseModel):
     updated_by: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
         ..., description='The user who last updated the metric.'
     )
-    metric_value: MetricValueOutput = Field(
-        ..., description='The type of value that this metric is measuring.'
+    metric_value: Union[
+        BooleanMetricValue, NumericalMetricValue, CategoricalMetricValueOutput
+    ] = Field(
+        ...,
+        description='The type of value that this metric is measuring.',
+        discriminator='type',
     )
     is_deleted: bool = Field(
         ..., description='Whether the metric has been deleted.', title='Is Deleted'
@@ -14891,7 +14725,9 @@ class MetricEvaluationResultInstance(BaseModel):
         description='The justification for the result of the metric evaluation.',
         title='Justification',
     )
-    source: EvaluationSource
+    source: Union[
+        PostSessionEvaluationSource, ManualEvaluationSource, SimulationEvaluationSource
+    ] = Field(..., discriminator='type')
     timestamp: AwareDatetime = Field(
         ...,
         description='The time at which the metric evaluation result was created.',
@@ -14925,8 +14761,12 @@ class MetricInstance(BaseModel):
     updated_by: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
         ..., description='The user who last updated the metric.'
     )
-    metric_value: MetricValueOutput = Field(
-        ..., description='The type of value that this metric is measuring.'
+    metric_value: Union[
+        BooleanMetricValue, NumericalMetricValue, CategoricalMetricValueOutput
+    ] = Field(
+        ...,
+        description='The type of value that this metric is measuring.',
+        discriminator='type',
     )
     is_deleted: bool = Field(
         ..., description='Whether the metric has been deleted.', title='Is Deleted'
@@ -14940,14 +14780,20 @@ class MetricToEvaluate(BaseModel):
         description='The name of the success criterion.',
         title='Success Criterion Name',
     )
-    success_criterion: SuccessCriterionDescriptionOutput = Field(
-        ..., description='The description of the success criterion.'
+    success_criterion: Union[
+        BooleanMetricSuccessCriterion,
+        NumericalMetricSuccessCriterion,
+        CategoricalMetricSuccessCriterionOutput,
+    ] = Field(
+        ...,
+        description='The description of the success criterion.',
+        discriminator='type',
     )
 
 
 class ReflectionStateInput(BaseModel):
     type: Literal['reflection'] = Field(..., title='Type')
-    problem: AmigoLibPydanticBaseModelStrippedNonemptyString1
+    problem: str = Field(..., min_length=1)
     word_limit: int = Field(..., gt=0, title='Word Limit')
     next_state: Union[StateOrRefName, ReflectionStateInputNextState] = Field(
         ..., title='Next State'
@@ -14980,9 +14826,17 @@ class ServiceHierarchicalStateMachineInstance(BaseModel):
     description: str = Field(
         ..., description='A description of the state machine.', title='Description'
     )
-    states: Dict[str, State] = Field(
-        ..., description='The states in this state machine.', title='States'
-    )
+    states: Dict[
+        str,
+        Union[
+            ActionStateOutput,
+            DecisionStateOutput,
+            RecallStateOutput,
+            AnnotationStateOutput,
+            ReflectionStateOutput,
+            SideEffectState,
+        ],
+    ] = Field(..., description='The states in this state machine.', title='States')
     new_user_initial_state: str = Field(
         ...,
         description='The state a new user will be in when a session starts. This must be an action state.',
@@ -15048,7 +14902,11 @@ class ServiceInstance(BaseModel):
 class SuccessCriterionOutput(BaseModel):
     name: str = Field(..., title='Name')
     metric_id: str = Field(..., title='Metric Id')
-    criterion: SuccessCriterionDescriptionOutput
+    criterion: Union[
+        BooleanMetricSuccessCriterion,
+        NumericalMetricSuccessCriterion,
+        CategoricalMetricSuccessCriterionOutput,
+    ] = Field(..., discriminator='type')
 
 
 class UnitTestRunSpec(BaseModel):
@@ -15140,8 +14998,8 @@ class ConversationGetConversationsResponse(BaseModel):
 
 
 class DynamicBehaviorSetCreateDynamicBehaviorSetRequest(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The name of the dynamic behavior set.'
+    name: str = Field(
+        ..., description='The name of the dynamic behavior set.', min_length=1
     )
     tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
         ...,
@@ -15241,8 +15099,10 @@ class OrganizationCreateAgentVersionRequest(BaseModel):
 
 
 class OrganizationCreateServiceHierarchicalStateMachineVersionRequest(BaseModel):
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='A description of the service hierarchical state machine.'
+    description: str = Field(
+        ...,
+        description='A description of the service hierarchical state machine.',
+        min_length=1,
     )
     states: Dict[
         str,
@@ -15258,17 +15118,20 @@ class OrganizationCreateServiceHierarchicalStateMachineVersionRequest(BaseModel)
         description='The internal states in this service hierarchical state machine.',
         title='States',
     )
-    new_user_initial_state: StateOrRefName = Field(
+    new_user_initial_state: str = Field(
         ...,
         description='The state a new user will be in when a session starts. This must be an action state, and must be an internal state.',
+        pattern='^[A-Za-z0-9_]+$',
     )
-    returning_user_initial_state: StateOrRefName = Field(
+    returning_user_initial_state: str = Field(
         ...,
         description='The state a returning user will be in when a session starts. This must be an action state, and must be an internal state.',
+        pattern='^[A-Za-z0-9_]+$',
     )
-    terminal_state: StateOrRefName = Field(
+    terminal_state: str = Field(
         ...,
         description='The state the user will be in when the session ends. This must be an action state, and must be an internal state.',
+        pattern='^[A-Za-z0-9_]+$',
     )
     references: Dict[str, List] = Field(
         ...,
@@ -15353,8 +15216,8 @@ class ServiceGetServicesResponse(BaseModel):
 
 
 class SimulationCreateSimulationScenarioRequest(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The name of the simulation scenario.'
+    name: str = Field(
+        ..., description='The name of the simulation scenario.', min_length=1
     )
     tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
         ...,
@@ -15367,11 +15230,9 @@ class SimulationCreateSimulationScenarioRequest(BaseModel):
 
 
 class SimulationCreateSimulationUnitTestRequest(BaseModel):
-    name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='The name of the unit test.'
-    )
-    description: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
-        ..., description='A description of the unit test.'
+    name: str = Field(..., description='The name of the unit test.', min_length=1)
+    description: str = Field(
+        ..., description='A description of the unit test.', min_length=1
     )
     service_id: str = Field(
         ...,
@@ -15469,24 +15330,6 @@ class SimulationUpdateSimulationUnitTestRequest(BaseModel):
     )
 
 
-class ConversationEvent(
-    RootModel[
-        Union[
-            InteractionCompleteEvent,
-            EndSessionEvent,
-            NewMessageEvent,
-            CurrentAgentActionEvent,
-        ]
-    ]
-):
-    root: Union[
-        InteractionCompleteEvent,
-        EndSessionEvent,
-        NewMessageEvent,
-        CurrentAgentActionEvent,
-    ]
-
-
 class SimulationUnitTest(BaseModel):
     field_id: str = Field(
         ..., alias='_id', description='The ID of the document.', title='Id'
@@ -15624,9 +15467,29 @@ class SimulationUnitTestSetRunInstance(BaseModel):
 
 
 class ConversationInteractWithConversationResponse(
-    RootModel[Union[ConversationEvent, UserMessageAvailableEvent, ErrorEvent]]
+    RootModel[
+        Union[
+            UserMessageAvailableEvent,
+            ErrorEvent,
+            Union[
+                InteractionCompleteEvent,
+                EndSessionEvent,
+                NewMessageEvent,
+                CurrentAgentActionEvent,
+            ],
+        ]
+    ]
 ):
-    root: Union[ConversationEvent, UserMessageAvailableEvent, ErrorEvent]
+    root: Union[
+        UserMessageAvailableEvent,
+        ErrorEvent,
+        Union[
+            InteractionCompleteEvent,
+            EndSessionEvent,
+            NewMessageEvent,
+            CurrentAgentActionEvent,
+        ],
+    ]
 
 
 class SimulationGetSimulationUnitTestSetRunsResponse(BaseModel):

{amigo_sdk-0.11.0.dist-info → amigo_sdk-0.12.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amigo_sdk
-Version: 0.11.0
+Version: 0.12.0
 Summary: Amigo AI Python SDK
 Author: Amigo AI
 License-File: LICENSE

{amigo_sdk-0.11.0.dist-info → amigo_sdk-0.12.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-amigo_sdk/__init__.py,sha256=MNEge8Pwx6KfzTaKsnEQdwJ4t9xpa5JWOMZude9UL70,139
+amigo_sdk/__init__.py,sha256=IEpHGle8VT0uoku9xjW1s95C5r-iuRhIXGH8muORDlY,139
 amigo_sdk/_retry_utils.py,sha256=kFjw9Wqye6MB5-B4rjLxsbSNcfYBIztcollIoncd1hY,2142
 amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
 amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
@@ -6,13 +6,13 @@ amigo_sdk/errors.py,sha256=RkRyF5eAASd8fIOS6YvL9rLDvLAYWqHfpHSCR7jqvl4,4840
 amigo_sdk/http_client.py,sha256=v25UoUbXcMeHTnfJMcrl8RSSwCVkKUL1Jv-0HoXP1B4,13507
 amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
 amigo_sdk/sdk_client.py,sha256=Kr9M9o66pOLu0T2VDvqdYMmPZzgKJyTELu7BSPgGrYQ,6152
-amigo_sdk/generated/model.py,sha256=1iw7NL8cMOWN8KcYP_P3oai0JMWgwhjSWVEZAb2BWy4,427233
+amigo_sdk/generated/model.py,sha256=_AwOzuy1mAzX8yNzK35ojD8OYKLUE768A8uDTJECKMY,421774
 amigo_sdk/resources/conversation.py,sha256=5PkJOvLKqnriSS9K9hKw2VRPxRLTuABEbCyPy1fz1r0,14817
 amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
 amigo_sdk/resources/service.py,sha256=SiwEHXCQk4r1b_tGv47M08VuB7RALDHJQzWlpuD937g,1571
 amigo_sdk/resources/user.py,sha256=i4t5aVzBI37KwAtLKSDWTMwf4D4KQdSDoUiblFe1u7o,3529
-amigo_sdk-0.11.0.dist-info/METADATA,sha256=blllxaFKFVLxLkoEmft0TnniC8_6Qg_yEi_Xwgc2uO4,8350
-amigo_sdk-0.11.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-amigo_sdk-0.11.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.11.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.11.0.dist-info/RECORD,,
+amigo_sdk-0.12.0.dist-info/METADATA,sha256=cVAo-SpuzcIHoBeomVa7KFCAKiWIxZThdcQbWj7lDqw,8350
+amigo_sdk-0.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+amigo_sdk-0.12.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.12.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.12.0.dist-info/RECORD,,