amigo_sdk 0.36.0__py3-none-any.whl → 0.38.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.36.0"
+__version__ = "0.38.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-10-02T02:40:25+00:00
+#   timestamp: 2025-10-02T18:06:48+00:00
 
 from __future__ import annotations
 
@@ -240,7 +240,7 @@ class EngageUserDynamicBehaviorOverrideEvent(BaseModel):
     )
 
 
-class EngageUserInvocationMetadataInput(BaseModel):
+class EngageUserInvocationMetadata(BaseModel):
     type: Literal['engage-user'] = Field('engage-user', title='Type')
     current_state_machine_and_version: List = Field(
         ...,
@@ -256,9 +256,6 @@ class EngageUserInvocationMetadataInput(BaseModel):
     )
 
 
-EngageUserInvocationMetadataOutput = EngageUserInvocationMetadataInput
-
-
 class EngageUserMessageFragmentGeneratedEvent(BaseModel):
     type: Literal['engage-user-message-fragment-generated'] = Field(
         'engage-user-message-fragment-generated', title='Type'
@@ -8871,24 +8868,6 @@ class ReflectionStateOutputNextState(AnnotationStateOutputNextState):
     pass
 
 
-class RegularConversationInvocationSourceInputConversationId(RootModel[str]):
-    root: str = Field(
-        ...,
-        description='The ID of the conversation where the tool was invoked. This conversation might not actually exist if later in the call the conversation is rolled back.',
-        pattern='^[a-f0-9]{24}$',
-        title='Conversation Id',
-    )
-
-
-class RegularConversationInvocationSourceInputInteractionId(RootModel[str]):
-    root: str = Field(
-        ...,
-        description='The ID of the interaction where the tool was invoked. This interaction might not actually exist if later in the call the interaction is rolled back.',
-        pattern='^[a-f0-9]{24}$',
-        title='Interaction Id',
-    )
-
-
 class RelationshipToDeveloperOutput(BaseModel):
     ownership: str = Field(
         ...,
@@ -9018,18 +8997,6 @@ class SideEffectStateTransitionLog(BaseModel):
     )
 
 
-class SimulationConversationInvocationSourceInputSimulationUnitTestSetRunId(
-    RootModel[str]
-):
-    root: str = Field(
-        ..., pattern='^[a-f0-9]{24}$', title='Simulation Unit Test Set Run Id'
-    )
-
-
-class SimulationConversationInvocationSourceInputUnitTestId(RootModel[str]):
-    root: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
-
-
 class SimulationEvaluationSource(BaseModel):
     type: Literal['simulation'] = Field('simulation', title='Type')
     simulation_unit_test_set_run_id: str = Field(
@@ -9100,7 +9067,7 @@ class StateOrRefName(RootModel[str]):
     root: str = Field(..., pattern='^[A-Za-z0-9_]+$')
 
 
-class StateTransitionInvocationMetadataInput(BaseModel):
+class StateTransitionInvocationMetadata(BaseModel):
     type: Literal['state-transition'] = Field('state-transition', title='Type')
     current_state_machine_and_version: List = Field(
         ...,
@@ -9126,9 +9093,6 @@ class StateTransitionInvocationMetadataInput(BaseModel):
     )
 
 
-StateTransitionInvocationMetadataOutput = StateTransitionInvocationMetadataInput
-
-
 class StrippedNonemptyStringWS(RootModel[str]):
     root: str = Field(..., min_length=1, pattern='^[\\w\\s]+$')
 
@@ -11344,6 +11308,38 @@ class ToolPublishToolVersionResponse(BaseModel):
     )
 
 
+class ToolTestToolRequest(BaseModel):
+    inputs: List[Input] = Field(
+        ...,
+        description='The list of inputs for the tool invocation.',
+        max_length=10,
+        min_length=1,
+        title='Inputs',
+    )
+    commit_hash: str = Field(
+        ...,
+        description='The commit hash of the tool to test.',
+        pattern='^[a-f0-9]{40}$',
+        title='Commit Hash',
+    )
+    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 UserCreateInvitedUserRequestLoginLink(RootModel[AnyUrl]):
     root: AnyUrl = Field(
         ...,
@@ -11875,7 +11871,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         [], description='The IDs of the messages to retrieve.', title='Id'
     )
     message_type: Optional[List[MessageType]] = Field(
-        ['external-event', 'agent-message', 'user-message'],
+        ['agent-message', 'external-event', 'user-message'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13077,6 +13073,11 @@ class GetToolInvocationsParametersQueryToolIdItem(
     pass
 
 
+class GetToolInvocationsParametersQueryInvocationSourceTypeEnum(Enum):
+    regular_conversation = 'regular-conversation'
+    simulation_conversation = 'simulation-conversation'
+
+
 class GetToolInvocationsParametersQueryConversationIdItem(
     DynamicBehaviorSetCreateDynamicBehaviorSetRequestAppliedToService
 ):
@@ -13089,6 +13090,56 @@ class GetToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem(
     pass
 
 
+class GetToolInvocationsParametersQuery(BaseModel):
+    tool_id: Optional[List[GetToolInvocationsParametersQueryToolIdItem]] = Field(
+        [], description='The ID of the tool to get invocations for.', title='Tool Id'
+    )
+    version: Optional[List[str]] = Field(
+        [],
+        description='A list of semver constraint that specifies the version to retrieve in the [Python packaging specification](https://packaging.python.org/en/latest/specifications/version-specifiers/#id5). This must be the exact same length as `tool_id`, and each entry corresponds to the tool ID at the same index.',
+        title='Version',
+    )
+    invocation_source_type: Optional[
+        List[GetToolInvocationsParametersQueryInvocationSourceTypeEnum]
+    ] = Field(
+        [], description='The source of the invocation', title='Invocation Source Type'
+    )
+    conversation_id: Optional[
+        List[GetToolInvocationsParametersQueryConversationIdItem]
+    ] = Field(
+        [],
+        description="The conversation ID associated with the invocation if it's of invocation source `regular-conversation`",
+        title='Conversation Id',
+    )
+    simulation_unit_test_set_run_id: Optional[
+        List[GetToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem]
+    ] = Field(
+        [],
+        description="The simulation unit test set run ID associated with the invocation if it's of invocation source `simulation-conversation`",
+        title='Simulation Unit Test Set Run Id',
+    )
+    succeeded: Optional[bool] = Field(
+        None, description='Whether the invocation succeeded', title='Succeeded'
+    )
+    limit: Optional[int] = Field(
+        100,
+        description='The maximum number of tool invocations to return.',
+        ge=0,
+        le=100,
+        title='Limit',
+    )
+    continuation_token: Optional[int] = Field(
+        0,
+        description='The continuation token from the previous request used to retrieve the next page of tool invocations.',
+        title='Continuation Token',
+    )
+    sort_by: Optional[List[str]] = Field(
+        [],
+        description='The fields to sort the versions by. Supported fields are `created_at`, `version.major`, `version.minor`, `version.patch`, `tool_id`, `invocation_source.type`, and `invocation_status.succeeded`. Specify a `+` before the field name to indicate ascending sorting and `-` for descending sorting. Multiple fields can be specified to break ties.',
+        title='Sort By',
+    )
+
+
 class SearchToolInvocationsParametersQueryQueryAgainstEnum(Enum):
     invocation_status_exception_message = 'invocation_status.exception_message'
     invocation_status_exception_type = 'invocation_status.exception_type'
@@ -13114,6 +13165,41 @@ class SearchToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem(
     pass
 
 
+class SearchToolInvocationsParametersQuery(BaseModel):
+    query: str = Field(..., description='The search query.', title='Query')
+    query_against: List[SearchToolInvocationsParametersQueryQueryAgainstEnum] = Field(
+        ...,
+        description='The fields to query against. By default, all supported fields are queried against.',
+        min_length=1,
+        title='Query Against',
+    )
+    tool_id: Optional[List[SearchToolInvocationsParametersQueryToolIdItem]] = Field(
+        [], description='The ID of the tool to get invocations for.', title='Tool Id'
+    )
+    invocation_source_type: Optional[
+        List[GetToolInvocationsParametersQueryInvocationSourceTypeEnum]
+    ] = Field(
+        [], description='The source of the invocation', title='Invocation Source Type'
+    )
+    conversation_id: Optional[
+        List[SearchToolInvocationsParametersQueryConversationIdItem]
+    ] = Field(
+        [],
+        description="The conversation ID associated with the invocation if it's of invocation source `regular-conversation`",
+        title='Conversation Id',
+    )
+    simulation_unit_test_set_run_id: Optional[
+        List[SearchToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem]
+    ] = Field(
+        [],
+        description="The simulation unit test set run ID associated with the invocation if it's of invocation source `simulation-conversation`",
+        title='Simulation Unit Test Set Run Id',
+    )
+    succeeded: Optional[bool] = Field(
+        None, description='Whether the invocation succeeded', title='Succeeded'
+    )
+
+
 class APIKey(BaseModel):
     field_id: str = Field(
         ..., alias='_id', description='The ID of the document.', title='Id'
@@ -13412,6 +13498,14 @@ class InsertRequest(BaseModel):
     )
 
 
+class InvocationResult(
+    RootModel[Union[SucceededInvocationResult, FailedInvocationResult]]
+):
+    root: Union[SucceededInvocationResult, FailedInvocationResult] = Field(
+        ..., discriminator='succeeded'
+    )
+
+
 class LLMConfigInput(BaseModel):
     llm_name: LLMType
     top_p: float = Field(..., description='The preferred `top_p` value.', title='Top P')
@@ -13608,33 +13702,7 @@ class ReflectionStateTransitionLog(BaseModel):
     )
 
 
-class RegularConversationInvocationSourceInput(BaseModel):
-    type: Literal['regular-conversation'] = Field('regular-conversation', title='Type')
-    user_id: str = Field(
-        ...,
-        description='The ID of the user whose conversation invoked the tool.',
-        title='User Id',
-    )
-    conversation_id: Optional[
-        RegularConversationInvocationSourceInputConversationId
-    ] = Field(
-        ...,
-        description='The ID of the conversation where the tool was invoked. This conversation might not actually exist if later in the call the conversation is rolled back.',
-        title='Conversation Id',
-    )
-    interaction_id: Optional[RegularConversationInvocationSourceInputInteractionId] = (
-        Field(
-            ...,
-            description='The ID of the interaction where the tool was invoked. This interaction might not actually exist if later in the call the interaction is rolled back.',
-            title='Interaction Id',
-        )
-    )
-    invocation_metadata: Union[
-        StateTransitionInvocationMetadataInput, EngageUserInvocationMetadataInput
-    ] = Field(..., description='Metadata about the invocation.', discriminator='type')
-
-
-class RegularConversationInvocationSourceOutput(BaseModel):
+class RegularConversationInvocationSource(BaseModel):
     type: Literal['regular-conversation'] = Field('regular-conversation', title='Type')
     user_id: str = Field(
         ...,
@@ -13652,7 +13720,7 @@ class RegularConversationInvocationSourceOutput(BaseModel):
         title='Interaction Id',
     )
     invocation_metadata: Union[
-        StateTransitionInvocationMetadataOutput, EngageUserInvocationMetadataOutput
+        StateTransitionInvocationMetadata, EngageUserInvocationMetadata
     ] = Field(..., description='Metadata about the invocation.', discriminator='type')
 
 
@@ -13690,23 +13758,7 @@ class RoleInstance(BaseModel):
     )
 
 
-class SimulationConversationInvocationSourceInput(BaseModel):
-    type: Literal['simulation-conversation'] = Field(
-        'simulation-conversation', title='Type'
-    )
-    simulation_unit_test_set_run_id: Optional[
-        SimulationConversationInvocationSourceInputSimulationUnitTestSetRunId
-    ] = Field(..., title='Simulation Unit Test Set Run Id')
-    unit_test_id: Optional[SimulationConversationInvocationSourceInputUnitTestId] = (
-        Field(..., title='Unit Test Id')
-    )
-    run_index: Optional[int] = Field(..., title='Run Index')
-    invocation_metadata: Union[
-        StateTransitionInvocationMetadataInput, EngageUserInvocationMetadataInput
-    ] = Field(..., description='Metadata about the invocation.', discriminator='type')
-
-
-class SimulationConversationInvocationSourceOutput(BaseModel):
+class SimulationConversationInvocationSource(BaseModel):
     type: Literal['simulation-conversation'] = Field(
         'simulation-conversation', title='Type'
     )
@@ -13716,7 +13768,7 @@ class SimulationConversationInvocationSourceOutput(BaseModel):
     unit_test_id: Optional[str] = Field(..., title='Unit Test Id')
     run_index: Optional[int] = Field(..., title='Run Index')
     invocation_metadata: Union[
-        StateTransitionInvocationMetadataOutput, EngageUserInvocationMetadataOutput
+        StateTransitionInvocationMetadata, EngageUserInvocationMetadata
     ] = Field(..., description='Metadata about the invocation.', discriminator='type')
 
 
@@ -14849,6 +14901,9 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
     deletes: Optional[List[DeleteRequest]] = Field([], title='Deletes')
 
 
+ToolTestToolResponse = ToolInvokeToolVersionResponse
+
+
 class UserCreateInvitedUserRequest(BaseModel):
     first_name: str = Field(
         ..., description='The first name of the user.', min_length=1
@@ -15293,8 +15348,7 @@ class ToolInvocationInstance(BaseModel):
         ..., description='The status of the tool invocation.', discriminator='succeeded'
     )
     invocation_source: Union[
-        RegularConversationInvocationSourceOutput,
-        SimulationConversationInvocationSourceOutput,
+        RegularConversationInvocationSource, SimulationConversationInvocationSource
     ] = Field(
         ..., description='The source of the tool invocation.', discriminator='type'
     )
@@ -15755,101 +15809,6 @@ class ToolSearchToolInvocationsResponse(BaseModel):
     )
 
 
-class GetToolInvocationsParametersQuery(BaseModel):
-    tool_id: Optional[List[GetToolInvocationsParametersQueryToolIdItem]] = Field(
-        [], description='The ID of the tool to get invocations for.', title='Tool Id'
-    )
-    version: Optional[List[Optional[str]]] = Field(
-        [],
-        description='A list of semver constraint that specifies the version to retrieve in the [Python packaging specification](https://packaging.python.org/en/latest/specifications/version-specifiers/#id5). This must be the exact same length as `tool_id`, and each entry corresponds to the tool ID at the same index. If an entry is null, all versions of the corresponding tool are retrieved.',
-        title='Version',
-    )
-    invocation_source_type: Optional[
-        List[
-            Union[
-                RegularConversationInvocationSourceInput,
-                SimulationConversationInvocationSourceInput,
-            ]
-        ]
-    ] = Field(
-        [], description='The source of the invocation', title='Invocation Source Type'
-    )
-    conversation_id: Optional[
-        List[GetToolInvocationsParametersQueryConversationIdItem]
-    ] = Field(
-        [],
-        description="The conversation ID associated with the invocation if it's of invocation source `regular-conversation`",
-        title='Conversation Id',
-    )
-    simulation_unit_test_set_run_id: Optional[
-        List[GetToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem]
-    ] = Field(
-        [],
-        description="The simulation unit test set run ID associated with the invocation if it's of invocation source `simulation-conversation`",
-        title='Simulation Unit Test Set Run Id',
-    )
-    succeeded: Optional[bool] = Field(
-        None, description='Whether the invocation succeeded', title='Succeeded'
-    )
-    limit: Optional[int] = Field(
-        100,
-        description='The maximum number of tool invocations to return.',
-        ge=0,
-        le=100,
-        title='Limit',
-    )
-    continuation_token: Optional[int] = Field(
-        0,
-        description='The continuation token from the previous request used to retrieve the next page of tool invocations.',
-        title='Continuation Token',
-    )
-    sort_by: Optional[List[str]] = Field(
-        [],
-        description='The fields to sort the versions by. Supported fields are `created_at`, `version.major`, `version.minor`, `version.patch`, `tool_id`, `invocation_source.type`, and `invocation_status.succeeded`. Specify a `+` before the field name to indicate ascending sorting and `-` for descending sorting. Multiple fields can be specified to break ties.',
-        title='Sort By',
-    )
-
-
-class SearchToolInvocationsParametersQuery(BaseModel):
-    query: str = Field(..., description='The search query.', title='Query')
-    query_against: List[SearchToolInvocationsParametersQueryQueryAgainstEnum] = Field(
-        ...,
-        description='The fields to query against. By default, all supported fields are queried against.',
-        min_length=1,
-        title='Query Against',
-    )
-    tool_id: Optional[List[SearchToolInvocationsParametersQueryToolIdItem]] = Field(
-        [], description='The ID of the tool to get invocations for.', title='Tool Id'
-    )
-    invocation_source_type: Optional[
-        List[
-            Union[
-                RegularConversationInvocationSourceInput,
-                SimulationConversationInvocationSourceInput,
-            ]
-        ]
-    ] = Field(
-        [], description='The source of the invocation', title='Invocation Source Type'
-    )
-    conversation_id: Optional[
-        List[SearchToolInvocationsParametersQueryConversationIdItem]
-    ] = Field(
-        [],
-        description="The conversation ID associated with the invocation if it's of invocation source `regular-conversation`",
-        title='Conversation Id',
-    )
-    simulation_unit_test_set_run_id: Optional[
-        List[SearchToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem]
-    ] = Field(
-        [],
-        description="The simulation unit test set run ID associated with the invocation if it's of invocation source `simulation-conversation`",
-        title='Simulation Unit Test Set Run Id',
-    )
-    succeeded: Optional[bool] = Field(
-        None, description='Whether the invocation succeeded', title='Succeeded'
-    )
-
-
 class SimulationUnitTest(BaseModel):
     field_id: str = Field(
         ..., alias='_id', description='The ID of the document.', title='Id'

{amigo_sdk-0.36.0.dist-info → amigo_sdk-0.38.0.dist-info}/METADATA

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

{amigo_sdk-0.36.0.dist-info → amigo_sdk-0.38.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-amigo_sdk/__init__.py,sha256=TB9w73-cvjBX_Mt1CSVwihd37z3iIjDbMXSELcaUxr0,139
+amigo_sdk/__init__.py,sha256=89EKxyTu_-s0FXnZoNSaB82ujJk5RRB6XO38laDVRoE,139
 amigo_sdk/_retry_utils.py,sha256=kFjw9Wqye6MB5-B4rjLxsbSNcfYBIztcollIoncd1hY,2142
 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=RkRyF5eAASd8fIOS6YvL9rLDvLAYWqHfpHSCR7jqvl4,4840
 amigo_sdk/http_client.py,sha256=v25UoUbXcMeHTnfJMcrl8RSSwCVkKUL1Jv-0HoXP1B4,13507
 amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
 amigo_sdk/sdk_client.py,sha256=Kr9M9o66pOLu0T2VDvqdYMmPZzgKJyTELu7BSPgGrYQ,6152
-amigo_sdk/generated/model.py,sha256=cr8-tGnCcFR5uODc0Y7-RTI9hd98vjW3eH3VB64b_9g,442041
+amigo_sdk/generated/model.py,sha256=3QouG75JA8Q4x1VGmp6Pz_v_e-ju2mW4qviOR5fZsRg,439992
 amigo_sdk/resources/conversation.py,sha256=5PkJOvLKqnriSS9K9hKw2VRPxRLTuABEbCyPy1fz1r0,14817
 amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
 amigo_sdk/resources/service.py,sha256=SiwEHXCQk4r1b_tGv47M08VuB7RALDHJQzWlpuD937g,1571
 amigo_sdk/resources/user.py,sha256=i4t5aVzBI37KwAtLKSDWTMwf4D4KQdSDoUiblFe1u7o,3529
-amigo_sdk-0.36.0.dist-info/METADATA,sha256=8Ef8FFXf8prskcl98v5gg6SjUpcZzZCN7O1bod3h3ZQ,8350
-amigo_sdk-0.36.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-amigo_sdk-0.36.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.36.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.36.0.dist-info/RECORD,,
+amigo_sdk-0.38.0.dist-info/METADATA,sha256=dyaANkUnsFIOMFDMs8zCQFIrrbFgJVcqwYq9w07WKYY,8350
+amigo_sdk-0.38.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+amigo_sdk-0.38.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.38.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.38.0.dist-info/RECORD,,