amigo_sdk 0.60.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.60.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-13T01:47:51+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):
@@ -1601,11 +1615,6 @@ class AmigoLibMongoCollectionsDynamicBehaviorSetVersionAction(BaseModel):
1601
1615
  )
1602
1616
 
1603
1617
 
1604
- class AmigoLibMongoCollectionsMetricMetricTag(BaseModel):
1605
- key: str = Field(..., title='Key')
1606
- value: Optional[str] = Field(..., title='Value')
1607
-
1608
-
1609
1618
  AmigoLibMongoCollectionsMetricMetricUserInfo = (
1610
1619
  AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1611
1620
  )
@@ -1652,6 +1661,11 @@ AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
1652
1661
  )
1653
1662
 
1654
1663
 
1664
+ class AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag(BaseModel):
1665
+ key: str = Field(..., title='Key')
1666
+ value: Optional[str] = Field(..., title='Value')
1667
+
1668
+
1655
1669
  AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo = (
1656
1670
  AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1657
1671
  )
@@ -1693,12 +1707,6 @@ AmigoLibMongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo
1693
1707
  )
1694
1708
 
1695
1709
 
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,
@@ -1937,6 +1951,20 @@ class DynamicBehaviorSetGetDynamicBehaviorInvocationsResponse(BaseModel):
1937
1951
  )
1938
1952
 
1939
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
+
1940
1968
  class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
1941
1969
  applied_to_services_ids: List[str] = Field(
1942
1970
  ...,
@@ -1957,18 +1985,14 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
1957
1985
  )
1958
1986
 
1959
1987
 
1960
- class DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
1961
- BaseModel
1962
- ):
1963
- id: str = Field(..., title='Id')
1964
- name: str = Field(..., title='Name')
1965
- is_active: bool = Field(..., title='Is Active')
1966
- applied_to_services: List[str] = Field(..., title='Applied To Services')
1967
- tags: Dict[str, Optional[str]] = Field(..., title='Tags')
1968
- created_at: AwareDatetime = Field(..., title='Created At')
1969
- creator: AmigoLibMongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1970
- updated_at: AwareDatetime = Field(..., title='Updated At')
1971
- 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
+ )
1972
1996
 
1973
1997
 
1974
1998
  class DynamicBehaviorSetUpdateDynamicBehaviorSetRequestAppliedToService(
@@ -10754,19 +10778,19 @@ class OrganizationCreateServiceHierarchicalStateMachineResponse(BaseModel):
10754
10778
  )
10755
10779
 
10756
10780
 
10757
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState(
10781
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState(
10758
10782
  AnnotationStateInputNextState
10759
10783
  ):
10760
10784
  pass
10761
10785
 
10762
10786
 
10763
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition(
10787
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition(
10764
10788
  BaseModel
10765
10789
  ):
10766
10790
  description: str = Field(..., min_length=1)
10767
10791
  next_state: Union[
10768
10792
  StateOrRefName,
10769
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState,
10793
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState,
10770
10794
  ] = Field(..., title='Next State')
10771
10795
 
10772
10796
 
@@ -10934,19 +10958,6 @@ class RoleModifyRoleResponse(BaseModel):
10934
10958
  )
10935
10959
 
10936
10960
 
10937
- class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
10938
- llm_name: LLMType
10939
- top_p: float = Field(..., description='The preferred `top_p` value.', title='Top P')
10940
- temperature: float = Field(
10941
- ..., description='The preferred temperature value.', title='Temperature'
10942
- )
10943
- top_k: int = Field(
10944
- ...,
10945
- description='The preferred `top_k` value. Note that not all LLMs support this.',
10946
- title='Top K',
10947
- )
10948
-
10949
-
10950
10961
  class ServiceCreateServiceRequestVersionSetLLMLoadBalancingSetConfig(BaseModel):
10951
10962
  llm_load_balancing_set_name: LLMLoadBalancingSetType
10952
10963
  configs: Dict[str, PerLLMConfig] = Field(..., title='Configs')
@@ -11011,6 +11022,19 @@ class ServiceUpdateServiceRequest(BaseModel):
11011
11022
  )
11012
11023
 
11013
11024
 
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
+ )
11036
+
11037
+
11014
11038
  class SimulationCreateSimulationPersonaRequestInitialVersion(BaseModel):
11015
11039
  background: str = Field(
11016
11040
  ..., description='The background of the simulation persona.', min_length=1
@@ -11129,16 +11153,6 @@ class SimulationCreateSimulationUnitTestResponse(BaseModel):
11129
11153
  )
11130
11154
 
11131
11155
 
11132
- class SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
11133
- unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
11134
- run_count: int = Field(
11135
- ...,
11136
- description='The number of times to run the unit test.',
11137
- gt=0,
11138
- title='Run Count',
11139
- )
11140
-
11141
-
11142
11156
  class SimulationCreateSimulationUnitTestSetResponse(BaseModel):
11143
11157
  simulation_unit_test_set_id: str = Field(
11144
11158
  ...,
@@ -11449,39 +11463,13 @@ class SimulationUpdateSimulationUnitTestRequestRunCount(RootModel[int]):
11449
11463
  )
11450
11464
 
11451
11465
 
11452
- class SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns(
11453
- RootModel[List[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
11454
- ):
11455
- root: List[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
11456
- Field(
11457
- ...,
11458
- description='The unit test runs that are part of this set.',
11459
- min_length=1,
11460
- title='Unit Test Runs',
11461
- )
11462
- )
11463
-
11464
-
11465
- class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
11466
- name: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
11467
- None, description='The name of the simulation unit test set.'
11468
- )
11469
- description: Optional[
11470
- Union[AmigoLibPydanticBaseModelStrippedNonemptyString1, FieldNotSet]
11471
- ] = Field(
11472
- {},
11473
- description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
11474
- title='Description',
11475
- )
11476
- unit_test_runs: Optional[
11477
- SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns
11478
- ] = Field(
11479
- None,
11480
- description='The unit test runs that are part of this set.',
11481
- title='Unit Test Runs',
11482
- )
11483
- tags: Optional[Dict[str, Optional[str]]] = Field(
11484
- None, description='The tags of the simulation unit test set.', title='Tags'
11466
+ class SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
11467
+ unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
11468
+ run_count: int = Field(
11469
+ ...,
11470
+ description='The number of times to run the unit test.',
11471
+ gt=0,
11472
+ title='Run Count',
11485
11473
  )
11486
11474
 
11487
11475
 
@@ -11518,6 +11506,22 @@ class ToolGetToolsResponseFilterValues(BaseModel):
11518
11506
  )
11519
11507
 
11520
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
+
11521
11525
  class ToolModifyToolRequest(BaseModel):
11522
11526
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
11523
11527
  None, description='A description of this tool. Only updates if not-null.'
@@ -11560,19 +11564,34 @@ class ToolPublishToolVersionResponse(BaseModel):
11560
11564
  )
11561
11565
 
11562
11566
 
11563
- class ToolTestToolRequestInputInvocationMode(Enum):
11564
- regular = 'regular'
11565
- conversation_simulation = 'conversation-simulation'
11566
-
11567
-
11568
- class ToolTestToolRequestInput(BaseModel):
11569
- input_parameters: Dict[str, Any] = Field(
11567
+ class ToolTestToolRequest(BaseModel):
11568
+ inputs: List[ToolInvokeToolVersionRequestInput] = Field(
11570
11569
  ...,
11571
- description='The input parameters for the tool invocation.',
11572
- title='Input Parameters',
11570
+ description='The list of inputs for the tool invocation.',
11571
+ max_length=10,
11572
+ min_length=1,
11573
+ title='Inputs',
11573
11574
  )
11574
- invocation_mode: ToolTestToolRequestInputInvocationMode = Field(
11575
- ..., 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',
11576
11595
  )
11577
11596
 
11578
11597
 
@@ -11644,7 +11663,7 @@ class UserGetMemoriesResponse(BaseModel):
11644
11663
  )
11645
11664
 
11646
11665
 
11647
- class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11666
+ class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11648
11667
  num_conversations: int = Field(
11649
11668
  ...,
11650
11669
  description='The number of conversations the user has created.',
@@ -11662,6 +11681,31 @@ class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11662
11681
  )
11663
11682
 
11664
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
+
11665
11709
  class UserSignInWithApiKeyResponse(BaseModel):
11666
11710
  id_token: str = Field(
11667
11711
  ...,
@@ -13840,7 +13884,7 @@ class ServiceInstance(BaseModel):
13840
13884
  agent_id: str = Field(
13841
13885
  ..., description='The ID of the agent that this service uses.', title='Agent Id'
13842
13886
  )
13843
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
13887
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13844
13888
  ..., description='The tags of the service.', title='Tags'
13845
13889
  )
13846
13890
 
@@ -13893,7 +13937,7 @@ class SimulationUnitTestSet(BaseModel):
13893
13937
  description='Whether the unit test set has been deleted.',
13894
13938
  title='Is Deleted',
13895
13939
  )
13896
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
13940
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13897
13941
  ..., description='The tags of the simulation unit test set.', title='Tags'
13898
13942
  )
13899
13943
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13932,7 +13976,7 @@ class SimulationUnitTestSetInstance(BaseModel):
13932
13976
  description='Whether the unit test set has been deleted.',
13933
13977
  title='Is Deleted',
13934
13978
  )
13935
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
13979
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
13936
13980
  ..., description='The tags of the simulation unit test set.', title='Tags'
13937
13981
  )
13938
13982
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13987,6 +14031,9 @@ class ToolCallSpecInput(BaseModel):
13987
14031
  le=10.0,
13988
14032
  title='Audio Filler Triggered After',
13989
14033
  )
14034
+ result_persistence: Optional[ResultPersistence] = Field(
14035
+ 'persisted', description='The type of result persistence for this tool call.'
14036
+ )
13990
14037
 
13991
14038
 
13992
14039
  class ToolCallStateInput(BaseModel):
@@ -14044,7 +14091,7 @@ class ToolInstance(BaseModel):
14044
14091
  description='A list of encrypted environment variables that some versions of this tool uses.',
14045
14092
  title='Secret Envvars',
14046
14093
  )
14047
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
14094
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
14048
14095
  ..., description='The tags of the simulation persona.', title='Tags'
14049
14096
  )
14050
14097
 
@@ -14059,7 +14106,7 @@ class ToolVersionInstance(BaseModel):
14059
14106
  description='The ID of the tool that this version belongs to.',
14060
14107
  title='Tool Id',
14061
14108
  )
14062
- version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
14109
+ version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
14063
14110
  ..., description='The version of the tool.'
14064
14111
  )
14065
14112
  required_envvars: List[str] = Field(
@@ -14378,7 +14425,7 @@ class DynamicBehaviorSetGetDynamicBehaviorSetVersionsResponse(BaseModel):
14378
14425
 
14379
14426
  class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
14380
14427
  dynamic_behavior_sets: List[
14381
- DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14428
+ DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14382
14429
  ] = Field(
14383
14430
  ...,
14384
14431
  description='The retrieved dynamic behavior sets.',
@@ -14402,16 +14449,6 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
14402
14449
  )
14403
14450
 
14404
14451
 
14405
- class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
14406
- dynamic_behavior_sets: List[
14407
- DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14408
- ] = Field(
14409
- ...,
14410
- description='The retrieved dynamic behavior sets.',
14411
- title='Dynamic Behavior Sets',
14412
- )
14413
-
14414
-
14415
14452
  class MetricCreateMetricRequest(BaseModel):
14416
14453
  name: str = Field(
14417
14454
  ...,
@@ -14700,7 +14737,7 @@ class ServiceCreateServiceRequestVersionSet(BaseModel):
14700
14737
  llm_model_preferences: Dict[
14701
14738
  str,
14702
14739
  Union[
14703
- ServiceCreateServiceRequestVersionSetLLMConfig,
14740
+ ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig,
14704
14741
  ServiceCreateServiceRequestVersionSetLLMLoadBalancingSetConfig,
14705
14742
  ],
14706
14743
  ] = Field(..., title='Llm Model Preferences')
@@ -14781,7 +14818,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
14781
14818
  ..., description='The description of the simulation unit test set.'
14782
14819
  )
14783
14820
  unit_test_runs: List[
14784
- SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor
14821
+ SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor
14785
14822
  ] = Field(
14786
14823
  ...,
14787
14824
  description='The unit test runs that are part of this set.',
@@ -14888,6 +14925,42 @@ SimulationUpdateSimulationUnitTestRequestSuccessCriterion = (
14888
14925
  )
14889
14926
 
14890
14927
 
14928
+ class SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns(
14929
+ RootModel[List[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
14930
+ ):
14931
+ root: List[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
14932
+ Field(
14933
+ ...,
14934
+ description='The unit test runs that are part of this set.',
14935
+ min_length=1,
14936
+ title='Unit Test Runs',
14937
+ )
14938
+ )
14939
+
14940
+
14941
+ class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
14942
+ name: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14943
+ None, description='The name of the simulation unit test set.'
14944
+ )
14945
+ description: Optional[
14946
+ Union[AmigoLibPydanticBaseModelStrippedNonemptyString1, FieldNotSet]
14947
+ ] = Field(
14948
+ {},
14949
+ description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
14950
+ title='Description',
14951
+ )
14952
+ unit_test_runs: Optional[
14953
+ SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns
14954
+ ] = Field(
14955
+ None,
14956
+ description='The unit test runs that are part of this set.',
14957
+ title='Unit Test Runs',
14958
+ )
14959
+ tags: Optional[Dict[str, Optional[str]]] = Field(
14960
+ None, description='The tags of the simulation unit test set.', title='Tags'
14961
+ )
14962
+
14963
+
14891
14964
  class ToolGetToolVersionsResponse(BaseModel):
14892
14965
  tool_versions: List[ToolVersionInstance] = Field(
14893
14966
  ...,
@@ -14929,7 +15002,7 @@ class ToolGetToolsResponse(BaseModel):
14929
15002
 
14930
15003
 
14931
15004
  class ToolInvokeToolVersionRequest(BaseModel):
14932
- inputs: List[ToolTestToolRequestInput] = Field(
15005
+ inputs: List[ToolInvokeToolVersionRequestInput] = Field(
14933
15006
  ...,
14934
15007
  description='The list of inputs for the tool invocation.',
14935
15008
  max_length=10,
@@ -14950,37 +15023,6 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
14950
15023
  deletes: Optional[List[DeleteRequest]] = Field([], title='Deletes')
14951
15024
 
14952
15025
 
14953
- class ToolTestToolRequest(BaseModel):
14954
- inputs: List[ToolTestToolRequestInput] = Field(
14955
- ...,
14956
- description='The list of inputs for the tool invocation.',
14957
- max_length=10,
14958
- min_length=1,
14959
- title='Inputs',
14960
- )
14961
- commit_branch: str = Field(
14962
- ...,
14963
- description='The branch in the tools repo whose tip will be tested.',
14964
- title='Commit Branch',
14965
- )
14966
- project_path: str = Field(
14967
- ...,
14968
- description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
14969
- pattern='^[-\\w\\d_/]+$',
14970
- title='Project Path',
14971
- )
14972
- required_envvars: Dict[str, str] = Field(
14973
- ...,
14974
- description='The environment variables required for the tool to run.',
14975
- title='Required Envvars',
14976
- )
14977
- required_secret_envvars: Dict[str, str] = Field(
14978
- ...,
14979
- description='The secret environment variables required for the tool to run.',
14980
- title='Required Secret Envvars',
14981
- )
14982
-
14983
-
14984
15026
  ToolTestToolResponse = ToolInvokeToolVersionResponse
14985
15027
 
14986
15028
 
@@ -15021,33 +15063,22 @@ class UserGetUserModelResponse(BaseModel):
15021
15063
  )
15022
15064
 
15023
15065
 
15024
- class UserGetUsersResponseUserInstance(BaseModel):
15025
- org_id: str = Field(
15026
- ...,
15027
- description='The ID of the organization that this user belongs to.',
15028
- title='Org Id',
15029
- )
15030
- user_id: str = Field(
15031
- ..., description='The identifier of the user.', title='User Id'
15032
- )
15033
- first_name: str = Field(
15034
- ..., description='The first name of the user.', title='First Name'
15035
- )
15036
- last_name: str = Field(
15037
- ..., description='The last name of the user.', title='Last Name'
15066
+ class UserGetUsersResponse(BaseModel):
15067
+ users: List[UserSearchUsersResponseUserInstance] = Field(
15068
+ ..., description='Users in this organization.', title='Users'
15038
15069
  )
15039
- email: str = Field(..., description='Email of the user.', title='Email')
15040
- user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
15041
- ..., description="Statistics about the user's usage of the Amigo platform."
15070
+ has_more: bool = Field(
15071
+ ..., description='Whether there are more users to retrieve.', title='Has More'
15042
15072
  )
15043
- role: str = Field(..., description='The ID of the role of the user.', title='Role')
15044
- preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
15045
- ..., description='The preferences of the user.'
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',
15046
15077
  )
15047
15078
 
15048
15079
 
15049
15080
  class UserSearchUsersResponse(BaseModel):
15050
- users: List[UserGetUsersResponseUserInstance] = Field(
15081
+ users: List[UserSearchUsersResponseUserInstance] = Field(
15051
15082
  ..., description='Users in this organization.', title='Users'
15052
15083
  )
15053
15084
 
@@ -15085,7 +15116,7 @@ class ActionStateInput(BaseModel):
15085
15116
  Field(..., title='Boundary Constraints')
15086
15117
  )
15087
15118
  exit_conditions: List[
15088
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15119
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15089
15120
  ] = Field(..., title='Exit Conditions')
15090
15121
  action_tool_call_specs: List[ToolCallSpecInput] = Field(
15091
15122
  ..., title='Action Tool Call Specs'
@@ -15194,7 +15225,7 @@ class DecisionStateInput(BaseModel):
15194
15225
  type: Literal['decision'] = Field(..., title='Type')
15195
15226
  name: str = Field(..., pattern='^[A-Za-z0-9_]+$')
15196
15227
  exit_conditions: List[
15197
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15228
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15198
15229
  ] = Field(..., title='Exit Conditions')
15199
15230
  decision_guidelines: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
15200
15231
  ..., title='Decision Guidelines'
@@ -15235,7 +15266,7 @@ class Metric(BaseModel):
15235
15266
  additional_notes: Optional[str] = Field(
15236
15267
  ..., description='Additional notes about the metric.', title='Additional Notes'
15237
15268
  )
15238
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
15269
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15239
15270
  ..., description='The tags of the metric.', title='Tags'
15240
15271
  )
15241
15272
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15305,7 +15336,7 @@ class MetricInstance(BaseModel):
15305
15336
  additional_notes: Optional[str] = Field(
15306
15337
  ..., description='Additional notes about the metric.', title='Additional Notes'
15307
15338
  )
15308
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
15339
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15309
15340
  ..., description='The tags of the metric.', title='Tags'
15310
15341
  )
15311
15342
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15507,7 +15538,7 @@ class SimulationUnitTest(BaseModel):
15507
15538
  is_deleted: bool = Field(
15508
15539
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15509
15540
  )
15510
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
15541
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15511
15542
  ..., description='The tags of the simulation persona.', title='Tags'
15512
15543
  )
15513
15544
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15552,7 +15583,7 @@ class SimulationUnitTestInstance(BaseModel):
15552
15583
  is_deleted: bool = Field(
15553
15584
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15554
15585
  )
15555
- tags: List[AmigoLibMongoCollectionsMetricMetricTag] = Field(
15586
+ tags: List[AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestTag] = Field(
15556
15587
  ..., description='The tags of the simulation persona.', title='Tags'
15557
15588
  )
15558
15589
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -16068,7 +16099,7 @@ class SimulationUpdateSimulationUnitTestRequest(BaseModel):
16068
16099
  )
16069
16100
 
16070
16101
 
16071
- class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
16102
+ class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
16072
16103
  id: str = Field(..., description='The ID of the tool invocation.', title='Id')
16073
16104
  org_id: str = Field(..., description='The ID of the organization.', title='Org Id')
16074
16105
  created_at: Optional[AwareDatetime] = Field(None, title='Created At')
@@ -16090,22 +16121,16 @@ class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
16090
16121
  tool_id: str = Field(
16091
16122
  ..., description='The ID of the tool that was invoked.', title='Tool Id'
16092
16123
  )
16093
- tool_version: AmigoLibMongoCollectionsToolInvocationToolInvocationVersion = Field(
16124
+ tool_version: AmigoLibMongoCollectionsToolVersionToolVersionVersion = Field(
16094
16125
  ..., description='The version of the tool that was invoked.'
16095
16126
  )
16096
16127
 
16097
16128
 
16098
- class UserGetUsersResponse(BaseModel):
16099
- users: List[UserGetUsersResponseUserInstance] = Field(
16100
- ..., description='Users in this organization.', title='Users'
16101
- )
16102
- has_more: bool = Field(
16103
- ..., description='Whether there are more users to retrieve.', title='Has More'
16104
- )
16105
- continuation_token: Optional[int] = Field(
16106
- ...,
16107
- description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
16108
- title='Continuation Token',
16129
+ class ToolSearchToolInvocationsResponse(BaseModel):
16130
+ tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
16131
+ Field(
16132
+ ..., description='The list of tool invocations.', title='Tool Invocations'
16133
+ )
16109
16134
  )
16110
16135
 
16111
16136
 
@@ -16208,7 +16233,7 @@ class SimulationGetSimulationUnitTestSetRunsResponse(BaseModel):
16208
16233
 
16209
16234
 
16210
16235
  class ToolGetToolInvocationsResponse(BaseModel):
16211
- tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
16236
+ tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
16212
16237
  Field(
16213
16238
  ..., description='The list of tool invocations.', title='Tool Invocations'
16214
16239
  )
@@ -16223,11 +16248,3 @@ class ToolGetToolInvocationsResponse(BaseModel):
16223
16248
  description='The continuation token to retrieve the next page of tool invocations, or null if there are no more tool invocations.',
16224
16249
  title='Continuation Token',
16225
16250
  )
16226
-
16227
-
16228
- class ToolSearchToolInvocationsResponse(BaseModel):
16229
- tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
16230
- Field(
16231
- ..., description='The list of tool invocations.', title='Tool Invocations'
16232
- )
16233
- )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: amigo_sdk
3
- Version: 0.60.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=rgp1sJXXCKEVwuxoIMTj5ymrLUO0G_eYGZSjSkHbvow,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=dM-xfybcf5JOrabD7ZgriOLdjv7mx97wmPQdLc4UD9A,452054
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.60.0.dist-info/METADATA,sha256=4fvaIAJU51NqGEP_ZVYYQuUDVE9I1rRJwhsdwZru_0w,8350
15
- amigo_sdk-0.60.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- amigo_sdk-0.60.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
- amigo_sdk-0.60.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
- amigo_sdk-0.60.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,,