amigo_sdk 0.69.0__py3-none-any.whl → 0.71.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.69.0"
1
+ __version__ = "0.71.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-12-03T11:17:35+00:00
3
+ # timestamp: 2025-12-03T17:33:48+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
  ...,
@@ -1704,6 +1699,11 @@ AmigoLibMongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo
1704
1699
  )
1705
1700
 
1706
1701
 
1702
+ class AmigoLibMongoCollectionsToolToolTag(BaseModel):
1703
+ key: str = Field(..., title='Key')
1704
+ value: Optional[str] = Field(..., title='Value')
1705
+
1706
+
1707
1707
  class AmigoLibMongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
1708
1708
  major: int = Field(..., title='Major')
1709
1709
  minor: int = Field(..., title='Minor')
@@ -10752,13 +10752,6 @@ class OrganizationCreateOrganizationRequestPreferences(BaseModel):
10752
10752
  )
10753
10753
 
10754
10754
 
10755
- class OrganizationCreateOrganizationRequestUserDimension(BaseModel):
10756
- description: str = Field(..., min_length=1)
10757
- tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
10758
- ..., title='Tags'
10759
- )
10760
-
10761
-
10762
10755
  class OrganizationCreateOrganizationResponse(BaseModel):
10763
10756
  org_id: str = Field(
10764
10757
  ...,
@@ -10782,19 +10775,19 @@ class OrganizationCreateServiceHierarchicalStateMachineResponse(BaseModel):
10782
10775
  )
10783
10776
 
10784
10777
 
10785
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState(
10778
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState(
10786
10779
  AnnotationStateInputNextState
10787
10780
  ):
10788
10781
  pass
10789
10782
 
10790
10783
 
10791
- class OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition(
10784
+ class OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition(
10792
10785
  BaseModel
10793
10786
  ):
10794
10787
  description: str = Field(..., min_length=1)
10795
10788
  next_state: Union[
10796
10789
  StateOrRefName,
10797
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitConditionNextState,
10790
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitConditionNextState,
10798
10791
  ] = Field(..., title='Next State')
10799
10792
 
10800
10793
 
@@ -10900,22 +10893,18 @@ class OrganizationGetUserDimensionsResponse(BaseModel):
10900
10893
  )
10901
10894
 
10902
10895
 
10903
- class OrganizationModifyOrganizationRequestUserDimensions(
10904
- RootModel[List[OrganizationCreateOrganizationRequestUserDimension]]
10905
- ):
10906
- root: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
10907
- ...,
10908
- description='User dimensions for the organization. If not set or `null`, this field is not updated.',
10909
- min_length=1,
10910
- title='User Dimensions',
10911
- )
10912
-
10913
-
10914
10896
  OrganizationModifyOrganizationRequestPreferences = (
10915
10897
  OrganizationCreateOrganizationRequestPreferences
10916
10898
  )
10917
10899
 
10918
10900
 
10901
+ class OrganizationModifyOrganizationRequestUserDimension(BaseModel):
10902
+ description: str = Field(..., min_length=1)
10903
+ tags: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
10904
+ ..., title='Tags'
10905
+ )
10906
+
10907
+
10919
10908
  class RoleAssignRoleRequest(BaseModel):
10920
10909
  user_id: str = Field(
10921
10910
  ...,
@@ -10962,13 +10951,6 @@ class RoleModifyRoleResponse(BaseModel):
10962
10951
  )
10963
10952
 
10964
10953
 
10965
- class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
10966
- llm_name: LLMType
10967
- params: Optional[Dict[str, Any]] = Field(
10968
- {}, description='LLM-specific parameters to use.', title='Params'
10969
- )
10970
-
10971
-
10972
10954
  class ServiceCreateServiceResponse(BaseModel):
10973
10955
  id: str = Field(..., description='The ID of the newly created service.', title='Id')
10974
10956
 
@@ -11028,6 +11010,13 @@ class ServiceUpdateServiceRequest(BaseModel):
11028
11010
  )
11029
11011
 
11030
11012
 
11013
+ class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
11014
+ llm_name: LLMType
11015
+ params: Optional[Dict[str, Any]] = Field(
11016
+ {}, description='LLM-specific parameters to use.', title='Params'
11017
+ )
11018
+
11019
+
11031
11020
  class SimulationCreateSimulationPersonaRequestInitialVersion(BaseModel):
11032
11021
  background: str = Field(
11033
11022
  ..., description='The background of the simulation persona.', min_length=1
@@ -11146,16 +11135,6 @@ class SimulationCreateSimulationUnitTestResponse(BaseModel):
11146
11135
  )
11147
11136
 
11148
11137
 
11149
- class SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
11150
- unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
11151
- run_count: int = Field(
11152
- ...,
11153
- description='The number of times to run the unit test.',
11154
- gt=0,
11155
- title='Run Count',
11156
- )
11157
-
11158
-
11159
11138
  class SimulationCreateSimulationUnitTestSetResponse(BaseModel):
11160
11139
  simulation_unit_test_set_id: str = Field(
11161
11140
  ...,
@@ -11280,37 +11259,6 @@ class SimulationGetSimulationScenariosResponseFilterValues(BaseModel):
11280
11259
  )
11281
11260
 
11282
11261
 
11283
- class SimulationGetSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
11284
- id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
11285
- name: str = Field(
11286
- ..., description='The name of the simulation scenario.', title='Name'
11287
- )
11288
- tags: Dict[str, Optional[str]] = Field(
11289
- ..., description='The tags of the simulation scenario.', title='Tags'
11290
- )
11291
- is_deleted: bool = Field(
11292
- ...,
11293
- description='Whether the simulation scenario is deleted.',
11294
- title='Is Deleted',
11295
- )
11296
- created_at: AwareDatetime = Field(
11297
- ...,
11298
- description='The timestamp when the simulation scenario was created.',
11299
- title='Created At',
11300
- )
11301
- creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11302
- Field(..., description='The user who created the simulation scenario.')
11303
- )
11304
- updated_at: AwareDatetime = Field(
11305
- ...,
11306
- description='The timestamp when the simulation scenario was last updated.',
11307
- title='Updated At',
11308
- )
11309
- updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11310
- Field(..., description='The user who last updated the simulation scenario.')
11311
- )
11312
-
11313
-
11314
11262
  class SimulationGetSimulationUnitTestSetRunArtifactsResponse(BaseModel):
11315
11263
  presigned_url: str = Field(
11316
11264
  ...,
@@ -11407,13 +11355,34 @@ class SimulationSearchSimulationPersonasResponse(BaseModel):
11407
11355
  )
11408
11356
 
11409
11357
 
11410
- class SimulationSearchSimulationScenariosResponse(BaseModel):
11411
- simulation_scenarios: List[
11412
- SimulationGetSimulationScenariosResponseSimulationScenarioInstance
11413
- ] = Field(
11358
+ class SimulationSearchSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
11359
+ id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
11360
+ name: str = Field(
11361
+ ..., description='The name of the simulation scenario.', title='Name'
11362
+ )
11363
+ tags: Dict[str, Optional[str]] = Field(
11364
+ ..., description='The tags of the simulation scenario.', title='Tags'
11365
+ )
11366
+ is_deleted: bool = Field(
11414
11367
  ...,
11415
- description='The list of simulation scenarios.',
11416
- title='Simulation Scenarios',
11368
+ description='Whether the simulation scenario is deleted.',
11369
+ title='Is Deleted',
11370
+ )
11371
+ created_at: AwareDatetime = Field(
11372
+ ...,
11373
+ description='The timestamp when the simulation scenario was created.',
11374
+ title='Created At',
11375
+ )
11376
+ creator: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11377
+ Field(..., description='The user who created the simulation scenario.')
11378
+ )
11379
+ updated_at: AwareDatetime = Field(
11380
+ ...,
11381
+ description='The timestamp when the simulation scenario was last updated.',
11382
+ title='Updated At',
11383
+ )
11384
+ updated_by: AmigoLibMongoCollectionsSimulationScenarioSimulationScenarioUserInfo = (
11385
+ Field(..., description='The user who last updated the simulation scenario.')
11417
11386
  )
11418
11387
 
11419
11388
 
@@ -11474,39 +11443,13 @@ class SimulationUpdateSimulationUnitTestRequestRunCount(RootModel[int]):
11474
11443
  )
11475
11444
 
11476
11445
 
11477
- class SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns(
11478
- RootModel[List[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
11479
- ):
11480
- root: List[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
11481
- Field(
11482
- ...,
11483
- description='The unit test runs that are part of this set.',
11484
- min_length=1,
11485
- title='Unit Test Runs',
11486
- )
11487
- )
11488
-
11489
-
11490
- class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
11491
- name: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
11492
- None, description='The name of the simulation unit test set.'
11493
- )
11494
- description: Optional[
11495
- Union[AmigoLibPydanticBaseModelStrippedNonemptyString1, FieldNotSet]
11496
- ] = Field(
11497
- {},
11498
- description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
11499
- title='Description',
11500
- )
11501
- unit_test_runs: Optional[
11502
- SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns
11503
- ] = Field(
11504
- None,
11505
- description='The unit test runs that are part of this set.',
11506
- title='Unit Test Runs',
11507
- )
11508
- tags: Optional[Dict[str, Optional[str]]] = Field(
11509
- None, description='The tags of the simulation unit test set.', title='Tags'
11446
+ class SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
11447
+ unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
11448
+ run_count: int = Field(
11449
+ ...,
11450
+ description='The number of times to run the unit test.',
11451
+ gt=0,
11452
+ title='Run Count',
11510
11453
  )
11511
11454
 
11512
11455
 
@@ -11543,6 +11486,22 @@ class ToolGetToolsResponseFilterValues(BaseModel):
11543
11486
  )
11544
11487
 
11545
11488
 
11489
+ class ToolInvokeToolVersionRequestInputInvocationMode(Enum):
11490
+ regular = 'regular'
11491
+ conversation_simulation = 'conversation-simulation'
11492
+
11493
+
11494
+ class ToolInvokeToolVersionRequestInput(BaseModel):
11495
+ input_parameters: Dict[str, Any] = Field(
11496
+ ...,
11497
+ description='The input parameters for the tool invocation.',
11498
+ title='Input Parameters',
11499
+ )
11500
+ invocation_mode: ToolInvokeToolVersionRequestInputInvocationMode = Field(
11501
+ ..., description='The mode of invocation.', title='Invocation Mode'
11502
+ )
11503
+
11504
+
11546
11505
  class ToolModifyToolRequest(BaseModel):
11547
11506
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
11548
11507
  None, description='A description of this tool. Only updates if not-null.'
@@ -11585,19 +11544,34 @@ class ToolPublishToolVersionResponse(BaseModel):
11585
11544
  )
11586
11545
 
11587
11546
 
11588
- class ToolTestToolRequestInputInvocationMode(Enum):
11589
- regular = 'regular'
11590
- conversation_simulation = 'conversation-simulation'
11591
-
11592
-
11593
- class ToolTestToolRequestInput(BaseModel):
11594
- input_parameters: Dict[str, Any] = Field(
11547
+ class ToolTestToolRequest(BaseModel):
11548
+ inputs: List[ToolInvokeToolVersionRequestInput] = Field(
11595
11549
  ...,
11596
- description='The input parameters for the tool invocation.',
11597
- title='Input Parameters',
11550
+ description='The list of inputs for the tool invocation.',
11551
+ max_length=10,
11552
+ min_length=1,
11553
+ title='Inputs',
11598
11554
  )
11599
- invocation_mode: ToolTestToolRequestInputInvocationMode = Field(
11600
- ..., description='The mode of invocation.', title='Invocation Mode'
11555
+ commit_branch: str = Field(
11556
+ ...,
11557
+ description='The branch in the tools repo whose tip will be tested.',
11558
+ title='Commit Branch',
11559
+ )
11560
+ project_path: str = Field(
11561
+ ...,
11562
+ description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
11563
+ pattern='^[-\\w\\d_/]+$',
11564
+ title='Project Path',
11565
+ )
11566
+ required_envvars: Dict[str, str] = Field(
11567
+ ...,
11568
+ description='The environment variables required for the tool to run.',
11569
+ title='Required Envvars',
11570
+ )
11571
+ required_secret_envvars: Dict[str, str] = Field(
11572
+ ...,
11573
+ description='The secret environment variables required for the tool to run.',
11574
+ title='Required Secret Envvars',
11601
11575
  )
11602
11576
 
11603
11577
 
@@ -11669,7 +11643,7 @@ class UserGetMemoriesResponse(BaseModel):
11669
11643
  )
11670
11644
 
11671
11645
 
11672
- class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11646
+ class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11673
11647
  num_conversations: int = Field(
11674
11648
  ...,
11675
11649
  description='The number of conversations the user has created.',
@@ -11687,31 +11661,6 @@ class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11687
11661
  )
11688
11662
 
11689
11663
 
11690
- class UserSearchUsersResponseUserInstance(BaseModel):
11691
- org_id: str = Field(
11692
- ...,
11693
- description='The ID of the organization that this user belongs to.',
11694
- title='Org Id',
11695
- )
11696
- user_id: str = Field(
11697
- ..., description='The identifier of the user.', title='User Id'
11698
- )
11699
- first_name: str = Field(
11700
- ..., description='The first name of the user.', title='First Name'
11701
- )
11702
- last_name: str = Field(
11703
- ..., description='The last name of the user.', title='Last Name'
11704
- )
11705
- email: str = Field(..., description='Email of the user.', title='Email')
11706
- user_stats: UserGetUsersResponseUserInstanceUserStats = Field(
11707
- ..., description="Statistics about the user's usage of the Amigo platform."
11708
- )
11709
- role: str = Field(..., description='The ID of the role of the user.', title='Role')
11710
- preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
11711
- ..., description='The preferences of the user.'
11712
- )
11713
-
11714
-
11715
11664
  class UserSignInWithApiKeyResponse(BaseModel):
11716
11665
  id_token: str = Field(
11717
11666
  ...,
@@ -13939,7 +13888,7 @@ class ServiceInstance(BaseModel):
13939
13888
  agent_id: str = Field(
13940
13889
  ..., description='The ID of the agent that this service uses.', title='Agent Id'
13941
13890
  )
13942
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
13891
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
13943
13892
  ..., description='The tags of the service.', title='Tags'
13944
13893
  )
13945
13894
 
@@ -13992,7 +13941,7 @@ class SimulationUnitTestSet(BaseModel):
13992
13941
  description='Whether the unit test set has been deleted.',
13993
13942
  title='Is Deleted',
13994
13943
  )
13995
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
13944
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
13996
13945
  ..., description='The tags of the simulation unit test set.', title='Tags'
13997
13946
  )
13998
13947
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14031,7 +13980,7 @@ class SimulationUnitTestSetInstance(BaseModel):
14031
13980
  description='Whether the unit test set has been deleted.',
14032
13981
  title='Is Deleted',
14033
13982
  )
14034
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
13983
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
14035
13984
  ..., description='The tags of the simulation unit test set.', title='Tags'
14036
13985
  )
14037
13986
  creator: AmigoLibMongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -14147,7 +14096,7 @@ class ToolInstance(BaseModel):
14147
14096
  description='A list of encrypted environment variables that some versions of this tool uses.',
14148
14097
  title='Secret Envvars',
14149
14098
  )
14150
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
14099
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
14151
14100
  ..., description='The tags of the simulation persona.', title='Tags'
14152
14101
  )
14153
14102
 
@@ -14558,7 +14507,7 @@ class OrganizationCreateOrganizationRequest(BaseModel):
14558
14507
  description='Additional descriptions of the services offered by this organization that is displayed below `main_description` in a smaller font.',
14559
14508
  min_length=1,
14560
14509
  )
14561
- user_dimensions: List[OrganizationCreateOrganizationRequestUserDimension] = Field(
14510
+ user_dimensions: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
14562
14511
  ...,
14563
14512
  description='User dimensions for the organization.',
14564
14513
  title='User Dimensions',
@@ -14702,6 +14651,17 @@ class OrganizationGetOrganizationMetricsResponse(BaseModel):
14702
14651
  )
14703
14652
 
14704
14653
 
14654
+ class OrganizationModifyOrganizationRequestUserDimensions(
14655
+ RootModel[List[OrganizationModifyOrganizationRequestUserDimension]]
14656
+ ):
14657
+ root: List[OrganizationModifyOrganizationRequestUserDimension] = Field(
14658
+ ...,
14659
+ description='User dimensions for the organization. If not set or `null`, this field is not updated.',
14660
+ min_length=1,
14661
+ title='User Dimensions',
14662
+ )
14663
+
14664
+
14705
14665
  class OrganizationModifyOrganizationRequest(BaseModel):
14706
14666
  title: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14707
14667
  None,
@@ -14759,7 +14719,7 @@ class OrganizationModifyOrganizationRequest(BaseModel):
14759
14719
  )
14760
14720
 
14761
14721
 
14762
- class RoleCreateRoleRequestPermissionGrant(BaseModel):
14722
+ class RoleModifyRoleRequestPermissionGrant(BaseModel):
14763
14723
  action: AmigoLibMongoCollectionsRoleRolePermissionGrantAction = Field(
14764
14724
  ...,
14765
14725
  description='Whether this grant allows or denies the specified access.',
@@ -14776,45 +14736,6 @@ class RoleCreateRoleRequestPermissionGrant(BaseModel):
14776
14736
  description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = None
14777
14737
 
14778
14738
 
14779
- class RoleModifyRoleRequest(BaseModel):
14780
- description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14781
- None,
14782
- description='A description about the role. Only updated if specified. This field is a mutable field.',
14783
- )
14784
- permission_grants: Optional[List[RoleCreateRoleRequestPermissionGrant]] = Field(
14785
- None,
14786
- description='A list of permission grants associated with this role. Only updated if specified. This field is an immutable field.',
14787
- title='Permission Grants',
14788
- )
14789
- frontend_view: Optional[FrontendView] = Field(
14790
- None,
14791
- description='The frontend view for the user of this role. Only updated if specified. This field is an immutable field.',
14792
- )
14793
- inherited_from: Optional[Union[RoleModifyRoleRequestInheritedFrom, FieldNotSet]] = (
14794
- Field(
14795
- {},
14796
- description='The ID of the role that this role inherits from. Only updated if specified. This field is an immutable field.',
14797
- title='Inherited From',
14798
- )
14799
- )
14800
-
14801
-
14802
- class ServiceCreateServiceRequestVersionSet(BaseModel):
14803
- agent_version_number: Optional[int] = Field(
14804
- ...,
14805
- description='The version number of the agent to be used. If None, the latest agent version will be used.',
14806
- title='Agent Version Number',
14807
- )
14808
- service_hierarchical_state_machine_version_number: Optional[int] = Field(
14809
- ...,
14810
- description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
14811
- title='Service Hierarchical State Machine Version Number',
14812
- )
14813
- llm_model_preferences: Dict[str, ServiceCreateServiceRequestVersionSetLLMConfig] = (
14814
- Field(..., title='Llm Model Preferences')
14815
- )
14816
-
14817
-
14818
14739
  class ServiceGetServicesResponse(BaseModel):
14819
14740
  services: List[ServiceInstance] = Field(
14820
14741
  ..., description='The services that are found.', title='Services'
@@ -14835,10 +14756,20 @@ class ServiceGetServicesResponse(BaseModel):
14835
14756
  )
14836
14757
 
14837
14758
 
14838
- class ServiceUpsertServiceVersionSetRequest(BaseModel):
14839
- version_set: ServiceCreateServiceRequestVersionSet = Field(
14840
- ..., description='The version set to upsert.'
14759
+ class ServiceUpsertServiceVersionSetRequestVersionSet(BaseModel):
14760
+ agent_version_number: Optional[int] = Field(
14761
+ ...,
14762
+ description='The version number of the agent to be used. If None, the latest agent version will be used.',
14763
+ title='Agent Version Number',
14764
+ )
14765
+ service_hierarchical_state_machine_version_number: Optional[int] = Field(
14766
+ ...,
14767
+ description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
14768
+ title='Service Hierarchical State Machine Version Number',
14841
14769
  )
14770
+ llm_model_preferences: Dict[
14771
+ str, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
14772
+ ] = Field(..., title='Llm Model Preferences')
14842
14773
 
14843
14774
 
14844
14775
  class SimulationCreateSimulationPersonaRequest(BaseModel):
@@ -14890,7 +14821,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
14890
14821
  ..., description='The description of the simulation unit test set.'
14891
14822
  )
14892
14823
  unit_test_runs: List[
14893
- SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor
14824
+ SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor
14894
14825
  ] = Field(
14895
14826
  ...,
14896
14827
  description='The unit test runs that are part of this set.',
@@ -14928,7 +14859,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
14928
14859
 
14929
14860
  class SimulationGetSimulationScenariosResponse(BaseModel):
14930
14861
  simulation_scenarios: List[
14931
- SimulationGetSimulationScenariosResponseSimulationScenarioInstance
14862
+ SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
14932
14863
  ] = Field(
14933
14864
  ...,
14934
14865
  description='The list of simulation scenarios.',
@@ -14976,6 +14907,16 @@ class SimulationGetSimulationUnitTestSetsResponse(BaseModel):
14976
14907
  )
14977
14908
 
14978
14909
 
14910
+ class SimulationSearchSimulationScenariosResponse(BaseModel):
14911
+ simulation_scenarios: List[
14912
+ SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
14913
+ ] = Field(
14914
+ ...,
14915
+ description='The list of simulation scenarios.',
14916
+ title='Simulation Scenarios',
14917
+ )
14918
+
14919
+
14979
14920
  class SimulationSearchSimulationUnitTestSetResponse(BaseModel):
14980
14921
  simulation_unit_test_sets: List[SimulationUnitTestSet] = Field(
14981
14922
  ...,
@@ -14989,6 +14930,42 @@ SimulationUpdateSimulationUnitTestRequestSuccessCriterion = (
14989
14930
  )
14990
14931
 
14991
14932
 
14933
+ class SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns(
14934
+ RootModel[List[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
14935
+ ):
14936
+ root: List[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
14937
+ Field(
14938
+ ...,
14939
+ description='The unit test runs that are part of this set.',
14940
+ min_length=1,
14941
+ title='Unit Test Runs',
14942
+ )
14943
+ )
14944
+
14945
+
14946
+ class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
14947
+ name: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
14948
+ None, description='The name of the simulation unit test set.'
14949
+ )
14950
+ description: Optional[
14951
+ Union[AmigoLibPydanticBaseModelStrippedNonemptyString1, FieldNotSet]
14952
+ ] = Field(
14953
+ {},
14954
+ description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
14955
+ title='Description',
14956
+ )
14957
+ unit_test_runs: Optional[
14958
+ SimulationUpdateSimulationUnitTestSetRequestUnitTestRuns
14959
+ ] = Field(
14960
+ None,
14961
+ description='The unit test runs that are part of this set.',
14962
+ title='Unit Test Runs',
14963
+ )
14964
+ tags: Optional[Dict[str, Optional[str]]] = Field(
14965
+ None, description='The tags of the simulation unit test set.', title='Tags'
14966
+ )
14967
+
14968
+
14992
14969
  class ToolGetToolVersionsResponse(BaseModel):
14993
14970
  tool_versions: List[ToolVersionInstance] = Field(
14994
14971
  ...,
@@ -15030,7 +15007,7 @@ class ToolGetToolsResponse(BaseModel):
15030
15007
 
15031
15008
 
15032
15009
  class ToolInvokeToolVersionRequest(BaseModel):
15033
- inputs: List[ToolTestToolRequestInput] = Field(
15010
+ inputs: List[ToolInvokeToolVersionRequestInput] = Field(
15034
15011
  ...,
15035
15012
  description='The list of inputs for the tool invocation.',
15036
15013
  max_length=10,
@@ -15051,37 +15028,6 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
15051
15028
  deletes: Optional[List[DeleteRequest]] = Field([], title='Deletes')
15052
15029
 
15053
15030
 
15054
- class ToolTestToolRequest(BaseModel):
15055
- inputs: List[ToolTestToolRequestInput] = Field(
15056
- ...,
15057
- description='The list of inputs for the tool invocation.',
15058
- max_length=10,
15059
- min_length=1,
15060
- title='Inputs',
15061
- )
15062
- commit_branch: str = Field(
15063
- ...,
15064
- description='The branch in the tools repo whose tip will be tested.',
15065
- title='Commit Branch',
15066
- )
15067
- project_path: str = Field(
15068
- ...,
15069
- description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
15070
- pattern='^[-\\w\\d_/]+$',
15071
- title='Project Path',
15072
- )
15073
- required_envvars: Dict[str, str] = Field(
15074
- ...,
15075
- description='The environment variables required for the tool to run.',
15076
- title='Required Envvars',
15077
- )
15078
- required_secret_envvars: Dict[str, str] = Field(
15079
- ...,
15080
- description='The secret environment variables required for the tool to run.',
15081
- title='Required Secret Envvars',
15082
- )
15083
-
15084
-
15085
15031
  ToolTestToolResponse = ToolInvokeToolVersionResponse
15086
15032
 
15087
15033
 
@@ -15122,23 +15068,28 @@ class UserGetUserModelResponse(BaseModel):
15122
15068
  )
15123
15069
 
15124
15070
 
15125
- class UserGetUsersResponse(BaseModel):
15126
- users: List[UserSearchUsersResponseUserInstance] = Field(
15127
- ..., description='Users in this organization.', title='Users'
15071
+ class UserSearchUsersResponseUserInstance(BaseModel):
15072
+ org_id: str = Field(
15073
+ ...,
15074
+ description='The ID of the organization that this user belongs to.',
15075
+ title='Org Id',
15128
15076
  )
15129
- has_more: bool = Field(
15130
- ..., description='Whether there are more users to retrieve.', title='Has More'
15077
+ user_id: str = Field(
15078
+ ..., description='The identifier of the user.', title='User Id'
15131
15079
  )
15132
- continuation_token: Optional[int] = Field(
15133
- ...,
15134
- description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
15135
- title='Continuation Token',
15080
+ first_name: str = Field(
15081
+ ..., description='The first name of the user.', title='First Name'
15136
15082
  )
15137
-
15138
-
15139
- class UserSearchUsersResponse(BaseModel):
15140
- users: List[UserSearchUsersResponseUserInstance] = Field(
15141
- ..., description='Users in this organization.', title='Users'
15083
+ last_name: str = Field(
15084
+ ..., description='The last name of the user.', title='Last Name'
15085
+ )
15086
+ email: str = Field(..., description='Email of the user.', title='Email')
15087
+ user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
15088
+ ..., description="Statistics about the user's usage of the Amigo platform."
15089
+ )
15090
+ role: str = Field(..., description='The ID of the role of the user.', title='Role')
15091
+ preferences: AmigoLibMongoCollectionsUserUserPreferences = Field(
15092
+ ..., description='The preferences of the user.'
15142
15093
  )
15143
15094
 
15144
15095
 
@@ -15237,7 +15188,7 @@ class ActionStateInput(BaseModel):
15237
15188
  Field(..., title='Boundary Constraints')
15238
15189
  )
15239
15190
  exit_conditions: List[
15240
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15191
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15241
15192
  ] = Field(..., title='Exit Conditions')
15242
15193
  action_tool_call_specs: List[ToolCallSpecInput] = Field(
15243
15194
  ..., title='Action Tool Call Specs'
@@ -15346,7 +15297,7 @@ class DecisionStateInput(BaseModel):
15346
15297
  type: Literal['decision'] = Field(..., title='Type')
15347
15298
  name: str = Field(..., pattern='^[A-Za-z0-9_]+$')
15348
15299
  exit_conditions: List[
15349
- OrganizationCreateServiceHierarchicalStateMachineVersionRequestDecisionStateExitCondition
15300
+ OrganizationCreateServiceHierarchicalStateMachineVersionRequestActionStateExitCondition
15350
15301
  ] = Field(..., title='Exit Conditions')
15351
15302
  decision_guidelines: List[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
15352
15303
  ..., title='Decision Guidelines'
@@ -15387,7 +15338,7 @@ class Metric(BaseModel):
15387
15338
  additional_notes: Optional[str] = Field(
15388
15339
  ..., description='Additional notes about the metric.', title='Additional Notes'
15389
15340
  )
15390
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15341
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15391
15342
  ..., description='The tags of the metric.', title='Tags'
15392
15343
  )
15393
15344
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15457,7 +15408,7 @@ class MetricInstance(BaseModel):
15457
15408
  additional_notes: Optional[str] = Field(
15458
15409
  ..., description='Additional notes about the metric.', title='Additional Notes'
15459
15410
  )
15460
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15411
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15461
15412
  ..., description='The tags of the metric.', title='Tags'
15462
15413
  )
15463
15414
  creator: AmigoLibMongoCollectionsMetricMetricUserInfo = Field(
@@ -15659,7 +15610,7 @@ class SimulationUnitTest(BaseModel):
15659
15610
  is_deleted: bool = Field(
15660
15611
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15661
15612
  )
15662
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15613
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15663
15614
  ..., description='The tags of the simulation persona.', title='Tags'
15664
15615
  )
15665
15616
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15704,7 +15655,7 @@ class SimulationUnitTestInstance(BaseModel):
15704
15655
  is_deleted: bool = Field(
15705
15656
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15706
15657
  )
15707
- tags: List[AmigoLibMongoCollectionsServiceServiceTag] = Field(
15658
+ tags: List[AmigoLibMongoCollectionsToolToolTag] = Field(
15708
15659
  ..., description='The tags of the simulation persona.', title='Tags'
15709
15660
  )
15710
15661
  creator: AmigoLibMongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -16004,7 +15955,7 @@ class RoleCreateRoleRequest(BaseModel):
16004
15955
  description: str = Field(
16005
15956
  ..., description='A description about the role.', min_length=1
16006
15957
  )
16007
- permission_grants: List[RoleCreateRoleRequestPermissionGrant] = Field(
15958
+ permission_grants: List[RoleModifyRoleRequestPermissionGrant] = Field(
16008
15959
  ...,
16009
15960
  description='A list of permission grants associated with this role.',
16010
15961
  title='Permission Grants',
@@ -16030,6 +15981,29 @@ class RoleGetRolesResponse(BaseModel):
16030
15981
  )
16031
15982
 
16032
15983
 
15984
+ class RoleModifyRoleRequest(BaseModel):
15985
+ description: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString1] = Field(
15986
+ None,
15987
+ description='A description about the role. Only updated if specified. This field is a mutable field.',
15988
+ )
15989
+ permission_grants: Optional[List[RoleModifyRoleRequestPermissionGrant]] = Field(
15990
+ None,
15991
+ description='A list of permission grants associated with this role. Only updated if specified. This field is an immutable field.',
15992
+ title='Permission Grants',
15993
+ )
15994
+ frontend_view: Optional[FrontendView] = Field(
15995
+ None,
15996
+ description='The frontend view for the user of this role. Only updated if specified. This field is an immutable field.',
15997
+ )
15998
+ inherited_from: Optional[Union[RoleModifyRoleRequestInheritedFrom, FieldNotSet]] = (
15999
+ Field(
16000
+ {},
16001
+ description='The ID of the role that this role inherits from. Only updated if specified. This field is an immutable field.',
16002
+ title='Inherited From',
16003
+ )
16004
+ )
16005
+
16006
+
16033
16007
  class ServiceCreateServiceRequest(BaseModel):
16034
16008
  service_hierarchical_state_machine_id: str = Field(
16035
16009
  ...,
@@ -16052,15 +16026,23 @@ class ServiceCreateServiceRequest(BaseModel):
16052
16026
  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.',
16053
16027
  title='Is Active',
16054
16028
  )
16055
- release_version_set: Optional[ServiceCreateServiceRequestVersionSet] = Field(
16056
- None,
16057
- 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.',
16029
+ release_version_set: Optional[ServiceUpsertServiceVersionSetRequestVersionSet] = (
16030
+ Field(
16031
+ None,
16032
+ 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.',
16033
+ )
16058
16034
  )
16059
16035
  tags: Dict[str, Optional[StrippedNonemptyStringWS]] = Field(
16060
16036
  ..., description='The tags of this service.', title='Tags'
16061
16037
  )
16062
16038
 
16063
16039
 
16040
+ class ServiceUpsertServiceVersionSetRequest(BaseModel):
16041
+ version_set: ServiceUpsertServiceVersionSetRequestVersionSet = Field(
16042
+ ..., description='The version set to upsert.'
16043
+ )
16044
+
16045
+
16064
16046
  class SimulationCreateSimulationUnitTestRequest(BaseModel):
16065
16047
  name: str = Field(..., description='The name of the unit test.', min_length=1)
16066
16048
  description: str = Field(
@@ -16194,7 +16176,7 @@ class SimulationUpdateSimulationUnitTestRequest(BaseModel):
16194
16176
  )
16195
16177
 
16196
16178
 
16197
- class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
16179
+ class ToolGetToolInvocationsResponseToolInvocationInstance(BaseModel):
16198
16180
  id: str = Field(..., description='The ID of the tool invocation.', title='Id')
16199
16181
  org_id: str = Field(..., description='The ID of the organization.', title='Org Id')
16200
16182
  created_at: Optional[AwareDatetime] = Field(None, title='Created At')
@@ -16221,6 +16203,34 @@ class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
16221
16203
  )
16222
16204
 
16223
16205
 
16206
+ class ToolSearchToolInvocationsResponse(BaseModel):
16207
+ tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
16208
+ Field(
16209
+ ..., description='The list of tool invocations.', title='Tool Invocations'
16210
+ )
16211
+ )
16212
+
16213
+
16214
+ class UserGetUsersResponse(BaseModel):
16215
+ users: List[UserSearchUsersResponseUserInstance] = Field(
16216
+ ..., description='Users in this organization.', title='Users'
16217
+ )
16218
+ has_more: bool = Field(
16219
+ ..., description='Whether there are more users to retrieve.', title='Has More'
16220
+ )
16221
+ continuation_token: Optional[int] = Field(
16222
+ ...,
16223
+ description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
16224
+ title='Continuation Token',
16225
+ )
16226
+
16227
+
16228
+ class UserSearchUsersResponse(BaseModel):
16229
+ users: List[UserSearchUsersResponseUserInstance] = Field(
16230
+ ..., description='Users in this organization.', title='Users'
16231
+ )
16232
+
16233
+
16224
16234
  class SimulationUnitTestSetRunInstance(BaseModel):
16225
16235
  org_id: str = Field(..., title='Org Id')
16226
16236
  id: str = Field(..., title='Id')
@@ -16320,7 +16330,7 @@ class SimulationGetSimulationUnitTestSetRunsResponse(BaseModel):
16320
16330
 
16321
16331
 
16322
16332
  class ToolGetToolInvocationsResponse(BaseModel):
16323
- tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
16333
+ tool_invocations: List[ToolGetToolInvocationsResponseToolInvocationInstance] = (
16324
16334
  Field(
16325
16335
  ..., description='The list of tool invocations.', title='Tool Invocations'
16326
16336
  )
@@ -16335,11 +16345,3 @@ class ToolGetToolInvocationsResponse(BaseModel):
16335
16345
  description='The continuation token to retrieve the next page of tool invocations, or null if there are no more tool invocations.',
16336
16346
  title='Continuation Token',
16337
16347
  )
16338
-
16339
-
16340
- class ToolSearchToolInvocationsResponse(BaseModel):
16341
- tool_invocations: List[ToolSearchToolInvocationsResponseToolInvocationInstance] = (
16342
- Field(
16343
- ..., description='The list of tool invocations.', title='Tool Invocations'
16344
- )
16345
- )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: amigo_sdk
3
- Version: 0.69.0
3
+ Version: 0.71.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=f92ou06jG_dF-bjS0-u3e5y3NCawwALx_NrF_xmcEnw,139
1
+ amigo_sdk/__init__.py,sha256=HDbivAb29t5GGkU9dPkkh0GQ6qDToJIqNdRgkpdSMYo,139
2
2
  amigo_sdk/_retry_utils.py,sha256=VbA_Uwv268J80d11R9XfnCjZiV-NwE1aXRsSlRjjRio,2108
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=Yn12AxIQ6zE9jnGLWO0MUVbBsmw77Kzh82AijyreOkA,4818
6
6
  amigo_sdk/http_client.py,sha256=tfMXHqMTmlz2ja7aJhOVgbwAiJNlKCzPcK4UGNqXHO8,13488
7
7
  amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
8
8
  amigo_sdk/sdk_client.py,sha256=5nj8hzjHSFRs9JMNabdYIhQAcFE2G1jI_r7SHVGr9No,6106
9
- amigo_sdk/generated/model.py,sha256=D0il2vJlghD_VpR9544TJLTunouNRIshwInQV9jNFBI,456861
9
+ amigo_sdk/generated/model.py,sha256=AMOa2wehaiY0um98EUEV_thJJXDBM1em6PK4_MjFoAk,456924
10
10
  amigo_sdk/resources/conversation.py,sha256=u4kOx5LPbhX92L41UnmPmaF9TkQA5bGev8N00dMoP9I,14805
11
11
  amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
12
12
  amigo_sdk/resources/service.py,sha256=t1iA3nS9co-wuR-x5jBhAXXTWfMeGsLwfQcLycwVrCA,1536
13
13
  amigo_sdk/resources/user.py,sha256=zikijiuGXgmkBVtrldor6XQGQK6jWXonDvNFuQs6XK4,3472
14
- amigo_sdk-0.69.0.dist-info/METADATA,sha256=J5S8JJ1vkUxh-b00MjaGH_60uwEA6lS7Ies9BFf-kRM,8499
15
- amigo_sdk-0.69.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
16
- amigo_sdk-0.69.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
- amigo_sdk-0.69.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
- amigo_sdk-0.69.0.dist-info/RECORD,,
14
+ amigo_sdk-0.71.0.dist-info/METADATA,sha256=sbLqyfisJQa45uQja9_jnH95bR46KVTu2DQ5JrM1TU0,8499
15
+ amigo_sdk-0.71.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
16
+ amigo_sdk-0.71.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
17
+ amigo_sdk-0.71.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
18
+ amigo_sdk-0.71.0.dist-info/RECORD,,