ibm-watsonx-orchestrate 1.4.2__py3-none-any.whl → 1.5.0b1__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 (40) hide show
  1. ibm_watsonx_orchestrate/__init__.py +1 -1
  2. ibm_watsonx_orchestrate/agent_builder/agents/types.py +10 -1
  3. ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py +14 -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 +57 -9
  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 +140 -258
  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/toolkit/toolkit_client.py +13 -8
  22. ibm_watsonx_orchestrate/client/tools/tool_client.py +2 -1
  23. ibm_watsonx_orchestrate/docker/compose-lite.yml +2 -0
  24. ibm_watsonx_orchestrate/docker/default.env +10 -11
  25. ibm_watsonx_orchestrate/experimental/flow_builder/data_map.py +19 -0
  26. ibm_watsonx_orchestrate/experimental/flow_builder/flows/__init__.py +4 -3
  27. ibm_watsonx_orchestrate/experimental/flow_builder/flows/constants.py +3 -1
  28. ibm_watsonx_orchestrate/experimental/flow_builder/flows/decorators.py +3 -2
  29. ibm_watsonx_orchestrate/experimental/flow_builder/flows/flow.py +245 -223
  30. ibm_watsonx_orchestrate/experimental/flow_builder/node.py +34 -15
  31. ibm_watsonx_orchestrate/experimental/flow_builder/resources/flow_status.openapi.yml +7 -39
  32. ibm_watsonx_orchestrate/experimental/flow_builder/types.py +285 -12
  33. ibm_watsonx_orchestrate/experimental/flow_builder/utils.py +3 -1
  34. {ibm_watsonx_orchestrate-1.4.2.dist-info → ibm_watsonx_orchestrate-1.5.0b1.dist-info}/METADATA +1 -1
  35. {ibm_watsonx_orchestrate-1.4.2.dist-info → ibm_watsonx_orchestrate-1.5.0b1.dist-info}/RECORD +38 -35
  36. ibm_watsonx_orchestrate/cli/commands/models/env_file_model_provider_mapper.py +0 -180
  37. ibm_watsonx_orchestrate/experimental/flow_builder/flows/data_map.py +0 -91
  38. {ibm_watsonx_orchestrate-1.4.2.dist-info → ibm_watsonx_orchestrate-1.5.0b1.dist-info}/WHEEL +0 -0
  39. {ibm_watsonx_orchestrate-1.4.2.dist-info → ibm_watsonx_orchestrate-1.5.0b1.dist-info}/entry_points.txt +0 -0
  40. {ibm_watsonx_orchestrate-1.4.2.dist-info → ibm_watsonx_orchestrate-1.5.0b1.dist-info}/licenses/LICENSE +0 -0
@@ -1,23 +1,27 @@
1
- ibm_watsonx_orchestrate/__init__.py,sha256=m2r_4Q5VtNQnnYLu_7IR2e1LoLeMNTbpjxOE8LoZWSo,425
1
+ ibm_watsonx_orchestrate/__init__.py,sha256=-O_eW7OaX1sLXk-oyJB6kCDMPZdHGXzyM-omTW_94Z8,427
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=l_bRSkjgR4GEn1Ev2XQq00jpUo-M_wIweV8MIDzN7Ps,7593
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=AyisLeVXBNfeNxqYp2xklPnrTKmz1t020dR7OveJy5U,12150
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,36 +76,34 @@ 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=VeIsru_jCPon9lcuWDoCJ4isTzBMD6dKtCOQSoLqERM,7928
84
- ibm_watsonx_orchestrate/client/toolkit/toolkit_client.py,sha256=g5RYZblvOZ7sEYkS2feu4jQ900_SfQTExb-VbcYI4EI,3286
86
+ ibm_watsonx_orchestrate/client/models/models_client.py,sha256=ZvP3iPgUFw_SXp41kJuzWQDsgQOF5oqJURH7dFEnPz8,2164
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=aGkSGIr0cHX9TlME0PMzsGtAmqrIU0IKgcwdZ_zkYOY,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=VHU1ZkUndqaKYU3DKegTZsw-kJYWRC3SBSsbFP_zvV8,5117
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
92
95
  ibm_watsonx_orchestrate/docker/tempus/common-config.yaml,sha256=Zo3F36F5DV4VO_vUg1RG-r4WhcukVh79J2fXhGl6j0A,22
93
96
  ibm_watsonx_orchestrate/experimental/flow_builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
94
- ibm_watsonx_orchestrate/experimental/flow_builder/node.py,sha256=aw9DGk87YaiX7S-Fs-PqzFmmUC1mLA-yWMnAwNUsHvY,3153
95
- ibm_watsonx_orchestrate/experimental/flow_builder/types.py,sha256=awXOYlwZ-HfhRo0vdWIhAY-LWeIV9l9H870ZqswLcHc,17244
96
- ibm_watsonx_orchestrate/experimental/flow_builder/utils.py,sha256=3hrSmkHh86PL6wDRO0tkUp6rSfeX0uPzgvK2Zml3y4Q,4380
97
- ibm_watsonx_orchestrate/experimental/flow_builder/flows/__init__.py,sha256=yQUNdjEcU-A8DSPREndAwjQ006-F5Vxq20-XbDBk16E,865
98
- ibm_watsonx_orchestrate/experimental/flow_builder/flows/constants.py,sha256=k-6wAiOiaQGtVGdWaSODzVE5py4RheM6oIBOuyP5QTI,258
99
- ibm_watsonx_orchestrate/experimental/flow_builder/flows/data_map.py,sha256=tLYTIK5XVlJJsj0SmyPkk2CpC6huOkegbzQfUWPneEs,2524
100
- ibm_watsonx_orchestrate/experimental/flow_builder/flows/decorators.py,sha256=Qnu2at5EEaQlfZ442E2a8i9Y4s3nqOxppCsW3IQ7pSw,5345
97
+ ibm_watsonx_orchestrate/experimental/flow_builder/data_map.py,sha256=1brmWWFERDsNG2XGais-5-r58LKUUwBtqwdaLQIFRhE,503
98
+ ibm_watsonx_orchestrate/experimental/flow_builder/node.py,sha256=tp_ssBOSDDi8q-ET2bP7xRPwNOLmLUgQ7v9zZZqn9SU,4090
99
+ ibm_watsonx_orchestrate/experimental/flow_builder/types.py,sha256=SEl4v2jUJtL8hfE5T6p93B-na9_FI_WJk9AKd_TU2V0,27324
100
+ ibm_watsonx_orchestrate/experimental/flow_builder/utils.py,sha256=XoKzocDzOuUp4q8JkCG0YHni1MkL-ddO-OviF90A4-4,4425
101
+ ibm_watsonx_orchestrate/experimental/flow_builder/flows/__init__.py,sha256=MkPxvsXyMCd9jE6Vh70j0YEHQIKmwllPd635vmeTyeU,888
102
+ ibm_watsonx_orchestrate/experimental/flow_builder/flows/constants.py,sha256=XXYtL5Y1OTjj3jKo2zoJiwb7FCUNu_M43fgU8w591TY,322
103
+ ibm_watsonx_orchestrate/experimental/flow_builder/flows/decorators.py,sha256=krJHviV7rfPlW2TDveKA0XA9yrnaQeQiNpwcxg7SP4o,5441
101
104
  ibm_watsonx_orchestrate/experimental/flow_builder/flows/events.py,sha256=tfTOiBP1MbaTpe89PKeHuo96amrPKOBqYh8cU9M_YdA,2743
102
- ibm_watsonx_orchestrate/experimental/flow_builder/flows/flow.py,sha256=_JESCMyyiqEOsOBk6p3oz-rJmLmIJRWlOhFSYS_Vn-U,50742
103
- ibm_watsonx_orchestrate/experimental/flow_builder/resources/flow_status.openapi.yml,sha256=reGTaFLd_nW7w3wAZoMv0fCZtkAZ_qs-7K5NFPSZMXs,2323
105
+ ibm_watsonx_orchestrate/experimental/flow_builder/flows/flow.py,sha256=_UDDGVshez3PpJXA2t1ISjnvlrQZKO7i6vBDTUg-IFI,51186
106
+ ibm_watsonx_orchestrate/experimental/flow_builder/resources/flow_status.openapi.yml,sha256=UkQ4FD_ZhvZuMOjrgLm7cx8zJFZD7Ri-MPCe_zktU-8,1664
104
107
  ibm_watsonx_orchestrate/run/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
105
108
  ibm_watsonx_orchestrate/run/connections.py,sha256=o5TjSqxLG8zLk6UGWCcz_UdylMWZOyUT1o3i1XkL5Uc,1844
106
109
  ibm_watsonx_orchestrate/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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.4.2.dist-info/METADATA,sha256=RB99KQLCKIVlp_zpPR3DglpD8n5PjgVSGao2TZcL8Bc,1370
112
- ibm_watsonx_orchestrate-1.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
113
- ibm_watsonx_orchestrate-1.4.2.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
114
- ibm_watsonx_orchestrate-1.4.2.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
115
- ibm_watsonx_orchestrate-1.4.2.dist-info/RECORD,,
114
+ ibm_watsonx_orchestrate-1.5.0b1.dist-info/METADATA,sha256=fK06S-FUjO--gB2U4hToj003wab9Mr34zdHLQx456WY,1372
115
+ ibm_watsonx_orchestrate-1.5.0b1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
116
+ ibm_watsonx_orchestrate-1.5.0b1.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
117
+ ibm_watsonx_orchestrate-1.5.0b1.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
118
+ ibm_watsonx_orchestrate-1.5.0b1.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
@@ -1,91 +0,0 @@
1
- from typing import Any, List, cast
2
- import uuid
3
- import json
4
- import yaml
5
- from pydantic import BaseModel, Field, SerializeAsAny
6
-
7
- from ibm_watsonx_orchestrate.agent_builder.tools.types import (
8
- ToolRequestBody, ToolResponseBody, JsonSchemaObject
9
- )
10
- from ..types import (
11
- _to_json_from_input_schema, _to_json_from_output_schema, SchemaRef, Assignment
12
- )
13
-
14
- class DataMapSpec(BaseModel):
15
-
16
- name: str
17
-
18
- def __init__(self, **data: Any) -> None:
19
- if 'id' not in data:
20
- data['id'] = str(uuid.uuid4())
21
- super().__init__(**data)
22
-
23
- def to_json(self) -> dict[Any, dict]:
24
- '''Create a JSON object representing the data'''
25
- model_spec = {}
26
- model_spec["name"] = self.name
27
- return model_spec
28
-
29
-
30
-
31
- class AssignmentDataMapSpec(DataMapSpec):
32
-
33
- maps: List[Assignment]
34
-
35
- def to_json(self) -> dict[str, Any]:
36
- '''Create a JSON object representing the data'''
37
- model_spec = super().to_json()
38
-
39
- if self.maps:
40
- model_spec["maps"] = [assignment.model_dump() for assignment in self.maps]
41
-
42
- return model_spec
43
-
44
-
45
- class DataMap(BaseModel):
46
-
47
- spec: SerializeAsAny[DataMapSpec]
48
-
49
- def __call__(self, **kwargs):
50
- pass
51
-
52
- def dump_spec(self, file: str) -> None:
53
-
54
- dumped = self.spec.model_dump(mode='json',
55
- exclude_unset=True, exclude_none=True, by_alias=True)
56
- with open(file, 'w', encoding="utf-8") as f:
57
- if file.endswith('.yaml') or file.endswith('.yml'):
58
- yaml.dump(dumped, f)
59
- elif file.endswith('.json'):
60
- json.dump(dumped, f, indent=2)
61
- else:
62
- raise ValueError('file must end in .json, .yaml, or .yml')
63
-
64
- def dumps_spec(self) -> str:
65
-
66
- dumped = self.spec.model_dump(mode='json',
67
- exclude_unset=True, exclude_none=True, by_alias=True)
68
- return json.dumps(dumped, indent=2)
69
-
70
- def __repr__(self):
71
- return f"DataMap(name='{self.spec.name}', description='{self.spec.description}')"
72
-
73
- def to_json(self) -> dict[str, Any]:
74
-
75
- obj = self.get_spec().to_json()
76
-
77
- return { "spec": obj }
78
-
79
- def get_spec(self) -> DataMapSpec:
80
-
81
- return self.spec
82
-
83
- class AssignmentDataMap(DataMap):
84
- def get_spec(self) -> AssignmentDataMapSpec:
85
- return cast(AssignmentDataMapSpec, self.spec)
86
-
87
- def to_json(self) -> dict[str, Any]:
88
-
89
- obj = super().to_json()
90
- return obj
91
-