amigo_sdk 0.103.0__py3-none-any.whl → 0.104.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.103.0"
1
+ __version__ = "0.104.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: 2026-01-11T23:34:54+00:00
3
+ # timestamp: 2026-01-12T16:41:39+00:00
4
4
 
5
5
  from __future__ import annotations
6
6
 
@@ -1397,6 +1397,11 @@ class Action(Enum):
1397
1397
  deny = 'Deny'
1398
1398
 
1399
1399
 
1400
+ class MongoCollectionsServiceServiceTag(BaseModel):
1401
+ key: str = Field(..., title='Key')
1402
+ value: str | None = Field(..., title='Value')
1403
+
1404
+
1400
1405
  class MongoCollectionsServiceServiceVersionSet(BaseModel):
1401
1406
  agent_version_number: int | None = Field(
1402
1407
  ...,
@@ -1497,11 +1502,6 @@ class MongoCollectionsSimulationUnitTestSetRunSimulationUnitTestSetRunUserInfo(
1497
1502
  pass
1498
1503
 
1499
1504
 
1500
- class MongoCollectionsToolToolTag(BaseModel):
1501
- key: str = Field(..., title='Key')
1502
- value: str | None = Field(..., title='Value')
1503
-
1504
-
1505
1505
  class MongoCollectionsToolInvocationToolInvocationVersion(BaseModel):
1506
1506
  major: int = Field(..., title='Major')
1507
1507
  minor: int = Field(..., title='Minor')
@@ -1641,6 +1641,7 @@ class ConversationGenerateConversationStarterRequest(BaseModel):
1641
1641
  facets: list[PydanticBaseModelStrippedNonemptyString1] = Field(
1642
1642
  ...,
1643
1643
  description='The facets that the generated conversation starter should be about. Each generated starter is related to some of these facets.',
1644
+ min_length=1,
1644
1645
  title='Facets',
1645
1646
  )
1646
1647
  min_count: int = Field(
@@ -1795,6 +1796,20 @@ class DynamicBehaviorSetGetDynamicBehaviorInvocationsResponse(BaseModel):
1795
1796
  )
1796
1797
 
1797
1798
 
1799
+ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
1800
+ BaseModel
1801
+ ):
1802
+ id: str = Field(..., title='Id')
1803
+ name: str = Field(..., title='Name')
1804
+ is_active: bool = Field(..., title='Is Active')
1805
+ applied_to_services: list[str] = Field(..., title='Applied To Services')
1806
+ tags: dict[str, str | None] = Field(..., title='Tags')
1807
+ created_at: AwareDatetime = Field(..., title='Created At')
1808
+ creator: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1809
+ updated_at: AwareDatetime = Field(..., title='Updated At')
1810
+ updated_by: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1811
+
1812
+
1798
1813
  class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
1799
1814
  applied_to_services_ids: list[str] = Field(
1800
1815
  ...,
@@ -1815,18 +1830,14 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponseFilterValues(BaseModel):
1815
1830
  )
1816
1831
 
1817
1832
 
1818
- class DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance(
1819
- BaseModel
1820
- ):
1821
- id: str = Field(..., title='Id')
1822
- name: str = Field(..., title='Name')
1823
- is_active: bool = Field(..., title='Is Active')
1824
- applied_to_services: list[str] = Field(..., title='Applied To Services')
1825
- tags: dict[str, str | None] = Field(..., title='Tags')
1826
- created_at: AwareDatetime = Field(..., title='Created At')
1827
- creator: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1828
- updated_at: AwareDatetime = Field(..., title='Updated At')
1829
- updated_by: MongoCollectionsDynamicBehaviorSetDynamicBehaviorSetUserInfo
1833
+ class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
1834
+ dynamic_behavior_sets: list[
1835
+ DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
1836
+ ] = Field(
1837
+ ...,
1838
+ description='The retrieved dynamic behavior sets.',
1839
+ title='Dynamic Behavior Sets',
1840
+ )
1830
1841
 
1831
1842
 
1832
1843
  class DynamicBehaviorSetUpdateDynamicBehaviorSetRequest(BaseModel):
@@ -10565,6 +10576,11 @@ class OrganizationCreateOrganizationRequestPreferences(BaseModel):
10565
10576
  )
10566
10577
 
10567
10578
 
10579
+ class OrganizationCreateOrganizationRequestUserDimension(BaseModel):
10580
+ description: str = Field(..., min_length=1)
10581
+ tags: list[PydanticBaseModelStrippedNonemptyString1] = Field(..., title='Tags')
10582
+
10583
+
10568
10584
  class OrganizationCreateOrganizationResponse(BaseModel):
10569
10585
  org_id: str = Field(
10570
10586
  ...,
@@ -10739,17 +10755,23 @@ class OrganizationGetUserDimensionsResponse(BaseModel):
10739
10755
  )
10740
10756
 
10741
10757
 
10758
+ class UserDimensions(
10759
+ RootModel[list[OrganizationCreateOrganizationRequestUserDimension]]
10760
+ ):
10761
+ root: list[OrganizationCreateOrganizationRequestUserDimension] = Field(
10762
+ ...,
10763
+ description='User dimensions for the organization. If not set or `null`, this field is not updated.',
10764
+ min_length=1,
10765
+ title='User Dimensions',
10766
+ )
10767
+
10768
+
10742
10769
  class OrganizationModifyOrganizationRequestPreferences(
10743
10770
  OrganizationCreateOrganizationRequestPreferences
10744
10771
  ):
10745
10772
  pass
10746
10773
 
10747
10774
 
10748
- class OrganizationModifyOrganizationRequestUserDimension(BaseModel):
10749
- description: str = Field(..., min_length=1)
10750
- tags: list[PydanticBaseModelStrippedNonemptyString1] = Field(..., title='Tags')
10751
-
10752
-
10753
10775
  class RoleAssignRoleRequest(BaseModel):
10754
10776
  user_id: str = Field(
10755
10777
  ...,
@@ -10778,13 +10800,6 @@ class RoleModifyRoleResponse(BaseModel):
10778
10800
  )
10779
10801
 
10780
10802
 
10781
- class ServiceCreateServiceRequestVersionSetLLMConfig(BaseModel):
10782
- llm_name: LLMType
10783
- params: dict[str, Any] | None = Field(
10784
- {}, description='LLM-specific parameters to use.', title='Params'
10785
- )
10786
-
10787
-
10788
10803
  class ServiceCreateServiceResponse(BaseModel):
10789
10804
  id: str = Field(..., description='The ID of the newly created service.', title='Id')
10790
10805
 
@@ -10860,19 +10875,10 @@ class ServiceUpdateServiceRequest(BaseModel):
10860
10875
  )
10861
10876
 
10862
10877
 
10863
- class ServiceUpsertServiceVersionSetRequestVersionSet(BaseModel):
10864
- agent_version_number: int | None = Field(
10865
- ...,
10866
- description='The version number of the agent to be used. If None, the latest agent version will be used.',
10867
- title='Agent Version Number',
10868
- )
10869
- service_hierarchical_state_machine_version_number: int | None = Field(
10870
- ...,
10871
- description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
10872
- title='Service Hierarchical State Machine Version Number',
10873
- )
10874
- llm_model_preferences: dict[str, ServiceCreateServiceRequestVersionSetLLMConfig] = (
10875
- Field(..., title='Llm Model Preferences')
10878
+ class ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig(BaseModel):
10879
+ llm_name: LLMType
10880
+ params: dict[str, Any] | None = Field(
10881
+ {}, description='LLM-specific parameters to use.', title='Params'
10876
10882
  )
10877
10883
 
10878
10884
 
@@ -10981,6 +10987,16 @@ class SimulationCreateSimulationUnitTestResponse(BaseModel):
10981
10987
  )
10982
10988
 
10983
10989
 
10990
+ class SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
10991
+ unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
10992
+ run_count: int = Field(
10993
+ ...,
10994
+ description='The number of times to run the unit test.',
10995
+ gt=0,
10996
+ title='Run Count',
10997
+ )
10998
+
10999
+
10984
11000
  class SimulationCreateSimulationUnitTestSetResponse(BaseModel):
10985
11001
  simulation_unit_test_set_id: str = Field(
10986
11002
  ...,
@@ -11069,6 +11085,37 @@ class SimulationGetSimulationScenariosResponseFilterValues(BaseModel):
11069
11085
  )
11070
11086
 
11071
11087
 
11088
+ class SimulationGetSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
11089
+ id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
11090
+ name: str = Field(
11091
+ ..., description='The name of the simulation scenario.', title='Name'
11092
+ )
11093
+ tags: dict[str, str | None] = Field(
11094
+ ..., description='The tags of the simulation scenario.', title='Tags'
11095
+ )
11096
+ is_deleted: bool = Field(
11097
+ ...,
11098
+ description='Whether the simulation scenario is deleted.',
11099
+ title='Is Deleted',
11100
+ )
11101
+ created_at: AwareDatetime = Field(
11102
+ ...,
11103
+ description='The timestamp when the simulation scenario was created.',
11104
+ title='Created At',
11105
+ )
11106
+ creator: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = Field(
11107
+ ..., description='The user who created the simulation scenario.'
11108
+ )
11109
+ updated_at: AwareDatetime = Field(
11110
+ ...,
11111
+ description='The timestamp when the simulation scenario was last updated.',
11112
+ title='Updated At',
11113
+ )
11114
+ updated_by: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = Field(
11115
+ ..., description='The user who last updated the simulation scenario.'
11116
+ )
11117
+
11118
+
11072
11119
  class SimulationGetSimulationUnitTestSetRunArtifactsResponse(BaseModel):
11073
11120
  presigned_url: str = Field(
11074
11121
  ...,
@@ -11191,34 +11238,13 @@ class SimulationSearchSimulationPersonasResponseSimulationPersonaInstance(BaseMo
11191
11238
  )
11192
11239
 
11193
11240
 
11194
- class SimulationSearchSimulationScenariosResponseSimulationScenarioInstance(BaseModel):
11195
- id: str = Field(..., description='The ID of the simulation scenario.', title='Id')
11196
- name: str = Field(
11197
- ..., description='The name of the simulation scenario.', title='Name'
11198
- )
11199
- tags: dict[str, str | None] = Field(
11200
- ..., description='The tags of the simulation scenario.', title='Tags'
11201
- )
11202
- is_deleted: bool = Field(
11203
- ...,
11204
- description='Whether the simulation scenario is deleted.',
11205
- title='Is Deleted',
11206
- )
11207
- created_at: AwareDatetime = Field(
11208
- ...,
11209
- description='The timestamp when the simulation scenario was created.',
11210
- title='Created At',
11211
- )
11212
- creator: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = Field(
11213
- ..., description='The user who created the simulation scenario.'
11214
- )
11215
- updated_at: AwareDatetime = Field(
11241
+ class SimulationSearchSimulationScenariosResponse(BaseModel):
11242
+ simulation_scenarios: list[
11243
+ SimulationGetSimulationScenariosResponseSimulationScenarioInstance
11244
+ ] = Field(
11216
11245
  ...,
11217
- description='The timestamp when the simulation scenario was last updated.',
11218
- title='Updated At',
11219
- )
11220
- updated_by: MongoCollectionsSimulationScenarioSimulationScenarioUserInfo = Field(
11221
- ..., description='The user who last updated the simulation scenario.'
11246
+ description='The list of simulation scenarios.',
11247
+ title='Simulation Scenarios',
11222
11248
  )
11223
11249
 
11224
11250
 
@@ -11284,13 +11310,35 @@ class RunCount(RootModel[int]):
11284
11310
  )
11285
11311
 
11286
11312
 
11287
- class SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor(BaseModel):
11288
- unit_test_id: str = Field(..., pattern='^[a-f0-9]{24}$', title='Unit Test Id')
11289
- run_count: int = Field(
11290
- ...,
11291
- description='The number of times to run the unit test.',
11292
- gt=0,
11293
- title='Run Count',
11313
+ class UnitTestRuns(
11314
+ RootModel[list[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
11315
+ ):
11316
+ root: list[SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
11317
+ Field(
11318
+ ...,
11319
+ description='The unit test runs that are part of this set.',
11320
+ min_length=1,
11321
+ title='Unit Test Runs',
11322
+ )
11323
+ )
11324
+
11325
+
11326
+ class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
11327
+ name: PydanticBaseModelStrippedNonemptyString1 | None = Field(
11328
+ None, description='The name of the simulation unit test set.'
11329
+ )
11330
+ description: PydanticBaseModelStrippedNonemptyString1 | FieldNotSet | None = Field(
11331
+ default_factory=lambda: PydanticBaseModelStrippedNonemptyString1({}),
11332
+ description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
11333
+ title='Description',
11334
+ )
11335
+ unit_test_runs: UnitTestRuns | None = Field(
11336
+ None,
11337
+ description='The unit test runs that are part of this set.',
11338
+ title='Unit Test Runs',
11339
+ )
11340
+ tags: dict[str, str | None] | None = Field(
11341
+ None, description='The tags of the simulation unit test set.', title='Tags'
11294
11342
  )
11295
11343
 
11296
11344
 
@@ -11327,6 +11375,22 @@ class ToolGetToolsResponseFilterValues(BaseModel):
11327
11375
  )
11328
11376
 
11329
11377
 
11378
+ class InvocationMode(Enum):
11379
+ regular = 'regular'
11380
+ conversation_simulation = 'conversation-simulation'
11381
+
11382
+
11383
+ class ToolInvokeToolVersionRequestInput(BaseModel):
11384
+ input_parameters: dict[str, Any] = Field(
11385
+ ...,
11386
+ description='The input parameters for the tool invocation.',
11387
+ title='Input Parameters',
11388
+ )
11389
+ invocation_mode: InvocationMode = Field(
11390
+ ..., description='The mode of invocation.', title='Invocation Mode'
11391
+ )
11392
+
11393
+
11330
11394
  class ToolModifyToolRequest(BaseModel):
11331
11395
  description: PydanticBaseModelStrippedNonemptyString1 | None = Field(
11332
11396
  None, description='A description of this tool. Only updates if not-null.'
@@ -11371,19 +11435,34 @@ class ToolPublishToolVersionResponse(BaseModel):
11371
11435
  )
11372
11436
 
11373
11437
 
11374
- class InvocationMode(Enum):
11375
- regular = 'regular'
11376
- conversation_simulation = 'conversation-simulation'
11377
-
11378
-
11379
- class ToolTestToolRequestInput(BaseModel):
11380
- input_parameters: dict[str, Any] = Field(
11438
+ class ToolTestToolRequest(BaseModel):
11439
+ inputs: list[ToolInvokeToolVersionRequestInput] = Field(
11381
11440
  ...,
11382
- description='The input parameters for the tool invocation.',
11383
- title='Input Parameters',
11441
+ description='The list of inputs for the tool invocation.',
11442
+ max_length=10,
11443
+ min_length=1,
11444
+ title='Inputs',
11384
11445
  )
11385
- invocation_mode: InvocationMode = Field(
11386
- ..., description='The mode of invocation.', title='Invocation Mode'
11446
+ commit_branch: str = Field(
11447
+ ...,
11448
+ description='The branch in the tools repo whose tip will be tested.',
11449
+ title='Commit Branch',
11450
+ )
11451
+ project_path: str = Field(
11452
+ ...,
11453
+ description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
11454
+ pattern='^[-\\w\\d_/]+$',
11455
+ title='Project Path',
11456
+ )
11457
+ required_envvars: dict[str, str] = Field(
11458
+ ...,
11459
+ description='The environment variables required for the tool to run.',
11460
+ title='Required Envvars',
11461
+ )
11462
+ required_secret_envvars: dict[str, str] = Field(
11463
+ ...,
11464
+ description='The secret environment variables required for the tool to run.',
11465
+ title='Required Secret Envvars',
11387
11466
  )
11388
11467
 
11389
11468
 
@@ -11453,7 +11532,7 @@ class UserGetMemoriesResponse(BaseModel):
11453
11532
  )
11454
11533
 
11455
11534
 
11456
- class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11535
+ class UserSearchUsersResponseUserInstanceUserStats(BaseModel):
11457
11536
  num_conversations: int = Field(
11458
11537
  ...,
11459
11538
  description='The number of conversations the user has created.',
@@ -11471,31 +11550,6 @@ class UserGetUsersResponseUserInstanceUserStats(BaseModel):
11471
11550
  )
11472
11551
 
11473
11552
 
11474
- class UserSearchUsersResponseUserInstance(BaseModel):
11475
- org_id: str = Field(
11476
- ...,
11477
- description='The ID of the organization that this user belongs to.',
11478
- title='Org Id',
11479
- )
11480
- user_id: str = Field(
11481
- ..., description='The identifier of the user.', title='User Id'
11482
- )
11483
- first_name: str = Field(
11484
- ..., description='The first name of the user.', title='First Name'
11485
- )
11486
- last_name: str = Field(
11487
- ..., description='The last name of the user.', title='Last Name'
11488
- )
11489
- email: str = Field(..., description='Email of the user.', title='Email')
11490
- user_stats: UserGetUsersResponseUserInstanceUserStats = Field(
11491
- ..., description="Statistics about the user's usage of the Amigo platform."
11492
- )
11493
- role: str = Field(..., description='The ID of the role of the user.', title='Role')
11494
- preferences: MongoCollectionsUserUserPreferences = Field(
11495
- ..., description='The preferences of the user.'
11496
- )
11497
-
11498
-
11499
11553
  class UserSignInWithApiKeyResponse(BaseModel):
11500
11554
  id_token: str = Field(
11501
11555
  ...,
@@ -11905,7 +11959,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
11905
11959
  title='Id',
11906
11960
  )
11907
11961
  message_type: list[MessageType] | None = Field(
11908
- ['user-message', 'external-event', 'agent-message'],
11962
+ ['agent-message', 'user-message', 'external-event'],
11909
11963
  description='The type of messages to retrieve.',
11910
11964
  title='Message Type',
11911
11965
  )
@@ -13568,7 +13622,7 @@ class ServiceInstance(BaseModel):
13568
13622
  agent_id: str = Field(
13569
13623
  ..., description='The ID of the agent that this service uses.', title='Agent Id'
13570
13624
  )
13571
- tags: list[MongoCollectionsToolToolTag] = Field(
13625
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
13572
13626
  ..., description='The tags of the service.', title='Tags'
13573
13627
  )
13574
13628
 
@@ -13621,7 +13675,7 @@ class SimulationUnitTestSet(BaseModel):
13621
13675
  description='Whether the unit test set has been deleted.',
13622
13676
  title='Is Deleted',
13623
13677
  )
13624
- tags: list[MongoCollectionsToolToolTag] = Field(
13678
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
13625
13679
  ..., description='The tags of the simulation unit test set.', title='Tags'
13626
13680
  )
13627
13681
  creator: MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13658,7 +13712,7 @@ class SimulationUnitTestSetInstance(BaseModel):
13658
13712
  description='Whether the unit test set has been deleted.',
13659
13713
  title='Is Deleted',
13660
13714
  )
13661
- tags: list[MongoCollectionsToolToolTag] = Field(
13715
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
13662
13716
  ..., description='The tags of the simulation unit test set.', title='Tags'
13663
13717
  )
13664
13718
  creator: MongoCollectionsSimulationUnitTestSetSimulationUnitTestSetUserInfo
@@ -13786,7 +13840,7 @@ class ToolInstance(BaseModel):
13786
13840
  description='A list of encrypted environment variables that some versions of this tool uses.',
13787
13841
  title='Secret Envvars',
13788
13842
  )
13789
- tags: list[MongoCollectionsToolToolTag] = Field(
13843
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
13790
13844
  ..., description='The tags of the simulation persona.', title='Tags'
13791
13845
  )
13792
13846
 
@@ -14081,7 +14135,7 @@ class DynamicBehaviorSetCreateDynamicBehaviorSetVersionRequestChangeToolCandidat
14081
14135
 
14082
14136
  class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
14083
14137
  dynamic_behavior_sets: list[
14084
- DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14138
+ DynamicBehaviorSetGetDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14085
14139
  ] = Field(
14086
14140
  ...,
14087
14141
  description='The retrieved dynamic behavior sets.',
@@ -14105,16 +14159,6 @@ class DynamicBehaviorSetGetDynamicBehaviorSetsResponse(BaseModel):
14105
14159
  )
14106
14160
 
14107
14161
 
14108
- class DynamicBehaviorSetSearchDynamicBehaviorSetsResponse(BaseModel):
14109
- dynamic_behavior_sets: list[
14110
- DynamicBehaviorSetSearchDynamicBehaviorSetsResponseDynamicBehaviorSetInstance
14111
- ] = Field(
14112
- ...,
14113
- description='The retrieved dynamic behavior sets.',
14114
- title='Dynamic Behavior Sets',
14115
- )
14116
-
14117
-
14118
14162
  class MetricCreateMetricRequest(BaseModel):
14119
14163
  name: str = Field(
14120
14164
  ...,
@@ -14171,7 +14215,7 @@ class OrganizationCreateOrganizationRequest(BaseModel):
14171
14215
  description='Additional descriptions of the services offered by this organization that is displayed below `main_description` in a smaller font.',
14172
14216
  min_length=1,
14173
14217
  )
14174
- user_dimensions: list[OrganizationModifyOrganizationRequestUserDimension] = Field(
14218
+ user_dimensions: list[OrganizationCreateOrganizationRequestUserDimension] = Field(
14175
14219
  ...,
14176
14220
  description='User dimensions for the organization.',
14177
14221
  title='User Dimensions',
@@ -14341,17 +14385,6 @@ class OrganizationGetApiKeysResponse(BaseModel):
14341
14385
  )
14342
14386
 
14343
14387
 
14344
- class UserDimensions(
14345
- RootModel[list[OrganizationModifyOrganizationRequestUserDimension]]
14346
- ):
14347
- root: list[OrganizationModifyOrganizationRequestUserDimension] = Field(
14348
- ...,
14349
- description='User dimensions for the organization. If not set or `null`, this field is not updated.',
14350
- min_length=1,
14351
- title='User Dimensions',
14352
- )
14353
-
14354
-
14355
14388
  class OrganizationModifyOrganizationRequest(BaseModel):
14356
14389
  title: PydanticBaseModelStrippedNonemptyString1 | None = Field(
14357
14390
  None,
@@ -14436,41 +14469,20 @@ class RoleModifyRoleRequest(BaseModel):
14436
14469
  )
14437
14470
 
14438
14471
 
14439
- class ServiceCreateServiceRequest(BaseModel):
14440
- service_hierarchical_state_machine_id: str = Field(
14441
- ...,
14442
- description='The ID of the state machine that this service uses.',
14443
- pattern='^[a-f0-9]{24}$',
14444
- title='Service Hierarchical State Machine Id',
14445
- )
14446
- agent_id: str = Field(
14472
+ class ServiceCreateServiceRequestVersionSet(BaseModel):
14473
+ agent_version_number: int | None = Field(
14447
14474
  ...,
14448
- description='The ID of the agent that this service uses.',
14449
- pattern='^[a-f0-9]{24}$',
14450
- title='Agent Id',
14451
- )
14452
- name: str = Field(..., description='The name of this service.', min_length=1)
14453
- description: str = Field(
14454
- ..., description='A description of this service.', min_length=1
14475
+ description='The version number of the agent to be used. If None, the latest agent version will be used.',
14476
+ title='Agent Version Number',
14455
14477
  )
14456
- is_active: bool = Field(
14478
+ service_hierarchical_state_machine_version_number: int | None = Field(
14457
14479
  ...,
14458
- 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.',
14459
- title='Is Active',
14460
- )
14461
- release_version_set: ServiceUpsertServiceVersionSetRequestVersionSet | None = Field(
14462
- None,
14463
- 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.',
14464
- )
14465
- keyterms: list[StrippedNonemptyStringAZAZ02] | None = Field(
14466
- default_factory=list,
14467
- description='A list of keyterms that are easy to get wrong during audio transcriptions that tend to occur commonly in audio sessions using this service.',
14468
- max_length=20,
14469
- title='Keyterms',
14470
- )
14471
- tags: dict[StrippedNonemptyStringWS, StrippedNonemptyStringWS | None] = Field(
14472
- ..., description='The tags of this service.', title='Tags'
14480
+ description='The version number of the state machine to be used. If None, the latest state machine version will be used.',
14481
+ title='Service Hierarchical State Machine Version Number',
14473
14482
  )
14483
+ llm_model_preferences: dict[
14484
+ str, ServiceUpsertServiceVersionSetRequestVersionSetLLMConfig
14485
+ ] = Field(..., title='Llm Model Preferences')
14474
14486
 
14475
14487
 
14476
14488
  class ServiceGetServicesResponse(BaseModel):
@@ -14494,7 +14506,7 @@ class ServiceGetServicesResponse(BaseModel):
14494
14506
 
14495
14507
 
14496
14508
  class ServiceUpsertServiceVersionSetRequest(BaseModel):
14497
- version_set: ServiceUpsertServiceVersionSetRequestVersionSet = Field(
14509
+ version_set: ServiceCreateServiceRequestVersionSet = Field(
14498
14510
  ..., description='The version set to upsert.'
14499
14511
  )
14500
14512
 
@@ -14548,7 +14560,7 @@ class SimulationCreateSimulationUnitTestSetRequest(BaseModel):
14548
14560
  ..., description='The description of the simulation unit test set.'
14549
14561
  )
14550
14562
  unit_test_runs: list[
14551
- SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor
14563
+ SimulationCreateSimulationUnitTestSetRequestUnitTestRunDescriptor
14552
14564
  ] = Field(
14553
14565
  ...,
14554
14566
  description='The unit test runs that are part of this set.',
@@ -14584,7 +14596,7 @@ class SimulationGetSimulationPersonasResponse(BaseModel):
14584
14596
 
14585
14597
  class SimulationGetSimulationScenariosResponse(BaseModel):
14586
14598
  simulation_scenarios: list[
14587
- SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
14599
+ SimulationGetSimulationScenariosResponseSimulationScenarioInstance
14588
14600
  ] = Field(
14589
14601
  ...,
14590
14602
  description='The list of simulation scenarios.',
@@ -14638,16 +14650,6 @@ class SimulationSearchSimulationPersonasResponse(BaseModel):
14638
14650
  )
14639
14651
 
14640
14652
 
14641
- class SimulationSearchSimulationScenariosResponse(BaseModel):
14642
- simulation_scenarios: list[
14643
- SimulationSearchSimulationScenariosResponseSimulationScenarioInstance
14644
- ] = Field(
14645
- ...,
14646
- description='The list of simulation scenarios.',
14647
- title='Simulation Scenarios',
14648
- )
14649
-
14650
-
14651
14653
  class SimulationSearchSimulationUnitTestSetResponse(BaseModel):
14652
14654
  simulation_unit_test_sets: list[SimulationUnitTestSet] = Field(
14653
14655
  ...,
@@ -14662,38 +14664,6 @@ class SimulationUpdateSimulationUnitTestRequestSuccessCriterion(
14662
14664
  pass
14663
14665
 
14664
14666
 
14665
- class UnitTestRuns(
14666
- RootModel[list[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor]]
14667
- ):
14668
- root: list[SimulationUpdateSimulationUnitTestSetRequestUnitTestRunDescriptor] = (
14669
- Field(
14670
- ...,
14671
- description='The unit test runs that are part of this set.',
14672
- min_length=1,
14673
- title='Unit Test Runs',
14674
- )
14675
- )
14676
-
14677
-
14678
- class SimulationUpdateSimulationUnitTestSetRequest(BaseModel):
14679
- name: PydanticBaseModelStrippedNonemptyString1 | None = Field(
14680
- None, description='The name of the simulation unit test set.'
14681
- )
14682
- description: PydanticBaseModelStrippedNonemptyString1 | FieldNotSet | None = Field(
14683
- default_factory=lambda: PydanticBaseModelStrippedNonemptyString1({}),
14684
- description='The description of the simulation unit test set. If not specified or an empty object is provided, the description will not be updated.',
14685
- title='Description',
14686
- )
14687
- unit_test_runs: UnitTestRuns | None = Field(
14688
- None,
14689
- description='The unit test runs that are part of this set.',
14690
- title='Unit Test Runs',
14691
- )
14692
- tags: dict[str, str | None] | None = Field(
14693
- None, description='The tags of the simulation unit test set.', title='Tags'
14694
- )
14695
-
14696
-
14697
14667
  class ToolGetToolVersionsResponse(BaseModel):
14698
14668
  tool_versions: list[ToolVersionInstance] = Field(
14699
14669
  ...,
@@ -14735,7 +14705,7 @@ class ToolGetToolsResponse(BaseModel):
14735
14705
 
14736
14706
 
14737
14707
  class ToolInvokeToolVersionRequest(BaseModel):
14738
- inputs: list[ToolTestToolRequestInput] = Field(
14708
+ inputs: list[ToolInvokeToolVersionRequestInput] = Field(
14739
14709
  ...,
14740
14710
  description='The list of inputs for the tool invocation.',
14741
14711
  max_length=10,
@@ -14759,37 +14729,6 @@ class ToolModifyToolEnvvarsRequest(BaseModel):
14759
14729
  deletes: list[DeleteRequest] | None = Field(default_factory=list, title='Deletes')
14760
14730
 
14761
14731
 
14762
- class ToolTestToolRequest(BaseModel):
14763
- inputs: list[ToolTestToolRequestInput] = Field(
14764
- ...,
14765
- description='The list of inputs for the tool invocation.',
14766
- max_length=10,
14767
- min_length=1,
14768
- title='Inputs',
14769
- )
14770
- commit_branch: str = Field(
14771
- ...,
14772
- description='The branch in the tools repo whose tip will be tested.',
14773
- title='Commit Branch',
14774
- )
14775
- project_path: str = Field(
14776
- ...,
14777
- description='The directory in the tools repo that contains the `pyproject.toml` file corresponding to the tool version to publish.',
14778
- pattern='^[-\\w\\d_/]+$',
14779
- title='Project Path',
14780
- )
14781
- required_envvars: dict[str, str] = Field(
14782
- ...,
14783
- description='The environment variables required for the tool to run.',
14784
- title='Required Envvars',
14785
- )
14786
- required_secret_envvars: dict[str, str] = Field(
14787
- ...,
14788
- description='The secret environment variables required for the tool to run.',
14789
- title='Required Secret Envvars',
14790
- )
14791
-
14792
-
14793
14732
  class ToolTestToolResponse(ToolInvokeToolVersionResponse):
14794
14733
  pass
14795
14734
 
@@ -14831,23 +14770,28 @@ class UserGetUserModelResponse(BaseModel):
14831
14770
  )
14832
14771
 
14833
14772
 
14834
- class UserGetUsersResponse(BaseModel):
14835
- users: list[UserSearchUsersResponseUserInstance] = Field(
14836
- ..., description='Users in this organization.', title='Users'
14773
+ class UserSearchUsersResponseUserInstance(BaseModel):
14774
+ org_id: str = Field(
14775
+ ...,
14776
+ description='The ID of the organization that this user belongs to.',
14777
+ title='Org Id',
14837
14778
  )
14838
- has_more: bool = Field(
14839
- ..., description='Whether there are more users to retrieve.', title='Has More'
14779
+ user_id: str = Field(
14780
+ ..., description='The identifier of the user.', title='User Id'
14840
14781
  )
14841
- continuation_token: int | None = Field(
14842
- ...,
14843
- description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
14844
- title='Continuation Token',
14782
+ first_name: str = Field(
14783
+ ..., description='The first name of the user.', title='First Name'
14845
14784
  )
14846
-
14847
-
14848
- class UserSearchUsersResponse(BaseModel):
14849
- users: list[UserSearchUsersResponseUserInstance] = Field(
14850
- ..., description='Users in this organization.', title='Users'
14785
+ last_name: str = Field(
14786
+ ..., description='The last name of the user.', title='Last Name'
14787
+ )
14788
+ email: str = Field(..., description='Email of the user.', title='Email')
14789
+ user_stats: UserSearchUsersResponseUserInstanceUserStats = Field(
14790
+ ..., description="Statistics about the user's usage of the Amigo platform."
14791
+ )
14792
+ role: str = Field(..., description='The ID of the role of the user.', title='Role')
14793
+ preferences: MongoCollectionsUserUserPreferences = Field(
14794
+ ..., description='The preferences of the user.'
14851
14795
  )
14852
14796
 
14853
14797
 
@@ -14986,7 +14930,7 @@ class Metric(BaseModel):
14986
14930
  additional_notes: str | None = Field(
14987
14931
  ..., description='Additional notes about the metric.', title='Additional Notes'
14988
14932
  )
14989
- tags: list[MongoCollectionsToolToolTag] = Field(
14933
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
14990
14934
  ..., description='The tags of the metric.', title='Tags'
14991
14935
  )
14992
14936
  creator: MongoCollectionsMetricMetricUserInfo = Field(
@@ -15059,7 +15003,7 @@ class MetricInstance(BaseModel):
15059
15003
  additional_notes: str | None = Field(
15060
15004
  ..., description='Additional notes about the metric.', title='Additional Notes'
15061
15005
  )
15062
- tags: list[MongoCollectionsToolToolTag] = Field(
15006
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
15063
15007
  ..., description='The tags of the metric.', title='Tags'
15064
15008
  )
15065
15009
  creator: MongoCollectionsMetricMetricUserInfo = Field(
@@ -15155,7 +15099,7 @@ class SimulationUnitTest(BaseModel):
15155
15099
  is_deleted: bool = Field(
15156
15100
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15157
15101
  )
15158
- tags: list[MongoCollectionsToolToolTag] = Field(
15102
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
15159
15103
  ..., description='The tags of the simulation persona.', title='Tags'
15160
15104
  )
15161
15105
  creator: MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15200,7 +15144,7 @@ class SimulationUnitTestInstance(BaseModel):
15200
15144
  is_deleted: bool = Field(
15201
15145
  ..., description='Whether the unit test has been deleted.', title='Is Deleted'
15202
15146
  )
15203
- tags: list[MongoCollectionsToolToolTag] = Field(
15147
+ tags: list[MongoCollectionsServiceServiceTag] = Field(
15204
15148
  ..., description='The tags of the simulation persona.', title='Tags'
15205
15149
  )
15206
15150
  creator: MongoCollectionsSimulationUnitTestSimulationUnitTestUserInfo
@@ -15468,6 +15412,43 @@ class RoleGetRolesResponse(BaseModel):
15468
15412
  )
15469
15413
 
15470
15414
 
15415
+ class ServiceCreateServiceRequest(BaseModel):
15416
+ service_hierarchical_state_machine_id: str = Field(
15417
+ ...,
15418
+ description='The ID of the state machine that this service uses.',
15419
+ pattern='^[a-f0-9]{24}$',
15420
+ title='Service Hierarchical State Machine Id',
15421
+ )
15422
+ agent_id: str = Field(
15423
+ ...,
15424
+ description='The ID of the agent that this service uses.',
15425
+ pattern='^[a-f0-9]{24}$',
15426
+ title='Agent Id',
15427
+ )
15428
+ name: str = Field(..., description='The name of this service.', min_length=1)
15429
+ description: str = Field(
15430
+ ..., description='A description of this service.', min_length=1
15431
+ )
15432
+ is_active: bool = Field(
15433
+ ...,
15434
+ 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.',
15435
+ title='Is Active',
15436
+ )
15437
+ release_version_set: ServiceCreateServiceRequestVersionSet | None = Field(
15438
+ None,
15439
+ 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.',
15440
+ )
15441
+ keyterms: list[StrippedNonemptyStringAZAZ02] | None = Field(
15442
+ default_factory=list,
15443
+ description='A list of keyterms that are easy to get wrong during audio transcriptions that tend to occur commonly in audio sessions using this service.',
15444
+ max_length=20,
15445
+ title='Keyterms',
15446
+ )
15447
+ tags: dict[StrippedNonemptyStringWS, StrippedNonemptyStringWS | None] = Field(
15448
+ ..., description='The tags of this service.', title='Tags'
15449
+ )
15450
+
15451
+
15471
15452
  class SimulationCreateSimulationUnitTestRequest(BaseModel):
15472
15453
  name: str = Field(..., description='The name of the unit test.', min_length=1)
15473
15454
  description: str = Field(
@@ -15622,6 +15603,26 @@ class ToolSearchToolInvocationsResponseToolInvocationInstance(BaseModel):
15622
15603
  )
15623
15604
 
15624
15605
 
15606
+ class UserGetUsersResponse(BaseModel):
15607
+ users: list[UserSearchUsersResponseUserInstance] = Field(
15608
+ ..., description='Users in this organization.', title='Users'
15609
+ )
15610
+ has_more: bool = Field(
15611
+ ..., description='Whether there are more users to retrieve.', title='Has More'
15612
+ )
15613
+ continuation_token: int | None = Field(
15614
+ ...,
15615
+ description='A token to supply to the next request to retrieve the next page of users. Only populated if `has_more` is `True`.',
15616
+ title='Continuation Token',
15617
+ )
15618
+
15619
+
15620
+ class UserSearchUsersResponse(BaseModel):
15621
+ users: list[UserSearchUsersResponseUserInstance] = Field(
15622
+ ..., description='Users in this organization.', title='Users'
15623
+ )
15624
+
15625
+
15625
15626
  class ServiceHierarchicalStateMachineInstance(BaseModel):
15626
15627
  id: str = Field(..., description='The ID of the state machine version.', title='Id')
15627
15628
  org_id: str = Field(
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: amigo_sdk
3
- Version: 0.103.0
3
+ Version: 0.104.0
4
4
  Summary: Amigo AI Python SDK
5
5
  Author: Amigo AI
6
6
  License-File: LICENSE
@@ -2,7 +2,7 @@ scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
2
  scripts/aliases.json,sha256=c1_bkZyvNDQ8iMrKtUuZQqdP87GdLnRPrb3wjX533eQ,322
3
3
  scripts/check.py,sha256=ikFavw1IjX51FNjO-kIzXjYdep3on0dKZHYvPElPFtc,2872
4
4
  scripts/gen_models.py,sha256=c4J_gHYQIBL3CcKg5z8oUusLvPePytfvLdZ7STHBcNk,3553
5
- amigo_sdk/__init__.py,sha256=alcG7WzR2ZlGlcUfiPYGTcPdeQtDbFYiGL88Pv763dM,140
5
+ amigo_sdk/__init__.py,sha256=iolFD9HiwaCApwA9XPByWngpt0x-TKchVyAi54_jF9g,140
6
6
  amigo_sdk/_retry_utils.py,sha256=VbA_Uwv268J80d11R9XfnCjZiV-NwE1aXRsSlRjjRio,2108
7
7
  amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
8
8
  amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
@@ -10,13 +10,13 @@ amigo_sdk/errors.py,sha256=Yn12AxIQ6zE9jnGLWO0MUVbBsmw77Kzh82AijyreOkA,4818
10
10
  amigo_sdk/http_client.py,sha256=tfMXHqMTmlz2ja7aJhOVgbwAiJNlKCzPcK4UGNqXHO8,13488
11
11
  amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
12
12
  amigo_sdk/sdk_client.py,sha256=5nj8hzjHSFRs9JMNabdYIhQAcFE2G1jI_r7SHVGr9No,6106
13
- amigo_sdk/generated/model.py,sha256=-6o9IrmpS_ENAnfvXP0xKnQ2V8bL4Gn8eppbQK5RFkg,442617
13
+ amigo_sdk/generated/model.py,sha256=bCMB3cgF9NlJMhN1FB30GFMOUDHKDOpMKLz58TirWxE,442696
14
14
  amigo_sdk/resources/conversation.py,sha256=X47PbPA6NqdkqXgxAPOjlO1E0W7t695tn1tPWSikMOk,14807
15
15
  amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
16
16
  amigo_sdk/resources/service.py,sha256=t1iA3nS9co-wuR-x5jBhAXXTWfMeGsLwfQcLycwVrCA,1536
17
17
  amigo_sdk/resources/user.py,sha256=P14lN-3fmIvELUjn_vpSaYpNsK-xd8zNmUV_KfcZPAU,4138
18
- amigo_sdk-0.103.0.dist-info/METADATA,sha256=Hdx6Ol0pB4nphqDk6QulKTNGd_7svOge-6sRy6qekVY,8500
19
- amigo_sdk-0.103.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
20
- amigo_sdk-0.103.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
21
- amigo_sdk-0.103.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
22
- amigo_sdk-0.103.0.dist-info/RECORD,,
18
+ amigo_sdk-0.104.0.dist-info/METADATA,sha256=OmL3cpSgmbOHoaG39xbqN263nnr4t3nhQ05IvFDP9WI,8500
19
+ amigo_sdk-0.104.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
20
+ amigo_sdk-0.104.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
21
+ amigo_sdk-0.104.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
22
+ amigo_sdk-0.104.0.dist-info/RECORD,,