amigo_sdk 0.59.0__py3-none-any.whl → 0.61.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.59.0"
1
+ __version__ = "0.61.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-12T02:24:09+00:00
3
+ # timestamp: 2025-11-13T18:32:12+00:00
4
4
 
5
5
  from __future__ import annotations
6
6
 
@@ -917,6 +917,12 @@ class ReflectionStateOutputNextState(AnnotationStateOutputNextState):
917
917
  pass
918
918
 
919
919
 
920
+ class ResultPersistence(Enum):
921
+ ephemral = 'ephemral'
922
+ persisted_preferred = 'persisted-preferred'
923
+ persisted = 'persisted'
924
+
925
+
920
926
  class SelectDynamicBehaviorCompletedEventSelectedDynamicBehaviorSetVersionInfo(
921
927
  RootModel[List]
922
928
  ):
@@ -1157,6 +1163,11 @@ class ToolCallLog(BaseModel):
1157
1163
  duration: float = Field(
1158
1164
  ..., description='The duration of the tool call in seconds.', title='Duration'
1159
1165
  )
1166
+ hidden: Optional[bool] = Field(
1167
+ False,
1168
+ description="Whether the tool call should be hidden in interaction logs based on the tool call spec's `result_persistence` setting.",
1169
+ title='Hidden',
1170
+ )
1160
1171
 
1161
1172
 
1162
1173
  class ToolCallSpecOutput(BaseModel):
@@ -1181,6 +1192,9 @@ class ToolCallSpecOutput(BaseModel):
1181
1192
  description='The number of seconds to wait before playing an audio filler.',
1182
1193
  title='Audio Filler Triggered After',
1183
1194
  )
1195
+ result_persistence: Optional[ResultPersistence] = Field(
1196
+ 'persisted', description='The type of result persistence for this tool call.'
1197
+ )
1184
1198
 
1185
1199
 
1186
1200
  class ToolCallStartedEvent(BaseModel):
@@ -1647,6 +1661,11 @@ AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
1647
1661
  )
1648
1662
 
1649
1663
 
1664
+ class AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag(BaseModel):
1665
+ key: str = Field(..., title='Key')
1666
+ value: Optional[str] = Field(..., title='Value')
1667
+
1668
+
1650
1669
  AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo = (
1651
1670
  AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1652
1671
  )
@@ -1688,17 +1707,6 @@ AmigoLibMongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo
1688
1707
  )
1689
1708
 
1690
1709
 
1691
- class AmigoLibMongoCollectionsToolToolTag(BaseModel):
1692
- key: str = Field(..., title='Key')
1693
- value: Optional[str] = Field(..., title='Value')
1694
-
1695
-
1696
- class AmigoLibMongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
1697
- major: int = Field(..., title='Major')
1698
- minor: int = Field(..., title='Minor')
1699
- patch: int = Field(..., title='Patch')
1700
-
1701
-
1702
1710
  class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
1703
1711
  user_id: str = Field(..., description='The ID of the user.', title='User Id')
1704
1712
  user_org_id: str = Field(
@@ -1706,6 +1714,12 @@ class AmigoLibMongoCollectionsToolVersionToolVersionUserInfo(BaseModel):
1706
1714
  )
1707
1715
 
1708
1716
 
1717
+ class AmigoLibMongoCollectionsToolVersionToolVersionVersion(BaseModel):
1718
+ major: int = Field(..., title='Major')
1719
+ minor: int = Field(..., title='Minor')
1720
+ patch: int = Field(..., title='Patch')
1721
+
1722
+
1709
1723
  class AmigoLibMongoCollectionsUserUserPreferences(BaseModel):
1710
1724
  enable_response_recommendation: Optional[bool] = Field(
1711
1725
  False,
@@ -10734,6 +10748,13 @@ class OrganizationCreateOrganizationRequestPreferences(BaseModel):
10734
10748
  )
10735
10749
 
10736
10750
 
10751
+ class OrganizationCreateOrganizationRequestUserDimension(BaseModel):
10752
+ description: str = Field(..., min_length=1)
10753
+ tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
10754
+ ..., title='Tags'
10755
+ )
10756
+
10757
+
10737
10758
  class OrganizationCreateOrganizationResponse(BaseModel):
10738
10759
  org_id: str = Field(
10739
10760
  ...,
@@ -10757,19 +10778,19 @@ class OrganizationCreateServiceHierarchicalStateMachineResponse(BaseModel):
10757
10778
  )
10758
10779
 
10759
10780
 
10760
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState(
10781
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState(
10761
10782
  AnnotationStateInputNextState
10762
10783
  ):
10763
10784
  pass
10764
10785
 
10765
10786
 
10766
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition(
10787
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition(
10767
10788
  BaseModel
10768
10789
  ):
10769
10790
  description: str = Field(..., min_length=1)
10770
10791
  next_state: Union[
10771
10792
  StateOrRefName,
10772
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState,
10793
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState,
10773
10794
  ] = Field(..., title='Next State')
10774
10795
 
10775
10796
 
@@ -10875,18 +10896,22 @@ class OrganizationGetUserDimensionsResponse(BaseModel):
10875
10896
  )
10876
10897
 
10877
10898
 
10899
+ class OrganizationModifyOrganizationRequestUserDimensions(
10900
+ RootModel[List[OrganizationCreateOrganizationRequestUserDimension]]
10901
+ ):
10902
+ root: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
10903
+ ...,
10904
+ description='User dimensions for the organization. If not set or `null`, this field is not updated.',
10905
+ min_length=1,
10906
+ title='User Dimensions',
10907
+ )
10908
+
10909
+
10878
10910
  OrganizationModifyOrganizationRequestPreferences = (
10879
10911
  OrganizationCreateOrganizationRequestPreferences
10880
10912
  )
10881
10913
 
10882
10914
 
10883
- class OrganizationModifyOrganizationRequestUserDimension(BaseModel):
10884
- description: str = Field(..., min_length=1)
10885
- tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
10886
- ..., title='Tags'
10887
- )
10888
-
10889
-
10890
10915
  class RoleAssignRoleRequest(BaseModel):
10891
10916
  user_id: str = Field(
10892
10917
  ...,
@@ -10933,17 +10958,9 @@ class RoleModifyRoleResponse(BaseModel):
10933
10958
  )
10934
10959
 
10935
10960
 
10936
- class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
10937
- llm_name: LLMType
10938
- top_p: float = Field(..., description='The preferred `top_p` value.', title='Top P')
10939
- temperature: float = Field(
10940
- ..., description='The preferred temperature value.', title='Temperature'
10941
- )
10942
- top_k: int = Field(
10943
- ...,
10944
- description='The preferred `top_k` value. Note that not all LLMs support this.',
10945
- title='Top K',
10946
- )
10961
+ class ServiceCreateServiceRequestVersionSetLLMLoadBalancingSetConfig(BaseModel):
10962
+ llm_load_balancing_set_name: LLMLoadBalancingSetType
10963
+ configs: Dict[str, PerLLMConfig] = Field(..., title='Configs')
10947
10964
 
10948
10965
 
10949
10966
  class ServiceCreateServiceResponse(BaseModel):
@@ -11005,11 +11022,17 @@ class ServiceUpdateServiceRequest(BaseModel):
11005
11022
  )
11006
11023
 
11007
11024
 
11008
- class ServiceUpsertServiceVersionSetRequestVersionSetLLMLoadBalancingSetConfig(
11009
- BaseModel
11010
- ):
11011
- llm_load_balancing_set_name: LLMLoadBalancingSetType
11012
- configs: Dict[str, PerLLMConfig] = Field(..., title='Configs')
11025
+ class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
11026
+ llm_name: LLMType
11027
+ top_p: float = Field(..., description='The preferred `top_p` value.', title='Top P')
11028
+ temperature: float = Field(
11029
+ ..., description='The preferred temperature value.', title='Temperature'
11030
+ )
11031
+ top_k: int = Field(
11032
+ ...,
11033
+ description='The preferred `top_k` value. Note that not all LLMs support this.',
11034
+ title='Top K',
11035
+ )
11013
11036
 
11014
11037
 
11015
11038
  class SimulationCreateSimulationPersonaRequestInitialVersion(BaseModel):
@@ -11220,6 +11243,37 @@ class SimulationGetSimulationScenariosResponseFilterValues(BaseModel):
11220
11243
  )
11221
11244
 
11222
11245
 
11246
+ class SimulationGetSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
11247
+ id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
11248
+ name: str = Field(
11249
+ ..., description='The name of the simulation scenario.', title='Name'
11250
+ )
11251
+ tags: Dict[str, Optional[str]] = Field(
11252
+ ..., description='The tags of the simulation scenario.', title='Tags'
11253
+ )
11254
+ is_deleted: bool = Field(
11255
+ ...,
11256
+ description='Whether the simulation scenario is deleted.',
11257
+ title='Is Deleted',
11258
+ )
11259
+ created_at: AwareDatetime = Field(
11260
+ ...,
11261
+ description='The timestamp when the simulation scenario was created.',
11262
+ title='Created At',
11263
+ )
11264
+ creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11265
+ Field(..., description='The user who created the simulation scenario.')
11266
+ )
11267
+ updated_at: AwareDatetime = Field(
11268
+ ...,
11269
+ description='The timestamp when the simulation scenario was last updated.',
11270
+ title='Updated At',
11271
+ )
11272
+ updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11273
+ Field(..., description='The user who last updated the simulation scenario.')
11274
+ )
11275
+
11276
+
11223
11277
  class SimulationGetSimulationUnitTestSetRunArtifactsResponse(BaseModel):
11224
11278
  presigned_url: str = Field(
11225
11279
  ...,
@@ -11342,34 +11396,13 @@ class SimulationSearchSimulationPersonasResponseSimulationPersonaInstance(BaseMo
11342
11396
  )
11343
11397
 
11344
11398
 
11345
- class SimulationSearchSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
11346
- id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
11347
- name: str = Field(
11348
- ..., description='The name of the simulation scenario.', title='Name'
11349
- )
11350
- tags: Dict[str, Optional[str]] = Field(
11351
- ..., description='The tags of the simulation scenario.', title='Tags'
11352
- )
11353
- is_deleted: bool = Field(
11354
- ...,
11355
- description='Whether the simulation scenario is deleted.',
11356
- title='Is Deleted',
11357
- )
11358
- created_at: AwareDatetime = Field(
11359
- ...,
11360
- description='The timestamp when the simulation scenario was created.',
11361
- title='Created At',
11362
- )
11363
- creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11364
- Field(..., description='The user who created the simulation scenario.')
11365
- )
11366
- updated_at: AwareDatetime = Field(
11399
+ class SimulationSearchSimulationScenariosResponse(BaseModel):
11400
+ simulation_scenarios: List[
11401
+ SimulationGetSimulationScenariosResponseSimulationScenarioInstance
11402
+ ] = Field(
11367
11403
  ...,
11368
- description='The timestamp when the simulation scenario was last updated.',
11369
- title='Updated At',
11370
- )
11371
- updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11372
- Field(..., description='The user who last updated the simulation scenario.')
11404
+ description='The list of simulation scenarios.',
11405
+ title='Simulation Scenarios',
11373
11406
  )
11374
11407
 
11375
11408
 
@@ -11473,6 +11506,22 @@ class ToolGetToolsResponseFilterValues(BaseModel):
11473
11506
  )
11474
11507
 
11475
11508
 
11509
+ class ToolInvokeToolVersionRequestInputInvocationMode(Enum):
11510
+ regular = 'regular'
11511
+ conversation_simulation = 'conversation-simulation'
11512
+
11513
+
11514
+ class ToolInvokeToolVersionRequestInput(BaseModel):
11515
+ input_parameters: Dict[str, Any] = Field(
11516
+ ...,
11517
+ description='The input parameters for the tool invocation.',
11518
+ title='Input Parameters',
11519
+ )
11520
+ invocation_mode: ToolInvokeToolVersionRequestInputInvocationMode = Field(
11521
+ ..., description='The mode of invocation.', title='Invocation Mode'
11522
+ )
11523
+
11524
+
11476
11525
  class ToolModifyToolRequest(BaseModel):
11477
11526
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
11478
11527
  None, description='A description of this tool. Only updates if not-null.'
@@ -11515,19 +11564,34 @@ class ToolPublishToolVersionResponse(BaseModel):
11515
11564
  )
11516
11565
 
11517
11566
 
11518
- class ToolTestToolRequestInputInvocationMode(Enum):
11519
- regular = 'regular'
11520
- conversation_simulation = 'conversation-simulation'
11521
-
11522
-
11523
- class ToolTestToolRequestInput(BaseModel):
11524
- input_parameters: Dict[str, Any] = Field(
11567
+ class ToolTestToolRequest(BaseModel):
11568
+ inputs: List[ToolInvokeToolVersionRequestInput] = Field(
11525
11569
  ...,
11526
- description='The input parameters for the tool invocation.',
11527
- title='Input Parameters',
11570
+ description='The list of inputs for the tool invocation.',
11571
+ max_length=10,
11572
+ min_length=1,
11573
+ title='Inputs',
11528
11574
  )
11529
- invocation_mode: ToolTestToolRequestInputInvocationMode = Field(
11530
- ..., description='The mode of invocation.', title='Invocation Mode'
11575
+ commit_branch: str = Field(
11576
+ ...,
11577
+ description='The branch in the tools repo whose tip will be tested.',
11578
+ title='Commit Branch',
11579
+ )
11580
+ project_path: str = Field(
11581
+ ...,
11582
+ description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
11583
+ pattern='^[-\\w\\d_/]+$',
11584
+ title='Project Path',
11585
+ )
11586
+ required_envvars: Dict[str, str] = Field(
11587
+ ...,
11588
+ description='The environment variables required for the tool to run.',
11589
+ title='Required Envvars',
11590
+ )
11591
+ required_secret_envvars: Dict[str, str] = Field(
11592
+ ...,
11593
+ description='The secret environment variables required for the tool to run.',
11594
+ title='Required Secret Envvars',
11531
11595
  )
11532
11596
 
11533
11597
 
@@ -11599,7 +11663,7 @@ class UserGetMemoriesResponse(BaseModel):
11599
11663
  )
11600
11664
 
11601
11665
 
11602
- class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11666
+ class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11603
11667
  num_conversations: int = Field(
11604
11668
  ...,
11605
11669
  description='The number of conversations the user has created.',
@@ -11617,6 +11681,31 @@ class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11617
11681
  )
11618
11682
 
11619
11683
 
11684
+ class UserSearchUsersResponseUserInstance(BaseModel):
11685
+ org_id: str = Field(
11686
+ ...,
11687
+ description='The ID of the organization that this user belongs to.',
11688
+ title='Org Id',
11689
+ )
11690
+ user_id: str = Field(
11691
+ ..., description='The identifier of the user.', title='User Id'
11692
+ )
11693
+ first_name: str = Field(
11694
+ ..., description='The first name of the user.', title='First Name'
11695
+ )
11696
+ last_name: str = Field(
11697
+ ..., description='The last name of the user.', title='Last Name'
11698
+ )
11699
+ email: str = Field(..., description='Email of the user.', title='Email')
11700
+ user_stats: UserGetUsersResponseUserInstanceUserStats = Field(
11701
+ ..., description="Statistics about the user's usage of the Amigo platform."
11702
+ )
11703
+ role: str = Field(..., description='The ID of the role of the user.', title='Role')
11704
+ preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
11705
+ ..., description='The preferences of the user.'
11706
+ )
11707
+
11708
+
11620
11709
  class UserSignInWithApiKeyResponse(BaseModel):
11621
11710
  id_token: str = Field(
11622
11711
  ...,
@@ -12064,7 +12153,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
12064
12153
  [], description='The IDs of the messages to retrieve.', title='Id'
12065
12154
  )
12066
12155
  message_type: Optional[List[MessageType]] = Field(
12067
- ['agent-message', 'external-event', 'user-message'],
12156
+ ['external-event', 'agent-message', 'user-message'],
12068
12157
  description='The type of messages to retrieve.',
12069
12158
  title='Message Type',
12070
12159
  )
@@ -13795,7 +13884,7 @@ class ServiceInstance(BaseModel):
13795
13884
  agent_id: str = Field(
13796
13885
  ..., description='The ID of the agent that this service uses.', title='Agent Id'
13797
13886
  )
13798
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
13887
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13799
13888
  ..., description='The tags of the service.', title='Tags'
13800
13889
  )
13801
13890
 
@@ -13848,7 +13937,7 @@ class SimulationUnitTestSet(BaseModel):
13848
13937
  description='Whether the unit test set has been deleted.',
13849
13938
  title='Is Deleted',
13850
13939
  )
13851
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
13940
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13852
13941
  ..., description='The tags of the simulation unit test set.', title='Tags'
13853
13942
  )
13854
13943
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13887,7 +13976,7 @@ class SimulationUnitTestSetInstance(BaseModel):
13887
13976
  description='Whether the unit test set has been deleted.',
13888
13977
  title='Is Deleted',
13889
13978
  )
13890
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
13979
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13891
13980
  ..., description='The tags of the simulation unit test set.', title='Tags'
13892
13981
  )
13893
13982
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13942,6 +14031,9 @@ class ToolCallSpecInput(BaseModel):
13942
14031
  le=10.0,
13943
14032
  title='Audio Filler Triggered After',
13944
14033
  )
14034
+ result_persistence: Optional[ResultPersistence] = Field(
14035
+ 'persisted', description='The type of result persistence for this tool call.'
14036
+ )
13945
14037
 
13946
14038
 
13947
14039
  class ToolCallStateInput(BaseModel):
@@ -13999,7 +14091,7 @@ class ToolInstance(BaseModel):
13999
14091
  description='A list of encrypted environment variables that some versions of this tool uses.',
14000
14092
  title='Secret Envvars',
14001
14093
  )
14002
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
14094
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
14003
14095
  ..., description='The tags of the simulation persona.', title='Tags'
14004
14096
  )
14005
14097
 
@@ -14014,7 +14106,7 @@ class ToolVersionInstance(BaseModel):
14014
14106
  description='The ID of the tool that this version belongs to.',
14015
14107
  title='Tool Id',
14016
14108
  )
14017
- version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
14109
+ version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
14018
14110
  ..., description='The version of the tool.'
14019
14111
  )
14020
14112
  required_envvars: List[str] = Field(
@@ -14413,7 +14505,7 @@ class OrganizationCreateOrganizationRequest(BaseModel):
14413
14505
  description='Additional descriptions of the services offered by this organization that is displayed below `main_description` in a smaller font.',
14414
14506
  min_length=1,
14415
14507
  )
14416
- user_dimensions: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
14508
+ user_dimensions: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
14417
14509
  ...,
14418
14510
  description='User dimensions for the organization.',
14419
14511
  title='User Dimensions',
@@ -14557,17 +14649,6 @@ class OrganizationGetOrganizationMetricsResponse(BaseModel):
14557
14649
  )
14558
14650
 
14559
14651
 
14560
- class OrganizationModifyOrganizationRequestUserDimensions(
14561
- RootModel[List[OrganizationModifyOrganizationRequestUserDimension]]
14562
- ):
14563
- root: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
14564
- ...,
14565
- description='User dimensions for the organization. If not set or `null`, this field is not updated.',
14566
- min_length=1,
14567
- title='User Dimensions',
14568
- )
14569
-
14570
-
14571
14652
  class OrganizationModifyOrganizationRequest(BaseModel):
14572
14653
  title: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14573
14654
  None,
@@ -14642,6 +14723,26 @@ class RoleModifyRoleRequestPermissionGrant(BaseModel):
14642
14723
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = None
14643
14724
 
14644
14725
 
14726
+ class ServiceCreateServiceRequestVersionSet(BaseModel):
14727
+ agent_version_number: Optional[int] = Field(
14728
+ ...,
14729
+ description='The version number of the agent to be used. If None, the latest agent version will be used.',
14730
+ title='Agent Version Number',
14731
+ )
14732
+ service_hierarchical_state_machine_version_number: Optional[int] = Field(
14733
+ ...,
14734
+ description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
14735
+ title='Service Hierarchical State Machine Version Number',
14736
+ )
14737
+ llm_model_preferences: Dict[
14738
+ str,
14739
+ Union[
14740
+ ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig,
14741
+ ServiceCreateServiceRequestVersionSetLLMLoadBalancingSetConfig,
14742
+ ],
14743
+ ] = Field(..., title='Llm Model Preferences')
14744
+
14745
+
14645
14746
  class ServiceGetServicesResponse(BaseModel):
14646
14747
  services: List[ServiceInstance] = Field(
14647
14748
  ..., description='The services that are found.', title='Services'
@@ -14662,24 +14763,10 @@ class ServiceGetServicesResponse(BaseModel):
14662
14763
  )
14663
14764
 
14664
14765
 
14665
- class ServiceUpsertServiceVersionSetRequestVersionSet(BaseModel):
14666
- agent_version_number: Optional[int] = Field(
14667
- ...,
14668
- description='The version number of the agent to be used. If None, the latest agent version will be used.',
14669
- title='Agent Version Number',
14670
- )
14671
- service_hierarchical_state_machine_version_number: Optional[int] = Field(
14672
- ...,
14673
- description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
14674
- title='Service Hierarchical State Machine Version Number',
14766
+ class ServiceUpsertServiceVersionSetRequest(BaseModel):
14767
+ version_set: ServiceCreateServiceRequestVersionSet = Field(
14768
+ ..., description='The version set to upsert.'
14675
14769
  )
14676
- llm_model_preferences: Dict[
14677
- str,
14678
- Union[
14679
- ServiceCreateServiceRequestVersionSetLLMConfig,
14680
- ServiceUpsertServiceVersionSetRequestVersionSetLLMLoadBalancingSetConfig,
14681
- ],
14682
- ] = Field(..., title='Llm Model Preferences')
14683
14770
 
14684
14771
 
14685
14772
  class SimulationCreateSimulationPersonaRequest(BaseModel):
@@ -14769,7 +14856,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
14769
14856
 
14770
14857
  class SimulationGetSimulationScenariosResponse(BaseModel):
14771
14858
  simulation_scenarios: List[
14772
- SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
14859
+ SimulationGetSimulationScenariosResponseSimulationScenarioInstance
14773
14860
  ] = Field(
14774
14861
  ...,
14775
14862
  description='The list of simulation scenarios.',
@@ -14825,16 +14912,6 @@ class SimulationSearchSimulationPersonasResponse(BaseModel):
14825
14912
  )
14826
14913
 
14827
14914
 
14828
- class SimulationSearchSimulationScenariosResponse(BaseModel):
14829
- simulation_scenarios: List[
14830
- SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
14831
- ] = Field(
14832
- ...,
14833
- description='The list of simulation scenarios.',
14834
- title='Simulation Scenarios',
14835
- )
14836
-
14837
-
14838
14915
  class SimulationSearchSimulationUnitTestSetResponse(BaseModel):
14839
14916
  simulation_unit_test_sets: List[SimulationUnitTestSet] = Field(
14840
14917
  ...,
@@ -14925,7 +15002,7 @@ class ToolGetToolsResponse(BaseModel):
14925
15002
 
14926
15003
 
14927
15004
  class ToolInvokeToolVersionRequest(BaseModel):
14928
- inputs: List[ToolTestToolRequestInput] = Field(
15005
+ inputs: List[ToolInvokeToolVersionRequestInput] = Field(
14929
15006
  ...,
14930
15007
  description='The list of inputs for the tool invocation.',
14931
15008
  max_length=10,
@@ -14946,37 +15023,6 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
14946
15023
  deletes: Optional[List[DeleteRequest]] = Field([], title='Deletes')
14947
15024
 
14948
15025
 
14949
- class ToolTestToolRequest(BaseModel):
14950
- inputs: List[ToolTestToolRequestInput] = Field(
14951
- ...,
14952
- description='The list of inputs for the tool invocation.',
14953
- max_length=10,
14954
- min_length=1,
14955
- title='Inputs',
14956
- )
14957
- commit_branch: str = Field(
14958
- ...,
14959
- description='The branch in the tools repo whose tip will be tested.',
14960
- title='Commit Branch',
14961
- )
14962
- project_path: str = Field(
14963
- ...,
14964
- description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
14965
- pattern='^[-\\w\\d_/]+$',
14966
- title='Project Path',
14967
- )
14968
- required_envvars: Dict[str, str] = Field(
14969
- ...,
14970
- description='The environment variables required for the tool to run.',
14971
- title='Required Envvars',
14972
- )
14973
- required_secret_envvars: Dict[str, str] = Field(
14974
- ...,
14975
- description='The secret environment variables required for the tool to run.',
14976
- title='Required Secret Envvars',
14977
- )
14978
-
14979
-
14980
15026
  ToolTestToolResponse = ToolInvokeToolVersionResponse
14981
15027
 
14982
15028
 
@@ -15017,28 +15063,23 @@ class UserGetUserModelResponse(BaseModel):
15017
15063
  )
15018
15064
 
15019
15065
 
15020
- class UserSearchUsersResponseUserInstance(BaseModel):
15021
- org_id: str = Field(
15022
- ...,
15023
- description='The ID of the organization that this user belongs to.',
15024
- title='Org Id',
15025
- )
15026
- user_id: str = Field(
15027
- ..., description='The identifier of the user.', title='User Id'
15028
- )
15029
- first_name: str = Field(
15030
- ..., description='The first name of the user.', title='First Name'
15066
+ class UserGetUsersResponse(BaseModel):
15067
+ users: List[UserSearchUsersResponseUserInstance] = Field(
15068
+ ..., description='Users in this organization.', title='Users'
15031
15069
  )
15032
- last_name: str = Field(
15033
- ..., description='The last name of the user.', title='Last Name'
15070
+ has_more: bool = Field(
15071
+ ..., description='Whether there are more users to retrieve.', title='Has More'
15034
15072
  )
15035
- email: str = Field(..., description='Email of the user.', title='Email')
15036
- user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
15037
- ..., description="Statistics about the user's usage of the Amigo platform."
15073
+ continuation_token: Optional[int] = Field(
15074
+ ...,
15075
+ description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
15076
+ title='Continuation Token',
15038
15077
  )
15039
- role: str = Field(..., description='The ID of the role of the user.', title='Role')
15040
- preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
15041
- ..., description='The preferences of the user.'
15078
+
15079
+
15080
+ class UserSearchUsersResponse(BaseModel):
15081
+ users: List[UserSearchUsersResponseUserInstance] = Field(
15082
+ ..., description='Users in this organization.', title='Users'
15042
15083
  )
15043
15084
 
15044
15085
 
@@ -15075,7 +15116,7 @@ class ActionStateInput(BaseModel):
15075
15116
  Field(..., title='Boundary Constraints')
15076
15117
  )
15077
15118
  exit_conditions: List[
15078
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15119
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15079
15120
  ] = Field(..., title='Exit Conditions')
15080
15121
  action_tool_call_specs: List[ToolCallSpecInput] = Field(
15081
15122
  ..., title='Action Tool Call Specs'
@@ -15184,7 +15225,7 @@ class DecisionStateInput(BaseModel):
15184
15225
  type: Literal['decision'] = Field(..., title='Type')
15185
15226
  name: str = Field(..., pattern='^[A-Za-z0-9_]+$')
15186
15227
  exit_conditions: List[
15187
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15228
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15188
15229
  ] = Field(..., title='Exit Conditions')
15189
15230
  decision_guidelines: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
15190
15231
  ..., title='Decision Guidelines'
@@ -15225,7 +15266,7 @@ class Metric(BaseModel):
15225
15266
  additional_notes: Optional[str] = Field(
15226
15267
  ..., description='Additional notes about the metric.', title='Additional Notes'
15227
15268
  )
15228
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15269
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15229
15270
  ..., description='The tags of the metric.', title='Tags'
15230
15271
  )
15231
15272
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15295,7 +15336,7 @@ class MetricInstance(BaseModel):
15295
15336
  additional_notes: Optional[str] = Field(
15296
15337
  ..., description='Additional notes about the metric.', title='Additional Notes'
15297
15338
  )
15298
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15339
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15299
15340
  ..., description='The tags of the metric.', title='Tags'
15300
15341
  )
15301
15342
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15497,7 +15538,7 @@ class SimulationUnitTest(BaseModel):
15497
15538
  is_deleted: bool = Field(
15498
15539
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15499
15540
  )
15500
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15541
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15501
15542
  ..., description='The tags of the simulation persona.', title='Tags'
15502
15543
  )
15503
15544
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15542,7 +15583,7 @@ class SimulationUnitTestInstance(BaseModel):
15542
15583
  is_deleted: bool = Field(
15543
15584
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15544
15585
  )
15545
- tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15586
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15546
15587
  ..., description='The tags of the simulation persona.', title='Tags'
15547
15588
  )
15548
15589
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15916,23 +15957,15 @@ class ServiceCreateServiceRequest(BaseModel):
15916
15957
  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.',
15917
15958
  title='Is Active',
15918
15959
  )
15919
- release_version_set: Optional[ServiceUpsertServiceVersionSetRequestVersionSet] = (
15920
- Field(
15921
- None,
15922
- 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.',
15923
- )
15960
+ release_version_set: Optional[ServiceCreateServiceRequestVersionSet] = Field(
15961
+ None,
15962
+ 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.',
15924
15963
  )
15925
15964
  tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
15926
15965
  ..., description='The tags of this service.', title='Tags'
15927
15966
  )
15928
15967
 
15929
15968
 
15930
- class ServiceUpsertServiceVersionSetRequest(BaseModel):
15931
- version_set: ServiceUpsertServiceVersionSetRequestVersionSet = Field(
15932
- ..., description='The version set to upsert.'
15933
- )
15934
-
15935
-
15936
15969
  class SimulationCreateSimulationUnitTestRequest(BaseModel):
15937
15970
  name: str = Field(..., description='The name of the unit test.', min_length=1)
15938
15971
  description: str = Field(
@@ -16088,7 +16121,7 @@ class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
16088
16121
  tool_id: str = Field(
16089
16122
  ..., description='The ID of the tool that was invoked.', title='Tool Id'
16090
16123
  )
16091
- tool_version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
16124
+ tool_version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
16092
16125
  ..., description='The version of the tool that was invoked.'
16093
16126
  )
16094
16127
 
@@ -16101,26 +16134,6 @@ class ToolSearchToolInvocationsResponse(BaseModel):
16101
16134
  )
16102
16135
 
16103
16136
 
16104
- class UserGetUsersResponse(BaseModel):
16105
- users: List[UserSearchUsersResponseUserInstance] = Field(
16106
- ..., description='Users in this organization.', title='Users'
16107
- )
16108
- has_more: bool = Field(
16109
- ..., description='Whether there are more users to retrieve.', title='Has More'
16110
- )
16111
- continuation_token: Optional[int] = Field(
16112
- ...,
16113
- description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
16114
- title='Continuation Token',
16115
- )
16116
-
16117
-
16118
- class UserSearchUsersResponse(BaseModel):
16119
- users: List[UserSearchUsersResponseUserInstance] = Field(
16120
- ..., description='Users in this organization.', title='Users'
16121
- )
16122
-
16123
-
16124
16137
  class SimulationUnitTestSetRunInstance(BaseModel):
16125
16138
  org_id: str = Field(..., title='Org Id')
16126
16139
  id: str = Field(..., title='Id')
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: amigo_sdk
3
- Version: 0.59.0
3
+ Version: 0.61.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=YfC2yqJ-vgGRTmduEHhH5Oc35gDmkfaqfl8-O16Vdgo,139
1
+ amigo_sdk/__init__.py,sha256=-clUEoMLj9P89QvxZSOg6hC2zVSbmICf4ZVLXTaFEnM,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=4t6KgjiEXxRXMAKf0tiUJwkTUmjpUeqH16oTl5UHzWE,452102
9
+ amigo_sdk/generated/model.py,sha256=lAsbnObxCkjQZP0eBDX_OHkvYAzDgb3T-3ZUERBzRxk,452956
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.59.0.dist-info/METADATA,sha256=RVHMmG4MS7xKbhgpcp2Upvqyofypm_dnyqCbHnND_gI,8350
15
- amigo_sdk-0.59.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- amigo_sdk-0.59.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
- amigo_sdk-0.59.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
- amigo_sdk-0.59.0.dist-info/RECORD,,
14
+ amigo_sdk-0.61.0.dist-info/METADATA,sha256=WgMx4jQFf3yLBrPlE4FLHI350iRY3e_aFtGc90FG0iw,8350
15
+ amigo_sdk-0.61.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
+ amigo_sdk-0.61.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
+ amigo_sdk-0.61.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
+ amigo_sdk-0.61.0.dist-info/RECORD,,