amigo_sdk 0.97.0__py3-none-any.whl → 0.99.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.97.0"
+__version__ = "0.99.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: 2026-01-08T19:54:56+00:00
+#   timestamp: 2026-01-09T15:07:06+00:00
 
 from __future__ import annotations
 
@@ -398,6 +398,16 @@ class InitialMessageType(Enum):
     skip = 'skip'
 
 
+class InjectInstructionActionOutput(BaseModel):
+    type: Literal['inject-instruction'] = Field(..., title='Type')
+    instruction: str = Field(..., title='Instruction')
+    overrides_instructions: bool = Field(
+        ...,
+        description='During injection, whether the original instruction of the state is overriden with this instruction.',
+        title='Overrides Instructions',
+    )
+
+
 class InteractionCompleteEvent(BaseModel):
     type: Literal['interaction-complete'] = Field('interaction-complete', title='Type')
     message_id: str = Field(
@@ -1397,18 +1407,6 @@ class MongoCollectionsDynamicBehaviorSetVersionDynamicBehaviorSetVersionChangeTo
     )
 
 
-class MongoCollectionsDynamicBehaviorSetVersionDynamicBehaviorSetVersionInjectInstructionAction(
-    BaseModel
-):
-    type: Literal['inject-instruction'] = Field(..., title='Type')
-    instruction: str = Field(..., title='Instruction')
-    overrides_instructions: bool = Field(
-        ...,
-        description='During injection, whether the original instruction of the state is overriden with this instruction.',
-        title='Overrides Instructions',
-    )
-
-
 class MongoCollectionsMetricMetricUserInfo(
     MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
 ):
@@ -1426,11 +1424,6 @@ class Action(Enum):
     deny = 'Deny'
 
 
-class MongoCollectionsServiceServiceTag(BaseModel):
-    key: str = Field(..., title='Key')
-    value: str | None = Field(..., title='Value')
-
-
 class MongoCollectionsServiceServiceVersionSet(BaseModel):
     agent_version_number: int | None = Field(
         ...,
@@ -1497,6 +1490,11 @@ class MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo(
     pass
 
 
+class MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag(BaseModel):
+    key: str = Field(..., title='Key')
+    value: str | None = Field(..., title='Value')
+
+
 class MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUnitTestRunDescriptor(
     BaseModel
 ):
@@ -1772,18 +1770,6 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionChangeToolC
     )
 
 
-class DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionInjectInstructionAction(
-    BaseModel
-):
-    type: Literal['inject-instruction'] = Field(..., title='Type')
-    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.',
-        title='Overrides Instructions',
-    )
-
-
 class DynamicBehaviorSetCreateDynamicBehaviorSetResponse(BaseModel):
     dynamic_behavior_set_id: str = Field(
         ...,
@@ -1798,6 +1784,18 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestChangeToolCandidat
     pass
 
 
+class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestInjectInstructionAction(
+    BaseModel
+):
+    type: Literal['inject-instruction'] = Field(..., title='Type')
+    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.',
+        title='Overrides Instructions',
+    )
+
+
 class DynamicBehaviorSetCreateDynamicBehaviorSetVersionResponse(BaseModel):
     new_version_number: int = Field(
         ...,
@@ -10604,11 +10602,6 @@ class OrganizationCreateOrganizationRequestPreferences(BaseModel):
     )
 
 
-class OrganizationCreateOrganizationRequestUserDimension(BaseModel):
-    description: str = Field(..., min_length=1)
-    tags: list[PydanticBaseModelStrippedNonemptyString1] = Field(..., title='Tags')
-
-
 class OrganizationCreateOrganizationResponse(BaseModel):
     org_id: str = Field(
         ...,
@@ -10640,7 +10633,7 @@ class NextState9(NextState7):
     pass
 
 
-class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition(
+class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition(
     BaseModel
 ):
     description: str = Field(..., min_length=1)
@@ -10783,23 +10776,17 @@ class OrganizationGetUserDimensionsResponse(BaseModel):
     )
 
 
-class UserDimensions(
-    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',
-    )
-
-
 class OrganizationModifyOrganizationRequestPreferences(
     OrganizationCreateOrganizationRequestPreferences
 ):
     pass
 
 
+class OrganizationModifyOrganizationRequestUserDimension(BaseModel):
+    description: str = Field(..., min_length=1)
+    tags: list[PydanticBaseModelStrippedNonemptyString1] = Field(..., title='Tags')
+
+
 class RoleAssignRoleRequest(BaseModel):
     user_id: str = Field(
         ...,
@@ -10816,36 +10803,25 @@ class RoleAssignRoleResponse(BaseModel):
     )
 
 
-class InheritedFrom(RootModel[str]):
-    root: str = Field(
-        ...,
-        description='The ID of the role that this role inherits from.',
-        pattern='^[a-f0-9]{24}$',
-        title='Inherited From',
-    )
-
-
 class RoleCreateRoleResponse(BaseModel):
     role_id: str = Field(
         ..., description='The ID of the newly created role.', title='Role Id'
     )
 
 
-class InheritedFrom1(RootModel[str]):
-    root: str = Field(
-        ...,
-        description='The ID of the role that this role inherits from. Only updated if specified. This field is an immutable field.',
-        pattern='^[a-f0-9]{24}$',
-        title='Inherited From',
-    )
-
-
 class RoleModifyRoleResponse(BaseModel):
     role_id: str = Field(
         ..., description='The identifier of the updated role.', title='Role Id'
     )
 
 
+class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
+    llm_name: LLMType
+    params: dict[str, Any] | None = 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')
 
@@ -10921,13 +10897,6 @@ class ServiceUpdateServiceRequest(BaseModel):
     )
 
 
-class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
-    llm_name: LLMType
-    params: dict[str, Any] | None = Field(
-        {}, description='LLM-specific parameters to use.', title='Params'
-    )
-
-
 class SimulationCreateSimulationPersonaRequestInitialVersion(BaseModel):
     background: str = Field(
         ..., description='The background of the simulation persona.', min_length=1
@@ -11100,6 +11069,40 @@ class SimulationGetSimulationPersonasResponseFilterValues(BaseModel):
     )
 
 
+class SimulationGetSimulationPersonasResponseSimulationPersonaInstance(BaseModel):
+    id: str = Field(..., description='The ID of the simulation persona.', title='Id')
+    name: str = Field(
+        ..., description='The name of the simulation persona.', title='Name'
+    )
+    role: str = Field(
+        ..., description='The role of the simulation persona.', title='Role'
+    )
+    tags: dict[str, str | None] = Field(
+        ..., description='The tags of the simulation persona.', title='Tags'
+    )
+    is_deleted: bool = Field(
+        ...,
+        description='Whether the simulation persona is deleted.',
+        title='Is Deleted',
+    )
+    created_at: AwareDatetime = Field(
+        ...,
+        description='The timestamp when the simulation persona was created.',
+        title='Created At',
+    )
+    creator: MongoCollectionsSimulationPersonaSimulationPersonaUserInfo = Field(
+        ..., description='The user who created the simulation persona.'
+    )
+    updated_at: AwareDatetime = Field(
+        ...,
+        description='The timestamp when the simulation persona was last updated.',
+        title='Updated At',
+    )
+    updated_by: MongoCollectionsSimulationPersonaSimulationPersonaUserInfo = Field(
+        ..., description='The user who last updated the simulation persona.'
+    )
+
+
 class SimulationGetSimulationScenarioVersionsResponse(BaseModel):
     simulation_scenario_versions: list[SimulationScenarioVersion] = Field(
         ...,
@@ -11250,37 +11253,11 @@ class SimulationGetSimulationUnitTestsResponseFilterValues(BaseModel):
     )
 
 
-class SimulationSearchSimulationPersonasResponseSimulationPersonaInstance(BaseModel):
-    id: str = Field(..., description='The ID of the simulation persona.', title='Id')
-    name: str = Field(
-        ..., description='The name of the simulation persona.', title='Name'
-    )
-    role: str = Field(
-        ..., description='The role of the simulation persona.', title='Role'
-    )
-    tags: dict[str, str | None] = Field(
-        ..., description='The tags of the simulation persona.', title='Tags'
-    )
-    is_deleted: bool = Field(
-        ...,
-        description='Whether the simulation persona is deleted.',
-        title='Is Deleted',
-    )
-    created_at: AwareDatetime = Field(
-        ...,
-        description='The timestamp when the simulation persona was created.',
-        title='Created At',
-    )
-    creator: MongoCollectionsSimulationPersonaSimulationPersonaUserInfo = Field(
-        ..., description='The user who created the simulation persona.'
-    )
-    updated_at: AwareDatetime = Field(
-        ...,
-        description='The timestamp when the simulation persona was last updated.',
-        title='Updated At',
-    )
-    updated_by: MongoCollectionsSimulationPersonaSimulationPersonaUserInfo = Field(
-        ..., description='The user who last updated the simulation persona.'
+class SimulationSearchSimulationPersonasResponse(BaseModel):
+    simulation_personas: list[
+        SimulationGetSimulationPersonasResponseSimulationPersonaInstance
+    ] = Field(
+        ..., description='The list of simulation personas.', title='Simulation Personas'
     )
 
 
@@ -11578,7 +11555,7 @@ class UserGetMemoriesResponse(BaseModel):
     )
 
 
-class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
+class UserGetUsersResponseUserInstanceUserStats(BaseModel):
     num_conversations: int = Field(
         ...,
         description='The number of conversations the user has created.',
@@ -11596,6 +11573,31 @@ class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
     )
 
 
+class UserSearchUsersResponseUserInstance(BaseModel):
+    org_id: str = Field(
+        ...,
+        description='The ID of the organization that this user belongs to.',
+        title='Org Id',
+    )
+    user_id: str = Field(
+        ..., description='The identifier of the user.', title='User Id'
+    )
+    first_name: str = Field(
+        ..., description='The first name of the user.', title='First Name'
+    )
+    last_name: str = Field(
+        ..., description='The last name of the user.', title='Last Name'
+    )
+    email: str = Field(..., description='Email of the user.', title='Email')
+    user_stats: UserGetUsersResponseUserInstanceUserStats = 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')
+    preferences: MongoCollectionsUserUserPreferences = Field(
+        ..., description='The preferences of the user.'
+    )
+
+
 class UserSignInWithApiKeyResponse(BaseModel):
     id_token: str = Field(
         ...,
@@ -11997,7 +11999,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         title='Id',
     )
     message_type: list[MessageType] | None = Field(
-        ['agent-message', 'external-event', 'user-message'],
+        ['external-event', 'agent-message', 'user-message'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13157,7 +13159,7 @@ class ActionStateInput(BaseModel):
         ..., title='Boundary Constraints'
     )
     exit_conditions: list[
-        OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
+        OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
     ] = Field(..., title='Exit Conditions')
     action_tool_call_specs: list[
         OrganizationCreateServiceHierarchicalStateMachineVersionRequestToolCallSpec
@@ -13303,7 +13305,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[PydanticBaseModelStrippedNonemptyString1] = Field(
         ..., title='Decision Guidelines'
@@ -13635,7 +13637,7 @@ class ServiceInstance(BaseModel):
     agent_id: str = Field(
         ..., description='The ID of the agent that this service uses.', title='Agent Id'
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the service.', title='Tags'
     )
 
@@ -13688,7 +13690,7 @@ class SimulationUnitTestSet(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13725,7 +13727,7 @@ class SimulationUnitTestSetInstance(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13853,7 +13855,7 @@ class ToolInstance(BaseModel):
         description='A list of encrypted environment variables that some versions of this tool uses.',
         title='Secret Envvars',
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
 
@@ -14228,7 +14230,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',
@@ -14398,6 +14400,17 @@ class OrganizationGetApiKeysResponse(BaseModel):
     )
 
 
+class UserDimensions(
+    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: PydanticBaseModelStrippedNonemptyString1 | None = Field(
         None,
@@ -14451,7 +14464,7 @@ class OrganizationModifyOrganizationRequest(BaseModel):
     )
 
 
-class RoleModifyRoleRequestPermissionGrant(BaseModel):
+class RoleCreateRoleRequestPermissionGrant(BaseModel):
     action: Action = Field(
         ...,
         description='Whether this grant allows or denies the specified access.',
@@ -14466,6 +14479,22 @@ class RoleModifyRoleRequestPermissionGrant(BaseModel):
     description: PydanticBaseModelStrippedNonemptyString1 | None = None
 
 
+class RoleModifyRoleRequest(BaseModel):
+    description: PydanticBaseModelStrippedNonemptyString1 | None = Field(
+        None,
+        description='A description about the role. Only updated if specified. This field is a mutable field.',
+    )
+    permission_grants: list[RoleCreateRoleRequestPermissionGrant] | None = Field(
+        None,
+        description='A list of permission grants associated with this role. Only updated if specified. This field is an immutable field.',
+        title='Permission Grants',
+    )
+    frontend_view: FrontendView | None = Field(
+        None,
+        description='The frontend view for the user of this role. Only updated if specified. This field is an immutable field.',
+    )
+
+
 class ServiceCreateServiceRequestVersionSet(BaseModel):
     agent_version_number: int | None = Field(
         ...,
@@ -14477,9 +14506,9 @@ class ServiceCreateServiceRequestVersionSet(BaseModel):
         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, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
-    ] = Field(..., title='Llm Model Preferences')
+    llm_model_preferences: dict[str, ServiceCreateServiceRequestVersionSetLLMConfig] = (
+        Field(..., title='Llm Model Preferences')
+    )
 
 
 class ServiceGetServicesResponse(BaseModel):
@@ -14571,7 +14600,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
 
 class SimulationGetSimulationPersonasResponse(BaseModel):
     simulation_personas: list[
-        SimulationSearchSimulationPersonasResponseSimulationPersonaInstance
+        SimulationGetSimulationPersonasResponseSimulationPersonaInstance
     ] = Field(
         ..., description='The list of simulation personas.', title='Simulation Personas'
     )
@@ -14639,14 +14668,6 @@ class SimulationGetSimulationUnitTestSetsResponse(BaseModel):
     )
 
 
-class SimulationSearchSimulationPersonasResponse(BaseModel):
-    simulation_personas: list[
-        SimulationSearchSimulationPersonasResponseSimulationPersonaInstance
-    ] = Field(
-        ..., description='The list of simulation personas.', title='Simulation Personas'
-    )
-
-
 class SimulationSearchSimulationUnitTestSetResponse(BaseModel):
     simulation_unit_test_sets: list[SimulationUnitTestSet] = Field(
         ...,
@@ -14767,33 +14788,22 @@ class UserGetUserModelResponse(BaseModel):
     )
 
 
-class UserGetUsersResponseUserInstance(BaseModel):
-    org_id: str = Field(
-        ...,
-        description='The ID of the organization that this user belongs to.',
-        title='Org Id',
-    )
-    user_id: str = Field(
-        ..., description='The identifier of the user.', title='User Id'
-    )
-    first_name: str = Field(
-        ..., description='The first name of the user.', title='First Name'
-    )
-    last_name: str = Field(
-        ..., description='The last name of the user.', title='Last Name'
+class UserGetUsersResponse(BaseModel):
+    users: list[UserSearchUsersResponseUserInstance] = Field(
+        ..., description='Users in this organization.', title='Users'
     )
-    email: str = Field(..., description='Email of the user.', title='Email')
-    user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
-        ..., description="Statistics about the user's usage of the Amigo platform."
+    has_more: bool = Field(
+        ..., description='Whether there are more users to retrieve.', title='Has More'
     )
-    role: str = Field(..., description='The ID of the role of the user.', title='Role')
-    preferences: MongoCollectionsUserUserPreferences = Field(
-        ..., description='The preferences of the user.'
+    continuation_token: int | None = Field(
+        ...,
+        description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
+        title='Continuation Token',
     )
 
 
 class UserSearchUsersResponse(BaseModel):
-    users: list[UserGetUsersResponseUserInstance] = Field(
+    users: list[UserSearchUsersResponseUserInstance] = Field(
         ..., description='Users in this organization.', title='Users'
     )
 
@@ -14954,7 +14964,7 @@ class CurrentAgentActionEvent(BaseModel):
     ) = Field(..., description='The type of action the agent is performing right now.')
 
 
-class DynamicBehaviorSetVersionInstance(BaseModel):
+class DynamicBehaviorSetVersion(BaseModel):
     field_id: str = Field(
         ..., alias='_id', description='The ID of the document.', title='Id'
     )
@@ -14966,15 +14976,10 @@ class DynamicBehaviorSetVersionInstance(BaseModel):
     conversation_triggers: list[str] = Field(..., title='Conversation Triggers')
     actions: list[
         Annotated[
-            MongoCollectionsDynamicBehaviorSetVersionDynamicBehaviorSetVersionInjectInstructionAction
-            | ChangeToolCandidatesActionOutput,
+            InjectInstructionActionOutput | ChangeToolCandidatesActionOutput,
             Field(discriminator='type'),
         ]
     ] = Field(..., title='Actions')
-    action: (
-        MongoCollectionsDynamicBehaviorSetVersionDynamicBehaviorSetVersionInjectInstructionAction
-        | None
-    )
 
 
 class Metric(BaseModel):
@@ -14996,7 +15001,7 @@ class Metric(BaseModel):
     additional_notes: str | None = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
     creator: MongoCollectionsMetricMetricUserInfo = Field(
@@ -15069,7 +15074,7 @@ class MetricInstance(BaseModel):
     additional_notes: str | None = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
     creator: MongoCollectionsMetricMetricUserInfo = Field(
@@ -15125,11 +15130,13 @@ class RoleInstance(BaseModel):
     )
     inherited_from: str | None = Field(
         ...,
+        deprecated=True,
         description='The ID of the role that this role inherits from. Only returned if the role inherits from another role.',
         title='Inherited From',
     )
     is_base_role: bool = Field(
         ...,
+        deprecated=True,
         description='Whether this role is a base role. Base roles cannot inherit from other roles. One can only inherit from base roles.',
         title='Is Base Role',
     )
@@ -15175,7 +15182,7 @@ class SimulationUnitTest(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15220,7 +15227,7 @@ class SimulationUnitTestInstance(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: list[MongoCollectionsServiceServiceTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15320,16 +15327,13 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersion(BaseModel)
         min_length=1,
         title='Conversation Triggers',
     )
-    actions: (
-        list[
-            Annotated[
-                DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionInjectInstructionAction
-                | DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionChangeToolCandidatesAction,
-                Field(discriminator='type'),
-            ]
+    actions: list[
+        Annotated[
+            DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestInjectInstructionAction
+            | DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionChangeToolCandidatesAction,
+            Field(discriminator='type'),
         ]
-        | DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionInjectInstructionAction
-    ) = Field(
+    ] = Field(
         ...,
         description='The action to perform when the dynamic behavior set version is activated.',
         title='Actions',
@@ -15344,16 +15348,13 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequest(BaseModel):
         min_length=1,
         title='Conversation Triggers',
     )
-    actions: (
-        list[
-            Annotated[
-                DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionInjectInstructionAction
-                | DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestChangeToolCandidatesAction,
-                Field(discriminator='type'),
-            ]
+    actions: list[
+        Annotated[
+            DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestInjectInstructionAction
+            | DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestChangeToolCandidatesAction,
+            Field(discriminator='type'),
         ]
-        | DynamicBehaviorSetCreateDynamicBehaviorSetRequestInitialVersionInjectInstructionAction
-    ) = Field(
+    ] = Field(
         ...,
         description='The action to perform when the dynamic behavior set version is activated.',
         title='Actions',
@@ -15361,7 +15362,7 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequest(BaseModel):
 
 
 class DynamicBehaviorSetGetDynamicBehaviorSetVersionsResponse(BaseModel):
-    dynamic_behavior_set_versions: list[DynamicBehaviorSetVersionInstance] = Field(
+    dynamic_behavior_set_versions: list[DynamicBehaviorSetVersion] = Field(
         ...,
         description='The retrieved dynamic behavior set versions.',
         title='Dynamic Behavior Set Versions',
@@ -15478,7 +15479,7 @@ class RoleCreateRoleRequest(BaseModel):
     description: str = Field(
         ..., description='A description about the role.', min_length=1
     )
-    permission_grants: list[RoleModifyRoleRequestPermissionGrant] = Field(
+    permission_grants: list[RoleCreateRoleRequestPermissionGrant] = Field(
         ...,
         description='A list of permission grants associated with this role.',
         title='Permission Grants',
@@ -15486,16 +15487,6 @@ class RoleCreateRoleRequest(BaseModel):
     frontend_view: FrontendView = Field(
         ..., description='The frontend view for users of this role.'
     )
-    is_base_role: bool = Field(
-        ...,
-        description='Whether this role is a base role. Base roles cannot inherit from other roles. One can only inherit from base roles.',
-        title='Is Base Role',
-    )
-    inherited_from: InheritedFrom | None = Field(
-        ...,
-        description='The ID of the role that this role inherits from.',
-        title='Inherited From',
-    )
 
 
 class RoleGetRolesResponse(BaseModel):
@@ -15504,27 +15495,6 @@ class RoleGetRolesResponse(BaseModel):
     )
 
 
-class RoleModifyRoleRequest(BaseModel):
-    description: PydanticBaseModelStrippedNonemptyString1 | None = Field(
-        None,
-        description='A description about the role. Only updated if specified. This field is a mutable field.',
-    )
-    permission_grants: list[RoleModifyRoleRequestPermissionGrant] | None = Field(
-        None,
-        description='A list of permission grants associated with this role. Only updated if specified. This field is an immutable field.',
-        title='Permission Grants',
-    )
-    frontend_view: FrontendView | None = Field(
-        None,
-        description='The frontend view for the user of this role. Only updated if specified. This field is an immutable field.',
-    )
-    inherited_from: InheritedFrom1 | FieldNotSet | None = Field(
-        default_factory=lambda: InheritedFrom1({}),
-        description='The ID of the role that this role inherits from. Only updated if specified. This field is an immutable field.',
-        title='Inherited From',
-    )
-
-
 class ServiceCreateServiceRequest(BaseModel):
     service_hierarchical_state_machine_id: str = Field(
         ...,
@@ -15689,7 +15659,7 @@ class SimulationUpdateSimulationUnitTestRequest(BaseModel):
     )
 
 
-class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
+class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
     id: str = Field(..., description='The ID of the tool invocation.', title='Id')
     org_id: str = Field(..., description='The ID of the organization.', title='Org Id')
     created_at: AwareDatetime | None = Field(None, title='Created At')
@@ -15716,17 +15686,11 @@ class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
     )
 
 
-class UserGetUsersResponse(BaseModel):
-    users: list[UserGetUsersResponseUserInstance] = Field(
-        ..., description='Users in this organization.', title='Users'
-    )
-    has_more: bool = Field(
-        ..., description='Whether there are more users to retrieve.', title='Has More'
-    )
-    continuation_token: int | None = Field(
-        ...,
-        description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
-        title='Continuation Token',
+class ToolSearchToolInvocationsResponse(BaseModel):
+    tool_invocations: list[ToolGetToolInvocationsResponseToolInvocationInstance] = (
+        Field(
+            ..., description='The list of tool invocations.', title='Tool Invocations'
+        )
     )
 
 
@@ -15934,7 +15898,7 @@ class SimulationGetSimulationUnitTestSetRunsResponse(BaseModel):
 
 
 class ToolGetToolInvocationsResponse(BaseModel):
-    tool_invocations: list[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
+    tool_invocations: list[ToolGetToolInvocationsResponseToolInvocationInstance] = (
         Field(
             ..., description='The list of tool invocations.', title='Tool Invocations'
         )
@@ -15949,11 +15913,3 @@ class ToolGetToolInvocationsResponse(BaseModel):
         description='The continuation token to retrieve the next page of tool invocations, or null if there are no more tool invocations.',
         title='Continuation Token',
     )
-
-
-class ToolSearchToolInvocationsResponse(BaseModel):
-    tool_invocations: list[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
-        Field(
-            ..., description='The list of tool invocations.', title='Tool Invocations'
-        )
-    )

amigo_sdk/resources/user.py

@@ -2,6 +2,7 @@ from amigo_sdk.generated.model import (
     GetUsersParametersQuery,
     UserCreateInvitedUserRequest,
     UserCreateInvitedUserResponse,
+    UserGetUserModelResponse,
     UserGetUsersResponse,
     UserUpdateUserInfoRequest,
 )
@@ -54,6 +55,14 @@ class AsyncUserResource:
             json=body.model_dump(mode="json", exclude_none=True),
         )
 
+    async def get_user_model(self, user_id: str) -> UserGetUserModelResponse:
+        """Get the latest user model for a user."""
+        response = await self._http.request(
+            "GET",
+            f"/v1/{self._organization_id}/user/{user_id}/user_model",
+        )
+        return UserGetUserModelResponse.model_validate_json(response.text)
+
 
 class UserResource:
     """User resource (synchronous)."""
@@ -93,3 +102,10 @@ class UserResource:
             f"/v1/{self._organization_id}/user/{user_id}",
             json=body.model_dump(mode="json", exclude_none=True),
         )
+
+    def get_user_model(self, user_id: str) -> UserGetUserModelResponse:
+        response = self._http.request(
+            "GET",
+            f"/v1/{self._organization_id}/user/{user_id}/user_model",
+        )
+        return UserGetUserModelResponse.model_validate_json(response.text)

{amigo_sdk-0.97.0.dist-info → amigo_sdk-0.99.0.dist-info}/METADATA

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

{amigo_sdk-0.97.0.dist-info → amigo_sdk-0.99.0.dist-info}/RECORD

@@ -2,7 +2,7 @@ scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 scripts/aliases.json,sha256=c1_bkZyvNDQ8iMrKtUuZQqdP87GdLnRPrb3wjX533eQ,322
 scripts/check.py,sha256=ikFavw1IjX51FNjO-kIzXjYdep3on0dKZHYvPElPFtc,2872
 scripts/gen_models.py,sha256=c4J_gHYQIBL3CcKg5z8oUusLvPePytfvLdZ7STHBcNk,3553
-amigo_sdk/__init__.py,sha256=dO3eJrvgigE3qeUKVhp5Zh4JLFJAAvSWS_T6OfpYxBw,139
+amigo_sdk/__init__.py,sha256=fOBc3DBkFkXroPOMRTiJxzkTPgX9MrMlvui8UAyHCDY,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
@@ -10,13 +10,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=UQoe67pqk63SVV_PNN4or9YBSO88UaC_NZb3Td8zokc,446306
+amigo_sdk/generated/model.py,sha256=N85ReSiut41HgVUydr8HTqB6DTJHRkhEZwJFIqheeYY,444818
 amigo_sdk/resources/conversation.py,sha256=X47PbPA6NqdkqXgxAPOjlO1E0W7t695tn1tPWSikMOk,14807
 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.97.0.dist-info/METADATA,sha256=ORIIPXTHiZFRdzJmdxkMmDq2sSkDhg5Q3J-9StH92zI,8499
-amigo_sdk-0.97.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-amigo_sdk-0.97.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.97.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.97.0.dist-info/RECORD,,
+amigo_sdk/resources/user.py,sha256=P14lN-3fmIvELUjn_vpSaYpNsK-xd8zNmUV_KfcZPAU,4138
+amigo_sdk-0.99.0.dist-info/METADATA,sha256=3HF-35030s8F6r2XGXe9krwkvGtRzo7RzxG2ztjfnP4,8499
+amigo_sdk-0.99.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+amigo_sdk-0.99.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.99.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.99.0.dist-info/RECORD,,