datarobot-genai 0.2.37__py3-none-any.whl → 0.3.1__py3-none-any.whl

{datarobot_genai-0.2.37.dist-info → datarobot_genai-0.3.1.dist-info}/RECORD

@@ -3,31 +3,30 @@ datarobot_genai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datarobot_genai/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datarobot_genai/core/custom_model.py,sha256=9rSJhKZ1Fo8BpVVLnRMUj3QV2AYUr5ADHEs0aW6s2tY,5813
 datarobot_genai/core/telemetry_agent.py,sha256=CxvoyResG3jXQ7ucU26NXCzWjWQyua-5qSYvVxpZJQg,5343
-datarobot_genai/core/agents/__init__.py,sha256=mTG_QVV5aoOWOgVA3KEq7KQLJllyxtG2ZQoq9wiUNYo,1542
-datarobot_genai/core/agents/base.py,sha256=63JRi0r7L94p6QHpor4JBFu_g5nxaQkqb_I4dsfrDc0,6722
+datarobot_genai/core/agents/__init__.py,sha256=hjsJ5BEqbMugkVR1oQxVUMm3AgnP74XAwQA3KkArgcs,1531
+datarobot_genai/core/agents/base.py,sha256=I4mS3LShG8g5n9UVL_hs-Q03BCeFYnV4u3ihKnciYwM,6796
 datarobot_genai/core/chat/__init__.py,sha256=kAxp4Dc-6HIM_cdBl-3IxwzJQr13UYYQ2Zc-hMwz2F8,638
 datarobot_genai/core/chat/auth.py,sha256=6qITKTHFtESsBc2NsA6cvJf78pPUrcA5XV3Vxlhb5us,5457
 datarobot_genai/core/chat/client.py,sha256=fk8MebXa8_R33VK0_DrXCS0Fgw3wFvPEvsuubC27c3s,6639
-datarobot_genai/core/chat/responses.py,sha256=vGxTA433f2AxGVlijV6O4EghyNPJCDmEqpAK2oWnsIs,10583
+datarobot_genai/core/chat/responses.py,sha256=4t3wHqaFpTbUud3lBpw833jX5O8DJm_oZ5n5AucuLZQ,10675
 datarobot_genai/core/cli/__init__.py,sha256=B93Yb6VavoZpatrh8ltCL6YglIfR5FHgytXbO9UuxBw,733
 datarobot_genai/core/cli/agent_environment.py,sha256=BJzQoiDvZF5gW4mFE71U0yeg-l72C--kxiE-fv6W194,1662
 datarobot_genai/core/cli/agent_kernel.py,sha256=Pan1l4tFwQBiJCK0u5jjJmxsyADVcicmiCdt5vVm6CI,7873
 datarobot_genai/core/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datarobot_genai/core/mcp/common.py,sha256=Y8SjuquUODKEfI7T9X-QuTMKdIlpCWFI1b3xs6tmHFA,7812
 datarobot_genai/core/utils/__init__.py,sha256=VxtRUz6iwb04eFQQy0zqTNXLAkYpPXcJxVoKV0nOdXk,59
-datarobot_genai/core/utils/auth.py,sha256=LpSoHdPD2siskYwG8q4f9cike4VQdgFWJpuJrpiszXU,8674
+datarobot_genai/core/utils/auth.py,sha256=2fRKGseIwtmztB8FO8MtX4K9uMq2bLiq7w46c3z27-A,14290
 datarobot_genai/core/utils/urls.py,sha256=tk0t13duDEPcmwz2OnS4vwEdatruiuX8lnxMMhSaJik,2289
-datarobot_genai/crewai/__init__.py,sha256=MtFnHA3EtmgiK_GjwUGPgQQ6G1MCEzz1SDBwQi9lE8M,706
-datarobot_genai/crewai/agent.py,sha256=vp8_2LExpeLls7Fpzo0R6ud5I6Ryfu3n3oVTN4Yyi6A,1417
-datarobot_genai/crewai/base.py,sha256=JLljEN7sj8zaH8OamYoevFBZzza5BjZ4f0CGHRp2jUU,6447
-datarobot_genai/crewai/events.py,sha256=K67bO1zwPrxmppz2wh8dFGNbVebyWGXAMD7oodFE2sQ,5462
+datarobot_genai/crewai/__init__.py,sha256=1o03TystuH6djv_KmHCYCymTypF5EuJ-bDF3KelGQFU,589
+datarobot_genai/crewai/agent.py,sha256=GMHPOPJ4j56ahZ2nD-Ohb8gYtlc6LFcnkwbmU7CHLZY,6910
+datarobot_genai/crewai/events.py,sha256=KCWhHLYaUFwIcARdLeHGhjMMjmJPCOs6KSBwvsT8fDw,5745
 datarobot_genai/crewai/mcp.py,sha256=AJTrs-8KdiRSjRECfBT1lJOsszWMoFoN9NIa1p5_wsM,2115
-datarobot_genai/drmcp/__init__.py,sha256=WyKW7p77kS63EsmuW8cnZpqgcxCzIv1TmRPo2qo-Z8A,2988
+datarobot_genai/drmcp/__init__.py,sha256=HPP4wkZSD1c5D_bnjnB5xkqbo-zYfeGmtnZbPgOMb0k,3092
 datarobot_genai/drmcp/server.py,sha256=KE4kjS5f9bfdYftG14HBHrfvxDfCD4pwCXePfvl1OvU,724
 datarobot_genai/drmcp/core/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
 datarobot_genai/drmcp/core/auth.py,sha256=E-5wrGbBFEBlD5377g6Exddrc7HsazamwX8tWr2RLXY,5815
 datarobot_genai/drmcp/core/clients.py,sha256=y-yG8617LbmiZ_L7FWfMrk4WjIekyr76u_Q80aLqGpI,5524
-datarobot_genai/drmcp/core/config.py,sha256=SWLhVKoqI4vmA-04TFKpKm1_G2yEMEN1e_8cv8d_XRM,13774
+datarobot_genai/drmcp/core/config.py,sha256=yuTMs1x6yDAlwBugM3FrPh9YieomtfGfVNgRFloEQa4,14442
 datarobot_genai/drmcp/core/config_utils.py,sha256=U-aieWw7MyP03cGDFIp97JH99ZUfr3vD9uuTzBzxn7w,6428
 datarobot_genai/drmcp/core/constants.py,sha256=lUwoW_PTrbaBGqRJifKqCn3EoFacoEgdO-CpoFVrUoU,739
 datarobot_genai/drmcp/core/credentials.py,sha256=PYEUDNMVw1BoMzZKLkPVTypNkVevEPtmk3scKnE-zYg,6706
@@ -35,14 +34,13 @@ datarobot_genai/drmcp/core/dr_mcp_server.py,sha256=czcjbwhZAeW9EtG_Bys0GARPOuQul
 datarobot_genai/drmcp/core/dr_mcp_server_logo.py,sha256=hib-nfR1SNTW6CnpFsFCkL9H_OMwa4YYyinV7VNOuLk,4708
 datarobot_genai/drmcp/core/exceptions.py,sha256=9zoNh5ph6QihWIYuw37ljZ73_iUfy38YVYyFSnEwivc,839
 datarobot_genai/drmcp/core/logging.py,sha256=rnUkws0vIDy_uLevwNj-wgA9uijW1Go774JPCrG0Yfw,3423
-datarobot_genai/drmcp/core/mcp_instance.py,sha256=nt4gOlAQklMcqmohRIKovYcyhgLdb08NHMo28DBYmOk,18362
-datarobot_genai/drmcp/core/routes.py,sha256=dqE2M0UzAyyN9vQjlyTjYW4rpju3LT039po5weuO__I,17936
+datarobot_genai/drmcp/core/mcp_instance.py,sha256=uzpHDpOIBIPox4P3MVReKc5caRBlNRRhdU9H0oskrY0,16925
+datarobot_genai/drmcp/core/routes.py,sha256=pSv8FxGlQdZ2p3u8L90yZ7XigX11r_s_P8U9nm8uvQY,21701
 datarobot_genai/drmcp/core/routes_utils.py,sha256=vSseXWlplMSnRgoJgtP_rHxWSAVYcx_tpTv4lyTpQoc,944
 datarobot_genai/drmcp/core/server_life_cycle.py,sha256=WKGJWGxalvqxupzJ2y67Kklc_9PgpZT0uyjlv_sr5wc,3419
 datarobot_genai/drmcp/core/telemetry.py,sha256=NEkSTC1w6uQgtukLHI-sWvR4EMgInysgATcvfQ5CplM,15378
-datarobot_genai/drmcp/core/tool_config.py,sha256=izUdM6dN3GRBzSBs-OagggM2dX5PGBnDbVv4N5bfWFI,3668
-datarobot_genai/drmcp/core/tool_filter.py,sha256=yKQlEtzyIeXGxZJkHbK36QI19vmgQkvqmfx5cTo2pp4,3156
-datarobot_genai/drmcp/core/utils.py,sha256=EvfpqKZ3tECMoxpIQ_tA_3rOgy6KJEYKC0lWZo_Daag,4517
+datarobot_genai/drmcp/core/tool_config.py,sha256=19lhmrAUka1urvMw-TIucSnfWh5n589SYikacUMIxjs,4235
+datarobot_genai/drmcp/core/utils.py,sha256=4h4UoJZbnzZHg3GcjbdiSVHTlWgnR1vs1DOJg-2cKxg,7448
 datarobot_genai/drmcp/core/dynamic_prompts/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
 datarobot_genai/drmcp/core/dynamic_prompts/controllers.py,sha256=AGJlKqgHRO0Kd7Gl-Ulw9KYBgzjTTFXWBvOUF-SuKUI,5454
 datarobot_genai/drmcp/core/dynamic_prompts/dr_lib.py,sha256=4j33AKmq7kQX_EE2_RWAbP8-K5KPVEvpUginTWn_MHs,2701
@@ -71,7 +69,7 @@ datarobot_genai/drmcp/test_utils/integration_mcp_server.py,sha256=YSk19tbaka_0zi
 datarobot_genai/drmcp/test_utils/mcp_utils_ete.py,sha256=mM52xKB-1fy_o3g6k4fNtFU2L_0jEi6GY-czdR3R5HE,5491
 datarobot_genai/drmcp/test_utils/mcp_utils_integration.py,sha256=sHA_BWtpgIAFp9IXiNkUeBartBMjLAauqkV9bYtCr-g,3874
 datarobot_genai/drmcp/test_utils/test_interactive.py,sha256=KAScFT65GUkOxuiiBcjli8HHvV1NusVN01nOib3xVCc,7939
-datarobot_genai/drmcp/test_utils/tool_base_ete.py,sha256=2RvVmwHYczl7F6qZHkKYiI77IoL-PaMT3y59t0aQtTE,9328
+datarobot_genai/drmcp/test_utils/tool_base_ete.py,sha256=VSo47v8VIgBA4kEe1KhFLixHh_P4sR-zXcb4fhfRUP0,10066
 datarobot_genai/drmcp/test_utils/utils.py,sha256=JF2W9J4Q8pCqro7dj_bHObHNP7dfybDXesTLFOUsIVM,3039
 datarobot_genai/drmcp/test_utils/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datarobot_genai/drmcp/test_utils/clients/anthropic.py,sha256=qFhLvLZHMpZa2tZwI8pZQaaeG1lsM56VaONt6a9VU8c,2333
@@ -82,44 +80,49 @@ datarobot_genai/drmcp/tools/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosH
 datarobot_genai/drmcp/tools/clients/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
 datarobot_genai/drmcp/tools/clients/atlassian.py,sha256=__M_uz7FrcbKCYRzeMn24DCEYD6OmFx_LuywHCxgXsA,6472
 datarobot_genai/drmcp/tools/clients/confluence.py,sha256=h_G0By_kDnJeWDT_d-IREsaZ5-0xB5GoLXOqblYP5MA,20706
-datarobot_genai/drmcp/tools/clients/gdrive.py,sha256=RK4IISpYb99aK6WgDthesDoglaZxwGpG_PPAAe6xsVM,33064
+datarobot_genai/drmcp/tools/clients/gdrive.py,sha256=UyY1qrggzu1GpOpWMk1oHKqYrB1bV3VAF9q4q23jE4s,33154
 datarobot_genai/drmcp/tools/clients/jira.py,sha256=Rm91JAyrNIqxu66-9rU1YqoRXVnWbEy-Ahvy6f6HlVg,9823
-datarobot_genai/drmcp/tools/clients/microsoft_graph.py,sha256=-g0EhaBVElKbujaO2cHdgc86hwFEkkyEyZVAw8pq7yM,24468
+datarobot_genai/drmcp/tools/clients/microsoft_graph.py,sha256=9FWl9HfWubDbVej9PuI2W8BccjsRWSrmvaPOmjg7PVI,30143
+datarobot_genai/drmcp/tools/clients/perplexity.py,sha256=86YvUi8zRVr4rs9iSIO1PvF_LIhYul5SbOro44DQN08,6219
 datarobot_genai/drmcp/tools/clients/s3.py,sha256=GmwzvurFdNfvxOooA8g5S4osRysHYU0S9ypg_177Glg,953
+datarobot_genai/drmcp/tools/clients/tavily.py,sha256=ekD552S3f9Et3xAbIo8HzI34pknlLSYa0rFswloG3jY,6717
 datarobot_genai/drmcp/tools/confluence/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
-datarobot_genai/drmcp/tools/confluence/tools.py,sha256=_-ws65WLK8KZP_mKkf4yJ7ZunR8qdyoiMwHQX47MSMw,12362
+datarobot_genai/drmcp/tools/confluence/tools.py,sha256=EtrlOQYiKkjdrSAfyIwH3vMEBYnKon0sTj5fokHrbDg,9912
 datarobot_genai/drmcp/tools/gdrive/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datarobot_genai/drmcp/tools/gdrive/tools.py,sha256=7bNrp7E3opKwsBDYfLIOsOGfPXW-Ae9KvcimEzetR0A,17631
+datarobot_genai/drmcp/tools/gdrive/tools.py,sha256=d65Ukm0c9rZKnVJIJRxmGH4cetJ86Bf_wUilR-A8iok,13570
 datarobot_genai/drmcp/tools/jira/__init__.py,sha256=0kq9vMkF7EBsS6lkEdiLibmUrghTQqosHbZ5k-V9a5g,578
-datarobot_genai/drmcp/tools/jira/tools.py,sha256=dfkqTU2HH-7n44hX80ODFacKq0p0LOchFcZtIIKFNMM,9687
+datarobot_genai/drmcp/tools/jira/tools.py,sha256=OGHwozgftXoQa50lOJvgmGNlzL_ULUEH5TT6QcdqSpQ,8334
 datarobot_genai/drmcp/tools/microsoft_graph/__init__.py,sha256=CuOaMt1AJo7cHx_GuhO3s_aqxZas_wlDsoBorBsvbeU,577
-datarobot_genai/drmcp/tools/microsoft_graph/tools.py,sha256=cNctozv_4lRC5Kva3D2j4taZfeQHDE6LTAjcmeQXwWA,10446
+datarobot_genai/drmcp/tools/microsoft_graph/tools.py,sha256=cWDEheHBS0f6UTsSuwtLle7SC-AwosuT2patDSMepVM,16413
+datarobot_genai/drmcp/tools/perplexity/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+datarobot_genai/drmcp/tools/perplexity/tools.py,sha256=ZIptd6w27eIlUZB_W4EmvZ2Jf5kuePmAX_xpfqRGEXc,4923
 datarobot_genai/drmcp/tools/predictive/__init__.py,sha256=WuOHlNNEpEmcF7gVnhckruJRKU2qtmJLE3E7zoCGLDo,1030
-datarobot_genai/drmcp/tools/predictive/data.py,sha256=VbGs8ERP8vNFtTTryGhI61JItNVaJsx1gxpRX1ZFZcg,4626
-datarobot_genai/drmcp/tools/predictive/deployment.py,sha256=Pc6lz9V2JOw3Ufw-SsGAhMKf6-YhvbjGoNLRFOIcSSY,3670
-datarobot_genai/drmcp/tools/predictive/deployment_info.py,sha256=BGEF_dmbxOBJR0n1Tt9TO2-iNTQSBTr-oQUyaxLZ0ZI,15297
-datarobot_genai/drmcp/tools/predictive/model.py,sha256=BVxOMHh3--liwBU4VB1OWRrqkhJ4y_Rq053f7y94TF8,6276
+datarobot_genai/drmcp/tools/predictive/data.py,sha256=-9zGKGiHFx5NwneHlnxZiX6BlBm2NqK7qQx8Qd2EQDE,4297
+datarobot_genai/drmcp/tools/predictive/deployment.py,sha256=LChISG8iibqNA7uq_HV1k0ZXi-bC5CnuZc2FRJ-kyrc,3288
+datarobot_genai/drmcp/tools/predictive/deployment_info.py,sha256=IY-xDI217RNjCbokLYMU9Lpl-k0cdrAp-Fk_dY7H_WI,13828
+datarobot_genai/drmcp/tools/predictive/model.py,sha256=E-JGZ-kQrGfezVMNtg68sv5NACbIHDpWV45XtzHwsWA,5289
 datarobot_genai/drmcp/tools/predictive/predict.py,sha256=Qoob2_t2crfWtyPzkXMRz2ITZumnczU6Dq4C7q9RBMI,9370
-datarobot_genai/drmcp/tools/predictive/predict_realtime.py,sha256=urq6rPyZFsAP-bPyclSNzrkvb6FTamdlFau8q0IWWJ0,13472
-datarobot_genai/drmcp/tools/predictive/project.py,sha256=Mzf7rQogBV6h1-MWQYTwtDHOsMWfjOyyJpSYmmvNNuc,3253
-datarobot_genai/drmcp/tools/predictive/training.py,sha256=jeZGPWJ69PPOd2MhUbACgbllQ0CK7Kz-hNl596mJujQ,25021
+datarobot_genai/drmcp/tools/predictive/predict_realtime.py,sha256=ICuXGgSrIBTYbjXv_S6itzK3pfgRA-ug9s5XSDIkOvc,13577
+datarobot_genai/drmcp/tools/predictive/project.py,sha256=IbdkuRehp69NogcQlnMliW9IYeBozCyFIyC9dgjw-Q0,2716
+datarobot_genai/drmcp/tools/predictive/training.py,sha256=CzIuH_276WLsve57ERjwX_L-8HZdv5Z4qtPSVTCFM_8,24324
+datarobot_genai/drmcp/tools/tavily/__init__.py,sha256=y4yapzp3KnFMzSR6HlNDS4uSuyNT7I1iPBvaCLsS0sU,577
+datarobot_genai/drmcp/tools/tavily/tools.py,sha256=6jWHFuN2vSE1f4Z1GMTaMDoTs5Svzko6KLoh5CzXls4,5434
 datarobot_genai/langgraph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datarobot_genai/langgraph/agent.py,sha256=DRnywmS9KDywyChtuIZZwNKbJs8BpC259EG_kxYbiQ8,15828
+datarobot_genai/langgraph/agent.py,sha256=g8NgheUxySUPhadASDSicmaOauAOQV4X4OriN_ZaiSg,16042
 datarobot_genai/langgraph/mcp.py,sha256=iA2_j46mZAaNaL7ntXT-LW6C-NMJkzr3VfKDDfe7mh8,2851
-datarobot_genai/llama_index/__init__.py,sha256=JEMkLQLuP8n14kNE3bZ2j08NdajnkJMfYjDQYqj7C0c,407
-datarobot_genai/llama_index/agent.py,sha256=V6ZsD9GcBDJS-RJo1tJtIHhyW69_78gM6_fOHFV-Piw,1829
-datarobot_genai/llama_index/base.py,sha256=ovcQQtC-djD_hcLrWdn93jg23AmD6NBEj7xtw4a6K6c,14481
+datarobot_genai/llama_index/__init__.py,sha256=kTmV6C_OSWFMi4vlGrKklR1fYHEUuwSMgraaMigeeBw,408
+datarobot_genai/llama_index/agent.py,sha256=d4EgDzaoScrbN4gLkAwO2CUJLjAUeZGlzDUsJ5k-UwU,15557
 datarobot_genai/llama_index/mcp.py,sha256=leXqF1C4zhuYEKFwNEfZHY4dsUuGZk3W7KArY-zxVL8,2645
 datarobot_genai/nat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datarobot_genai/nat/agent.py,sha256=DuGrgqt1FzvAE-cRH_P3LTFUlwuClvbVurdwA-RsbuY,11177
+datarobot_genai/nat/agent.py,sha256=MPgAwRgPr1R8CV1XS5JRwoDI-3kh8Ed-1zE8SdvQ9cg,11469
 datarobot_genai/nat/datarobot_auth_provider.py,sha256=Z4NSsrHxK8hUeiqtK_lryHsUuZC74ziNo_FHbsZgtiM,4230
 datarobot_genai/nat/datarobot_llm_clients.py,sha256=-_q_KlKOVQecIYJd8YRiYnS4ZNazQAiAdZBE1Zip_wQ,12684
 datarobot_genai/nat/datarobot_llm_providers.py,sha256=aDoQcTeGI-odqydPXEX9OGGNFbzAtpqzTvHHEkmJuEQ,4963
 datarobot_genai/nat/datarobot_mcp_client.py,sha256=jL8sXb8g4gvt0VYgB2tfMGsMjpB1GV2XIbN0iv_LxVU,10701
 datarobot_genai/nat/helpers.py,sha256=Q7E3ADZdtFfS8E6OQPyw2wgA6laQ58N3bhLj5CBWwJs,3265
-datarobot_genai-0.2.37.dist-info/METADATA,sha256=JFxYZKfbfrbePIywe1WQ1ZfjQ5W9HAq_GNYAuXxTcB8,6301
-datarobot_genai-0.2.37.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-datarobot_genai-0.2.37.dist-info/entry_points.txt,sha256=jEW3WxDZ8XIK9-ISmTyt5DbmBb047rFlzQuhY09rGrM,284
-datarobot_genai-0.2.37.dist-info/licenses/AUTHORS,sha256=isJGUXdjq1U7XZ_B_9AH8Qf0u4eX0XyQifJZ_Sxm4sA,80
-datarobot_genai-0.2.37.dist-info/licenses/LICENSE,sha256=U2_VkLIktQoa60Nf6Tbt7E4RMlfhFSjWjcJJfVC-YCE,11341
-datarobot_genai-0.2.37.dist-info/RECORD,,
+datarobot_genai-0.3.1.dist-info/METADATA,sha256=Ld9BgMGTyz3z_i8NUDGB7mR_0H82nCYrZ5urqMXj8kY,6419
+datarobot_genai-0.3.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+datarobot_genai-0.3.1.dist-info/entry_points.txt,sha256=jEW3WxDZ8XIK9-ISmTyt5DbmBb047rFlzQuhY09rGrM,284
+datarobot_genai-0.3.1.dist-info/licenses/AUTHORS,sha256=isJGUXdjq1U7XZ_B_9AH8Qf0u4eX0XyQifJZ_Sxm4sA,80
+datarobot_genai-0.3.1.dist-info/licenses/LICENSE,sha256=U2_VkLIktQoa60Nf6Tbt7E4RMlfhFSjWjcJJfVC-YCE,11341
+datarobot_genai-0.3.1.dist-info/RECORD,,

datarobot_genai/crewai/base.py

@@ -1,159 +0,0 @@
-# Copyright 2025 DataRobot, Inc. and its affiliates.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-"""
-Base class for CrewAI-based agents.
-
-Manages MCP tool lifecycle and standardizes kickoff flow.
-
-Note: This base does not capture pipeline interactions; it returns None by
-default. Subclasses may implement message capture if they need interactions.
-"""
-
-from __future__ import annotations
-
-import abc
-import asyncio
-from collections.abc import AsyncGenerator
-from typing import Any
-
-from crewai import Crew
-from crewai.events.event_bus import CrewAIEventsBus
-from crewai.tools import BaseTool
-from openai.types.chat import CompletionCreateParams
-from ragas import MultiTurnSample
-
-from datarobot_genai.core.agents.base import BaseAgent
-from datarobot_genai.core.agents.base import InvokeReturn
-from datarobot_genai.core.agents.base import UsageMetrics
-from datarobot_genai.core.agents.base import default_usage_metrics
-from datarobot_genai.core.agents.base import extract_user_prompt_content
-from datarobot_genai.core.agents.base import is_streaming
-
-from .agent import create_pipeline_interactions_from_messages
-from .mcp import mcp_tools_context
-
-
-class CrewAIAgent(BaseAgent[BaseTool], abc.ABC):
-    """Abstract base agent for CrewAI workflows.
-
-    Subclasses should define the ``agents`` and ``tasks`` properties
-    and may override ``build_crewai_workflow`` to customize the workflow
-    construction.
-    """
-
-    @property
-    @abc.abstractmethod
-    def agents(self) -> list[Any]:  # CrewAI Agent list
-        raise NotImplementedError
-
-    @property
-    @abc.abstractmethod
-    def tasks(self) -> list[Any]:  # CrewAI Task list
-        raise NotImplementedError
-
-    def build_crewai_workflow(self) -> Any:
-        """Create a CrewAI workflow instance.
-
-        Default implementation constructs a Crew with provided agents and tasks.
-        Subclasses can override to customize Crew options.
-        """
-        return Crew(agents=self.agents, tasks=self.tasks, verbose=self.verbose)
-
-    @abc.abstractmethod
-    def make_kickoff_inputs(self, user_prompt_content: str) -> dict[str, Any]:
-        """Build the inputs dict for ``Crew.kickoff``.
-
-        Subclasses must implement this to provide the exact inputs required
-        by their CrewAI tasks.
-        """
-        raise NotImplementedError
-
-    def _extract_pipeline_interactions(self) -> MultiTurnSample | None:
-        """Extract pipeline interactions from event listener if available."""
-        if not hasattr(self, "event_listener"):
-            return None
-        try:
-            listener = getattr(self, "event_listener", None)
-            messages = getattr(listener, "messages", None) if listener is not None else None
-            return create_pipeline_interactions_from_messages(messages)
-        except Exception:
-            return None
-
-    def _extract_usage_metrics(self, crew_output: Any) -> UsageMetrics:
-        """Extract usage metrics from crew output."""
-        token_usage = getattr(crew_output, "token_usage", None)
-        if token_usage is not None:
-            return {
-                "completion_tokens": int(getattr(token_usage, "completion_tokens", 0)),
-                "prompt_tokens": int(getattr(token_usage, "prompt_tokens", 0)),
-                "total_tokens": int(getattr(token_usage, "total_tokens", 0)),
-            }
-        return default_usage_metrics()
-
-    def _process_crew_output(
-        self, crew_output: Any
-    ) -> tuple[str, MultiTurnSample | None, UsageMetrics]:
-        """Process crew output into response tuple."""
-        response_text = str(crew_output.raw)
-        pipeline_interactions = self._extract_pipeline_interactions()
-        usage_metrics = self._extract_usage_metrics(crew_output)
-        return response_text, pipeline_interactions, usage_metrics
-
-    async def invoke(self, completion_create_params: CompletionCreateParams) -> InvokeReturn:
-        """Run the CrewAI workflow with the provided completion parameters."""
-        user_prompt_content = extract_user_prompt_content(completion_create_params)
-        # Preserve prior template startup print for CLI parity
-        try:
-            print("Running agent with user prompt:", user_prompt_content, flush=True)
-        except Exception:
-            # Printing is best-effort; proceed regardless
-            pass
-
-        # Use MCP context manager to handle connection lifecycle
-        with mcp_tools_context(
-            authorization_context=self._authorization_context,
-            forwarded_headers=self.forwarded_headers,
-        ) as mcp_tools:
-            # Set MCP tools for all agents if MCP is not configured this is effectively a no-op
-            self.set_mcp_tools(mcp_tools)
-
-            # If an event listener is provided by the subclass/template, register it
-            if hasattr(self, "event_listener") and CrewAIEventsBus is not None:
-                try:
-                    listener = getattr(self, "event_listener")
-                    setup_fn = getattr(listener, "setup_listeners", None)
-                    if callable(setup_fn):
-                        setup_fn(CrewAIEventsBus)
-                except Exception:
-                    # Listener is optional best-effort; proceed without failing invoke
-                    pass
-
-            crew = self.build_crewai_workflow()
-
-            if is_streaming(completion_create_params):
-
-                async def _gen() -> AsyncGenerator[
-                    tuple[str, MultiTurnSample | None, UsageMetrics]
-                ]:
-                    crew_output = await asyncio.to_thread(
-                        crew.kickoff,
-                        inputs=self.make_kickoff_inputs(user_prompt_content),
-                    )
-                    yield self._process_crew_output(crew_output)
-
-                return _gen()
-
-            crew_output = crew.kickoff(inputs=self.make_kickoff_inputs(user_prompt_content))
-            return self._process_crew_output(crew_output)

datarobot_genai/drmcp/core/tool_filter.py

@@ -1,117 +0,0 @@
-# Copyright 2025 DataRobot, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-
-from fastmcp.tools import Tool
-from mcp.types import Tool as MCPTool
-
-
-def filter_tools_by_tags(
-    tools: list[Tool | MCPTool],
-    tags: list[str] | None = None,
-    match_all: bool = False,
-) -> list[Tool | MCPTool]:
-    """
-    Filter tools by tags.
-
-    Args:
-        tools: List of tools to filter
-        tags: List of tags to filter by. If None, returns all tools
-        match_all: If True, tool must have all specified tags. If False, tool must have at least
-            one tag.
-
-    Returns
-    -------
-        List of tools that match the tag criteria
-    """
-    if not tags:
-        return tools
-
-    filtered_tools = []
-
-    for tool in tools:
-        tool_tags = get_tool_tags(tool)
-
-        if not tool_tags:
-            continue
-
-        if match_all:
-            # Tool must have all specified tags
-            if all(tag in tool_tags for tag in tags):
-                filtered_tools.append(tool)
-        elif any(tag in tool_tags for tag in tags):
-            # Tool must have at least one specified tag
-            filtered_tools.append(tool)
-
-    return filtered_tools
-
-
-def get_tool_tags(tool: Tool | MCPTool) -> list[str]:
-    """
-    Get tags for a specific tool.
-
-    Args:
-        tool: The tool to get tags for
-
-    Returns
-    -------
-        List of tags for the tool
-    """
-    # Primary: native FastMCP meta location
-    if hasattr(tool, "meta") and getattr(tool, "meta"):
-        fastmcp_meta = tool.meta.get("_fastmcp", {})
-        meta_tags = fastmcp_meta.get("tags", [])
-        if isinstance(meta_tags, list):
-            return meta_tags
-
-    # Fallback: annotations.tags (for compatibility during transition)
-    if tool.annotations and hasattr(tool.annotations, "tags"):
-        tags = getattr(tool.annotations, "tags", [])
-        return tags if isinstance(tags, list) else []
-
-    return []
-
-
-def list_all_tags(tools: list[Tool | MCPTool]) -> list[str]:
-    """
-    Get all unique tags from a list of tools.
-
-    Args:
-        tools: List of tools to extract tags from
-
-    Returns
-    -------
-        List of unique tags
-    """
-    all_tags = set()
-    for tool in tools:
-        tool_tags = get_tool_tags(tool)
-        all_tags.update(tool_tags)
-
-    return sorted(list(all_tags))
-
-
-def get_tools_by_tag(tools: list[Tool | MCPTool], tag: str) -> list[Tool | MCPTool]:
-    """
-    Get all tools that have a specific tag.
-
-    Args:
-        tools: List of tools to search
-        tag: The tag to search for
-
-    Returns
-    -------
-        List of tools with the specified tag
-    """
-    return filter_tools_by_tags(tools, [tag])