amigo_sdk 0.90.0__py3-none-any.whl → 0.91.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.90.0"
+__version__ = "0.91.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-13T11:10:09+00:00
+#   timestamp: 2025-12-14T08:50:01+00:00
 
 from __future__ import annotations
 
@@ -1635,11 +1635,6 @@ class AmigoLibMongoCollectionsRoleRolePermissionGrantAction(Enum):
     deny = 'Deny'
 
 
-class AmigoLibMongoCollectionsServiceServiceTag(BaseModel):
-    key: str = Field(..., title='Key')
-    value: Optional[str] = Field(..., title='Value')
-
-
 class AmigoLibMongoCollectionsServiceServiceVersionSet(BaseModel):
     agent_version_number: Optional[int] = Field(
         ...,
@@ -1668,6 +1663,11 @@ AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
 )
 
 
+class AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag(BaseModel):
+    key: str = Field(..., title='Key')
+    value: Optional[str] = Field(..., title='Value')
+
+
 AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo = (
     AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
 )
@@ -1709,6 +1709,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(
@@ -1716,12 +1722,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,
@@ -1965,20 +1965,6 @@ 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(
         ...,
@@ -1999,14 +1985,18 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
     )
 
 
-class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
-    dynamic_behavior_sets: List[
-        DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
-    ] = Field(
-        ...,
-        description='The retrieved dynamic behavior sets.',
-        title='Dynamic Behavior Sets',
-    )
+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 DynamicBehaviorSetUpdateDynamicBehaviorSetRequestAppliedToService(
@@ -10767,13 +10757,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(
         ...,
@@ -10915,22 +10898,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(
         ...,
@@ -10977,13 +10956,6 @@ 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')
 
@@ -11059,6 +11031,13 @@ class ServiceUpdateServiceRequest(BaseModel):
     )
 
 
+class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
+    llm_name: LLMType
+    params: Optional[Dict[str, Any]] = 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
@@ -11177,6 +11156,16 @@ class SimulationCreateSimulationUnitTestResponse(BaseModel):
     )
 
 
+class SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
+    unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
+    run_count: int = Field(
+        ...,
+        description='The number of times to run the unit test.',
+        gt=0,
+        title='Run Count',
+    )
+
+
 class SimulationCreateSimulationUnitTestSetResponse(BaseModel):
     simulation_unit_test_set_id: str = Field(
         ...,
@@ -11236,6 +11225,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, Optional[str]] = 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: AmigoLibMongoCollectionsSimulationPersonaSimulationPersonaUserInfo = 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: AmigoLibMongoCollectionsSimulationPersonaSimulationPersonaUserInfo = (
+        Field(..., description='The user who last updated the simulation persona.')
+    )
+
+
 class SimulationGetSimulationScenarioVersionsResponse(BaseModel):
     simulation_scenario_versions: List[SimulationScenarioVersion] = Field(
         ...,
@@ -11267,6 +11290,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(
         ...,
@@ -11355,68 +11409,21 @@ 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, Optional[str]] = 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: AmigoLibMongoCollectionsSimulationPersonaSimulationPersonaUserInfo = 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: AmigoLibMongoCollectionsSimulationPersonaSimulationPersonaUserInfo = (
-        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'
     )
 
 
-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',
     )
 
 
@@ -11477,13 +11484,39 @@ class SimulationUpdateSimulationUnitTestRequestRunCount(RootModel[int]):
     )
 
 
-class SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
-    unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
-    run_count: int = Field(
-        ...,
-        description='The number of times to run the unit test.',
-        gt=0,
-        title='Run Count',
+class SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns(
+    RootModel[List[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
+):
+    root: List[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
+        Field(
+            ...,
+            description='The unit test runs that are part of this set.',
+            min_length=1,
+            title='Unit Test Runs',
+        )
+    )
+
+
+class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
+    name: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
+        None, description='The name of the simulation unit test set.'
+    )
+    description: Optional[
+        Union[AmigoLibPydanticBaseModelStrippedNonemptyString1, FieldNotSet]
+    ] = Field(
+        {},
+        description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
+        title='Description',
+    )
+    unit_test_runs: Optional[
+        SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns
+    ] = Field(
+        None,
+        description='The unit test runs that are part of this set.',
+        title='Unit Test Runs',
+    )
+    tags: Optional[Dict[str, Optional[str]]] = Field(
+        None, description='The tags of the simulation unit test set.', title='Tags'
     )
 
 
@@ -11682,7 +11715,7 @@ class UserGetMemoriesResponse(BaseModel):
     )
 
 
-class UserGetUsersResponseUserInstanceUserStats(BaseModel):
+class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
     num_conversations: int = Field(
         ...,
         description='The number of conversations the user has created.',
@@ -12137,7 +12170,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'],
+        ['external-event', 'agent-message', 'user-message'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13943,7 +13976,7 @@ class ServiceInstance(BaseModel):
     agent_id: str = Field(
         ..., description='The ID of the agent that this service uses.', title='Agent Id'
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the service.', title='Tags'
     )
 
@@ -13996,7 +14029,7 @@ class SimulationUnitTestSet(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14035,7 +14068,7 @@ class SimulationUnitTestSetInstance(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14151,7 +14184,7 @@ class ToolInstance(BaseModel):
         description='A list of encrypted environment variables that some versions of this tool uses.',
         title='Secret Envvars',
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
 
@@ -14166,7 +14199,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(
@@ -14482,7 +14515,7 @@ class DynamicBehaviorSetGetDynamicBehaviorSetVersionsResponse(BaseModel):
 
 class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
     dynamic_behavior_sets: List[
-        DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
+        DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
     ] = Field(
         ...,
         description='The retrieved dynamic behavior sets.',
@@ -14506,6 +14539,16 @@ 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(
         ...,
@@ -14562,7 +14605,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',
@@ -14707,6 +14750,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,
@@ -14804,22 +14858,6 @@ class RoleModifyRoleRequest(BaseModel):
     )
 
 
-class ServiceCreateServiceRequestVersionSet(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')
-    )
-
-
 class ServiceGetServicesResponse(BaseModel):
     services: List[ServiceInstance] = Field(
         ..., description='The services that are found.', title='Services'
@@ -14840,10 +14878,20 @@ class ServiceGetServicesResponse(BaseModel):
     )
 
 
-class ServiceUpsertServiceVersionSetRequest(BaseModel):
-    version_set: ServiceCreateServiceRequestVersionSet = Field(
-        ..., description='The version set to upsert.'
+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, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
+    ] = Field(..., title='Llm Model Preferences')
 
 
 class SimulationCreateSimulationPersonaRequest(BaseModel):
@@ -14895,7 +14943,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
         ..., description='The description of the simulation unit test set.'
     )
     unit_test_runs: List[
-        SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor
+        SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor
     ] = Field(
         ...,
         description='The unit test runs that are part of this set.',
@@ -14909,7 +14957,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
 
 class SimulationGetSimulationPersonasResponse(BaseModel):
     simulation_personas: List[
-        SimulationSearchSimulationPersonasResponseSimulationPersonaInstance
+        SimulationGetSimulationPersonasResponseSimulationPersonaInstance
     ] = Field(
         ..., description='The list of simulation personas.', title='Simulation Personas'
     )
@@ -14933,7 +14981,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
 
 class SimulationGetSimulationScenariosResponse(BaseModel):
     simulation_scenarios: List[
-        SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
+        SimulationGetSimulationScenariosResponseSimulationScenarioInstance
     ] = Field(
         ...,
         description='The list of simulation scenarios.',
@@ -14981,24 +15029,6 @@ class SimulationGetSimulationUnitTestSetsResponse(BaseModel):
     )
 
 
-class SimulationSearchSimulationPersonasResponse(BaseModel):
-    simulation_personas: List[
-        SimulationSearchSimulationPersonasResponseSimulationPersonaInstance
-    ] = Field(
-        ..., description='The list of simulation personas.', title='Simulation Personas'
-    )
-
-
-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(
         ...,
@@ -15012,42 +15042,6 @@ SimulationUpdateSimulationUnitTestRequestSuccessCriterion = (
 )
 
 
-class SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns(
-    RootModel[List[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
-):
-    root: List[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
-        Field(
-            ...,
-            description='The unit test runs that are part of this set.',
-            min_length=1,
-            title='Unit Test Runs',
-        )
-    )
-
-
-class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
-    name: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
-        None, description='The name of the simulation unit test set.'
-    )
-    description: Optional[
-        Union[AmigoLibPydanticBaseModelStrippedNonemptyString1, FieldNotSet]
-    ] = Field(
-        {},
-        description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
-        title='Description',
-    )
-    unit_test_runs: Optional[
-        SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns
-    ] = Field(
-        None,
-        description='The unit test runs that are part of this set.',
-        title='Unit Test Runs',
-    )
-    tags: Optional[Dict[str, Optional[str]]] = Field(
-        None, description='The tags of the simulation unit test set.', title='Tags'
-    )
-
-
 class ToolGetToolVersionsResponse(BaseModel):
     tool_versions: List[ToolVersionInstance] = Field(
         ...,
@@ -15150,7 +15144,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.',
@@ -15166,7 +15160,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')
@@ -15175,12 +15169,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'
@@ -15422,7 +15410,7 @@ class Metric(BaseModel):
     additional_notes: Optional[str] = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15492,7 +15480,7 @@ class MetricInstance(BaseModel):
     additional_notes: Optional[str] = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15694,7 +15682,7 @@ class SimulationUnitTest(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15739,7 +15727,7 @@ class SimulationUnitTestInstance(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -16087,9 +16075,11 @@ class ServiceCreateServiceRequest(BaseModel):
         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.',
+    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.',
+        )
     )
     keyterms: Optional[List[StrippedNonemptyStringAZAZ02]] = Field(
         [],
@@ -16102,6 +16092,12 @@ class ServiceCreateServiceRequest(BaseModel):
     )
 
 
+class ServiceUpsertServiceVersionSetRequest(BaseModel):
+    version_set: ServiceUpsertServiceVersionSetRequestVersionSet = Field(
+        ..., description='The version set to upsert.'
+    )
+
+
 class SimulationCreateSimulationUnitTestRequest(BaseModel):
     name: str = Field(..., description='The name of the unit test.', min_length=1)
     description: str = Field(
@@ -16257,7 +16253,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.'
     )
 
@@ -16271,7 +16267,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(
@@ -16284,6 +16280,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.90.0.dist-info → amigo_sdk-0.91.0.dist-info}/METADATA

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

{amigo_sdk-0.90.0.dist-info → amigo_sdk-0.91.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-amigo_sdk/__init__.py,sha256=Mxe_TpbSCMUnWJ10nBK8qAyp4UYKGdxV7vy9wfe0L-g,139
+amigo_sdk/__init__.py,sha256=EdXtEYEF9IRNsV36jRztrW_x6L2qnkHmu0CfOZ0-Pno,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=dDPj9a9EKB686jU0GAkgIZ2S6ck7gF62__tSuetlqbE,459444
+amigo_sdk/generated/model.py,sha256=h51VyTn94b0e6wpXHC00DaRROOzHGmKlyo1pDjxhJ7k,459742
 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.90.0.dist-info/METADATA,sha256=3uCo-lU99nVRntaSm9B4-oJmYD0c8m0phNMMpfLpBdI,8499
-amigo_sdk-0.90.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-amigo_sdk-0.90.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.90.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.90.0.dist-info/RECORD,,
+amigo_sdk-0.91.0.dist-info/METADATA,sha256=QAWIxow4fHlgsrhxYbY9vBWdVvIESjS9foraPksCEZs,8499
+amigo_sdk-0.91.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+amigo_sdk-0.91.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.91.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.91.0.dist-info/RECORD,,