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

amigo_sdk/__init__.py

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

amigo_sdk/generated/model.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  <stdin>
-#   timestamp: 2025-12-02T08:21:19+00:00
+#   timestamp: 2025-12-03T10:03:06+00:00
 
 from __future__ import annotations
 
@@ -33,12 +33,12 @@ class ActionTooLongEvent(BaseModel):
     )
 
 
-class AnnotationStateInputNextState(RootModel[List]):
-    root: List = Field(..., max_length=2, min_length=2, title='Next State')
+class AnnotationStateInputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(..., max_length=2, min_length=2, title='Next State')
 
 
-class AnnotationStateOutputNextState(RootModel[List]):
-    root: List = Field(
+class AnnotationStateOutputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description="The state to transition to after the inner thought message is generated. If it's a string, the session will transition to a state within this contextual graph. If it's a tuple, the first element\nmust be of the form `{external_contextual_graph_reference_name}.{state_name}` and represents a state in another contextual graph that the session will transition to, and the second element must be a state\nin this contextual graph that the session will transition to after reaching the terminal state of the external graph.",
         max_length=2,
@@ -68,7 +68,7 @@ class AnnotationStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -78,7 +78,7 @@ class AnnotationStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -154,8 +154,8 @@ class ConversationByTimeMetricInput(BaseModel):
     )
 
 
-class ConversationByTimeMetricOutputConversationCount(RootModel[List]):
-    root: List = Field(..., max_length=2, min_length=2)
+class ConversationByTimeMetricOutputConversationCount(RootModel[List[Any]]):
+    root: List[Any] = Field(..., max_length=2, min_length=2)
 
 
 class ConversationByTimeMetricOutput(BaseModel):
@@ -224,7 +224,7 @@ class DynamicBehaviorInvocation(BaseModel):
         description='The IDs of the agent inner thoughts that led to the dynamic behavior invocation.',
         title='Agent Inner Thoughts Ids',
     )
-    dynamic_behavior_set_version_info: List = Field(
+    dynamic_behavior_set_version_info: List[Any] = Field(
         ...,
         description='The ID and version number of the dynamic behavior set version that was activated.',
         max_length=2,
@@ -260,7 +260,7 @@ class EngageUserDynamicBehaviorOverrideEvent(BaseModel):
 
 class EngageUserInvocationMetadata(BaseModel):
     type: Literal['engage-user'] = Field('engage-user', title='Type')
-    current_state_machine_and_version: List = Field(
+    current_state_machine_and_version: List[Any] = Field(
         ...,
         description='The ID and version of the state machine that is currently being executed.',
         max_length=2,
@@ -320,8 +320,8 @@ class EvaluationSourceType(Enum):
     simulation = 'simulation'
 
 
-class ExitConditionOutputNextState(RootModel[List]):
-    root: List = Field(
+class ExitConditionOutputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description="The state to transition to if this exit condition is met. If it's a string, the session will transition to a state within this contextual graph. If it's a tuple, the first element\nmust be of the form `{external_contextual_graph_reference_name}.{state_name}` and represents a state in another contextual graph that the session will transition to, and the second element must be a state\nin this contextual graph that the session will transition to after reaching the terminal state of the external graph.",
         max_length=2,
@@ -442,7 +442,7 @@ class GenerateToolParamsDesignatedToolRetrievedEvent(BaseModel):
 
 class InCondition(BaseModel):
     type: Literal['In'] = Field('In', title='Type')
-    values: List = Field(
+    values: List[Any] = Field(
         ...,
         description='A list of values, one of which the attribute must be equal to. All values must be of the same type as the type of this attribute in the permission.',
         title='Values',
@@ -483,7 +483,7 @@ class JumpbackStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -493,7 +493,7 @@ class JumpbackStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -610,7 +610,7 @@ class NavigateStateMachineExitConditionSelectedEvent(BaseModel):
         description='The exit condition that the agent has selected.',
         title='Exit Condition',
     )
-    next_state_state_machine_version_info: List = Field(
+    next_state_state_machine_version_info: List[Any] = Field(
         ...,
         description='The state machine version info of the next state.',
         max_length=2,
@@ -700,7 +700,7 @@ class NavigateStateMachineStateTransitionedEvent(BaseModel):
         'navigate-state-machine-state-transitioned', title='Type'
     )
     prev_state: str = Field(..., description='The previous state.', title='Prev State')
-    prev_state_state_machine_version_info: List = Field(
+    prev_state_state_machine_version_info: List[Any] = Field(
         ...,
         description='The state machine version info of the previous state.',
         max_length=2,
@@ -708,7 +708,7 @@ class NavigateStateMachineStateTransitionedEvent(BaseModel):
         title='Prev State State Machine Version Info',
     )
     next_state: str = Field(..., description='The next state.', title='Next State')
-    next_state_state_machine_version_info: List = Field(
+    next_state_state_machine_version_info: List[Any] = Field(
         ...,
         description='The state machine version info of the next state.',
         max_length=2,
@@ -834,8 +834,8 @@ class RecallStateInputNextState(AnnotationStateInputNextState):
     pass
 
 
-class RecallStateOutputNextState(RootModel[List]):
-    root: List = Field(
+class RecallStateOutputNextState(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description="The state to transition to after the active memory extraction is performed. If it's a string, the session will transition to a state within this contextual graph. If it's a tuple, the first element\nmust be of the form `{external_contextual_graph_reference_name}.{state_name}` and represents a state in another contextual graph that the session will transition to, and the second element must be a state\nin this contextual graph that the session will transition to after reaching the terminal state of the external graph.",
         max_length=2,
@@ -870,7 +870,7 @@ class RecallStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -880,7 +880,7 @@ class RecallStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -914,9 +914,9 @@ class ResultPersistence(Enum):
 
 
 class SelectDynamicBehaviorCompletedEventSelectedDynamicBehaviorSetVersionInfo(
-    RootModel[List]
+    RootModel[List[Any]]
 ):
-    root: List = Field(
+    root: List[Any] = Field(
         ...,
         max_length=2,
         min_length=2,
@@ -925,9 +925,9 @@ class SelectDynamicBehaviorCompletedEventSelectedDynamicBehaviorSetVersionInfo(
 
 
 class SelectDynamicBehaviorCompletedEventPreviousSelectedDynamicBehaviorSetVersionInfo(
-    RootModel[List]
+    RootModel[List[Any]]
 ):
-    root: List = Field(
+    root: List[Any] = Field(
         ...,
         max_length=2,
         min_length=2,
@@ -1064,7 +1064,7 @@ class StateOrRefName(RootModel[str]):
 
 class StateTransitionInvocationMetadata(BaseModel):
     type: Literal['state-transition'] = Field('state-transition', title='Type')
-    current_state_machine_and_version: List = Field(
+    current_state_machine_and_version: List[Any] = Field(
         ...,
         description='The ID and version of the state machine that is currently being executed.',
         max_length=2,
@@ -1263,7 +1263,7 @@ class ToolCallStateOutput(BaseModel):
 
 class ToolCallStateInvocationMetadata(BaseModel):
     type: Literal['tool-call-state'] = Field('tool-call-state', title='Type')
-    current_state_machine_and_version: List = Field(
+    current_state_machine_and_version: List[Any] = Field(
         ...,
         description='The ID and version of the state machine that is currently being executed.',
         max_length=2,
@@ -1283,7 +1283,7 @@ class ToolCallStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -1293,7 +1293,7 @@ class ToolCallStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -1396,8 +1396,8 @@ class TotalMessageCountMetricOutput(BaseModel):
     message_count: int = Field(..., title='Message Count')
 
 
-class UnitTestRunSpecScenarioVersionInfo(RootModel[List]):
-    root: List = Field(
+class UnitTestRunSpecScenarioVersionInfo(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description='The version of the scenario that the unit test runs on. This field is only populated if the authenticated user has the `Simulation:GetSimulationUnitTest` permission.',
         max_length=2,
@@ -1406,8 +1406,8 @@ class UnitTestRunSpecScenarioVersionInfo(RootModel[List]):
     )
 
 
-class UnitTestRunSpecPersonaVersionInfo(RootModel[List]):
-    root: List = Field(
+class UnitTestRunSpecPersonaVersionInfo(RootModel[List[Any]]):
+    root: List[Any] = Field(
         ...,
         description='The version of the persona that the unit test runs on. This field is only populated if the authenticated user has the `Simulation:GetSimulationUnitTest` permission.',
         max_length=2,
@@ -1658,16 +1658,16 @@ AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
 )
 
 
+class AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag(BaseModel):
+    key: str = Field(..., title='Key')
+    value: Optional[str] = Field(..., title='Value')
+
+
 AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo = (
     AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
 )
 
 
-class AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag(BaseModel):
-    key: str = Field(..., title='Key')
-    value: Optional[str] = Field(..., title='Value')
-
-
 class AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUnitTestRunDescriptor(
     BaseModel
 ):
@@ -1704,6 +1704,12 @@ AmigoLibMongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo
 )
 
 
+class AmigoLibMongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
+    major: int = Field(..., title='Major')
+    minor: int = Field(..., title='Minor')
+    patch: int = Field(..., title='Patch')
+
+
 class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
     user_id: str = Field(..., description='The ID of the user.', title='User Id')
     user_org_id: str = Field(
@@ -1711,12 +1717,6 @@ class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
     )
 
 
-class AmigoLibMongoCollectionsToolVersionToolVersionVersion(BaseModel):
-    major: int = Field(..., title='Major')
-    minor: int = Field(..., title='Minor')
-    patch: int = Field(..., title='Patch')
-
-
 class AmigoLibMongoCollectionsUserUserPreferences(BaseModel):
     enable_response_recommendation: Optional[bool] = Field(
         False,
@@ -1780,7 +1780,7 @@ class AdminSubmitSqlQueryResponse(BaseModel):
         description='The time taken to execute the query in milliseconds.',
         title='Execution Time Ms',
     )
-    result: List[List] = Field(
+    result: List[List[Any]] = Field(
         ...,
         description='The result of the query as a list of rows, where each row is a list of column values.',
         title='Result',
@@ -1870,9 +1870,9 @@ class ConversationGenerateConversationStarterResponse(BaseModel):
 
 
 class ConversationGetInteractionInsightsResponseTriggeredDynamicBehaviorSetVersionInfo(
-    RootModel[List]
+    RootModel[List[Any]]
 ):
-    root: List = Field(
+    root: List[Any] = Field(
         ...,
         description='The ID and version number of the dynamic behavior set that was activated during this interaction.',
         max_length=2,
@@ -11236,6 +11236,37 @@ class SimulationGetSimulationScenariosResponseFilterValues(BaseModel):
     )
 
 
+class SimulationGetSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
+    id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
+    name: str = Field(
+        ..., description='The name of the simulation scenario.', title='Name'
+    )
+    tags: Dict[str, Optional[str]] = Field(
+        ..., description='The tags of the simulation scenario.', title='Tags'
+    )
+    is_deleted: bool = Field(
+        ...,
+        description='Whether the simulation scenario is deleted.',
+        title='Is Deleted',
+    )
+    created_at: AwareDatetime = Field(
+        ...,
+        description='The timestamp when the simulation scenario was created.',
+        title='Created At',
+    )
+    creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
+        Field(..., description='The user who created the simulation scenario.')
+    )
+    updated_at: AwareDatetime = Field(
+        ...,
+        description='The timestamp when the simulation scenario was last updated.',
+        title='Updated At',
+    )
+    updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
+        Field(..., description='The user who last updated the simulation scenario.')
+    )
+
+
 class SimulationGetSimulationUnitTestSetRunArtifactsResponse(BaseModel):
     presigned_url: str = Field(
         ...,
@@ -11358,34 +11389,13 @@ class SimulationSearchSimulationPersonasResponseSimulationPersonaInstance(BaseMo
     )
 
 
-class SimulationSearchSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
-    id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
-    name: str = Field(
-        ..., description='The name of the simulation scenario.', title='Name'
-    )
-    tags: Dict[str, Optional[str]] = Field(
-        ..., description='The tags of the simulation scenario.', title='Tags'
-    )
-    is_deleted: bool = Field(
-        ...,
-        description='Whether the simulation scenario is deleted.',
-        title='Is Deleted',
-    )
-    created_at: AwareDatetime = Field(
-        ...,
-        description='The timestamp when the simulation scenario was created.',
-        title='Created At',
-    )
-    creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
-        Field(..., description='The user who created the simulation scenario.')
-    )
-    updated_at: AwareDatetime = Field(
+class SimulationSearchSimulationScenariosResponse(BaseModel):
+    simulation_scenarios: List[
+        SimulationGetSimulationScenariosResponseSimulationScenarioInstance
+    ] = Field(
         ...,
-        description='The timestamp when the simulation scenario was last updated.',
-        title='Updated At',
-    )
-    updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
-        Field(..., description='The user who last updated the simulation scenario.')
+        description='The list of simulation scenarios.',
+        title='Simulation Scenarios',
     )
 
 
@@ -11515,22 +11525,6 @@ class ToolGetToolsResponseFilterValues(BaseModel):
     )
 
 
-class ToolInvokeToolVersionRequestInputInvocationMode(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: ToolInvokeToolVersionRequestInputInvocationMode = Field(
-        ..., description='The mode of invocation.', title='Invocation Mode'
-    )
-
-
 class ToolModifyToolRequest(BaseModel):
     description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
         None, description='A description of this tool. Only updates if not-null.'
@@ -11573,34 +11567,19 @@ class ToolPublishToolVersionResponse(BaseModel):
     )
 
 
-class ToolTestToolRequest(BaseModel):
-    inputs: List[ToolInvokeToolVersionRequestInput] = 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(
+class ToolTestToolRequestInputInvocationMode(Enum):
+    regular = 'regular'
+    conversation_simulation = 'conversation-simulation'
+
+
+class ToolTestToolRequestInput(BaseModel):
+    input_parameters: Dict[str, Any] = Field(
         ...,
-        description='The environment variables required for the tool to run.',
-        title='Required Envvars',
+        description='The input parameters for the tool invocation.',
+        title='Input Parameters',
     )
-    required_secret_envvars: Dict[str, str] = Field(
-        ...,
-        description='The secret environment variables required for the tool to run.',
-        title='Required Secret Envvars',
+    invocation_mode: ToolTestToolRequestInputInvocationMode = Field(
+        ..., description='The mode of invocation.', title='Invocation Mode'
     )
 
 
@@ -12111,7 +12090,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         [], description='The IDs of the messages to retrieve.', title='Id'
     )
     message_type: Optional[List[MessageType]] = Field(
-        ['user-message', 'agent-message', 'external-event'],
+        ['user-message', 'external-event', 'agent-message'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13556,7 +13535,7 @@ class ActionStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -13566,7 +13545,7 @@ class ActionStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -13654,7 +13633,7 @@ class DecisionStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -13664,7 +13643,7 @@ class DecisionStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -13837,7 +13816,7 @@ class ReflectionStateTransitionLog(BaseModel):
         description='Name of the previous state to transition from.',
         title='Previous State',
     )
-    previous_service_hierarchical_state_machine_version_info: List = Field(
+    previous_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the previous state is in.',
         max_length=2,
@@ -13847,7 +13826,7 @@ class ReflectionStateTransitionLog(BaseModel):
     next_state: str = Field(
         ..., description='Name of the next state to transition to.', title='Next State'
     )
-    next_service_hierarchical_state_machine_version_info: List = Field(
+    next_service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The identifier of the service hierarchical state machine that the next state is in.',
         max_length=2,
@@ -13917,9 +13896,9 @@ class ServiceInstance(BaseModel):
     agent_id: str = Field(
         ..., description='The ID of the agent that this service uses.', title='Agent Id'
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(..., description='The tags of the service.', title='Tags')
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
+        ..., description='The tags of the service.', title='Tags'
+    )
 
 
 class SimulationConversationInvocationSource(BaseModel):
@@ -13970,9 +13949,7 @@ class SimulationUnitTestSet(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14011,9 +13988,7 @@ class SimulationUnitTestSetInstance(BaseModel):
         description='Whether the unit test set has been deleted.',
         title='Is Deleted',
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
         ..., description='The tags of the simulation unit test set.', title='Tags'
     )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14129,9 +14104,9 @@ class ToolInstance(BaseModel):
         description='A list of encrypted environment variables that some versions of this tool uses.',
         title='Secret Envvars',
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(..., description='The tags of the simulation persona.', title='Tags')
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
+        ..., description='The tags of the simulation persona.', title='Tags'
+    )
 
 
 class ToolVersionInstance(BaseModel):
@@ -14144,7 +14119,7 @@ class ToolVersionInstance(BaseModel):
         description='The ID of the tool that this version belongs to.',
         title='Tool Id',
     )
-    version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
+    version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
         ..., description='The version of the tool.'
     )
     required_envvars: List[str] = Field(
@@ -14225,14 +14200,14 @@ class VersionSetInfo(BaseModel):
         description='The name of the version set. If the value is `unknown`, it means the conversation was created before this field was introduced and the version set name cannot be retroactively determined.',
         title='Name',
     )
-    agent_version_info: List = Field(
+    agent_version_info: List[Any] = Field(
         ...,
         description='The ID of the agent and version number used in this conversation.',
         max_length=2,
         min_length=2,
         title='Agent Version Info',
     )
-    service_hierarchical_state_machine_version_info: List = Field(
+    service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The ID of the state machine and version number used in this conversation.',
         max_length=2,
@@ -14791,22 +14766,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, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
-    ] = Field(..., title='Llm Model Preferences')
-
-
 class ServiceGetServicesResponse(BaseModel):
     services: List[ServiceInstance] = Field(
         ..., description='The services that are found.', title='Services'
@@ -14827,10 +14786,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):
@@ -14920,7 +14889,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
 
 class SimulationGetSimulationScenariosResponse(BaseModel):
     simulation_scenarios: List[
-        SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
+        SimulationGetSimulationScenariosResponseSimulationScenarioInstance
     ] = Field(
         ...,
         description='The list of simulation scenarios.',
@@ -14976,16 +14945,6 @@ class SimulationSearchSimulationPersonasResponse(BaseModel):
     )
 
 
-class SimulationSearchSimulationScenariosResponse(BaseModel):
-    simulation_scenarios: List[
-        SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
-    ] = Field(
-        ...,
-        description='The list of simulation scenarios.',
-        title='Simulation Scenarios',
-    )
-
-
 class SimulationSearchSimulationUnitTestSetResponse(BaseModel):
     simulation_unit_test_sets: List[SimulationUnitTestSet] = Field(
         ...,
@@ -15040,7 +14999,7 @@ class ToolGetToolsResponse(BaseModel):
 
 
 class ToolInvokeToolVersionRequest(BaseModel):
-    inputs: List[ToolInvokeToolVersionRequestInput] = Field(
+    inputs: List[ToolTestToolRequestInput] = Field(
         ...,
         description='The list of inputs for the tool invocation.',
         max_length=10,
@@ -15061,6 +15020,37 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
     deletes: Optional[List[DeleteRequest]] = Field([], 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',
+    )
+
+
 ToolTestToolResponse = ToolInvokeToolVersionResponse
 
 
@@ -15377,9 +15367,9 @@ class Metric(BaseModel):
     additional_notes: Optional[str] = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(..., description='The tags of the metric.', title='Tags')
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
+        ..., description='The tags of the metric.', title='Tags'
+    )
     creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
         ..., description='The user who created the metric.'
     )
@@ -15447,9 +15437,9 @@ class MetricInstance(BaseModel):
     additional_notes: Optional[str] = Field(
         ..., description='Additional notes about the metric.', title='Additional Notes'
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(..., description='The tags of the metric.', title='Tags')
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
+        ..., description='The tags of the metric.', title='Tags'
+    )
     creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
         ..., description='The user who created the metric.'
     )
@@ -15587,7 +15577,7 @@ class ServiceHierarchicalStateMachineInstance(BaseModel):
         description='The state the user will be in when the session ends. This must be an action state.',
         title='Terminal State',
     )
-    references: Dict[str, List] = Field(
+    references: Dict[str, List[Any]] = Field(
         ...,
         description="A dictionary of other service hierarchical state machine versions that this references to. This field is a map of the reference name to the machine's ID and its version number.",
         title='References',
@@ -15649,9 +15639,9 @@ class SimulationUnitTest(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(..., description='The tags of the simulation persona.', title='Tags')
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
+        ..., description='The tags of the simulation persona.', title='Tags'
+    )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
     updated_by: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
 
@@ -15694,9 +15684,9 @@ class SimulationUnitTestInstance(BaseModel):
     is_deleted: bool = Field(
         ..., description='Whether the unit test has been deleted.', title='Is Deleted'
     )
-    tags: List[
-        AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetTag
-    ] = Field(..., description='The tags of the simulation persona.', title='Tags')
+    tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
+        ..., description='The tags of the simulation persona.', title='Tags'
+    )
     creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
     updated_by: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
 
@@ -15708,14 +15698,14 @@ class UnitTestRunSpec(BaseModel):
     run_count: int = Field(
         ..., description='The number of times to run the unit test.', title='Run Count'
     )
-    service_hierarchical_state_machine_version_info: List = Field(
+    service_hierarchical_state_machine_version_info: List[Any] = Field(
         ...,
         description='The version of the service hierarchical state machine that the unit test runs on.',
         max_length=2,
         min_length=2,
         title='Service Hierarchical State Machine Version Info',
     )
-    agent_version_info: List = Field(
+    agent_version_info: List[Any] = Field(
         ...,
         description='The version of the agent that the unit test runs on.',
         max_length=2,
@@ -15922,7 +15912,7 @@ class OrganizationCreateServiceHierarchicalStateMachineVersionRequest(BaseModel)
         description='The state the user will be in when the session ends. This must be an action state, and must be an internal state.',
         pattern='^[A-Za-z0-9_]+$',
     )
-    references: Dict[str, List] = Field(
+    references: Dict[str, List[Any]] = Field(
         ...,
         description="A dictionary of other service hierarchical state machines that this references to. This field is a map of the reference name to the machine's ID and its version number.",
         title='References',
@@ -16042,15 +16032,23 @@ 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.',
+        )
     )
     tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
         ..., description='The tags of this service.', title='Tags'
     )
 
 
+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(
@@ -16184,7 +16182,7 @@ class SimulationUpdateSimulationUnitTestRequest(BaseModel):
     )
 
 
-class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
+class ToolSearchToolInvocationsResponseToolInvocationInstance(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: Optional[AwareDatetime] = Field(None, title='Created At')
@@ -16206,19 +16204,11 @@ 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.'
     )
 
 
-class ToolSearchToolInvocationsResponse(BaseModel):
-    tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
-        Field(
-            ..., description='The list of tool invocations.', title='Tool Invocations'
-        )
-    )
-
-
 class UserGetUsersResponse(BaseModel):
     users: List[UserGetUsersResponseUserInstance] = Field(
         ..., description='Users in this organization.', title='Users'
@@ -16332,7 +16322,7 @@ class SimulationGetSimulationUnitTestSetRunsResponse(BaseModel):
 
 
 class ToolGetToolInvocationsResponse(BaseModel):
-    tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
+    tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
         Field(
             ..., description='The list of tool invocations.', title='Tool Invocations'
         )
@@ -16347,3 +16337,11 @@ 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-0.67.0.dist-info → amigo_sdk-0.68.0.dist-info}/METADATA

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

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

@@ -1,4 +1,4 @@
-amigo_sdk/__init__.py,sha256=_Vt20XUAV719JwAfR87fdDdt_AzUTzcNx0cHi5VVTdA,139
+amigo_sdk/__init__.py,sha256=wYrr7-kI3UXVygIXUsFCktVIZI4zZYVos0z-lkZBmgk,139
 amigo_sdk/_retry_utils.py,sha256=VbA_Uwv268J80d11R9XfnCjZiV-NwE1aXRsSlRjjRio,2108
 amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
 amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
@@ -6,13 +6,13 @@ amigo_sdk/errors.py,sha256=Yn12AxIQ6zE9jnGLWO0MUVbBsmw77Kzh82AijyreOkA,4818
 amigo_sdk/http_client.py,sha256=tfMXHqMTmlz2ja7aJhOVgbwAiJNlKCzPcK4UGNqXHO8,13488
 amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
 amigo_sdk/sdk_client.py,sha256=5nj8hzjHSFRs9JMNabdYIhQAcFE2G1jI_r7SHVGr9No,6106
-amigo_sdk/generated/model.py,sha256=byPdfHdBjkFPxltnyGsxQEo1yl8pF9x53IIF_OKJWzE,456940
+amigo_sdk/generated/model.py,sha256=AC2gU43CVs61eLPn6vwaoy_2qS4h9IHRTztw69QOCMM,457134
 amigo_sdk/resources/conversation.py,sha256=u4kOx5LPbhX92L41UnmPmaF9TkQA5bGev8N00dMoP9I,14805
 amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
 amigo_sdk/resources/service.py,sha256=t1iA3nS9co-wuR-x5jBhAXXTWfMeGsLwfQcLycwVrCA,1536
 amigo_sdk/resources/user.py,sha256=zikijiuGXgmkBVtrldor6XQGQK6jWXonDvNFuQs6XK4,3472
-amigo_sdk-0.67.0.dist-info/METADATA,sha256=qTyi_Kit19ZHi0PkFa6POvNZDZOoiBvsCCnetJZc40s,8499
-amigo_sdk-0.67.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-amigo_sdk-0.67.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.67.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.67.0.dist-info/RECORD,,
+amigo_sdk-0.68.0.dist-info/METADATA,sha256=DEJu0boyDaXm4W5LMyXWhlUsZHKWZPbih_hCauyiZC0,8499
+amigo_sdk-0.68.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+amigo_sdk-0.68.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.68.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.68.0.dist-info/RECORD,,