amigo_sdk 0.67.0__py3-none-any.whl → 0.70.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.67.0"
+__version__ = "0.70.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-12-02T08:21:19+00:00
+#   timestamp: 2025-12-03T16:58:45+00:00
 
 from __future__ import annotations
 
@@ -33,12 +33,12 @@ class ActionTooLongEvent(BaseModel):
     )
 
 
-class AnnotationStateInputNextState(RootModel[List]):
-    root: List = Field(..., max_length=2, min_length=2, title='Next State')
+class AnnotationStateInputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(..., max_length=2, min_length=2, title='Next State')
 
 
-class AnnotationStateOutputNextState(RootModel[List]):
-    root: List = Field(
+class AnnotationStateOutputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description="The state to transition to after the inner thought message is generated. If it's a string, the session will transition to a state within this contextual graph. If it's a tuple, the first element\nmust be of the form `{external_contextual_graph_reference_name}.{state_name}` and represents a state in another contextual graph that the session will transition to, and the second element must be a state\nin this contextual graph that the session will transition to after reaching the terminal state of the external graph.",
         max_length=2,
@@ -68,7 +68,7 @@ class AnnotationStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -78,7 +78,7 @@ class AnnotationStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -154,8 +154,8 @@ class ConversationByTimeMetricInput(BaseModel):
     )
 
 
-class ConversationByTimeMetricOutputConversationCount(RootModel[List]):
-    root: List = Field(..., max_length=2, min_length=2)
+class ConversationByTimeMetricOutputConversationCount(RootModel[List[Any]]):
+    root: List[Any] = Field(..., max_length=2, min_length=2)
 
 
 class ConversationByTimeMetricOutput(BaseModel):
@@ -224,7 +224,7 @@ class DynamicBehaviorInvocation(BaseModel):
         description='The IDs of the agent inner thoughts that led to the dynamic behavior invocation.',
         title='Agent Inner Thoughts Ids',
     )
-    dynamic_behavior_set_version_info: List = Field(
+    dynamic_behavior_set_version_info: List[Any] = Field(
         ...,
         description='The ID and version number of the dynamic behavior set version that was activated.',
         max_length=2,
@@ -260,7 +260,7 @@ class EngageUserDynamicBehaviorOverrideEvent(BaseModel):
 
 class EngageUserInvocationMetadata(BaseModel):
     type: Literal['engage-user'] = Field('engage-user', title='Type')
-    current_state_machine_and_version: List = Field(
+    current_state_machine_and_version: List[Any] = Field(
         ...,
         description='The ID and version of the state machine that is currently being executed.',
         max_length=2,
@@ -320,8 +320,8 @@ class EvaluationSourceType(Enum):
     simulation = 'simulation'
 
 
-class ExitConditionOutputNextState(RootModel[List]):
-    root: List = Field(
+class ExitConditionOutputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description="The state to transition to if this exit condition is met. If it's a string, the session will transition to a state within this contextual graph. If it's a tuple, the first element\nmust be of the form `{external_contextual_graph_reference_name}.{state_name}` and represents a state in another contextual graph that the session will transition to, and the second element must be a state\nin this contextual graph that the session will transition to after reaching the terminal state of the external graph.",
         max_length=2,
@@ -442,7 +442,7 @@ class GenerateToolParamsDesignatedToolRetrievedEvent(BaseModel):
 
 class InCondition(BaseModel):
     type: Literal['In'] = Field('In', title='Type')
-    values: List = Field(
+    values: List[Any] = Field(
         ...,
         description='A list of values, one of which the attribute must be equal to. All values must be of the same type as the type of this attribute in the permission.',
         title='Values',
@@ -483,7 +483,7 @@ class JumpbackStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -493,7 +493,7 @@ class JumpbackStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -610,7 +610,7 @@ class NavigateStateMachineExitConditionSelectedEvent(BaseModel):
         description='The exit condition that the agent has selected.',
         title='Exit Condition',
     )
-    next_state_state_machine_version_info: List = Field(
+    next_state_state_machine_version_info: List[Any] = Field(
         ...,
         description='The state machine version info of the next state.',
         max_length=2,
@@ -700,7 +700,7 @@ class NavigateStateMachineStateTransitionedEvent(BaseModel):
         'navigate-state-machine-state-transitioned', title='Type'
     )
     prev_state: str = Field(..., description='The previous state.', title='Prev State')
-    prev_state_state_machine_version_info: List = Field(
+    prev_state_state_machine_version_info: List[Any] = Field(
         ...,
         description='The state machine version info of the previous state.',
         max_length=2,
@@ -708,7 +708,7 @@ class NavigateStateMachineStateTransitionedEvent(BaseModel):
         title='Prev State State Machine Version Info',
     )
     next_state: str = Field(..., description='The next state.', title='Next State')
-    next_state_state_machine_version_info: List = Field(
+    next_state_state_machine_version_info: List[Any] = Field(
         ...,
         description='The state machine version info of the next state.',
         max_length=2,
@@ -834,8 +834,8 @@ class RecallStateInputNextState(AnnotationStateInputNextState):
     pass
 
 
-class RecallStateOutputNextState(RootModel[List]):
-    root: List = Field(
+class RecallStateOutputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description="The state to transition to after the active memory extraction is performed. If it's a string, the session will transition to a state within this contextual graph. If it's a tuple, the first element\nmust be of the form `{external_contextual_graph_reference_name}.{state_name}` and represents a state in another contextual graph that the session will transition to, and the second element must be a state\nin this contextual graph that the session will transition to after reaching the terminal state of the external graph.",
         max_length=2,
@@ -870,7 +870,7 @@ class RecallStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -880,7 +880,7 @@ class RecallStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -914,9 +914,9 @@ class ResultPersistence(Enum):
 
 
 class SelectDynamicBehaviorCompletedEventSelectedDynamicBehaviorSetVersionInfo(
-    RootModel[List]
+    RootModel[List[Any]]
 ):
-    root: List = Field(
+    root: List[Any] = Field(
         ...,
         max_length=2,
         min_length=2,
@@ -925,9 +925,9 @@ class SelectDynamicBehaviorCompletedEventSelectedDynamicBehaviorSetVersionInfo(
 
 
 class SelectDynamicBehaviorCompletedEventPreviousSelectedDynamicBehaviorSetVersionInfo(
-    RootModel[List]
+    RootModel[List[Any]]
 ):
-    root: List = Field(
+    root: List[Any] = Field(
         ...,
         max_length=2,
         min_length=2,
@@ -1064,7 +1064,7 @@ class StateOrRefName(RootModel[str]):
 
 class StateTransitionInvocationMetadata(BaseModel):
     type: Literal['state-transition'] = Field('state-transition', title='Type')
-    current_state_machine_and_version: List = Field(
+    current_state_machine_and_version: List[Any] = Field(
         ...,
         description='The ID and version of the state machine that is currently being executed.',
         max_length=2,
@@ -1263,7 +1263,7 @@ class ToolCallStateOutput(BaseModel):
 
 class ToolCallStateInvocationMetadata(BaseModel):
     type: Literal['tool-call-state'] = Field('tool-call-state', title='Type')
-    current_state_machine_and_version: List = Field(
+    current_state_machine_and_version: List[Any] = Field(
         ...,
         description='The ID and version of the state machine that is currently being executed.',
         max_length=2,
@@ -1283,7 +1283,7 @@ class ToolCallStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -1293,7 +1293,7 @@ class ToolCallStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -1396,8 +1396,8 @@ class TotalMessageCountMetricOutput(BaseModel):
     message_count: int = Field(..., title='Message Count')
 
 
-class UnitTestRunSpecScenarioVersionInfo(RootModel[List]):
-    root: List = Field(
+class UnitTestRunSpecScenarioVersionInfo(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description='The version of the scenario that the unit test runs on. This field is only populated if the authenticated user has the `Simulation:GetSimulationUnitTest` permission.',
         max_length=2,
@@ -1406,8 +1406,8 @@ class UnitTestRunSpecScenarioVersionInfo(RootModel[List]):
     )
 
 
-class UnitTestRunSpecPersonaVersionInfo(RootModel[List]):
-    root: List = Field(
+class UnitTestRunSpecPersonaVersionInfo(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description='The version of the persona that the unit test runs on. This field is only populated if the authenticated user has the `Simulation:GetSimulationUnitTest` permission.',
         max_length=2,
@@ -1704,6 +1704,12 @@ AmigoLibMongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo
 )
 
 
+class AmigoLibMongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
+    major: int = Field(..., title='Major')
+    minor: int = Field(..., title='Minor')
+    patch: int = Field(..., title='Patch')
+
+
 class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
     user_id: str = Field(..., description='The ID of the user.', title='User Id')
     user_org_id: str = Field(
@@ -1711,12 +1717,6 @@ class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
     )
 
 
-class AmigoLibMongoCollectionsToolVersionToolVersionVersion(BaseModel):
-    major: int = Field(..., title='Major')
-    minor: int = Field(..., title='Minor')
-    patch: int = Field(..., title='Patch')
-
-
 class AmigoLibMongoCollectionsUserUserPreferences(BaseModel):
     enable_response_recommendation: Optional[bool] = Field(
         False,
@@ -1780,7 +1780,7 @@ class AdminSubmitSqlQueryResponse(BaseModel):
         description='The time taken to execute the query in milliseconds.',
         title='Execution Time Ms',
     )
-    result: List[List] = Field(
+    result: List[List[Any]] = Field(
         ...,
         description='The result of the query as a list of rows, where each row is a list of column values.',
         title='Result',
@@ -1870,9 +1870,9 @@ class ConversationGenerateConversationStarterResponse(BaseModel):
 
 
 class ConversationGetInteractionInsightsResponseTriggeredDynamicBehaviorSetVersionInfo(
-    RootModel[List]
+    RootModel[List[Any]]
 ):
-    root: List = Field(
+    root: List[Any] = Field(
         ...,
         description='The ID and version number of the dynamic behavior set that was activated during this interaction.',
         max_length=2,
@@ -1955,6 +1955,20 @@ class DynamicBehaviorSetGetDynamicBehaviorInvocationsResponse(BaseModel):
     )
 
 
+class DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
+    BaseModel
+):
+    id: str = Field(..., title='Id')
+    name: str = Field(..., title='Name')
+    is_active: bool = Field(..., title='Is Active')
+    applied_to_services: List[str] = Field(..., title='Applied To Services')
+    tags: Dict[str, Optional[str]] = Field(..., title='Tags')
+    created_at: AwareDatetime = Field(..., title='Created At')
+    creator: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
+    updated_at: AwareDatetime = Field(..., title='Updated At')
+    updated_by: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
+
+
 class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
     applied_to_services_ids: List[str] = Field(
         ...,
@@ -1975,18 +1989,14 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
     )
 
 
-class DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
-    BaseModel
-):
-    id: str = Field(..., title='Id')
-    name: str = Field(..., title='Name')
-    is_active: bool = Field(..., title='Is Active')
-    applied_to_services: List[str] = Field(..., title='Applied To Services')
-    tags: Dict[str, Optional[str]] = Field(..., title='Tags')
-    created_at: AwareDatetime = Field(..., title='Created At')
-    creator: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
-    updated_at: AwareDatetime = Field(..., title='Updated At')
-    updated_by: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
+class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
+    dynamic_behavior_sets: List[
+        DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
+    ] = Field(
+        ...,
+        description='The retrieved dynamic behavior sets.',
+        title='Dynamic Behavior Sets',
+    )
 
 
 class DynamicBehaviorSetUpdateDynamicBehaviorSetRequestAppliedToService(
@@ -10742,13 +10752,6 @@ class OrganizationCreateOrganizationRequestPreferences(BaseModel):
     )
 
 
-class OrganizationCreateOrganizationRequestUserDimension(BaseModel):
-    description: str = Field(..., min_length=1)
-    tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
-        ..., title='Tags'
-    )
-
-
 class OrganizationCreateOrganizationResponse(BaseModel):
     org_id: str = Field(
         ...,
@@ -10772,19 +10775,19 @@ class OrganizationCreateServiceHierarchicalStateMachineResponse(BaseModel):
     )
 
 
-class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState(
+class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState(
     AnnotationStateInputNextState
 ):
     pass
 
 
-class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition(
+class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition(
     BaseModel
 ):
     description: str = Field(..., min_length=1)
     next_state: Union[
         StateOrRefName,
-        OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState,
+        OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState,
     ] = Field(..., title='Next State')
 
 
@@ -10890,22 +10893,18 @@ class OrganizationGetUserDimensionsResponse(BaseModel):
     )
 
 
-class OrganizationModifyOrganizationRequestUserDimensions(
-    RootModel[List[OrganizationCreateOrganizationRequestUserDimension]]
-):
-    root: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
-        ...,
-        description='User dimensions for the organization. If not set or `null`, this field is not updated.',
-        min_length=1,
-        title='User Dimensions',
-    )
-
-
 OrganizationModifyOrganizationRequestPreferences = (
     OrganizationCreateOrganizationRequestPreferences
 )
 
 
+class OrganizationModifyOrganizationRequestUserDimension(BaseModel):
+    description: str = Field(..., min_length=1)
+    tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
+        ..., title='Tags'
+    )
+
+
 class RoleAssignRoleRequest(BaseModel):
     user_id: str = Field(
         ...,
@@ -10952,6 +10951,13 @@ class RoleModifyRoleResponse(BaseModel):
     )
 
 
+class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
+    llm_name: LLMType
+    params: Optional[Dict[str, Any]] = Field(
+        {}, description='LLM-specific parameters to use.', title='Params'
+    )
+
+
 class ServiceCreateServiceResponse(BaseModel):
     id: str = Field(..., description='The ID of the newly created service.', title='Id')
 
@@ -11011,10 +11017,19 @@ class ServiceUpdateServiceRequest(BaseModel):
     )
 
 
-class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
-    llm_name: LLMType
-    params: Optional[Dict[str, Any]] = Field(
-        {}, description='LLM-specific parameters to use.', title='Params'
+class ServiceUpsertServiceVersionSetRequestVersionSet(BaseModel):
+    agent_version_number: Optional[int] = Field(
+        ...,
+        description='The version number of the agent to be used. If None, the latest agent version will be used.',
+        title='Agent Version Number',
+    )
+    service_hierarchical_state_machine_version_number: Optional[int] = Field(
+        ...,
+        description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
+        title='Service Hierarchical State Machine Version Number',
+    )
+    llm_model_preferences: Dict[str, ServiceCreateServiceRequestVersionSetLLMConfig] = (
+        Field(..., title='Llm Model Preferences')
     )
 
 
@@ -11236,6 +11251,37 @@ class SimulationGetSimulationScenariosResponseFilterValues(BaseModel):
     )
 
 
+class SimulationGetSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
+    id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
+    name: str = Field(
+        ..., description='The name of the simulation scenario.', title='Name'
+    )
+    tags: Dict[str, Optional[str]] = Field(
+        ..., description='The tags of the simulation scenario.', title='Tags'
+    )
+    is_deleted: bool = Field(
+        ...,
+        description='Whether the simulation scenario is deleted.',
+        title='Is Deleted',
+    )
+    created_at: AwareDatetime = Field(
+        ...,
+        description='The timestamp when the simulation scenario was created.',
+        title='Created At',
+    )
+    creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
+        Field(..., description='The user who created the simulation scenario.')
+    )
+    updated_at: AwareDatetime = Field(
+        ...,
+        description='The timestamp when the simulation scenario was last updated.',
+        title='Updated At',
+    )
+    updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
+        Field(..., description='The user who last updated the simulation scenario.')
+    )
+
+
 class SimulationGetSimulationUnitTestSetRunArtifactsResponse(BaseModel):
     presigned_url: str = Field(
         ...,
@@ -11358,34 +11404,13 @@ class SimulationSearchSimulationPersonasResponseSimulationPersonaInstance(BaseMo
     )
 
 
-class SimulationSearchSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
-    id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
-    name: str = Field(
-        ..., description='The name of the simulation scenario.', title='Name'
-    )
-    tags: Dict[str, Optional[str]] = Field(
-        ..., description='The tags of the simulation scenario.', title='Tags'
-    )
-    is_deleted: bool = Field(
-        ...,
-        description='Whether the simulation scenario is deleted.',
-        title='Is Deleted',
-    )
-    created_at: AwareDatetime = Field(
-        ...,
-        description='The timestamp when the simulation scenario was created.',
-        title='Created At',
-    )
-    creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
-        Field(..., description='The user who created the simulation scenario.')
-    )
-    updated_at: AwareDatetime = Field(
+class SimulationSearchSimulationScenariosResponse(BaseModel):
+    simulation_scenarios: List[
+        SimulationGetSimulationScenariosResponseSimulationScenarioInstance
+    ] = Field(
         ...,
-        description='The timestamp when the simulation scenario was last updated.',
-        title='Updated At',
-    )
-    updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
-        Field(..., description='The user who last updated the simulation scenario.')
+        description='The list of simulation scenarios.',
+        title='Simulation Scenarios',
     )
 
 
@@ -11672,7 +11697,7 @@ class UserGetMemoriesResponse(BaseModel):
     )
 
 
-class UserGetUsersResponseUserInstanceUserStats(BaseModel):
+class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
     num_conversations: int = Field(
         ...,
         description='The number of conversations the user has created.',
@@ -12111,7 +12136,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         [], description='The IDs of the messages to retrieve.', title='Id'
     )
     message_type: Optional[List[MessageType]] = Field(
-        ['user-message', 'agent-message', 'external-event'],
+        ['agent-message', 'user-message', 'external-event'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13556,7 +13581,7 @@ class ActionStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -13566,7 +13591,7 @@ class ActionStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -13654,7 +13679,7 @@ class DecisionStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -13664,7 +13689,7 @@ class DecisionStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -13837,7 +13862,7 @@ class ReflectionStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -13847,7 +13872,7 @@ class ReflectionStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -14144,7 +14169,7 @@ class ToolVersionInstance(BaseModel):
         description='The ID of the tool that this version belongs to.',
         title='Tool Id',
     )
-    version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
+    version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
         ..., description='The version of the tool.'
     )
     required_envvars: List[str] = Field(
@@ -14225,14 +14250,14 @@ class VersionSetInfo(BaseModel):
         description='The name of the version set. If the value is `unknown`, it means the conversation was created before this field was introduced and the version set name cannot be retroactively determined.',
         title='Name',
     )
-    agent_version_info: List = Field(
+    agent_version_info: List[Any] = Field(
         ...,
         description='The ID of the agent and version number used in this conversation.',
         max_length=2,
         min_length=2,
         title='Agent Version Info',
     )
-    service_hierarchical_state_machine_version_info: List = Field(
+    service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The ID of the state machine and version number used in this conversation.',
         max_length=2,
@@ -14460,7 +14485,7 @@ class DynamicBehaviorSetGetDynamicBehaviorSetVersionsResponse(BaseModel):
 
 class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
     dynamic_behavior_sets: List[
-        DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
+        DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
     ] = Field(
         ...,
         description='The retrieved dynamic behavior sets.',
@@ -14484,16 +14509,6 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
     )
 
 
-class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
-    dynamic_behavior_sets: List[
-        DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
-    ] = Field(
-        ...,
-        description='The retrieved dynamic behavior sets.',
-        title='Dynamic Behavior Sets',
-    )
-
-
 class MetricCreateMetricRequest(BaseModel):
     name: str = Field(
         ...,
@@ -14550,7 +14565,7 @@ class OrganizationCreateOrganizationRequest(BaseModel):
         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[OrganizationCreateOrganizationRequestUserDimension] = Field(
+    user_dimensions: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
         ...,
         description='User dimensions for the organization.',
         title='User Dimensions',
@@ -14694,6 +14709,17 @@ class OrganizationGetOrganizationMetricsResponse(BaseModel):
     )
 
 
+class OrganizationModifyOrganizationRequestUserDimensions(
+    RootModel[List[OrganizationModifyOrganizationRequestUserDimension]]
+):
+    root: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
+        ...,
+        description='User dimensions for the organization. If not set or `null`, this field is not updated.',
+        min_length=1,
+        title='User Dimensions',
+    )
+
+
 class OrganizationModifyOrganizationRequest(BaseModel):
     title: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         None,
@@ -14791,20 +14817,37 @@ class RoleModifyRoleRequest(BaseModel):
     )
 
 
-class ServiceCreateServiceRequestVersionSet(BaseModel):
-    agent_version_number: Optional[int] = Field(
+class ServiceCreateServiceRequest(BaseModel):
+    service_hierarchical_state_machine_id: str = Field(
         ...,
-        description='The version number of the agent to be used. If None, the latest agent version will be used.',
-        title='Agent Version Number',
+        description='The ID of the state machine that this service uses.',
+        pattern='^[a-f0-9]{24}$',
+        title='Service Hierarchical State Machine Id',
     )
-    service_hierarchical_state_machine_version_number: Optional[int] = Field(
+    agent_id: str = Field(
         ...,
-        description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
-        title='Service Hierarchical State Machine Version Number',
+        description='The ID of the agent that this service uses.',
+        pattern='^[a-f0-9]{24}$',
+        title='Agent Id',
+    )
+    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(
+        ...,
+        description='Whether the newly-created service is active. Only active services are visible to users on the dashboard. You can later adjust the activeness of this service.',
+        title='Is Active',
+    )
+    release_version_set: Optional[ServiceUpsertServiceVersionSetRequestVersionSet] = (
+        Field(
+            None,
+            description='The `release` version set to use for this service. If not specified, the `release` version set will be the same as the `edge` version set, which uses the\nlatest agent and state machine versions with no model preference.',
+        )
+    )
+    tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
+        ..., description='The tags of this service.', title='Tags'
     )
-    llm_model_preferences: Dict[
-        str, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
-    ] = Field(..., title='Llm Model Preferences')
 
 
 class ServiceGetServicesResponse(BaseModel):
@@ -14828,7 +14871,7 @@ class ServiceGetServicesResponse(BaseModel):
 
 
 class ServiceUpsertServiceVersionSetRequest(BaseModel):
-    version_set: ServiceCreateServiceRequestVersionSet = Field(
+    version_set: ServiceUpsertServiceVersionSetRequestVersionSet = Field(
         ..., description='The version set to upsert.'
     )
 
@@ -14920,7 +14963,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
 
 class SimulationGetSimulationScenariosResponse(BaseModel):
     simulation_scenarios: List[
-        SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
+        SimulationGetSimulationScenariosResponseSimulationScenarioInstance
     ] = Field(
         ...,
         description='The list of simulation scenarios.',
@@ -14976,16 +15019,6 @@ class SimulationSearchSimulationPersonasResponse(BaseModel):
     )
 
 
-class SimulationSearchSimulationScenariosResponse(BaseModel):
-    simulation_scenarios: List[
-        SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
-    ] = Field(
-        ...,
-        description='The list of simulation scenarios.',
-        title='Simulation Scenarios',
-    )
-
-
 class SimulationSearchSimulationUnitTestSetResponse(BaseModel):
     simulation_unit_test_sets: List[SimulationUnitTestSet] = Field(
         ...,
@@ -15101,7 +15134,7 @@ class UserGetUserModelResponse(BaseModel):
     )
 
 
-class UserGetUsersResponseUserInstance(BaseModel):
+class UserSearchUsersResponseUserInstance(BaseModel):
     org_id: str = Field(
         ...,
         description='The ID of the organization that this user belongs to.',
@@ -15117,7 +15150,7 @@ class UserGetUsersResponseUserInstance(BaseModel):
         ..., description='The last name of the user.', title='Last Name'
     )
     email: str = Field(..., description='Email of the user.', title='Email')
-    user_stats: UserGetUsersResponseUserInstanceUserStats = Field(
+    user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
         ..., description="Statistics about the user's usage of the Amigo platform."
     )
     role: str = Field(..., description='The ID of the role of the user.', title='Role')
@@ -15126,12 +15159,6 @@ class UserGetUsersResponseUserInstance(BaseModel):
     )
 
 
-class UserSearchUsersResponse(BaseModel):
-    users: List[UserGetUsersResponseUserInstance] = Field(
-        ..., description='Users in this organization.', title='Users'
-    )
-
-
 class WebhookDestinationGetWebhookDeliveriesResponse(BaseModel):
     webhook_deliveries: List[WebhookDeliveryInstance] = Field(
         ..., description='The retrieved webhook deliveries.', title='Webhook Deliveries'
@@ -15227,7 +15254,7 @@ class ActionStateInput(BaseModel):
         Field(..., title='Boundary Constraints')
     )
     exit_conditions: List[
-        OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
+        OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
     ] = Field(..., title='Exit Conditions')
     action_tool_call_specs: List[ToolCallSpecInput] = Field(
         ..., title='Action Tool Call Specs'
@@ -15336,7 +15363,7 @@ class DecisionStateInput(BaseModel):
     type: Literal['decision'] = Field(..., title='Type')
     name: str = Field(..., pattern='^[A-Za-z0-9_]+$')
     exit_conditions: List[
-        OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
+        OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
     ] = Field(..., title='Exit Conditions')
     decision_guidelines: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         ..., title='Decision Guidelines'
@@ -15587,7 +15614,7 @@ class ServiceHierarchicalStateMachineInstance(BaseModel):
         description='The state the user will be in when the session ends. This must be an action state.',
         title='Terminal State',
     )
-    references: Dict[str, List] = Field(
+    references: Dict[str, List[Any]] = Field(
         ...,
         description="A dictionary of other service hierarchical state machine versions that this references to. This field is a map of the reference name to the machine's ID and its version number.",
         title='References',
@@ -15708,14 +15735,14 @@ class UnitTestRunSpec(BaseModel):
     run_count: int = Field(
         ..., description='The number of times to run the unit test.', title='Run Count'
     )
-    service_hierarchical_state_machine_version_info: List = Field(
+    service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The version of the service hierarchical state machine that the unit test runs on.',
         max_length=2,
         min_length=2,
         title='Service Hierarchical State Machine Version Info',
     )
-    agent_version_info: List = Field(
+    agent_version_info: List[Any] = Field(
         ...,
         description='The version of the agent that the unit test runs on.',
         max_length=2,
@@ -15922,7 +15949,7 @@ class OrganizationCreateServiceHierarchicalStateMachineVersionRequest(BaseModel)
         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(
+    references: Dict[str, List[Any]] = Field(
         ...,
         description="A dictionary of other service hierarchical state machines that this references to. This field is a map of the reference name to the machine's ID and its version number.",
         title='References',
@@ -16020,37 +16047,6 @@ class RoleGetRolesResponse(BaseModel):
     )
 
 
-class ServiceCreateServiceRequest(BaseModel):
-    service_hierarchical_state_machine_id: str = Field(
-        ...,
-        description='The ID of the state machine that this service uses.',
-        pattern='^[a-f0-9]{24}$',
-        title='Service Hierarchical State Machine Id',
-    )
-    agent_id: str = Field(
-        ...,
-        description='The ID of the agent that this service uses.',
-        pattern='^[a-f0-9]{24}$',
-        title='Agent Id',
-    )
-    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(
-        ...,
-        description='Whether the newly-created service is active. Only active services are visible to users on the dashboard. You can later adjust the activeness of this service.',
-        title='Is Active',
-    )
-    release_version_set: Optional[ServiceCreateServiceRequestVersionSet] = Field(
-        None,
-        description='The `release` version set to use for this service. If not specified, the `release` version set will be the same as the `edge` version set, which uses the\nlatest agent and state machine versions with no model preference.',
-    )
-    tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
-        ..., description='The tags of this service.', title='Tags'
-    )
-
-
 class SimulationCreateSimulationUnitTestRequest(BaseModel):
     name: str = Field(..., description='The name of the unit test.', min_length=1)
     description: str = Field(
@@ -16206,7 +16202,7 @@ class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
     tool_id: str = Field(
         ..., description='The ID of the tool that was invoked.', title='Tool Id'
     )
-    tool_version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
+    tool_version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
         ..., description='The version of the tool that was invoked.'
     )
 
@@ -16220,7 +16216,7 @@ class ToolSearchToolInvocationsResponse(BaseModel):
 
 
 class UserGetUsersResponse(BaseModel):
-    users: List[UserGetUsersResponseUserInstance] = Field(
+    users: List[UserSearchUsersResponseUserInstance] = Field(
         ..., description='Users in this organization.', title='Users'
     )
     has_more: bool = Field(
@@ -16233,6 +16229,12 @@ class UserGetUsersResponse(BaseModel):
     )
 
 
+class UserSearchUsersResponse(BaseModel):
+    users: List[UserSearchUsersResponseUserInstance] = Field(
+        ..., description='Users in this organization.', title='Users'
+    )
+
+
 class SimulationUnitTestSetRunInstance(BaseModel):
     org_id: str = Field(..., title='Org Id')
     id: str = Field(..., title='Id')

{amigo_sdk-0.67.0.dist-info → amigo_sdk-0.70.0.dist-info}/METADATA

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

{amigo_sdk-0.67.0.dist-info → amigo_sdk-0.70.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-amigo_sdk/__init__.py,sha256=_Vt20XUAV719JwAfR87fdDdt_AzUTzcNx0cHi5VVTdA,139
+amigo_sdk/__init__.py,sha256=f5lR2WPI6sR2xJpKf3k4M0oUwFUzeqSBqcO3UipgrQc,139
 amigo_sdk/_retry_utils.py,sha256=VbA_Uwv268J80d11R9XfnCjZiV-NwE1aXRsSlRjjRio,2108
 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=Yn12AxIQ6zE9jnGLWO0MUVbBsmw77Kzh82AijyreOkA,4818
 amigo_sdk/http_client.py,sha256=tfMXHqMTmlz2ja7aJhOVgbwAiJNlKCzPcK4UGNqXHO8,13488
 amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
 amigo_sdk/sdk_client.py,sha256=5nj8hzjHSFRs9JMNabdYIhQAcFE2G1jI_r7SHVGr9No,6106
-amigo_sdk/generated/model.py,sha256=byPdfHdBjkFPxltnyGsxQEo1yl8pF9x53IIF_OKJWzE,456940
+amigo_sdk/generated/model.py,sha256=53HP0RAAxkl4msbceJG48IsQYARftJozpPxGMMavnO0,457250
 amigo_sdk/resources/conversation.py,sha256=u4kOx5LPbhX92L41UnmPmaF9TkQA5bGev8N00dMoP9I,14805
 amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
 amigo_sdk/resources/service.py,sha256=t1iA3nS9co-wuR-x5jBhAXXTWfMeGsLwfQcLycwVrCA,1536
 amigo_sdk/resources/user.py,sha256=zikijiuGXgmkBVtrldor6XQGQK6jWXonDvNFuQs6XK4,3472
-amigo_sdk-0.67.0.dist-info/METADATA,sha256=qTyi_Kit19ZHi0PkFa6POvNZDZOoiBvsCCnetJZc40s,8499
-amigo_sdk-0.67.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-amigo_sdk-0.67.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.67.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.67.0.dist-info/RECORD,,
+amigo_sdk-0.70.0.dist-info/METADATA,sha256=a3t-Y72PZyYcb7VTrKh8WL1zoJyDDZ371AO3t0FkZgQ,8499
+amigo_sdk-0.70.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+amigo_sdk-0.70.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.70.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.70.0.dist-info/RECORD,,