ibm-watsonx-orchestrate 1.5.0b0__py3-none-any.whl → 1.5.1__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.
Files changed (29) hide show
  1. ibm_watsonx_orchestrate/__init__.py +1 -2
  2. ibm_watsonx_orchestrate/agent_builder/agents/types.py +10 -1
  3. ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py +13 -0
  4. ibm_watsonx_orchestrate/agent_builder/model_policies/__init__.py +1 -0
  5. ibm_watsonx_orchestrate/{client → agent_builder}/model_policies/types.py +7 -8
  6. ibm_watsonx_orchestrate/agent_builder/models/__init__.py +1 -0
  7. ibm_watsonx_orchestrate/{client → agent_builder}/models/types.py +43 -7
  8. ibm_watsonx_orchestrate/agent_builder/tools/python_tool.py +46 -3
  9. ibm_watsonx_orchestrate/agent_builder/tools/types.py +47 -1
  10. ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py +17 -0
  11. ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py +86 -39
  12. ibm_watsonx_orchestrate/cli/commands/models/model_provider_mapper.py +191 -0
  13. ibm_watsonx_orchestrate/cli/commands/models/models_command.py +136 -259
  14. ibm_watsonx_orchestrate/cli/commands/models/models_controller.py +437 -0
  15. ibm_watsonx_orchestrate/cli/commands/server/server_command.py +2 -1
  16. ibm_watsonx_orchestrate/cli/commands/tools/tools_controller.py +1 -1
  17. ibm_watsonx_orchestrate/client/connections/__init__.py +2 -1
  18. ibm_watsonx_orchestrate/client/connections/utils.py +30 -0
  19. ibm_watsonx_orchestrate/client/model_policies/model_policies_client.py +23 -4
  20. ibm_watsonx_orchestrate/client/models/models_client.py +23 -3
  21. ibm_watsonx_orchestrate/client/tools/tool_client.py +2 -1
  22. ibm_watsonx_orchestrate/docker/compose-lite.yml +2 -0
  23. ibm_watsonx_orchestrate/docker/default.env +10 -11
  24. {ibm_watsonx_orchestrate-1.5.0b0.dist-info → ibm_watsonx_orchestrate-1.5.1.dist-info}/METADATA +1 -1
  25. {ibm_watsonx_orchestrate-1.5.0b0.dist-info → ibm_watsonx_orchestrate-1.5.1.dist-info}/RECORD +28 -25
  26. ibm_watsonx_orchestrate/cli/commands/models/env_file_model_provider_mapper.py +0 -180
  27. {ibm_watsonx_orchestrate-1.5.0b0.dist-info → ibm_watsonx_orchestrate-1.5.1.dist-info}/WHEEL +0 -0
  28. {ibm_watsonx_orchestrate-1.5.0b0.dist-info → ibm_watsonx_orchestrate-1.5.1.dist-info}/entry_points.txt +0 -0
  29. {ibm_watsonx_orchestrate-1.5.0b0.dist-info → ibm_watsonx_orchestrate-1.5.1.dist-info}/licenses/LICENSE +0 -0
@@ -53,38 +53,37 @@ EVENT_BROKER_TTL="-1"
53
53
  # See get_default_registry_env_vars_by_dev_edition_source() in src/ibm_watsonx_orchestrate/cli/commands/server/server_command.py for more details.
54
54
  REGISTRY_URL=
55
55
 
56
- SERVER_TAG=30-05-2025
56
+ SERVER_TAG=05-06-2025-v1
57
57
  SERVER_REGISTRY=
58
58
 
59
- WORKER_TAG=30-05-2025
59
+ WORKER_TAG=03-06-2025
60
60
  WORKER_REGISTRY=
61
61
 
62
- AI_GATEWAY_TAG=30-05-2025
62
+ AI_GATEWAY_TAG=03-06-2025
63
63
  AI_GATEWAY_REGISTRY=
64
64
 
65
65
  DB_REGISTRY=
66
66
  # If you build multiarch set all three of these to the same, we have a pr against main
67
67
  # to not have this separation, but we can merge it later
68
- DBTAG=30-05-2025
69
- AMDDBTAG=30-05-2025
70
- ARM64DBTAG=30-05-2025
68
+ DBTAG=07-06-2025
69
+ AMDDBTAG=07-06-2025
70
+ ARM64DBTAG=07-06-2025
71
71
 
72
72
  UI_REGISTRY=
73
- UITAG=23-05-2025
73
+ UITAG=03-06-2025
74
74
 
75
75
  CM_REGISTRY=
76
76
  CM_TAG=13-05-2025
77
77
 
78
- TRM_TAG=25-05-2025
78
+ TRM_TAG=11-06-2025-1-5-0
79
79
  TRM_REGISTRY=
80
80
 
81
- TR_TAG=25-05-2025
81
+ TR_TAG=11-06-2025-1-5-0
82
82
  TR_REGISTRY=
83
83
 
84
- BUILDER_TAG=27-05-2025
84
+ BUILDER_TAG=03-06-2025
85
85
  BUILDER_REGISTRY=
86
86
 
87
-
88
87
  FLOW_RUNTIME_TAG=13-05-2025
89
88
  FLOW_RUMTIME_REGISTRY=
90
89
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ibm-watsonx-orchestrate
3
- Version: 1.5.0b0
3
+ Version: 1.5.1
4
4
  Summary: IBM watsonx.orchestrate SDK
5
5
  Author-email: IBM <support@ibm.com>
6
6
  License: MIT License
@@ -1,23 +1,27 @@
1
- ibm_watsonx_orchestrate/__init__.py,sha256=CVO5jVx-PDoZ73vCs4cuFyBQ7XCDJbi0c5pkvEvXamU,428
1
+ ibm_watsonx_orchestrate/__init__.py,sha256=J54yOUYCYqBFQXgN1gkZULUyLgGU-5lexFmpZYh5G1w,425
2
2
  ibm_watsonx_orchestrate/agent_builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  ibm_watsonx_orchestrate/agent_builder/agents/__init__.py,sha256=v4G0MGh11eOCkUJP_4AMOcFgzW14oE41G3iFp7G2vvw,376
4
4
  ibm_watsonx_orchestrate/agent_builder/agents/agent.py,sha256=PcBg2dRi-IOzvl24u8fa3B0jLaM5hzgkpTS8k56L9Ag,919
5
5
  ibm_watsonx_orchestrate/agent_builder/agents/assistant_agent.py,sha256=U1cKr22umcWCMgI4G3r5JZXHPfWYbCbiTydH4YZy_Og,987
6
6
  ibm_watsonx_orchestrate/agent_builder/agents/external_agent.py,sha256=LQM-DfOg2ajzrEur_F-aSrqKqaX4TbeajRe9YWz6s7E,981
7
- ibm_watsonx_orchestrate/agent_builder/agents/types.py,sha256=iNZNxER9cIP48xazv5wvhFhJ59j33PnAfbZK_wAdOh4,8898
7
+ ibm_watsonx_orchestrate/agent_builder/agents/types.py,sha256=1I2Z2jaRsrd-vzSGO-K_MHXkeCeValbbUvNBeSaer88,9568
8
8
  ibm_watsonx_orchestrate/agent_builder/connections/__init__.py,sha256=T7rgkNrG5wx-DfZVb7ree5hw_fCV0FbxVFxvr0bnxdw,722
9
9
  ibm_watsonx_orchestrate/agent_builder/connections/connections.py,sha256=DImRGTXiky1KEVxM_5RmeLom74Lej9bBD8cXafkV6FY,4501
10
10
  ibm_watsonx_orchestrate/agent_builder/connections/types.py,sha256=znSOdikTv_yIZaVDwoHwOSPu6b9hajj8miCMKOBmi2w,9280
11
11
  ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base.py,sha256=_KuGF0RZpKpwdt31rzjlTjrhGRFz2RtLzleNkhMNX4k,1831
12
12
  ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base_requests.py,sha256=mRow0rf5EHeJCCsTrONeTq10jShs_yIBQjpgDY_mTrI,1641
13
- ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py,sha256=fPkjWhWEnQcZvZSBYDd_El2kF6LVMBnmQhS1dc8LW6s,7641
13
+ ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py,sha256=ttre2pRs_I_KBWSht60zT3dJw8twQml3m92ZkiRAoxs,7994
14
+ ibm_watsonx_orchestrate/agent_builder/model_policies/__init__.py,sha256=alJEjlneWlGpadmvOVlDjq5wulytKOmpkq3849fhKNc,131
15
+ ibm_watsonx_orchestrate/agent_builder/model_policies/types.py,sha256=ydDNRgMBUmd38SChwljV1zPVucvK8GiMRLdDwZqD7HM,875
16
+ ibm_watsonx_orchestrate/agent_builder/models/__init__.py,sha256=R5nTbyMBzahONdp5-bJFp-rbtTDnp2184k6doZqt67w,31
17
+ ibm_watsonx_orchestrate/agent_builder/models/types.py,sha256=OJ266ta2AcO5ecvhztf0d0gCvCgbB3BliOpRYzz4LnA,9533
14
18
  ibm_watsonx_orchestrate/agent_builder/toolkits/base_toolkit.py,sha256=KXRPgBK-F9Qa6IYqEslkN3ANj3cmZoZQnlSiy_-iXCk,1054
15
19
  ibm_watsonx_orchestrate/agent_builder/toolkits/types.py,sha256=yY-V4Hqct91-Rs4rJ3rY9OhzKkSMdOT63o224o-U9eg,959
16
20
  ibm_watsonx_orchestrate/agent_builder/tools/__init__.py,sha256=adkYX0wgB-RKFCUBw6LPJhNVelUjUdsxipGPk2ghLns,479
17
21
  ibm_watsonx_orchestrate/agent_builder/tools/base_tool.py,sha256=unJBOJUY8DAq3T3YX5d1H5KehJUCjObAdpGLVoWIfzw,1156
18
22
  ibm_watsonx_orchestrate/agent_builder/tools/openapi_tool.py,sha256=kYrHyvrOQ99T8mwmX4CYjsMfSfAh0OL9SoyBN7EZbJA,14875
19
- ibm_watsonx_orchestrate/agent_builder/tools/python_tool.py,sha256=c6c-zyWIG6s4XrvpJpvFcZkNV2AbsucHsS2pZn7aspU,8484
20
- ibm_watsonx_orchestrate/agent_builder/tools/types.py,sha256=XrwrI9CWmYqWPqySB63perfSntfQlS6G8zSTpXMFVxM,5856
23
+ ibm_watsonx_orchestrate/agent_builder/tools/python_tool.py,sha256=TEx8XCLxjD693clU-cQ229t6Gg7IpgdF1Ht0mqE4WfE,10531
24
+ ibm_watsonx_orchestrate/agent_builder/tools/types.py,sha256=C1xfGWQuvOCwOaoZqeWxO67bs3hIlZkIizCNR4yhPoc,7567
21
25
  ibm_watsonx_orchestrate/agent_builder/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
26
  ibm_watsonx_orchestrate/agent_builder/utils/pydantic_utils.py,sha256=QEanM6FpkmntvS02whdhWx1d4v6zT_1l9ipEbfTgHs8,7623
23
27
  ibm_watsonx_orchestrate/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -25,8 +29,8 @@ ibm_watsonx_orchestrate/cli/config.py,sha256=C_iSP6WSb5SO6cVPTueQ9lEX16gYOLXD12c
25
29
  ibm_watsonx_orchestrate/cli/init_helper.py,sha256=Lb0bj6b8EaeBxkfGyrBBvc581ORg_S5-tHcXWQ-mBmE,1466
26
30
  ibm_watsonx_orchestrate/cli/main.py,sha256=noyKaoJzXig4gNavxNRdEe9mknBqJdt4kS5o7kDljjo,2616
27
31
  ibm_watsonx_orchestrate/cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
- ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py,sha256=yLDxbyvUQU_eq0JIGb0qZUYAlsSR5fNoFEv_dG6ONxk,7602
29
- ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py,sha256=ouv_EaR5Efq4DNrQVSDOMBrtZUHB-mGbukhPemo3nZI,39900
32
+ ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py,sha256=mblQIcwOdzTU0Au8-6b0G2OVQIe86P1nNRD2n4evO88,8354
33
+ ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py,sha256=ErI1DMJkh6b4FaYZGL4s9MSWiLCq98PFfkuHftySVjo,42520
30
34
  ibm_watsonx_orchestrate/cli/commands/channels/channels_command.py,sha256=fVIFhPUTPdxsxIE10nWL-W5wvBR-BS8V8D6r__J8R98,822
31
35
  ibm_watsonx_orchestrate/cli/commands/channels/channels_controller.py,sha256=WjQxwJujvo28SsWgfJSXIpkcgniKcskJ2arL4MOz0Ys,455
32
36
  ibm_watsonx_orchestrate/cli/commands/channels/types.py,sha256=h8gA3EM7k5X08sk8T9b5siL-XL9uUE-hl_jS-0PDBHA,254
@@ -41,9 +45,10 @@ ibm_watsonx_orchestrate/cli/commands/environment/types.py,sha256=BEHkzhsXoWWZsJr
41
45
  ibm_watsonx_orchestrate/cli/commands/knowledge_bases/knowledge_bases_command.py,sha256=B-9cgqwdR6nyJdEkZHO0_-VqtM89Oy6w5P6ONKPpLgw,3089
42
46
  ibm_watsonx_orchestrate/cli/commands/knowledge_bases/knowledge_bases_controller.py,sha256=vLe1PV_qohLMctKl1Rajzyyhb0JfDC_XV_7vIfs_E8A,9595
43
47
  ibm_watsonx_orchestrate/cli/commands/login/login_command.py,sha256=xArMiojoozg7Exn6HTpbTcjDO2idZRA-y0WV-_Ic1Sk,651
44
- ibm_watsonx_orchestrate/cli/commands/models/env_file_model_provider_mapper.py,sha256=Ixz5TC5BbdoFx8zC5TAqHvDQWlmUI4clQxHN3JegIYI,7700
45
- ibm_watsonx_orchestrate/cli/commands/models/models_command.py,sha256=M1E6zFd1_0VytKsjuPZ_eqC8QYIMJjSAAKEpU8Dvec0,12315
46
- ibm_watsonx_orchestrate/cli/commands/server/server_command.py,sha256=b_AngUTqGTUbOMpzE8ZQhM4J41gK-oIOhmeby3kOg9s,31973
48
+ ibm_watsonx_orchestrate/cli/commands/models/model_provider_mapper.py,sha256=SIae_rFxc6s2Ui75Ffi6CVzRQYfu05ETuW0wd0p5crQ,7940
49
+ ibm_watsonx_orchestrate/cli/commands/models/models_command.py,sha256=5yzlQZ3tAF-wYJpIGZxcvBhzgtudjBkllAAric02J7w,6404
50
+ ibm_watsonx_orchestrate/cli/commands/models/models_controller.py,sha256=hsdaUCqk4ERroYfx0FCcF09nNcQaAG1eheWt1AjNGT4,18152
51
+ ibm_watsonx_orchestrate/cli/commands/server/server_command.py,sha256=TWjB0SR7gOwFDMmAtwhJlHyYe2gUTZbCcMWRynbVTHQ,31973
47
52
  ibm_watsonx_orchestrate/cli/commands/server/types.py,sha256=WaaIKB8IzDdxcetkdtHbupXr4u2VDcSRdHPrxh2oQ3E,4169
48
53
  ibm_watsonx_orchestrate/cli/commands/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
49
54
  ibm_watsonx_orchestrate/cli/commands/settings/settings_command.py,sha256=CzXRkd-97jXyS6LtaaNtMah-aZu0919dYl-mDwzGThc,344
@@ -54,7 +59,7 @@ ibm_watsonx_orchestrate/cli/commands/settings/observability/langfuse/langfuse_co
54
59
  ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_command.py,sha256=e8YKUr_8CodA5fgYjUKj-Dk856foQ5Iz3I9Jld5pU_o,4193
55
60
  ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_controller.py,sha256=T7rbkbDbfKDGVi2W7zL6NmmyNkUvlE73XmrHvbR7nzY,11526
56
61
  ibm_watsonx_orchestrate/cli/commands/tools/tools_command.py,sha256=Cuo1ZvlfsymojqbadCqdwwS0HUjaWpe2XQrV70g61_s,3943
57
- ibm_watsonx_orchestrate/cli/commands/tools/tools_controller.py,sha256=ZsOIGfWPnurRRmUGGqOnRRfpXLt7Rcb4JsOHJjZlLA8,36492
62
+ ibm_watsonx_orchestrate/cli/commands/tools/tools_controller.py,sha256=BIdhqGL-jAoaktOUnk4ET_AaRPLSECG2CL9--1Ns0k8,36495
58
63
  ibm_watsonx_orchestrate/cli/commands/tools/types.py,sha256=_md0GEa_cTH17NO_moWDY_LNdFvyEFQ1UVB9_FltYiA,173
59
64
  ibm_watsonx_orchestrate/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
60
65
  ibm_watsonx_orchestrate/client/base_api_client.py,sha256=5GbL_oL9rSMIuuge-qIjuKs0Hp6U-YW1HSumqPf9J10,4501
@@ -71,21 +76,19 @@ ibm_watsonx_orchestrate/client/agents/external_agent_client.py,sha256=aBGSQbLtdw
71
76
  ibm_watsonx_orchestrate/client/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
72
77
  ibm_watsonx_orchestrate/client/analytics/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
73
78
  ibm_watsonx_orchestrate/client/analytics/llm/analytics_llm_client.py,sha256=0YS_BCpmf5oGFawpZkJ38cuz5ArhKsZIbSydWRd194s,1340
74
- ibm_watsonx_orchestrate/client/connections/__init__.py,sha256=u821r2ZiYXLYNTknxdBYgc7glwTXsrQBd3z1I81f3ro,184
79
+ ibm_watsonx_orchestrate/client/connections/__init__.py,sha256=J7TOyVg38h71AlaJjlFs5fOuAXTceHvELtOJ9oz4Mvg,207
75
80
  ibm_watsonx_orchestrate/client/connections/connections_client.py,sha256=WwfwdykWByCItI14ltQVEOCXm1dnwOTW4xIlS4lbois,6820
76
- ibm_watsonx_orchestrate/client/connections/utils.py,sha256=XqgYoiehiqIZ-LnduIAiJ9DIIF7IR7QvkKTc9784Ii0,1326
81
+ ibm_watsonx_orchestrate/client/connections/utils.py,sha256=9F-35qnSEVweH4ZhBkbsqcQln_kl197K3BQlpHFKRd8,2566
77
82
  ibm_watsonx_orchestrate/client/knowledge_bases/knowledge_base_client.py,sha256=U-pG_H0I8992f0V13Li_e1dksKp54MrYX3X9bvr-09w,2181
78
83
  ibm_watsonx_orchestrate/client/model_policies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
79
- ibm_watsonx_orchestrate/client/model_policies/model_policies_client.py,sha256=qQhSHgTwVwmPnmLxCYIgnSCML5OlptVbUjlZ1SEDeIk,1398
80
- ibm_watsonx_orchestrate/client/model_policies/types.py,sha256=_EpN7IDIxPxO8ZzZmXdBHXUGsg0weusg0ITaSKnfVcE,881
84
+ ibm_watsonx_orchestrate/client/model_policies/model_policies_client.py,sha256=QE11XgQjoYGhCkj-HerWX7WEOnuUAgkdDbj0Ma14Ahk,2224
81
85
  ibm_watsonx_orchestrate/client/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
82
- ibm_watsonx_orchestrate/client/models/models_client.py,sha256=vy4dNWULRuRzER2WOqwsFQX1l4hKcTuDpA86TyuDkk8,1251
83
- ibm_watsonx_orchestrate/client/models/types.py,sha256=66ArfM01whmzpkr37yJnh1ZWpURY8f4OMpuGbPA5VIg,8175
86
+ ibm_watsonx_orchestrate/client/models/models_client.py,sha256=ZvP3iPgUFw_SXp41kJuzWQDsgQOF5oqJURH7dFEnPz8,2164
84
87
  ibm_watsonx_orchestrate/client/toolkit/toolkit_client.py,sha256=TLFNS39EeBD_t4Y-rX9sGp4sWBDr--mE5qVtHq8Q2hk,3121
85
88
  ibm_watsonx_orchestrate/client/tools/tempus_client.py,sha256=Q6h-ynbjU1GBYweeNORgqhsP6nvLE2sKjS7sk7VFZP0,1678
86
- ibm_watsonx_orchestrate/client/tools/tool_client.py,sha256=lcu6xhWfSVSC4ZuhWMGr__V9PErEMMPRlocRsFu2ZRE,1959
87
- ibm_watsonx_orchestrate/docker/compose-lite.yml,sha256=AQxBjE_FbyPpnE0ZcsnyO4C4f_VACwYx6GX7b2XaYGQ,26541
88
- ibm_watsonx_orchestrate/docker/default.env,sha256=It1j1nzfiyU915yMYz8fne3oAwOBBcBDenPjfPQQkPU,5106
89
+ ibm_watsonx_orchestrate/client/tools/tool_client.py,sha256=d3i3alVwa0TCN72w9sWOrM20GCbNmnpTnqEOJVbBIFM,1994
90
+ ibm_watsonx_orchestrate/docker/compose-lite.yml,sha256=aqfaWMiu7__6ogk0NpgkTm0nNgMHYcy90NRq2x-6Usc,26597
91
+ ibm_watsonx_orchestrate/docker/default.env,sha256=nYr389gj6Hg6jfQbSeKtVVoqbH2pf1IPvgXDBXW__Vw,5120
89
92
  ibm_watsonx_orchestrate/docker/start-up.sh,sha256=LTtwHp0AidVgjohis2LXGvZnkFQStOiUAxgGABOyeUI,1811
90
93
  ibm_watsonx_orchestrate/docker/sdk/ibm_watsonx_orchestrate-0.6.0-py3-none-any.whl,sha256=Hi3-owh5OM0Jz2ihX9nLoojnr7Ky1TV-GelyqLcewLE,2047417
91
94
  ibm_watsonx_orchestrate/docker/sdk/ibm_watsonx_orchestrate-0.6.0.tar.gz,sha256=e5T-q7XPAtiCyQljwZp6kk3Q_4Tg6y5sijHTkscmqqQ,2025466
@@ -108,8 +111,8 @@ ibm_watsonx_orchestrate/utils/utils.py,sha256=U7z_2iASoFiZ2zM0a_2Mc2Y-P5oOT7hOwu
108
111
  ibm_watsonx_orchestrate/utils/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
109
112
  ibm_watsonx_orchestrate/utils/logging/logger.py,sha256=FzeGnidXAjC7yHrvIaj4KZPeaBBSCniZFlwgr5yV3oA,1037
110
113
  ibm_watsonx_orchestrate/utils/logging/logging.yaml,sha256=9_TKfuFr1barnOKP0fZT5D6MhddiwsXVTFjtRbcOO5w,314
111
- ibm_watsonx_orchestrate-1.5.0b0.dist-info/METADATA,sha256=WulERVXfOWKXr2A35BZQ88aMjQtwlKa2XQBQdRn9wP4,1372
112
- ibm_watsonx_orchestrate-1.5.0b0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
113
- ibm_watsonx_orchestrate-1.5.0b0.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
114
- ibm_watsonx_orchestrate-1.5.0b0.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
115
- ibm_watsonx_orchestrate-1.5.0b0.dist-info/RECORD,,
114
+ ibm_watsonx_orchestrate-1.5.1.dist-info/METADATA,sha256=zS0sucqtprxssBJJobBw3oIjxofGyowUbFiDWiEi2TY,1370
115
+ ibm_watsonx_orchestrate-1.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
116
+ ibm_watsonx_orchestrate-1.5.1.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
117
+ ibm_watsonx_orchestrate-1.5.1.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
118
+ ibm_watsonx_orchestrate-1.5.1.dist-info/RECORD,,
@@ -1,180 +0,0 @@
1
- # https://portkey.ai/
2
- from dotenv import dotenv_values
3
-
4
- from ibm_watsonx_orchestrate.client.models.types import ProviderConfig, ModelProvider
5
-
6
- def _bpp(name: str, provider_config=None) -> dict:
7
- if provider_config is None:
8
- provider_config = {}
9
-
10
- name = name.upper()
11
-
12
- provider_config.update({
13
- f"{name}_API_KEY": 'api_key',
14
- f"{name}_CUSTOM_HOST": 'custom_host',
15
- f"{name}_URL_TO_FETCH": 'url_to_fetch',
16
- f"{name}_FORWARD_HEADERS": lambda value, config: setattr(config, 'forward_headers', list(map(lambda v: v.strip(), value.split(';')))),
17
- f"{name}_REQUEST_TIMEOUT": lambda value, config: setattr(config, 'request_timeout', int(value)),
18
- f"{name}_TRANSFORM_TO_FORM_DATA": 'transform_to_form_data'
19
- })
20
-
21
- return provider_config
22
-
23
- # model provider prefix => ENV_VAR key => provider config key
24
- PROVIDER_PROPERTIES_LUT = {
25
- ModelProvider.OPENAI: _bpp('OPENAI', {}),
26
- # ModelProvider.A21: _bpp('A21', {}),
27
- ModelProvider.ANTHROPIC: _bpp('ANTHROPIC', {
28
- 'ANTHROPIC_BETA': 'anthropic_beta',
29
- 'ANTHROPIC_VERSION': 'anthropic_version'
30
- }),
31
- # ModelProvider.ANYSCALE: _bpp('ANYSCALE', {}),
32
- # ModelProvider.AZURE_OPENAI: _bpp('AZURE_OPENAI', {}),
33
- # ModelProvider.AZURE_AI: _bpp('AZURE', {
34
- # 'AZURE_AI_RESOURCE_NAME': 'azure_resource_name',
35
- # 'AZURE_AI_DEPLOYMENT_ID': 'azure_deployment_id',
36
- # 'AZURE_AI_API_VERSION': 'azure_api_version',
37
- # 'AZURE_AI_AD_AUTH': 'ad_auth',
38
- # 'AZURE_AI_AUTH_MODE': 'azure_auth_mode',
39
- # 'AZURE_AI_MANAGED_CLIENT_ID': 'azure_managed_client_id',
40
- # 'AZURE_AI_ENTRA_CLIENT_ID': 'azure_entra_client_id',
41
- # 'AZURE_AI_ENTRA_CLIENT_SECRET': 'azure_entra_client_secret',
42
- # 'AZURE_AI_ENTRA_TENANT_ID': 'azure_entra_tenant_id',
43
- # 'AZURE_AI_AD_TOKEN': 'azure_ad_token',
44
- # 'AZURE_AI_MODEL_NAME': 'azure_model_name',
45
- # }),
46
- # ModelProvider.BEDROCK: _bpp('BEDROCK', {
47
- # 'AWS_SECRET_ACCESS_KEY': 'aws_secret_access_key',
48
- # 'AWS_ACCESS_KEY_ID': 'aws_access_key_id',
49
- # 'AWS_SESSION_TOKEN': 'aws_session_token',
50
- # 'AWS_REGION': 'aws_region',
51
- # 'AWS_AUTH_TYPE': 'aws_auth_type',
52
- # 'AWS_ROLE_ARN': 'aws_role_arn',
53
- # 'AWS_EXTERNAL_ID': 'aws_external_id',
54
- # 'AWS_S3_BUCKET': 'aws_s3_bucket',
55
- # 'AWS_S3_OBJECT_KEY': 'aws_s3_object_key',
56
- # 'AWS_BEDROCK_MODEL': 'aws_bedrock_model',
57
- # 'AWS_SERVER_SIDE_ENCRYPTION': 'aws_server_side_encryption',
58
- # 'AWS_SERVER_SIDE_ENCRYPTION_KMS_KEY_ID': 'aws_server_side_encryption_kms_key_id',
59
- # }),
60
- # ModelProvider.CEREBRAS: _bpp('COHERE', {}),
61
- # ModelProvider.COHERE: _bpp('COHERE', {}),
62
- ModelProvider.GOOGLE: _bpp('GOOGLE', {}),
63
- # ModelProvider.VERTEX_AI: _bpp('GOOGLE_VERTEX_AI', {
64
- # 'GOOGLE_VERTEX_AI_REGION': 'vertex_region',
65
- # 'GOOGLE_VERTEX_AI_PROJECT_ID': 'vertex_project_id',
66
- # 'GOOGLE_VERTEX_AI_SERVICE_ACCOUNT_JSON': 'vertex_service_account_json',
67
- # 'GOOGLE_VERTEX_AI_STORAGE_BUCKET_NAME': 'vertex_storage_bucket_name',
68
- # 'GOOGLE_VERTEX_AI_MODEL_NAME': 'vertex_model_name',
69
- # 'GOOGLE_VERTEX_AI_FILENAME': 'filename'
70
- # }),
71
- # ModelProvider.GROQ: _bpp('GROQ', {}),
72
- # ModelProvider.HUGGINGFACE: _bpp('HUGGINGFACE', {
73
- # 'HUGGINGFACE_BASE_URL': 'huggingfaceBaseUrl'
74
- # }),
75
- ModelProvider.MISTRAL_AI: _bpp('MISTRAL', {
76
- 'MISTRAL_FIM_COMPLETION': 'mistral_fim_completion'
77
- }),
78
- # ModelProvider.JINA: _bpp('JINA', {}),
79
- ModelProvider.OLLAMA: _bpp('OLLAMA', {}),
80
- ModelProvider.OPENROUTER: _bpp('OPENROUTER', {}),
81
- # ModelProvider.STABILITY_AI: _bpp('STABILITY', {
82
- # 'STABILITY_CLIENT_ID': 'stability_client_id',
83
- # 'STABILITY_CLIENT_USER_ID': 'stability_client_user_id',
84
- # 'STABILITY_CLIENT_VERSION': 'stability_client_version'
85
- # }),
86
- # ModelProvider.TOGETHER_AI: _bpp('TOGETHER_AI', {}),
87
- ModelProvider.WATSONX: _bpp('WATSONX', {
88
- 'WATSONX_VERSION': 'watsonx_version',
89
- 'WATSONX_SPACE_ID': 'watsonx_space_id',
90
- 'WATSONX_PROJECT_ID': 'watsonx_project_id',
91
- 'WATSONX_APIKEY': 'api_key'
92
- })
93
-
94
- # 'palm': _bpp('PALM', {}),
95
- # 'nomic': _bpp('NOMIC', {}),
96
- # 'perplexity-ai': _bpp('PERPLEXITY_AI', {}),
97
- # 'segmind': _bpp('SEGMIND', {}),
98
- # 'deepinfra': _bpp('DEEPINFRA', {}),
99
- # 'novita-ai': _bpp('NOVITA_AI', {}),
100
- # 'fireworks-ai': _bpp('FIREWORKS',{
101
- # 'FIREWORKS_ACCOUNT_ID': 'fireworks_account_id'
102
- # }),
103
- # 'deepseek': _bpp('DEEPSEEK', {}),
104
- # 'voyage': _bpp('VOYAGE', {}),
105
- # 'moonshot': _bpp('MOONSHOT', {}),
106
- # 'lingyi': _bpp('LINGYI', {}),
107
- # 'zhipu': _bpp('ZHIPU', {}),
108
- # 'monsterapi': _bpp('MONSTERAPI', {}),
109
- # 'predibase': _bpp('PREDIBASE', {}),
110
-
111
- # 'github': _bpp('GITHUB', {}),
112
- # 'deepbricks': _bpp('DEEPBRICKS', {}),
113
- # 'sagemaker': _bpp('AMZN_SAGEMAKER', {
114
- # 'AMZN_SAGEMAKER_CUSTOM_ATTRIBUTES': 'amzn_sagemaker_custom_attributes',
115
- # 'AMZN_SAGEMAKER_TARGET_MODEL': 'amzn_sagemaker_target_model',
116
- # 'AMZN_SAGEMAKER_TARGET_VARIANT': 'amzn_sagemaker_target_variant',
117
- # 'AMZN_SAGEMAKER_TARGET_CONTAINER_HOSTNAME': 'amzn_sagemaker_target_container_hostname',
118
- # 'AMZN_SAGEMAKER_INFERENCE_ID': 'amzn_sagemaker_inference_id',
119
- # 'AMZN_SAGEMAKER_ENABLE_EXPLANATIONS': 'amzn_sagemaker_enable_explanations',
120
- # 'AMZN_SAGEMAKER_INFERENCE_COMPONENT': 'amzn_sagemaker_inference_component',
121
- # 'AMZN_SAGEMAKER_SESSION_ID': 'amzn_sagemaker_session_id',
122
- # 'AMZN_SAGEMAKER_MODEL_NAME': 'amzn_sagemaker_model_name'
123
- # }),
124
- # '@cf': _bpp('WORKERS_AI', { # workers ai
125
- # 'WORKERS_AI_ACCOUNT_ID': 'workers_ai_account_id'
126
- # }),
127
- # 'snowflake': _bpp('SNOWFLAKE', { # no provider prefix found
128
- # 'SNOWFLAKE_ACCOUNT': 'snowflake_account'
129
- # })
130
- }
131
- PROVIDER_PROPERTIES_RLUT= {}
132
- for provider in PROVIDER_PROPERTIES_LUT.keys():
133
- PROVIDER_PROPERTIES_RLUT[provider] = {v:k for k,v in PROVIDER_PROPERTIES_LUT[provider].items()}
134
-
135
- PROVIDER_LUT = {k:k for k in PROVIDER_PROPERTIES_LUT.keys() }
136
- PROVIDER_LUT.update({
137
- # any overrides for the provider prefix to provider name can be provided here on PROVIDER_LUT
138
- })
139
-
140
-
141
-
142
- PROVIDER_REQUIRED_FIELDS = {k:['api_key'] if k not in ['ollama'] else [] for k in PROVIDER_PROPERTIES_LUT.keys()}
143
- PROVIDER_REQUIRED_FIELDS.update({
144
- # Mark the required fields for a provider
145
- })
146
-
147
- def env_file_to_model_ProviderConfig(model_name: str, env_file_path: str) -> ProviderConfig | None:
148
- provider = next(filter(lambda x: x not in ('virtual-policy', 'virtual-model'), model_name.split('/')))
149
- if provider not in PROVIDER_LUT:
150
- raise ValueError(f"Unsupported model provider {provider}")
151
-
152
- values = dotenv_values(str(env_file_path))
153
-
154
- if values is None:
155
- raise ValueError(f"No provider configuration in env file {env_file_path}")
156
-
157
- cfg = ProviderConfig()
158
- cfg.provider = PROVIDER_LUT[provider]
159
-
160
- cred_lut = PROVIDER_PROPERTIES_LUT[provider]
161
-
162
-
163
- consumed_credentials = []
164
- for key, value in values.items():
165
- if key in cred_lut:
166
- k = cred_lut[key]
167
- consumed_credentials.append(k)
168
- setattr(cfg, k, value)
169
-
170
-
171
- required_creds = PROVIDER_REQUIRED_FIELDS[provider]
172
- missing_credentials = []
173
- for cred in required_creds:
174
- if cred not in consumed_credentials:
175
- missing_credentials.append(cred)
176
-
177
- if len(missing_credentials) > 0:
178
- raise ValueError(f"Missing environment variable(s) {', '.join(map(lambda c: PROVIDER_PROPERTIES_RLUT[provider][c], missing_credentials))} required for the provider {provider}")
179
-
180
- return cfg