amigo_sdk 0.63.0__py3-none-any.whl → 0.64.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
amigo_sdk/__init__.py CHANGED
@@ -1,4 +1,4 @@
1
- __version__ = "0.63.0"
1
+ __version__ = "0.64.0"
2
2
  from .sdk_client import AmigoClient, AsyncAmigoClient
3
3
 
4
4
  __all__ = ["__version__", "AmigoClient", "AsyncAmigoClient"]
@@ -1,6 +1,6 @@
1
1
  # generated by datamodel-codegen:
2
2
  # filename: <stdin>
3
- # timestamp: 2025-11-14T16:56:28+00:00
3
+ # timestamp: 2025-11-14T21:52:42+00:00
4
4
 
5
5
  from __future__ import annotations
6
6
 
@@ -1630,11 +1630,6 @@ class AmigoLibMongoCollectionsRoleRolePermissionGrantAction(Enum):
1630
1630
  deny = 'Deny'
1631
1631
 
1632
1632
 
1633
- class AmigoLibMongoCollectionsServiceServiceTag(BaseModel):
1634
- key: str = Field(..., title='Key')
1635
- value: Optional[str] = Field(..., title='Value')
1636
-
1637
-
1638
1633
  class AmigoLibMongoCollectionsServiceServiceVersionSet(BaseModel):
1639
1634
  agent_version_number: Optional[int] = Field(
1640
1635
  ...,
@@ -1666,6 +1661,11 @@ AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
1666
1661
  )
1667
1662
 
1668
1663
 
1664
+ class AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag(BaseModel):
1665
+ key: str = Field(..., title='Key')
1666
+ value: Optional[str] = Field(..., title='Value')
1667
+
1668
+
1669
1669
  AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo = (
1670
1670
  AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1671
1671
  )
@@ -1707,6 +1707,12 @@ AmigoLibMongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo
1707
1707
  )
1708
1708
 
1709
1709
 
1710
+ class AmigoLibMongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
1711
+ major: int = Field(..., title='Major')
1712
+ minor: int = Field(..., title='Minor')
1713
+ patch: int = Field(..., title='Patch')
1714
+
1715
+
1710
1716
  class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
1711
1717
  user_id: str = Field(..., description='The ID of the user.', title='User Id')
1712
1718
  user_org_id: str = Field(
@@ -1714,12 +1720,6 @@ class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
1714
1720
  )
1715
1721
 
1716
1722
 
1717
- class AmigoLibMongoCollectionsToolVersionToolVersionVersion(BaseModel):
1718
- major: int = Field(..., title='Major')
1719
- minor: int = Field(..., title='Minor')
1720
- patch: int = Field(..., title='Patch')
1721
-
1722
-
1723
1723
  class AmigoLibMongoCollectionsUserUserPreferences(BaseModel):
1724
1724
  enable_response_recommendation: Optional[bool] = Field(
1725
1725
  False,
@@ -1951,6 +1951,20 @@ class DynamicBehaviorSetGetDynamicBehaviorInvocationsResponse(BaseModel):
1951
1951
  )
1952
1952
 
1953
1953
 
1954
+ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
1955
+ BaseModel
1956
+ ):
1957
+ id: str = Field(..., title='Id')
1958
+ name: str = Field(..., title='Name')
1959
+ is_active: bool = Field(..., title='Is Active')
1960
+ applied_to_services: List[str] = Field(..., title='Applied To Services')
1961
+ tags: Dict[str, Optional[str]] = Field(..., title='Tags')
1962
+ created_at: AwareDatetime = Field(..., title='Created At')
1963
+ creator: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1964
+ updated_at: AwareDatetime = Field(..., title='Updated At')
1965
+ updated_by: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1966
+
1967
+
1954
1968
  class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
1955
1969
  applied_to_services_ids: List[str] = Field(
1956
1970
  ...,
@@ -1971,18 +1985,14 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
1971
1985
  )
1972
1986
 
1973
1987
 
1974
- class DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
1975
- BaseModel
1976
- ):
1977
- id: str = Field(..., title='Id')
1978
- name: str = Field(..., title='Name')
1979
- is_active: bool = Field(..., title='Is Active')
1980
- applied_to_services: List[str] = Field(..., title='Applied To Services')
1981
- tags: Dict[str, Optional[str]] = Field(..., title='Tags')
1982
- created_at: AwareDatetime = Field(..., title='Created At')
1983
- creator: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1984
- updated_at: AwareDatetime = Field(..., title='Updated At')
1985
- updated_by: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1988
+ class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
1989
+ dynamic_behavior_sets: List[
1990
+ DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
1991
+ ] = Field(
1992
+ ...,
1993
+ description='The retrieved dynamic behavior sets.',
1994
+ title='Dynamic Behavior Sets',
1995
+ )
1986
1996
 
1987
1997
 
1988
1998
  class DynamicBehaviorSetUpdateDynamicBehaviorSetRequestAppliedToService(
@@ -10738,13 +10748,6 @@ class OrganizationCreateOrganizationRequestPreferences(BaseModel):
10738
10748
  )
10739
10749
 
10740
10750
 
10741
- class OrganizationCreateOrganizationRequestUserDimension(BaseModel):
10742
- description: str = Field(..., min_length=1)
10743
- tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
10744
- ..., title='Tags'
10745
- )
10746
-
10747
-
10748
10751
  class OrganizationCreateOrganizationResponse(BaseModel):
10749
10752
  org_id: str = Field(
10750
10753
  ...,
@@ -10768,19 +10771,19 @@ class OrganizationCreateServiceHierarchicalStateMachineResponse(BaseModel):
10768
10771
  )
10769
10772
 
10770
10773
 
10771
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState(
10774
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState(
10772
10775
  AnnotationStateInputNextState
10773
10776
  ):
10774
10777
  pass
10775
10778
 
10776
10779
 
10777
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition(
10780
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition(
10778
10781
  BaseModel
10779
10782
  ):
10780
10783
  description: str = Field(..., min_length=1)
10781
10784
  next_state: Union[
10782
10785
  StateOrRefName,
10783
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState,
10786
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState,
10784
10787
  ] = Field(..., title='Next State')
10785
10788
 
10786
10789
 
@@ -10886,22 +10889,18 @@ class OrganizationGetUserDimensionsResponse(BaseModel):
10886
10889
  )
10887
10890
 
10888
10891
 
10889
- class OrganizationModifyOrganizationRequestUserDimensions(
10890
- RootModel[List[OrganizationCreateOrganizationRequestUserDimension]]
10891
- ):
10892
- root: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
10893
- ...,
10894
- description='User dimensions for the organization. If not set or `null`, this field is not updated.',
10895
- min_length=1,
10896
- title='User Dimensions',
10897
- )
10898
-
10899
-
10900
10892
  OrganizationModifyOrganizationRequestPreferences = (
10901
10893
  OrganizationCreateOrganizationRequestPreferences
10902
10894
  )
10903
10895
 
10904
10896
 
10897
+ class OrganizationModifyOrganizationRequestUserDimension(BaseModel):
10898
+ description: str = Field(..., min_length=1)
10899
+ tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
10900
+ ..., title='Tags'
10901
+ )
10902
+
10903
+
10905
10904
  class RoleAssignRoleRequest(BaseModel):
10906
10905
  user_id: str = Field(
10907
10906
  ...,
@@ -10948,17 +10947,9 @@ class RoleModifyRoleResponse(BaseModel):
10948
10947
  )
10949
10948
 
10950
10949
 
10951
- class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
10952
- llm_name: LLMType
10953
- top_p: float = Field(..., description='The preferred `top_p` value.', title='Top P')
10954
- temperature: float = Field(
10955
- ..., description='The preferred temperature value.', title='Temperature'
10956
- )
10957
- top_k: int = Field(
10958
- ...,
10959
- description='The preferred `top_k` value. Note that not all LLMs support this.',
10960
- title='Top K',
10961
- )
10950
+ class ServiceCreateServiceRequestVersionSetLLMLoadBalancingSetConfig(BaseModel):
10951
+ llm_load_balancing_set_name: LLMLoadBalancingSetType
10952
+ configs: Dict[str, PerLLMConfig] = Field(..., title='Configs')
10962
10953
 
10963
10954
 
10964
10955
  class ServiceCreateServiceResponse(BaseModel):
@@ -11020,11 +11011,17 @@ class ServiceUpdateServiceRequest(BaseModel):
11020
11011
  )
11021
11012
 
11022
11013
 
11023
- class ServiceUpsertServiceVersionSetRequestVersionSetLLMLoadBalancingSetConfig(
11024
- BaseModel
11025
- ):
11026
- llm_load_balancing_set_name: LLMLoadBalancingSetType
11027
- configs: Dict[str, PerLLMConfig] = Field(..., title='Configs')
11014
+ class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
11015
+ llm_name: LLMType
11016
+ top_p: float = Field(..., description='The preferred `top_p` value.', title='Top P')
11017
+ temperature: float = Field(
11018
+ ..., description='The preferred temperature value.', title='Temperature'
11019
+ )
11020
+ top_k: int = Field(
11021
+ ...,
11022
+ description='The preferred `top_k` value. Note that not all LLMs support this.',
11023
+ title='Top K',
11024
+ )
11028
11025
 
11029
11026
 
11030
11027
  class SimulationCreateSimulationPersonaRequestInitialVersion(BaseModel):
@@ -11498,22 +11495,6 @@ class ToolGetToolsResponseFilterValues(BaseModel):
11498
11495
  )
11499
11496
 
11500
11497
 
11501
- class ToolInvokeToolVersionRequestInputInvocationMode(Enum):
11502
- regular = 'regular'
11503
- conversation_simulation = 'conversation-simulation'
11504
-
11505
-
11506
- class ToolInvokeToolVersionRequestInput(BaseModel):
11507
- input_parameters: Dict[str, Any] = Field(
11508
- ...,
11509
- description='The input parameters for the tool invocation.',
11510
- title='Input Parameters',
11511
- )
11512
- invocation_mode: ToolInvokeToolVersionRequestInputInvocationMode = Field(
11513
- ..., description='The mode of invocation.', title='Invocation Mode'
11514
- )
11515
-
11516
-
11517
11498
  class ToolModifyToolRequest(BaseModel):
11518
11499
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
11519
11500
  None, description='A description of this tool. Only updates if not-null.'
@@ -11556,34 +11537,19 @@ class ToolPublishToolVersionResponse(BaseModel):
11556
11537
  )
11557
11538
 
11558
11539
 
11559
- class ToolTestToolRequest(BaseModel):
11560
- inputs: List[ToolInvokeToolVersionRequestInput] = Field(
11561
- ...,
11562
- description='The list of inputs for the tool invocation.',
11563
- max_length=10,
11564
- min_length=1,
11565
- title='Inputs',
11566
- )
11567
- commit_branch: str = Field(
11568
- ...,
11569
- description='The branch in the tools repo whose tip will be tested.',
11570
- title='Commit Branch',
11571
- )
11572
- project_path: str = Field(
11573
- ...,
11574
- description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
11575
- pattern='^[-\\w\\d_/]+$',
11576
- title='Project Path',
11577
- )
11578
- required_envvars: Dict[str, str] = Field(
11540
+ class ToolTestToolRequestInputInvocationMode(Enum):
11541
+ regular = 'regular'
11542
+ conversation_simulation = 'conversation-simulation'
11543
+
11544
+
11545
+ class ToolTestToolRequestInput(BaseModel):
11546
+ input_parameters: Dict[str, Any] = Field(
11579
11547
  ...,
11580
- description='The environment variables required for the tool to run.',
11581
- title='Required Envvars',
11548
+ description='The input parameters for the tool invocation.',
11549
+ title='Input Parameters',
11582
11550
  )
11583
- required_secret_envvars: Dict[str, str] = Field(
11584
- ...,
11585
- description='The secret environment variables required for the tool to run.',
11586
- title='Required Secret Envvars',
11551
+ invocation_mode: ToolTestToolRequestInputInvocationMode = Field(
11552
+ ..., description='The mode of invocation.', title='Invocation Mode'
11587
11553
  )
11588
11554
 
11589
11555
 
@@ -11655,7 +11621,7 @@ class UserGetMemoriesResponse(BaseModel):
11655
11621
  )
11656
11622
 
11657
11623
 
11658
- class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11624
+ class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11659
11625
  num_conversations: int = Field(
11660
11626
  ...,
11661
11627
  description='The number of conversations the user has created.',
@@ -11673,6 +11639,31 @@ class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11673
11639
  )
11674
11640
 
11675
11641
 
11642
+ class UserSearchUsersResponseUserInstance(BaseModel):
11643
+ org_id: str = Field(
11644
+ ...,
11645
+ description='The ID of the organization that this user belongs to.',
11646
+ title='Org Id',
11647
+ )
11648
+ user_id: str = Field(
11649
+ ..., description='The identifier of the user.', title='User Id'
11650
+ )
11651
+ first_name: str = Field(
11652
+ ..., description='The first name of the user.', title='First Name'
11653
+ )
11654
+ last_name: str = Field(
11655
+ ..., description='The last name of the user.', title='Last Name'
11656
+ )
11657
+ email: str = Field(..., description='Email of the user.', title='Email')
11658
+ user_stats: UserGetUsersResponseUserInstanceUserStats = Field(
11659
+ ..., description="Statistics about the user's usage of the Amigo platform."
11660
+ )
11661
+ role: str = Field(..., description='The ID of the role of the user.', title='Role')
11662
+ preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
11663
+ ..., description='The preferences of the user.'
11664
+ )
11665
+
11666
+
11676
11667
  class UserSignInWithApiKeyResponse(BaseModel):
11677
11668
  id_token: str = Field(
11678
11669
  ...,
@@ -12120,7 +12111,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
12120
12111
  [], description='The IDs of the messages to retrieve.', title='Id'
12121
12112
  )
12122
12113
  message_type: Optional[List[MessageType]] = Field(
12123
- ['agent-message', 'external-event', 'user-message'],
12114
+ ['external-event', 'user-message', 'agent-message'],
12124
12115
  description='The type of messages to retrieve.',
12125
12116
  title='Message Type',
12126
12117
  )
@@ -12262,6 +12253,9 @@ class SearchUsersParametersQuery(BaseModel):
12262
12253
  description='The search query. Any users whose name or email contains the query are returned.',
12263
12254
  title='Query',
12264
12255
  )
12256
+ user_id: Optional[List[str]] = Field(
12257
+ [], description='The IDs of the users.', title='User Id'
12258
+ )
12265
12259
 
12266
12260
 
12267
12261
  class GetRolesV1OrganizationRoleGetParametersQueryIdItem(
@@ -13851,7 +13845,7 @@ class ServiceInstance(BaseModel):
13851
13845
  agent_id: str = Field(
13852
13846
  ..., description='The ID of the agent that this service uses.', title='Agent Id'
13853
13847
  )
13854
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
13848
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13855
13849
  ..., description='The tags of the service.', title='Tags'
13856
13850
  )
13857
13851
 
@@ -13904,7 +13898,7 @@ class SimulationUnitTestSet(BaseModel):
13904
13898
  description='Whether the unit test set has been deleted.',
13905
13899
  title='Is Deleted',
13906
13900
  )
13907
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
13901
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13908
13902
  ..., description='The tags of the simulation unit test set.', title='Tags'
13909
13903
  )
13910
13904
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13943,7 +13937,7 @@ class SimulationUnitTestSetInstance(BaseModel):
13943
13937
  description='Whether the unit test set has been deleted.',
13944
13938
  title='Is Deleted',
13945
13939
  )
13946
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
13940
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13947
13941
  ..., description='The tags of the simulation unit test set.', title='Tags'
13948
13942
  )
13949
13943
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14059,7 +14053,7 @@ class ToolInstance(BaseModel):
14059
14053
  description='A list of encrypted environment variables that some versions of this tool uses.',
14060
14054
  title='Secret Envvars',
14061
14055
  )
14062
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
14056
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
14063
14057
  ..., description='The tags of the simulation persona.', title='Tags'
14064
14058
  )
14065
14059
 
@@ -14074,7 +14068,7 @@ class ToolVersionInstance(BaseModel):
14074
14068
  description='The ID of the tool that this version belongs to.',
14075
14069
  title='Tool Id',
14076
14070
  )
14077
- version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
14071
+ version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
14078
14072
  ..., description='The version of the tool.'
14079
14073
  )
14080
14074
  required_envvars: List[str] = Field(
@@ -14393,7 +14387,7 @@ class DynamicBehaviorSetGetDynamicBehaviorSetVersionsResponse(BaseModel):
14393
14387
 
14394
14388
  class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
14395
14389
  dynamic_behavior_sets: List[
14396
- DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14390
+ DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14397
14391
  ] = Field(
14398
14392
  ...,
14399
14393
  description='The retrieved dynamic behavior sets.',
@@ -14417,16 +14411,6 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
14417
14411
  )
14418
14412
 
14419
14413
 
14420
- class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
14421
- dynamic_behavior_sets: List[
14422
- DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14423
- ] = Field(
14424
- ...,
14425
- description='The retrieved dynamic behavior sets.',
14426
- title='Dynamic Behavior Sets',
14427
- )
14428
-
14429
-
14430
14414
  class MetricCreateMetricRequest(BaseModel):
14431
14415
  name: str = Field(
14432
14416
  ...,
@@ -14483,7 +14467,7 @@ class OrganizationCreateOrganizationRequest(BaseModel):
14483
14467
  description='Additional descriptions of the services offered by this organization that is displayed below `main_description` in a smaller font.',
14484
14468
  min_length=1,
14485
14469
  )
14486
- user_dimensions: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
14470
+ user_dimensions: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
14487
14471
  ...,
14488
14472
  description='User dimensions for the organization.',
14489
14473
  title='User Dimensions',
@@ -14627,6 +14611,17 @@ class OrganizationGetOrganizationMetricsResponse(BaseModel):
14627
14611
  )
14628
14612
 
14629
14613
 
14614
+ class OrganizationModifyOrganizationRequestUserDimensions(
14615
+ RootModel[List[OrganizationModifyOrganizationRequestUserDimension]]
14616
+ ):
14617
+ root: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
14618
+ ...,
14619
+ description='User dimensions for the organization. If not set or `null`, this field is not updated.',
14620
+ min_length=1,
14621
+ title='User Dimensions',
14622
+ )
14623
+
14624
+
14630
14625
  class OrganizationModifyOrganizationRequest(BaseModel):
14631
14626
  title: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14632
14627
  None,
@@ -14684,7 +14679,7 @@ class OrganizationModifyOrganizationRequest(BaseModel):
14684
14679
  )
14685
14680
 
14686
14681
 
14687
- class RoleModifyRoleRequestPermissionGrant(BaseModel):
14682
+ class RoleCreateRoleRequestPermissionGrant(BaseModel):
14688
14683
  action: AmigoLibMongoCollectionsRoleRolePermissionGrantAction = Field(
14689
14684
  ...,
14690
14685
  description='Whether this grant allows or denies the specified access.',
@@ -14701,6 +14696,29 @@ class RoleModifyRoleRequestPermissionGrant(BaseModel):
14701
14696
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = None
14702
14697
 
14703
14698
 
14699
+ class RoleModifyRoleRequest(BaseModel):
14700
+ description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14701
+ None,
14702
+ description='A description about the role. Only updated if specified. This field is a mutable field.',
14703
+ )
14704
+ permission_grants: Optional[List[RoleCreateRoleRequestPermissionGrant]] = Field(
14705
+ None,
14706
+ description='A list of permission grants associated with this role. Only updated if specified. This field is an immutable field.',
14707
+ title='Permission Grants',
14708
+ )
14709
+ frontend_view: Optional[FrontendView] = Field(
14710
+ None,
14711
+ description='The frontend view for the user of this role. Only updated if specified. This field is an immutable field.',
14712
+ )
14713
+ inherited_from: Optional[Union[RoleModifyRoleRequestInheritedFrom, FieldNotSet]] = (
14714
+ Field(
14715
+ {},
14716
+ description='The ID of the role that this role inherits from. Only updated if specified. This field is an immutable field.',
14717
+ title='Inherited From',
14718
+ )
14719
+ )
14720
+
14721
+
14704
14722
  class ServiceGetServicesResponse(BaseModel):
14705
14723
  services: List[ServiceInstance] = Field(
14706
14724
  ..., description='The services that are found.', title='Services'
@@ -14735,8 +14753,8 @@ class ServiceUpsertServiceVersionSetRequestVersionSet(BaseModel):
14735
14753
  llm_model_preferences: Dict[
14736
14754
  str,
14737
14755
  Union[
14738
- ServiceCreateServiceRequestVersionSetLLMConfig,
14739
- ServiceUpsertServiceVersionSetRequestVersionSetLLMLoadBalancingSetConfig,
14756
+ ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig,
14757
+ ServiceCreateServiceRequestVersionSetLLMLoadBalancingSetConfig,
14740
14758
  ],
14741
14759
  ] = Field(..., title='Llm Model Preferences')
14742
14760
 
@@ -14974,7 +14992,7 @@ class ToolGetToolsResponse(BaseModel):
14974
14992
 
14975
14993
 
14976
14994
  class ToolInvokeToolVersionRequest(BaseModel):
14977
- inputs: List[ToolInvokeToolVersionRequestInput] = Field(
14995
+ inputs: List[ToolTestToolRequestInput] = Field(
14978
14996
  ...,
14979
14997
  description='The list of inputs for the tool invocation.',
14980
14998
  max_length=10,
@@ -14995,6 +15013,37 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
14995
15013
  deletes: Optional[List[DeleteRequest]] = Field([], title='Deletes')
14996
15014
 
14997
15015
 
15016
+ class ToolTestToolRequest(BaseModel):
15017
+ inputs: List[ToolTestToolRequestInput] = Field(
15018
+ ...,
15019
+ description='The list of inputs for the tool invocation.',
15020
+ max_length=10,
15021
+ min_length=1,
15022
+ title='Inputs',
15023
+ )
15024
+ commit_branch: str = Field(
15025
+ ...,
15026
+ description='The branch in the tools repo whose tip will be tested.',
15027
+ title='Commit Branch',
15028
+ )
15029
+ project_path: str = Field(
15030
+ ...,
15031
+ description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
15032
+ pattern='^[-\\w\\d_/]+$',
15033
+ title='Project Path',
15034
+ )
15035
+ required_envvars: Dict[str, str] = Field(
15036
+ ...,
15037
+ description='The environment variables required for the tool to run.',
15038
+ title='Required Envvars',
15039
+ )
15040
+ required_secret_envvars: Dict[str, str] = Field(
15041
+ ...,
15042
+ description='The secret environment variables required for the tool to run.',
15043
+ title='Required Secret Envvars',
15044
+ )
15045
+
15046
+
14998
15047
  ToolTestToolResponse = ToolInvokeToolVersionResponse
14999
15048
 
15000
15049
 
@@ -15035,28 +15084,23 @@ class UserGetUserModelResponse(BaseModel):
15035
15084
  )
15036
15085
 
15037
15086
 
15038
- class UserSearchUsersResponseUserInstance(BaseModel):
15039
- org_id: str = Field(
15040
- ...,
15041
- description='The ID of the organization that this user belongs to.',
15042
- title='Org Id',
15043
- )
15044
- user_id: str = Field(
15045
- ..., description='The identifier of the user.', title='User Id'
15046
- )
15047
- first_name: str = Field(
15048
- ..., description='The first name of the user.', title='First Name'
15087
+ class UserGetUsersResponse(BaseModel):
15088
+ users: List[UserSearchUsersResponseUserInstance] = Field(
15089
+ ..., description='Users in this organization.', title='Users'
15049
15090
  )
15050
- last_name: str = Field(
15051
- ..., description='The last name of the user.', title='Last Name'
15091
+ has_more: bool = Field(
15092
+ ..., description='Whether there are more users to retrieve.', title='Has More'
15052
15093
  )
15053
- email: str = Field(..., description='Email of the user.', title='Email')
15054
- user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
15055
- ..., description="Statistics about the user's usage of the Amigo platform."
15094
+ continuation_token: Optional[int] = Field(
15095
+ ...,
15096
+ description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
15097
+ title='Continuation Token',
15056
15098
  )
15057
- role: str = Field(..., description='The ID of the role of the user.', title='Role')
15058
- preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
15059
- ..., description='The preferences of the user.'
15099
+
15100
+
15101
+ class UserSearchUsersResponse(BaseModel):
15102
+ users: List[UserSearchUsersResponseUserInstance] = Field(
15103
+ ..., description='Users in this organization.', title='Users'
15060
15104
  )
15061
15105
 
15062
15106
 
@@ -15093,7 +15137,7 @@ class ActionStateInput(BaseModel):
15093
15137
  Field(..., title='Boundary Constraints')
15094
15138
  )
15095
15139
  exit_conditions: List[
15096
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15140
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15097
15141
  ] = Field(..., title='Exit Conditions')
15098
15142
  action_tool_call_specs: List[ToolCallSpecInput] = Field(
15099
15143
  ..., title='Action Tool Call Specs'
@@ -15202,7 +15246,7 @@ class DecisionStateInput(BaseModel):
15202
15246
  type: Literal['decision'] = Field(..., title='Type')
15203
15247
  name: str = Field(..., pattern='^[A-Za-z0-9_]+$')
15204
15248
  exit_conditions: List[
15205
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15249
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15206
15250
  ] = Field(..., title='Exit Conditions')
15207
15251
  decision_guidelines: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
15208
15252
  ..., title='Decision Guidelines'
@@ -15243,7 +15287,7 @@ class Metric(BaseModel):
15243
15287
  additional_notes: Optional[str] = Field(
15244
15288
  ..., description='Additional notes about the metric.', title='Additional Notes'
15245
15289
  )
15246
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15290
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15247
15291
  ..., description='The tags of the metric.', title='Tags'
15248
15292
  )
15249
15293
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15313,7 +15357,7 @@ class MetricInstance(BaseModel):
15313
15357
  additional_notes: Optional[str] = Field(
15314
15358
  ..., description='Additional notes about the metric.', title='Additional Notes'
15315
15359
  )
15316
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15360
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15317
15361
  ..., description='The tags of the metric.', title='Tags'
15318
15362
  )
15319
15363
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15515,7 +15559,7 @@ class SimulationUnitTest(BaseModel):
15515
15559
  is_deleted: bool = Field(
15516
15560
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15517
15561
  )
15518
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15562
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15519
15563
  ..., description='The tags of the simulation persona.', title='Tags'
15520
15564
  )
15521
15565
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15560,7 +15604,7 @@ class SimulationUnitTestInstance(BaseModel):
15560
15604
  is_deleted: bool = Field(
15561
15605
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15562
15606
  )
15563
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15607
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15564
15608
  ..., description='The tags of the simulation persona.', title='Tags'
15565
15609
  )
15566
15610
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15863,7 +15907,7 @@ class RoleCreateRoleRequest(BaseModel):
15863
15907
  description: str = Field(
15864
15908
  ..., description='A description about the role.', min_length=1
15865
15909
  )
15866
- permission_grants: List[RoleModifyRoleRequestPermissionGrant] = Field(
15910
+ permission_grants: List[RoleCreateRoleRequestPermissionGrant] = Field(
15867
15911
  ...,
15868
15912
  description='A list of permission grants associated with this role.',
15869
15913
  title='Permission Grants',
@@ -15889,29 +15933,6 @@ class RoleGetRolesResponse(BaseModel):
15889
15933
  )
15890
15934
 
15891
15935
 
15892
- class RoleModifyRoleRequest(BaseModel):
15893
- description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
15894
- None,
15895
- description='A description about the role. Only updated if specified. This field is a mutable field.',
15896
- )
15897
- permission_grants: Optional[List[RoleModifyRoleRequestPermissionGrant]] = Field(
15898
- None,
15899
- description='A list of permission grants associated with this role. Only updated if specified. This field is an immutable field.',
15900
- title='Permission Grants',
15901
- )
15902
- frontend_view: Optional[FrontendView] = Field(
15903
- None,
15904
- description='The frontend view for the user of this role. Only updated if specified. This field is an immutable field.',
15905
- )
15906
- inherited_from: Optional[Union[RoleModifyRoleRequestInheritedFrom, FieldNotSet]] = (
15907
- Field(
15908
- {},
15909
- description='The ID of the role that this role inherits from. Only updated if specified. This field is an immutable field.',
15910
- title='Inherited From',
15911
- )
15912
- )
15913
-
15914
-
15915
15936
  class ServiceCreateServiceRequest(BaseModel):
15916
15937
  service_hierarchical_state_machine_id: str = Field(
15917
15938
  ...,
@@ -16084,7 +16105,7 @@ class SimulationUpdateSimulationUnitTestRequest(BaseModel):
16084
16105
  )
16085
16106
 
16086
16107
 
16087
- class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
16108
+ class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
16088
16109
  id: str = Field(..., description='The ID of the tool invocation.', title='Id')
16089
16110
  org_id: str = Field(..., description='The ID of the organization.', title='Org Id')
16090
16111
  created_at: Optional[AwareDatetime] = Field(None, title='Created At')
@@ -16106,39 +16127,11 @@ class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
16106
16127
  tool_id: str = Field(
16107
16128
  ..., description='The ID of the tool that was invoked.', title='Tool Id'
16108
16129
  )
16109
- tool_version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
16130
+ tool_version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
16110
16131
  ..., description='The version of the tool that was invoked.'
16111
16132
  )
16112
16133
 
16113
16134
 
16114
- class ToolSearchToolInvocationsResponse(BaseModel):
16115
- tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
16116
- Field(
16117
- ..., description='The list of tool invocations.', title='Tool Invocations'
16118
- )
16119
- )
16120
-
16121
-
16122
- class UserGetUsersResponse(BaseModel):
16123
- users: List[UserSearchUsersResponseUserInstance] = Field(
16124
- ..., description='Users in this organization.', title='Users'
16125
- )
16126
- has_more: bool = Field(
16127
- ..., description='Whether there are more users to retrieve.', title='Has More'
16128
- )
16129
- continuation_token: Optional[int] = Field(
16130
- ...,
16131
- description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
16132
- title='Continuation Token',
16133
- )
16134
-
16135
-
16136
- class UserSearchUsersResponse(BaseModel):
16137
- users: List[UserSearchUsersResponseUserInstance] = Field(
16138
- ..., description='Users in this organization.', title='Users'
16139
- )
16140
-
16141
-
16142
16135
  class SimulationUnitTestSetRunInstance(BaseModel):
16143
16136
  org_id: str = Field(..., title='Org Id')
16144
16137
  id: str = Field(..., title='Id')
@@ -16238,7 +16231,7 @@ class SimulationGetSimulationUnitTestSetRunsResponse(BaseModel):
16238
16231
 
16239
16232
 
16240
16233
  class ToolGetToolInvocationsResponse(BaseModel):
16241
- tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
16234
+ tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
16242
16235
  Field(
16243
16236
  ..., description='The list of tool invocations.', title='Tool Invocations'
16244
16237
  )
@@ -16253,3 +16246,11 @@ class ToolGetToolInvocationsResponse(BaseModel):
16253
16246
  description='The continuation token to retrieve the next page of tool invocations, or null if there are no more tool invocations.',
16254
16247
  title='Continuation Token',
16255
16248
  )
16249
+
16250
+
16251
+ class ToolSearchToolInvocationsResponse(BaseModel):
16252
+ tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
16253
+ Field(
16254
+ ..., description='The list of tool invocations.', title='Tool Invocations'
16255
+ )
16256
+ )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: amigo_sdk
3
- Version: 0.63.0
3
+ Version: 0.64.0
4
4
  Summary: Amigo AI Python SDK
5
5
  Author: Amigo AI
6
6
  License-File: LICENSE
@@ -1,4 +1,4 @@
1
- amigo_sdk/__init__.py,sha256=67MLi4srbwW9vkJyLnDDEC6x6rLAQNh5IKfLkRJvLpk,139
1
+ amigo_sdk/__init__.py,sha256=eloZXJpddVuvbdsvt6IfmSlYGI2ifuvNgxQVElijx8k,139
2
2
  amigo_sdk/_retry_utils.py,sha256=kFjw9Wqye6MB5-B4rjLxsbSNcfYBIztcollIoncd1hY,2142
3
3
  amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
4
4
  amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
@@ -6,13 +6,13 @@ amigo_sdk/errors.py,sha256=RkRyF5eAASd8fIOS6YvL9rLDvLAYWqHfpHSCR7jqvl4,4840
6
6
  amigo_sdk/http_client.py,sha256=v25UoUbXcMeHTnfJMcrl8RSSwCVkKUL1Jv-0HoXP1B4,13507
7
7
  amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
8
8
  amigo_sdk/sdk_client.py,sha256=Kr9M9o66pOLu0T2VDvqdYMmPZzgKJyTELu7BSPgGrYQ,6152
9
- amigo_sdk/generated/model.py,sha256=Fiw9s5Eo1nqQPUIBGCncKR2j-DFoXu83l9h_eFcuDXE,453760
9
+ amigo_sdk/generated/model.py,sha256=mo9G2CZKtu9fbjsKvZGzX0KsP1-DwYIvI_1cXRzeCAQ,454042
10
10
  amigo_sdk/resources/conversation.py,sha256=5PkJOvLKqnriSS9K9hKw2VRPxRLTuABEbCyPy1fz1r0,14817
11
11
  amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
12
12
  amigo_sdk/resources/service.py,sha256=SiwEHXCQk4r1b_tGv47M08VuB7RALDHJQzWlpuD937g,1571
13
13
  amigo_sdk/resources/user.py,sha256=vZPCw8i-xvcDQik8ouJbm8qVHucRbfxCWI3yYMDj8hg,3517
14
- amigo_sdk-0.63.0.dist-info/METADATA,sha256=GNqgdLy0eqXw0HybQhKrdeCT4U9sQOIJ6SCTq8EOutM,8350
15
- amigo_sdk-0.63.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- amigo_sdk-0.63.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
- amigo_sdk-0.63.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
- amigo_sdk-0.63.0.dist-info/RECORD,,
14
+ amigo_sdk-0.64.0.dist-info/METADATA,sha256=OCBco9JUKtOjPZXliy1Vsl24g7aP2b5aGf2NJLVED8c,8350
15
+ amigo_sdk-0.64.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
+ amigo_sdk-0.64.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
+ amigo_sdk-0.64.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
+ amigo_sdk-0.64.0.dist-info/RECORD,,