amigo_sdk 0.35.0__py3-none-any.whl → 0.37.0__py3-none-any.whl

amigo_sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.35.0"
+__version__ = "0.37.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-01T22:56:24+00:00
+#   timestamp: 2025-10-02T15:05:02+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]+$')
 
@@ -11875,7 +11839,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'],
+        ['agent-message', 'external-event', 'user-message'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13077,6 +13041,11 @@ class GetToolInvocationsParametersQueryToolIdItem(
     pass
 
 
+class GetToolInvocationsParametersQueryInvocationSourceTypeEnum(Enum):
+    regular_conversation = 'regular-conversation'
+    simulation_conversation = 'simulation-conversation'
+
+
 class GetToolInvocationsParametersQueryConversationIdItem(
     DynamicBehaviorSetCreateDynamicBehaviorSetRequestAppliedToService
 ):
@@ -13089,6 +13058,116 @@ 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'
+    invocation_status_exception_stack_trace = 'invocation_status.exception_stack_trace'
+    invocation_status_output = 'invocation_status.output'
+
+
+class SearchToolInvocationsParametersQueryToolIdItem(
+    DynamicBehaviorSetCreateDynamicBehaviorSetRequestAppliedToService
+):
+    pass
+
+
+class SearchToolInvocationsParametersQueryConversationIdItem(
+    DynamicBehaviorSetCreateDynamicBehaviorSetRequestAppliedToService
+):
+    pass
+
+
+class SearchToolInvocationsParametersQuerySimulationUnitTestSetRunIdItem(
+    DynamicBehaviorSetCreateDynamicBehaviorSetRequestAppliedToService
+):
+    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'
@@ -13583,33 +13662,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(
         ...,
@@ -13627,7 +13680,7 @@ class RegularConversationInvocationSourceOutput(BaseModel):
         title='Interaction Id',
     )
     invocation_metadata: Union[
-        StateTransitionInvocationMetadataOutput, EngageUserInvocationMetadataOutput
+        StateTransitionInvocationMetadata, EngageUserInvocationMetadata
     ] = Field(..., description='Metadata about the invocation.', discriminator='type')
 
 
@@ -13665,23 +13718,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'
     )
@@ -13691,7 +13728,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')
 
 
@@ -15268,8 +15305,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'
     )
@@ -15724,58 +15760,9 @@ class ToolGetToolInvocationsResponse(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 ToolSearchToolInvocationsResponse(BaseModel):
+    tool_invocations: List[ToolInvocationInstance] = Field(
+        ..., description='The list of tool invocations.', title='Tool Invocations'
     )
 
 

{amigo_sdk-0.35.0.dist-info → amigo_sdk-0.37.0.dist-info}/METADATA

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

{amigo_sdk-0.35.0.dist-info → amigo_sdk-0.37.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-amigo_sdk/__init__.py,sha256=mfgCxw1nS6T4ulINXzGDvgjRvgObvgC6rgdUmxUcLPU,139
+amigo_sdk/__init__.py,sha256=SGoql7Zo7Kzvbv-zVkpkOVTxvbUKaTB8UEX3KQAOxuY,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=4715BsBiHr4YG5fpu0tWGGEKx8JbchIR9j8lJaXP2eU,439359
+amigo_sdk/generated/model.py,sha256=x4oSGAErmmOcVwfxDqQ59Eze7uw38rhh4YQhkt6C66s,438683
 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.35.0.dist-info/METADATA,sha256=Y5YYCVVbK6MCjW20YAVmEcwWaH05Q86mqBWCOwByjik,8350
-amigo_sdk-0.35.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-amigo_sdk-0.35.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.35.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.35.0.dist-info/RECORD,,
+amigo_sdk-0.37.0.dist-info/METADATA,sha256=LOP9t58u1u921IQue8nwyMNpUYYWnDu1MoDcMNFO67M,8350
+amigo_sdk-0.37.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+amigo_sdk-0.37.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.37.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.37.0.dist-info/RECORD,,