amigo_sdk 0.109.0__py3-none-any.whl → 0.111.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.109.0"
+__version__ = "0.111.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-02-03T19:54:09+00:00
+#   timestamp: 2026-02-03T22:52:28+00:00
 
 from __future__ import annotations
 
@@ -1377,11 +1377,6 @@ class MongoCollectionsDynamicBehaviorSetVersionDynamicBehaviorSetVersionChangeTo
     )
 
 
-class MongoCollectionsMetricMetricTag(BaseModel):
-    key: str = Field(..., title='Key')
-    value: str | None = Field(..., title='Value')
-
-
 class MongoCollectionsMetricMetricUserInfo(
     MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
 ):
@@ -1460,6 +1455,11 @@ class MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo(
     pass
 
 
+class MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag(BaseModel):
+    key: str = Field(..., title='Key')
+    value: str | None = Field(..., title='Value')
+
+
 class MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUnitTestRunDescriptor(
     BaseModel
 ):
@@ -1500,12 +1500,6 @@ class MongoCollectionsTemporaryPermissionGrantTemporaryPermissionGrantUserInfo(
     pass
 
 
-class MongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
-    major: int = Field(..., title='Major')
-    minor: int = Field(..., title='Minor')
-    patch: int = Field(..., title='Patch')
-
-
 class MongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
     user_id: str = Field(..., description='The ID of the user.', title='User Id')
     user_org_id: str = Field(
@@ -1513,6 +1507,12 @@ class MongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
     )
 
 
+class MongoCollectionsToolVersionToolVersionVersion(BaseModel):
+    major: int = Field(..., title='Major')
+    minor: int = Field(..., title='Minor')
+    patch: int = Field(..., title='Patch')
+
+
 class MongoCollectionsUserUserPreferences(BaseModel):
     enable_response_recommendation: bool | None = Field(
         False,
@@ -1816,20 +1816,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, str | None] = Field(..., title='Tags')
-    created_at: AwareDatetime = Field(..., title='Created At')
-    creator: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
-    updated_at: AwareDatetime = Field(..., title='Updated At')
-    updated_by: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
-
-
 class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
     applied_to_services_ids: list[str] = Field(
         ...,
@@ -1850,14 +1836,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, str | None] = Field(..., title='Tags')
+    created_at: AwareDatetime = Field(..., title='Created At')
+    creator: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
+    updated_at: AwareDatetime = Field(..., title='Updated At')
+    updated_by: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
 
 
 class DynamicBehaviorSetUpdateDynamicBehaviorSetRequest(BaseModel):
@@ -10828,6 +10818,13 @@ class RoleModifyRoleResponse(BaseModel):
     )
 
 
+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')
 
@@ -10903,13 +10900,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
@@ -11015,16 +11005,6 @@ 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(
         ...,
@@ -11082,6 +11062,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(
         ...,
@@ -11113,6 +11127,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, str | None] = 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: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = 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: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = Field(
+        ..., description='The user who last updated the simulation scenario.'
+    )
+
+
 class SimulationGetSimulationUnitTestSetRunArtifactsResponse(BaseModel):
     presigned_url: str = Field(
         ...,
@@ -11201,68 +11246,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, 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'
     )
 
 
-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, str | None] = 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: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = 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: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = Field(
-        ..., description='The user who last updated the simulation scenario.'
+        description='The list of simulation scenarios.',
+        title='Simulation Scenarios',
     )
 
 
@@ -11328,35 +11326,13 @@ class RunCount(RootModel[int]):
     )
 
 
-class UnitTestRuns(
-    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: PydanticBaseModelStrippedNonemptyString1 | None = Field(
-        None, description='The name of the simulation unit test set.'
-    )
-    description: PydanticBaseModelStrippedNonemptyString1 | FieldNotSet | None = Field(
-        default_factory=lambda: PydanticBaseModelStrippedNonemptyString1({}),
-        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: UnitTestRuns | None = Field(
-        None,
-        description='The unit test runs that are part of this set.',
-        title='Unit Test Runs',
-    )
-    tags: dict[str, str | None] | None = Field(
-        None, description='The tags of the simulation unit test set.', title='Tags'
+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',
     )
 
 
@@ -11393,6 +11369,22 @@ class ToolGetToolsResponseFilterValues(BaseModel):
     )
 
 
+class InvocationMode(Enum):
+    regular = 'regular'
+    conversation_simulation = 'conversation-simulation'
+
+
+class ToolInvokeToolVersionRequestInput(BaseModel):
+    input_parameters: dict[str, Any] = Field(
+        ...,
+        description='The input parameters for the tool invocation.',
+        title='Input Parameters',
+    )
+    invocation_mode: InvocationMode = Field(
+        ..., description='The mode of invocation.', title='Invocation Mode'
+    )
+
+
 class ToolModifyToolRequest(BaseModel):
     description: PydanticBaseModelStrippedNonemptyString1 | None = Field(
         None, description='A description of this tool. Only updates if not-null.'
@@ -11432,19 +11424,34 @@ class ToolPublishToolVersionResponse(BaseModel):
     )
 
 
-class InvocationMode(Enum):
-    regular = 'regular'
-    conversation_simulation = 'conversation-simulation'
-
-
-class ToolTestToolRequestInput(BaseModel):
-    input_parameters: dict[str, Any] = Field(
+class ToolTestToolRequest(BaseModel):
+    inputs: list[ToolInvokeToolVersionRequestInput] = Field(
         ...,
-        description='The input parameters for the tool invocation.',
-        title='Input Parameters',
+        description='The list of inputs for the tool invocation.',
+        max_length=10,
+        min_length=1,
+        title='Inputs',
     )
-    invocation_mode: InvocationMode = Field(
-        ..., description='The mode of invocation.', title='Invocation Mode'
+    commit_branch: str = Field(
+        ...,
+        description='The branch in the tools repo whose tip will be tested.',
+        title='Commit Branch',
+    )
+    project_path: str = Field(
+        ...,
+        description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
+        pattern='^[-\\w\\d_/]+$',
+        title='Project Path',
+    )
+    required_envvars: dict[str, str] = Field(
+        ...,
+        description='The environment variables required for the tool to run.',
+        title='Required Envvars',
+    )
+    required_secret_envvars: dict[str, str] = Field(
+        ...,
+        description='The secret environment variables required for the tool to run.',
+        title='Required Secret Envvars',
     )
 
 
@@ -11966,7 +11973,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         title='Id',
     )
     message_type: list[MessageType] | None = Field(
-        ['external-event', 'user-message', 'agent-message'],
+        ['user-message', 'agent-message', 'external-event'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13711,7 +13718,7 @@ class ServiceInstance(BaseModel):
     agent_id: str = Field(
         ..., description='The ID of the agent that this service uses.', title='Agent Id'
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the service.', title='Tags'
     )
     keyterms: list[str] = Field(
@@ -13769,7 +13776,7 @@ class SimulationUnitTestSet(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13806,7 +13813,7 @@ class SimulationUnitTestSetInstance(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13934,7 +13941,7 @@ class ToolInstance(BaseModel):
         description='A list of encrypted environment variables that some versions of this tool uses.',
         title='Secret Envvars',
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
 
@@ -13949,7 +13956,7 @@ class ToolVersionInstance(BaseModel):
         description='The ID of the tool that this version belongs to.',
         title='Tool Id',
     )
-    version: MongoCollectionsToolInvocationToolInvocationVersion = Field(
+    version: MongoCollectionsToolVersionToolVersionVersion = Field(
         ..., description='The version of the tool.'
     )
     required_envvars: list[str] = Field(
@@ -14213,7 +14220,7 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestChangeToolCandidat
 
 class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
     dynamic_behavior_sets: list[
-        DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
+        DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
     ] = Field(
         ...,
         description='The retrieved dynamic behavior sets.',
@@ -14237,6 +14244,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(
         ...,
@@ -14516,6 +14533,10 @@ class OrganizationModifyOrganizationRequest(BaseModel):
     )
 
 
+class RoleCreateRoleRequestPermissionGrant(PermissionGrantInstance):
+    pass
+
+
 class RoleCreateTemporaryPermissionGrantRequest(BaseModel):
     user_id: str = Field(
         ...,
@@ -14540,8 +14561,36 @@ class RoleCreateTemporaryPermissionGrantRequest(BaseModel):
     )
 
 
-class RoleModifyRoleRequestPermissionGrant(PermissionGrantInstance):
-    pass
+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(
+        ...,
+        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: int | None = 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):
@@ -14564,20 +14613,10 @@ class ServiceGetServicesResponse(BaseModel):
     )
 
 
-class ServiceUpsertServiceVersionSetRequestVersionSet(BaseModel):
-    agent_version_number: int | None = 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: int | None = 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',
+class ServiceUpsertServiceVersionSetRequest(BaseModel):
+    version_set: ServiceCreateServiceRequestVersionSet = Field(
+        ..., description='The version set to upsert.'
     )
-    llm_model_preferences: dict[
-        str, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
-    ] = Field(..., title='Llm Model Preferences')
 
 
 class SimulationCreateSimulationPersonaRequest(BaseModel):
@@ -14629,7 +14668,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
         ..., description='The description of the simulation unit test set.'
     )
     unit_test_runs: list[
-        SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor
+        SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor
     ] = Field(
         ...,
         description='The unit test runs that are part of this set.',
@@ -14643,7 +14682,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
 
 class SimulationGetSimulationPersonasResponse(BaseModel):
     simulation_personas: list[
-        SimulationSearchSimulationPersonasResponseSimulationPersonaInstance
+        SimulationGetSimulationPersonasResponseSimulationPersonaInstance
     ] = Field(
         ..., description='The list of simulation personas.', title='Simulation Personas'
     )
@@ -14665,7 +14704,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
 
 class SimulationGetSimulationScenariosResponse(BaseModel):
     simulation_scenarios: list[
-        SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
+        SimulationGetSimulationScenariosResponseSimulationScenarioInstance
     ] = Field(
         ...,
         description='The list of simulation scenarios.',
@@ -14711,24 +14750,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(
         ...,
@@ -14743,6 +14764,38 @@ class SimulationUpdateSimulationUnitTestRequestSuccessCriterion(
     pass
 
 
+class UnitTestRuns(
+    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: PydanticBaseModelStrippedNonemptyString1 | None = Field(
+        None, description='The name of the simulation unit test set.'
+    )
+    description: PydanticBaseModelStrippedNonemptyString1 | FieldNotSet | None = Field(
+        default_factory=lambda: PydanticBaseModelStrippedNonemptyString1({}),
+        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: UnitTestRuns | None = Field(
+        None,
+        description='The unit test runs that are part of this set.',
+        title='Unit Test Runs',
+    )
+    tags: dict[str, str | None] | None = Field(
+        None, description='The tags of the simulation unit test set.', title='Tags'
+    )
+
+
 class ToolGetToolVersionsResponse(BaseModel):
     tool_versions: list[ToolVersionInstance] = Field(
         ...,
@@ -14784,7 +14837,7 @@ class ToolGetToolsResponse(BaseModel):
 
 
 class ToolInvokeToolVersionRequest(BaseModel):
-    inputs: list[ToolTestToolRequestInput] = Field(
+    inputs: list[ToolInvokeToolVersionRequestInput] = Field(
         ...,
         description='The list of inputs for the tool invocation.',
         max_length=10,
@@ -14808,37 +14861,6 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
     deletes: list[DeleteRequest] | None = Field(default_factory=list, title='Deletes')
 
 
-class ToolTestToolRequest(BaseModel):
-    inputs: list[ToolTestToolRequestInput] = Field(
-        ...,
-        description='The list of inputs for the tool invocation.',
-        max_length=10,
-        min_length=1,
-        title='Inputs',
-    )
-    commit_branch: str = Field(
-        ...,
-        description='The branch in the tools repo whose tip will be tested.',
-        title='Commit Branch',
-    )
-    project_path: str = Field(
-        ...,
-        description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
-        pattern='^[-\\w\\d_/]+$',
-        title='Project Path',
-    )
-    required_envvars: dict[str, str] = Field(
-        ...,
-        description='The environment variables required for the tool to run.',
-        title='Required Envvars',
-    )
-    required_secret_envvars: dict[str, str] = Field(
-        ...,
-        description='The secret environment variables required for the tool to run.',
-        title='Required Secret Envvars',
-    )
-
-
 class ToolTestToolResponse(ToolInvokeToolVersionResponse):
     pass
 
@@ -15038,7 +15060,7 @@ class Metric(BaseModel):
     additional_notes: str | None = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
     creator: MongoCollectionsMetricMetricUserInfo = Field(
@@ -15111,7 +15133,7 @@ class MetricInstance(BaseModel):
     additional_notes: str | None = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the metric.', title='Tags'
     )
     creator: MongoCollectionsMetricMetricUserInfo = Field(
@@ -15207,7 +15229,7 @@ class SimulationUnitTest(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15252,7 +15274,7 @@ class SimulationUnitTestInstance(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: list[MongoCollectionsMetricMetricTag] = Field(
+    tags: list[MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag] = Field(
         ..., description='The tags of the simulation persona.', title='Tags'
     )
     creator: MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15537,7 +15559,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',
@@ -15571,22 +15593,6 @@ class RoleGetTemporaryPermissionGrantsResponse(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.',
-    )
-
-
 class ServiceCreateServiceRequest(BaseModel):
     service_hierarchical_state_machine_id: str = Field(
         ...,
@@ -15609,7 +15615,7 @@ 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: ServiceUpsertServiceVersionSetRequestVersionSet | None = Field(
+    release_version_set: ServiceCreateServiceRequestVersionSet | None = 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.',
     )
@@ -15624,12 +15630,6 @@ 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(
@@ -15779,7 +15779,7 @@ class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
     tool_id: str = Field(
         ..., description='The ID of the tool that was invoked.', title='Tool Id'
     )
-    tool_version: MongoCollectionsToolInvocationToolInvocationVersion = Field(
+    tool_version: MongoCollectionsToolVersionToolVersionVersion = Field(
         ..., description='The version of the tool that was invoked.'
     )
 

{amigo_sdk-0.109.0.dist-info → amigo_sdk-0.111.0.dist-info}/METADATA

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

{amigo_sdk-0.109.0.dist-info → amigo_sdk-0.111.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=rO5MZzslnWheO_CqeWF3yb7vFzKiKID8adpSMDTbN48,140
+amigo_sdk/__init__.py,sha256=CobfN_6KhuLR-MSdF_BIB0pDDqgzmVXfxSVgqtOlaoQ,140
 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=odS_Apfmq8utIL9uYJW2rtxX67SYlW99jboacdgK44I,448739
+amigo_sdk/generated/model.py,sha256=tlZz3E2AJFk6_8_mpxyBWEw_IfAsPM-2CvsjGXcVfy0,448961
 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=P14lN-3fmIvELUjn_vpSaYpNsK-xd8zNmUV_KfcZPAU,4138
-amigo_sdk-0.109.0.dist-info/METADATA,sha256=kRZ3AaZ66vlAdgD90iW44Ja7tq2w06diTzgXPhaTRMw,8500
-amigo_sdk-0.109.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-amigo_sdk-0.109.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.109.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.109.0.dist-info/RECORD,,
+amigo_sdk-0.111.0.dist-info/METADATA,sha256=Y-2v5WWsZlplYDhGZKxI4L70QXylUQZ7ts1PSx1Y_es,8500
+amigo_sdk-0.111.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+amigo_sdk-0.111.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.111.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.111.0.dist-info/RECORD,,