mito-ai 0.1.57__py3-none-any.whl → 0.1.59__py3-none-any.whl

{mito_ai-0.1.57.dist-info → mito_ai-0.1.59.dist-info}/RECORD

@@ -1,11 +1,12 @@
-mito_ai/__init__.py,sha256=iBzK_S6iBm403SNHhhFnuapvqmAH3h36XrQvY6ch0v8,5026
-mito_ai/_version.py,sha256=M1r8DtXf9kkJPFxKzTsjTVnijFXt6DfxUA0zzk6dz2s,172
-mito_ai/anthropic_client.py,sha256=f5O7lnQ3fC6LiRg4qT25SsGgcRCuO6fLXXicXFQ3iRY,14144
-mito_ai/constants.py,sha256=wP-J-srUgX2j8sbQ4nbFapalCrGZTX_k4u78ojYSHHY,2260
+mito_ai/__init__.py,sha256=Jk75F6q8wyyfPh1B6J17gACmQLdryRnj44-_TLLUQHQ,5054
+mito_ai/_version.py,sha256=ruca0MUTJPeI8V-_I_5oZK78ti_jkTckS43ACql-4nQ,172
+mito_ai/anthropic_client.py,sha256=D67YqjoS6A962V4E3r8pr0n1AMeQehYmkc8psvM_gPc,15238
+mito_ai/constants.py,sha256=CTYkasdAN_4mceqdsSHha6e2qEpnSCH9UEHnSYyhBuU,3500
 mito_ai/gemini_client.py,sha256=4L-h2yYc_p8cEOCMDYM0R-vD4aktfqN47siZH5ZjGJc,9771
 mito_ai/logger.py,sha256=ZJEC415ZHTodDYKgp8ezVg3c4O513oQZV5lSs708gp8,1285
-mito_ai/openai_client.py,sha256=ljLs-HmHX5IElVn-5GuLTLijVPzWKCzg0uA263K57Dw,14210
+mito_ai/openai_client.py,sha256=HLldWrCoiEjvJrUdHA7fujcOC5Xq94MbS_SoWDLh1rQ,12422
 mito_ai/path_utils.py,sha256=1tFZBRChMAxFQjLrpbwc40o7wPhsXG3H1ad5n1gFOzM,2573
+mito_ai/provider_manager.py,sha256=pll7dgYsHBRnNRbPTmYMZscV1NzmK3eJ_h1FKtR4tH4,18987
 mito_ai/version_check.py,sha256=871m9_Qj_UzYmxuByBcs8DIoonUb0uVqGTRmJYpw6Fk,2268
 mito_ai/app_deploy/__init__.py,sha256=Kn_8xI0wv_JVV6FRhLDKOuld2P01Lit8nR2zB0aIHN0,190
 mito_ai/app_deploy/app_deploy_utils.py,sha256=cnCayqP3WPfA4Lrz8Eo2h0lJbgBDgyfePriB-8wMeKA,1964
@@ -20,35 +21,35 @@ mito_ai/auth/__init__.py,sha256=CrNlpbSKMPcw40H7keblZ61aa_x8qqSXPAqrEq-yCqk,172
 mito_ai/auth/handlers.py,sha256=vpbLBsS_CIGVK3kZ_Pht4fjvRlde-z5kWQRc-jLjHe4,3888
 mito_ai/auth/urls.py,sha256=iO4ueOlmqt6Bc66mVX2OrSdFGWPcwqhvGU9Ur5__sUM,417
 mito_ai/chart_wizard/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
-mito_ai/chart_wizard/handlers.py,sha256=53coW_uR_bhq1zlgOtpEalyH1kYK7NI_dI5QLal-leo,2035
-mito_ai/chart_wizard/urls.py,sha256=gQjsDkTr7nEQOL4CIlfYM_y22wJQCZwSApR4c3mdirs,797
+mito_ai/chart_wizard/handlers.py,sha256=xte8fow6TH9ciJrwKPilk4N7C5dmJpFM5eWXmhCIW7E,3959
+mito_ai/chart_wizard/urls.py,sha256=lsMoB_k6aHXGEN4zgQR_TjzRBnq4wpBB7t9eNrUM1KE,906
 mito_ai/chat_history/handlers.py,sha256=zRH9oNiwf_x7bYaa0JPoodoMuWwcFzEvg9svAAzZtZc,2537
 mito_ai/chat_history/urls.py,sha256=WoXVkS3tn_wohT_6tQ5iLVF26llL697vmgkMb7xsvT0,1100
-mito_ai/completions/handlers.py,sha256=j806QnoGDPM1sf_t0gYbwMk9Eon-5atFBS4F7pGcaGw,18151
-mito_ai/completions/message_history.py,sha256=e-PZZd7fU2iQSH18tDsTb7J_QC56K29ZIIwivMQaVqk,17213
+mito_ai/completions/handlers.py,sha256=lh6xfVX_EmXwWly-k7JegsDDfckD44oAt408RYhhWzg,18654
+mito_ai/completions/message_history.py,sha256=kbZe_wwZD52CeDXPyxDo3VdRg4xT9yD6UGdPmK2wBVc,17079
 mito_ai/completions/models.py,sha256=DigsLKRefkJmov-lT_IbjSCerEM9jX3pCWSYKNxLHQ0,10846
-mito_ai/completions/providers.py,sha256=JcZZGxFDd5B2nwnwOuJICMhjLRpLFL8_QhomaepXb5c,11950
 mito_ai/completions/completion_handlers/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
-mito_ai/completions/completion_handlers/agent_auto_error_fixup_handler.py,sha256=Vu-SoOziFnWQjxomTpbnbZgxtLkvqPS4Etw2me6L94M,2810
-mito_ai/completions/completion_handlers/agent_execution_handler.py,sha256=QwZBnsLe__Tl1el5XMZgpz1AXKhA6n8dsXLZWd66shM,3027
-mito_ai/completions/completion_handlers/chat_completion_handler.py,sha256=4DMjn18KV7_YgwQVH5VmEPtBNver9M1qIh4vNzsqP_0,6210
-mito_ai/completions/completion_handlers/code_explain_handler.py,sha256=mkR-HJ0fEmzetJNgH20DKYSCL9Ax2o0WGu8limx9X_w,5074
-mito_ai/completions/completion_handlers/completion_handler.py,sha256=E1HFeY6aVGeVU5u9r3VSiogDf3MkGsUqDx5IMFckYe8,1747
-mito_ai/completions/completion_handlers/inline_completer_handler.py,sha256=IR9xPA6YggdHQkdXVEcWqXZy8GXphqYRHkTfOcdEphw,1808
-mito_ai/completions/completion_handlers/scratchpad_result_handler.py,sha256=WWkeUpILbASdchTpjPFoj8MjXeINKXjUCKLUHiw3Mi8,2959
-mito_ai/completions/completion_handlers/smart_debug_handler.py,sha256=5DH4NR9YB28YoQOUbFFcF7bip11cFzmzrcap_bZGLwA,6388
-mito_ai/completions/completion_handlers/utils.py,sha256=Bk-EX9gj_riRQrBC5Bjp5ceFbgymgBDmNXa1SZ9Vt1M,4389
+mito_ai/completions/completion_handlers/agent_auto_error_fixup_handler.py,sha256=q9xqnOEB2u0FeOly0sqWFVOfh3qxYzKdPznMAn6wmJE,2741
+mito_ai/completions/completion_handlers/agent_execution_handler.py,sha256=weB1zHGPoHGYvoiLgJF2bXsIi9yNXJVfIxYaU8ty22c,2958
+mito_ai/completions/completion_handlers/chat_completion_handler.py,sha256=P29PxfS4MHhC4jk_KDegzzVNqsktq4ISS84OUcnZFc8,6076
+mito_ai/completions/completion_handlers/code_explain_handler.py,sha256=xIrPn1lRBU2mG0cZo5515E8umIqYW0NwHlnOcYN8hV8,4940
+mito_ai/completions/completion_handlers/completion_handler.py,sha256=fKG0IIoJWJb4U4EYC25kF7YOXz2sCooI5jGoICbqe1Y,1669
+mito_ai/completions/completion_handlers/inline_completer_handler.py,sha256=1JBw-I6HhAScV4EExkI4Rv8i92N5z_jv2D_0fgv9CAw,1793
+mito_ai/completions/completion_handlers/scratchpad_result_handler.py,sha256=4GpZGDUPKH3lJpiCIFW9MHXH8Il0-De3BM8RXRvs2y4,2890
+mito_ai/completions/completion_handlers/smart_debug_handler.py,sha256=RqX0DxAaW2F8UHePa6lS8Jgci9mCOYZo9BLfS-vQnI4,6254
+mito_ai/completions/completion_handlers/utils.py,sha256=IxYj9cp_mfk9GdqmMwibNMDP3Qx620XdRtw0Orqgl2w,4313
 mito_ai/completions/prompt_builders/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
 mito_ai/completions/prompt_builders/agent_execution_prompt.py,sha256=onj6u6s6bF5XdN_2lSOnYyMIaype8Pk_d3HLHTZv0-U,1034
 mito_ai/completions/prompt_builders/agent_smart_debug_prompt.py,sha256=6uiosUsNNlzqFLnj2vNnTmWwISc3GY4lPTuZR8pJeFU,7483
 mito_ai/completions/prompt_builders/agent_system_message.py,sha256=n8Is8xk1sXW0iwO9KXFT6MCbIjBaYC_EDzD5IfL4fXA,36481
+mito_ai/completions/prompt_builders/chart_add_field_prompt.py,sha256=VM1WzeV1xta0-qT9yLNOQLodcQYrTBQm4LyRToxdkiI,2070
 mito_ai/completions/prompt_builders/chart_conversion_prompt.py,sha256=wulJt0I2xbb7ciZSXO80q6yTw1ZxXUUc7Liayywmq3g,1215
 mito_ai/completions/prompt_builders/chat_name_prompt.py,sha256=ERwgDUAnd7VH3drjbUh70zbP3iEUct1PdJ9EZvMi3qE,532
 mito_ai/completions/prompt_builders/chat_prompt.py,sha256=ozgbzjXaiXLFcmctu07Wmj1ZhK-7u9WV1-_fqyGV8Ww,1143
 mito_ai/completions/prompt_builders/chat_system_message.py,sha256=OoY_CHPD_uG3XIGQMMD4PAsueVyFVjg9UAWkpfO8owM,7775
 mito_ai/completions/prompt_builders/explain_code_prompt.py,sha256=jN5bX4TqdE0YbcaHxBQOFqryTiv_Gl-m8gNZN7Bju-Q,1178
 mito_ai/completions/prompt_builders/inline_completer_prompt.py,sha256=9j8tOn3wIxVTYyje-oadPidHtdrXdXY8Q89kN-nFXu8,5082
-mito_ai/completions/prompt_builders/prompt_constants.py,sha256=p94EnIW_LYC7TR2-_TYTyEoQc2LjeY33oVD9RrKezZs,9939
+mito_ai/completions/prompt_builders/prompt_constants.py,sha256=tzX4k8J3Menfm6xLHyr3u-XV_wj27J5U3DXR4TvILug,10143
 mito_ai/completions/prompt_builders/scratchpad_result_prompt.py,sha256=vpU7_vLXpzfqC1QIaLMPZ7mImKiqmbuzn5RS_uBdRDc,758
 mito_ai/completions/prompt_builders/smart_debug_prompt.py,sha256=2FdH_BAlXOMozbLRHN3q_fkcdhHRwGK7tTKpR1Xu98Y,6676
 mito_ai/completions/prompt_builders/prompt_section_registry/__init__.py,sha256=laS2z7BwYCQXeBzqXZEbbb1w8MOm7Rgko5yWUwNlW9A,2598
@@ -84,22 +85,24 @@ mito_ai/docker/oracle/compose.yml,sha256=6mgV7SmP3IenFwzY4f3JKCI1f1PDVlYqInDxjIq
 mito_ai/docker/oracle/init/setup.sql,sha256=d8xZ236OdMneN_ziZbOzxl5-rUI7eET1kmd2pDppbdg,1135
 mito_ai/docker/postgres/compose.yml,sha256=pp1fG2YfYbDpHuedkpcnAkRu0HJs_uibBn2CnQwrZYA,467
 mito_ai/docker/postgres/init/setup.sql,sha256=l--Ra5FNfR-24I-HdZWpqzqrERxF1jNOIzdrpVlMRFM,356
-mito_ai/enterprise/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
-mito_ai/enterprise/utils.py,sha256=Skdg9YahsnixKfBMXLneJmzn7yTi1QbvTjBxESV2dmw,652
+mito_ai/enterprise/__init__.py,sha256=qd39u6KdG0zeYRT2_aNlY13_lDqE6ZUSYv9JfgtT04s,115
+mito_ai/enterprise/litellm_client.py,sha256=hdMJIVFEu9_vRRZT93lMzgxyS_VXuulmS9o4CoHRyUM,5597
+mito_ai/enterprise/utils.py,sha256=l1oaHdomIihOGEGB2bYyl46p9fCFkcqXHGUjGdlriHg,957
 mito_ai/file_uploads/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
 mito_ai/file_uploads/handlers.py,sha256=_mmEmCoQTAWHzPXKScEdRIrp5imoZMYgmaD-O6OAhfo,9171
 mito_ai/file_uploads/urls.py,sha256=5U-Jx2GF7CsFIIsBNkvWBy_MU9pnk20G_BJGiAGENng,659
-mito_ai/log/handlers.py,sha256=RWkJFJ5OJ28Fm2AYgRbywxhgDZoTqjNwW-Z6tZJvUc4,1199
+mito_ai/log/handlers.py,sha256=Zf300WzOWK2UAVaVzhr9xkCkw0QUexgX7xEtO2-u56o,1197
 mito_ai/log/urls.py,sha256=OLqAQObEDAwMVmzXZVftR2YBw5D9eCn0bDWlW_52DvI,657
 mito_ai/rules/handlers.py,sha256=dWnusYlQkKGyEEahNm9wtTYdIJq-lxVchyEvTd_Jkeg,1600
 mito_ai/rules/urls.py,sha256=HkhjOcfuKPN35crvY_CBABclOG8d18OfexcmQR-trFE,697
 mito_ai/rules/utils.py,sha256=7Hj5ZOUZERLHMR7a5_7ipbJXaiTkvXjfiRjTHpCKdxU,1664
+mito_ai/settings/enterprise_handler.py,sha256=KAWEp0B-xvqiVWc4BfGDMK4TzxOXwMM14UktR8kTKyE,825
 mito_ai/settings/handlers.py,sha256=Ym1QieLf9VooP44w4mU2z4WtGSkThVhEgmtO7ENBg4c,1348
-mito_ai/settings/urls.py,sha256=4Oyd6Qtap2vMApyUevoUp_RjoL09IBIwPL6ZayGT3wM,648
+mito_ai/settings/urls.py,sha256=gW7wQbQj5vQrXVMrS9oSFkBMyb2tlEYl5gZEIFgwHPA,802
 mito_ai/settings/utils.py,sha256=ZiG-axUDFsIKK7iipZTMwunYtP7_dkThHL37mR8GGyA,1355
-mito_ai/streamlit_conversion/agent_utils.py,sha256=3i2Fdfj9NVCvIv19BbJc0akCYHA8thkOFtN414tLFlM,1563
+mito_ai/streamlit_conversion/agent_utils.py,sha256=MENz1M93w7kK96VzAMACT7GeX0xJULgfeUfM2sSU95Q,449
 mito_ai/streamlit_conversion/search_replace_utils.py,sha256=80tAohWalHemmDFbEtGcjFy-j-7u2-27c-pRooOxXWg,2955
-mito_ai/streamlit_conversion/streamlit_agent_handler.py,sha256=hqk_e7hC0BRraykIq6nANN1G1xmDDO0hZ-FoHaU5KMs,6973
+mito_ai/streamlit_conversion/streamlit_agent_handler.py,sha256=aAfGGxFfOQDOOjgNhldlbrharT6Q8-z40APaLupoIKA,7601
 mito_ai/streamlit_conversion/streamlit_utils.py,sha256=4VpHmSoSWpOlwQn4All7mXywFuAn9OzeLfu_IK6vqX0,3002
 mito_ai/streamlit_conversion/validate_streamlit_app.py,sha256=l-mowqwylASGsM-vCtEzErGfp5ZnvhNaEh8VG0jtNec,3997
 mito_ai/streamlit_conversion/prompts/prompt_constants.py,sha256=N8ShlrqKViI92s3a20KmlOxNo12pSu0DgztmwHZu2Ys,4521
@@ -110,17 +113,19 @@ mito_ai/streamlit_conversion/prompts/streamlit_finish_todo_prompt.py,sha256=JU9q
 mito_ai/streamlit_conversion/prompts/streamlit_system_prompt.py,sha256=ksf3Erx5H-vkUmgbCdoWUHIc_3TI5yzsLb0sRKREwcA,3517
 mito_ai/streamlit_conversion/prompts/update_existing_app_prompt.py,sha256=e3t7qp2nNqlcHpEjBRPdVku1w2_X88m_IBkf_Yz--SA,2302
 mito_ai/streamlit_preview/__init__.py,sha256=RLNxLYp76fZ-nezx75oQB7BDnX4F81GH4ITMS4s5Rpo,196
-mito_ai/streamlit_preview/handlers.py,sha256=0yt3hRNGDPh5W3P3sh68MnWpJjolEr0adDRkD7HAoPk,5554
+mito_ai/streamlit_preview/handlers.py,sha256=PvoWmXFZ44J4uVNqfhxmKe6ajv1d8FBf4VDOAhhC2v8,5704
 mito_ai/streamlit_preview/manager.py,sha256=Ey3Z7S9Bw576rrvshTC-CYDfOYJ0hKPhSL70foPtgTc,5451
-mito_ai/streamlit_preview/urls.py,sha256=IbcugX1H8u30Q6U8-9Ofdjmq_rdyAiesxw36qSB5giA,790
+mito_ai/streamlit_preview/urls.py,sha256=EsQPQXwhfiQ_ihGlyAbG_l2OPuCeedmSQUw6aLUlaG8,979
 mito_ai/streamlit_preview/utils.py,sha256=_Ns-C0gdSAm7M3KAGCfMIf7tyumnU0qPYBB0eEUX1Aw,1212
 mito_ai/tests/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
 mito_ai/tests/conftest.py,sha256=C4iW3L50rWLeI91gim3gyyJnKKb1zGJ1GLxlXZWfLNs,1494
 mito_ai/tests/create_agent_system_message_prompt_test.py,sha256=PmNBCEz3vuy43CPRfE2JWgSVt1dfGFtn8pzBkDxBlyc,1061
-mito_ai/tests/open_ai_utils_test.py,sha256=LSJ6sIQq5xQ7RR3cYqBUegUagVQSoLemAbennCJMhiE,6646
+mito_ai/tests/open_ai_utils_test.py,sha256=HQDUVy_m1FZWKl0zVNGjBh1LDKXIL7pyzezFHM4TFww,6372
 mito_ai/tests/performance_test.py,sha256=DwsvTTk51GZ1JOEMqvFHPGV6-8dfsL12GkEg9-7VlG8,12303
 mito_ai/tests/server_limits_test.py,sha256=BKIR75pUOntnQfmtif3XADHzcxzF9jMKEC4tqXefN-E,19521
-mito_ai/tests/test_constants.py,sha256=aBQRxuoNOax62gyGGHkk22CJZWhelmQnZXXdBLcY4go,1969
+mito_ai/tests/test_constants.py,sha256=7GHhYMY52LNprqPttppDmaPqqHQGxXrJJ8bKiAwngMk,6620
+mito_ai/tests/test_enterprise_mode.py,sha256=TjobAn5NQHqPWHuZhl-E3TCz1FdAcwLmiXahRq06IRQ,10315
+mito_ai/tests/test_model_utils.py,sha256=cSnorppC8_ruQUTEyMiVU6TGSYU9Yv-et09vF7JQkMY,16569
 mito_ai/tests/test_telemetry.py,sha256=p5p0-7EQKIT0Lf_lNgE9wdU0bs0wv-tO9eQVI_hYACI,349
 mito_ai/tests/version_check_test.py,sha256=ffEme3mS0sn2u6UjN7uf6R7MfwMda8gVMWFuks2lEXE,6753
 mito_ai/tests/chat_history/test_chat_history.py,sha256=G6vluuusKXA8WXqIrIzEQdeRjDYN979r_Ox0KctwZK0,7586
@@ -142,21 +147,21 @@ mito_ai/tests/db/test_db_constants.py,sha256=ilzEo8BHeI3lBQjXLfrVb3hdH7_Gkg9_YWd
 mito_ai/tests/deploy_app/test_app_deploy_utils.py,sha256=s8cqxno0jShLwdTAjf0L-YsN6nxVjd-UUlZLDkMlaLI,3724
 mito_ai/tests/file_uploads/__init__.py,sha256=Oq613SdOIx0n9iCsVSp-4p4zzbdgjewk1_ggS09rt4Q,111
 mito_ai/tests/file_uploads/test_handlers.py,sha256=WqicGVFr5-cvSm0MSY7_p03E4lap0RJf6W0OpXjYne0,8832
-mito_ai/tests/message_history/test_generate_short_chat_name.py,sha256=0nwoJ9BNZLwwqUMCUe7EFfgpogbZ5djhpOr0kwkqfJM,4814
+mito_ai/tests/message_history/test_generate_short_chat_name.py,sha256=0RnfAG_dcPwo9CF4PVKOOZxeJTekOP1jlNsgiKhFJqQ,10566
 mito_ai/tests/message_history/test_message_history_utils.py,sha256=6vp4GifG-h09oMFqmRgWXmIz1_eb_QfwSP_qBdFFUrQ,15229
-mito_ai/tests/providers/test_anthropic_client.py,sha256=R8orxW_o5uFx94BbtIXfiHG8PjwpNdHugp6sn3we-Jk,17504
-mito_ai/tests/providers/test_azure.py,sha256=6XJP7ZXmE6f41rEdimsAFgUF0I7bDB6jOKcu02sraak,27567
-mito_ai/tests/providers/test_capabilities.py,sha256=5pumT-CJeSzodxi9Wqt53SdudTmij5fapa6Zr3YMMbc,4621
-mito_ai/tests/providers/test_gemini_client.py,sha256=fhE06siTyluUlNyC3lJ2ffdkmG_1orCB5EU3mFsDOpg,7821
+mito_ai/tests/providers/test_anthropic_client.py,sha256=DLuwE4c0erSjFfvKk8pdNFfD_Lh68Qa9whAJhvuwALE,24239
+mito_ai/tests/providers/test_azure.py,sha256=02CcN6fXe-QVwqTom2mV6CnT5HYRfQ_6_lVxBCS_GJI,27602
+mito_ai/tests/providers/test_capabilities.py,sha256=YLr7n6825VmtvZYpiT3mgg1qiVBcdUeKMgbSV_m_Oc4,4508
+mito_ai/tests/providers/test_gemini_client.py,sha256=0Ndw-Aer4iWfAvzMkuDZkNtglJ-KYzkgkhBuVFXRO-c,7711
 mito_ai/tests/providers/test_mito_server_utils.py,sha256=34jjy-hUDwgav3g0CNe9ogCZwXXN7HdxgjRJKAad-_g,18885
-mito_ai/tests/providers/test_model_resolution.py,sha256=TXrdXD5qnJ1GnBqHZJR_6xCaTXTF1x-k0DE5QMQ09mk,6623
-mito_ai/tests/providers/test_openai_client.py,sha256=4f-B68JMZMIrvf_K7tEa8KDy-HyZ2Ov_ehqX7lzIWMc,2356
+mito_ai/tests/providers/test_model_resolution.py,sha256=ZH_RzFZ7vYckgEMPF3QGzqmtt4oJRgz-_a6NAZy0Rag,9000
+mito_ai/tests/providers/test_openai_client.py,sha256=vLf3YT0V5ipTG5rDaiCW2xx6yNVO0TbwQLbAo2zPY80,10603
 mito_ai/tests/providers/test_provider_completion_exception.py,sha256=BM09h8ZC7n1JGp9kqY4mE0xIJFBn7C6YcxAz6psyG5M,2645
-mito_ai/tests/providers/test_provider_limits.py,sha256=9GzNMV54pTTV1fzOtQci3cVQ0GwR6OI12hQYntyg-tI,1597
-mito_ai/tests/providers/test_providers.py,sha256=WEQRyNkGfyWE1sXJHxAsXyHaXKHRZIemVKE-DMNjjdk,14691
-mito_ai/tests/providers/test_retry_logic.py,sha256=8jizGx8A1R6ZYYAfH63LBdkry2xyT8NThfIH-xLx8-U,16081
+mito_ai/tests/providers/test_provider_limits.py,sha256=leSPsIf8uUqNLRD1GqAHFsz9Nd3h-q-YQMl9n9GXhVw,1594
+mito_ai/tests/providers/test_providers.py,sha256=1pAq8ELZssyRToaERd0_9zZSku5n3-pco3jYtOGqx5s,21859
+mito_ai/tests/providers/test_retry_logic.py,sha256=aY9Z9tnZuPaqbDpw1CB5EBQXNk-6yceirxXSq3KvYXY,15732
 mito_ai/tests/providers/test_stream_mito_server_utils.py,sha256=SOOb1vU9U7xtiuIKYQaBIG-QYQqnsWXYVb9gTbTInz8,3930
-mito_ai/tests/providers/utils.py,sha256=z8lVfmXDHVzRpz4L6ZWEaHcY7uTUn_5FsmSr7UoxXtQ,3456
+mito_ai/tests/providers/utils.py,sha256=lTlo4g49R0vIbZerV1JjjjV9hm57brZkYrKyrEEhg5o,3436
 mito_ai/tests/rules/conftest.py,sha256=DPW5JMtV5tZIKfDWq6R_DXTPJHZGMGfYsfA77cQi_ao,831
 mito_ai/tests/rules/rules_test.py,sha256=gaPGp3qMI-ZWW4J9nHEk9iXrAesB1mCyci7LDFN0K-g,3399
 mito_ai/tests/settings/conftest.py,sha256=dypIqQCssJ9sHArib7FTWQCrY2OLmkI3ZMK8Y3wtXDI,787
@@ -164,10 +169,10 @@ mito_ai/tests/settings/settings_test.py,sha256=dRmugyFeAfUZcvSLd9Bz2kQ96hnq0mrCv
 mito_ai/tests/settings/test_settings_constants.py,sha256=BBG7TCfl2pJzLcg825pRvEVgjP3D9sWyYAwqekQ10wo,349
 mito_ai/tests/streamlit_conversion/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
 mito_ai/tests/streamlit_conversion/test_apply_search_replace.py,sha256=PDyqApl79a8zjFHMDqrZn_2pKw1vyn1MTfbXt2bo_hA,5682
-mito_ai/tests/streamlit_conversion/test_streamlit_agent_handler.py,sha256=1WIadNoNSyGH9jdaXUpiumYlEiAydYl0JH47hd8DokI,11827
+mito_ai/tests/streamlit_conversion/test_streamlit_agent_handler.py,sha256=J1xJiXSae0-doYwDmn5IdNHWFm_9wRDtj8xsZupO3n4,10788
 mito_ai/tests/streamlit_conversion/test_streamlit_utils.py,sha256=bZZQw3MwBsy1m5wPmjT0mpLb3hSewbMvj3W-hy_s1j0,7387
 mito_ai/tests/streamlit_conversion/test_validate_streamlit_app.py,sha256=a8vEAXZ6UDG_IZ9NR4qQURKabgs3XKcbc-f3PrOFbe0,3860
-mito_ai/tests/streamlit_preview/test_streamlit_preview_handler.py,sha256=LoGm85VfQV-UfG7gQ6HvNInCQp78TALxYLQPXo6Yyf8,5256
+mito_ai/tests/streamlit_preview/test_streamlit_preview_handler.py,sha256=Xc13oKOsf0WG0zg-M3_dYqJGU5zFIzLRGvAse0RxGvk,5354
 mito_ai/tests/streamlit_preview/test_streamlit_preview_manager.py,sha256=0LueiqJj0lWkB8HRszXaT2UhZk-ZnardNkwMfZHDYQU,11655
 mito_ai/tests/user/__init__.py,sha256=Oq613SdOIx0n9iCsVSp-4p4zzbdgjewk1_ggS09rt4Q,111
 mito_ai/tests/user/test_user.py,sha256=5XA-lX2IYb9x_06sqAIJKTCnQqO1MytQxX7HDgcm3QU,3802
@@ -177,54 +182,56 @@ mito_ai/tests/utils/test_gemini_utils.py,sha256=DeN5392ZiuckeMw1agmHPaoa5y73TUSE
 mito_ai/user/handlers.py,sha256=wNDKWZZHzcl8SIFCmvR65HbU5NU5FBdfjBwzlBf000Y,1477
 mito_ai/user/urls.py,sha256=vUgE4r2QVbw275DvsRBjhJAOmiQ1pLG-F0JJcEJNZnE,626
 mito_ai/utils/__init__.py,sha256=10bPaPwGgcrId4d0uHyFr6ExhhbjO9Dog2oAfPHSMPs,112
-mito_ai/utils/anthropic_utils.py,sha256=HVf-OMw5xeneEdqIlQklX8yKQVsbn13ojrOPdqjx3f8,7293
+mito_ai/utils/anthropic_utils.py,sha256=mY-BNU7SWpAdp_guXV_cqM_5WOi_XtwJjHbSq6uaesQ,7403
 mito_ai/utils/create.py,sha256=Awd_WOxUZgUveSPmAJJL1-csoIyqTV_00AIRCGIaCio,3086
 mito_ai/utils/db.py,sha256=oeNjFVg5JET1epjmUUs9uhgGzzvbCn3Na69UbVP0Ib4,2433
 mito_ai/utils/error_classes.py,sha256=n3UBUJepE4ETIH3ENi_pAXeYP0HyY6zQhEHRe91dRNU,1602
-mito_ai/utils/gemini_utils.py,sha256=zZlEeaQvqvndVxgNNmIQfB7OGkUd292bIlH0pvW0k4M,4643
+mito_ai/utils/gemini_utils.py,sha256=ORpcVCl2bSg5lwZo-ee4WiQ2XhUl5AIpnE1elXDDiKw,4418
+mito_ai/utils/litellm_utils.py,sha256=S1JBGAvWa_bTzkDP5acAy_4a82aoRjLF9u9WWIE4Tmc,3104
 mito_ai/utils/message_history_utils.py,sha256=C6eWYY_AQcpvPRlEmBjPGdrIJGORYU9nc4kk9qeCvIQ,4884
 mito_ai/utils/mito_server_utils.py,sha256=bjKCq8I7MS8_5nDdNV-UOfeh3kb4tI0YkyJ93T4U7lE,9367
-mito_ai/utils/open_ai_utils.py,sha256=uyzxXA_YDtC8IBP8PGvS6xEKbbs4jRmgKIbLTyYM2AU,6640
-mito_ai/utils/provider_utils.py,sha256=7kwA60AAsgtIeYIXO3QirRXqlWtC0kJqlkHJCTTG6iw,1763
+mito_ai/utils/model_utils.py,sha256=N0Ny9DYRaS6qpwLweVFC9UnSxtL0fZaQPysh3CEJTb8,9540
+mito_ai/utils/open_ai_utils.py,sha256=bqjHUZjMgqqPWrXlxUJ57SXykLHNi3Kwg9bhrQd_n0U,6476
+mito_ai/utils/provider_utils.py,sha256=XZoxVycvFdZT65rRTnpcRou1nN2XstVMWR8EjG-s64Q,1103
 mito_ai/utils/schema.py,sha256=XAgw3GlAynGQDVIqWhKX_8trdlKuFVXatbUT-km_dhM,3106
 mito_ai/utils/server_limits.py,sha256=PAt1WpyaYRwV59UfHkt2IrGUJTF-NnE2PC143HxLJyA,6335
-mito_ai/utils/telemetry_utils.py,sha256=fsBoeGBLrCMMQ_l9AQ5G85KK8NhWB65BG3wTT86ZaRI,17121
+mito_ai/utils/telemetry_utils.py,sha256=Ymg_juxGHd6OpH08PFmXYhs5Pvf9nvsQkOgR123ponU,17417
 mito_ai/utils/tokens.py,sha256=EZhBnjue4JelL_sMc8n59gIwiFa4Je7XFOSu6o1SkRA,1168
 mito_ai/utils/utils.py,sha256=Ygl84xLOh06GIoqvAW4D5ZfUO_ilSEPcpkLPL4bm3Bg,3224
 mito_ai/utils/version_utils.py,sha256=yT1OJZpH7Q1eABR2BfYNV09hJ6xXv_aFJI58ltfpnvk,2409
 mito_ai/utils/websocket_base.py,sha256=O4xjMHYUhxKN_lOKtVEGl3mW8EXkzdPWt1tUtsbiHgM,2941
-mito_ai-0.1.57.data/data/etc/jupyter/jupyter_server_config.d/mito_ai.json,sha256=jnrJTWHAtiNWoD0APwyPSP0v4oXBO9aAqGwzTCq7ij8,82
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/build_log.json,sha256=nQ05lg2F401-j-bsVB4IUuaBPFfan-E7vnbVlowFkho,24285
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/package.json,sha256=BQWoHYV1ZM4vNB1-eKp5PG6z2bz1GMq8zeYojd_aKYU,7342
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/schemas/mito_ai/package.json.orig,sha256=8oPTdzu9mGaLBvwOjhM2cVrdN31rGuicsfjYAe3dgIY,7200
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/schemas/mito_ai/toolbar-buttons.json,sha256=ec8DpF_QTgroq8BgLhmo1G2ByPPF-_tCnG3qRVq4qRE,894
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.9d26322f3e78beb2b666.js,sha256=leyZnDqn_Ofuuxn2oRCDZMGC6NOfcDkZSMD3vuYIJEY,1629334
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.9d26322f3e78beb2b666.js.map,sha256=GpCQo-jYbrjK1axn6L0eQrYdvzFdwXa2tfluxAU-yZ0,1666950
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/node_modules_process_browser_js.4b128e94d31a81ebd209.js,sha256=5YLXH9YrEcyuTYXT9Ko_VJhj-CEJX9J2K9ILSlypLt8,5831
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/node_modules_process_browser_js.4b128e94d31a81ebd209.js.map,sha256=mUU_J-LI5MzCUorOfESVGeaop5hs344g-agqSgEkTak,6760
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/remoteEntry.79c1ea8a3cda73a4cb6f.js,sha256=TIrdv-vCRp5kNzwBNG0PYlLEekUPc0HpAfgNGRnqpd0,37928
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/remoteEntry.79c1ea8a3cda73a4cb6f.js.map,sha256=g2XuaBv2cxH1FGQeTOGe6oeXBoA0rTRLgqy7W-UONzk,36639
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/style.js,sha256=ymGkha4YxU1j3e7KeoKfqYpcsPp1u3RRuHzMaiUnOhw,150
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/style_index_js.f5d476ac514294615881.js,sha256=I5Tbj3I3be-EAs9zsatymPNQMwvt-OkXztMu9poDrto,34327
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/style_index_js.f5d476ac514294615881.js.map,sha256=QGA8jQzVFE2hVOKLeyhFlcZ_CC6JkgntOGzXpDXMP3Y,29252
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_apis_signOut_mjs-node_module-75790d.688c25857e7b81b1740f.js,sha256=yetgbLQTWpV9FiAdJhwM5jTWzAsTbqYhOckJCwDGwVw,33226
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_apis_signOut_mjs-node_module-75790d.688c25857e7b81b1740f.js.map,sha256=E2rwLG1ARY-9kexmFn88i3ADA-cellPzEZfjZTWegEE,19070
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_tokenProvider_tokenProvider_-72f1c8.a917210f057fcfe224ad.js,sha256=7qZt4qZgqoFq-T_fNAz0Z8cVPKQOHTCrcdOZFZazA8g,307327
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_tokenProvider_tokenProvider_-72f1c8.a917210f057fcfe224ad.js.map,sha256=TdWZQcuq4PCOo6dtkZNsG3mfSfQP0HsnFh_3ANvGDNs,244867
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_dist_esm_index_mjs.6bac1a8c4cc93f15f6b7.js,sha256=LTdWh-QP_Jfd7B1AD1ctH99ud2vuVy9a_F_Dj9qbLwc,57349
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_dist_esm_index_mjs.6bac1a8c4cc93f15f6b7.js.map,sha256=dBmkthEcsYp8b1aZ-SHek5VkKnfR0gegHxpxNWmQshk,39713
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_ui-react_dist_esm_index_mjs.4fcecd65bef9e9847609.js,sha256=HFbycQUh9rIqkJ1lWyKcuk5F8OfiB9-Ah_lnxDUdOk8,2858564
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_ui-react_dist_esm_index_mjs.4fcecd65bef9e9847609.js.map,sha256=5pW1Y1B6SSe6wS4209P-_LQlWDtBS9QsTrhHSVaDm4Y,2274714
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_react-dom_client_js-node_modules_aws-amplify_ui-react_dist_styles_css.b43d4249e4d3dac9ad7b.js,sha256=TXjBjPWd-496FSCl0JcUYIFhxGHByLSu_Po_gb8CasU,743178
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_react-dom_client_js-node_modules_aws-amplify_ui-react_dist_styles_css.b43d4249e4d3dac9ad7b.js.map,sha256=h9TcYmXGZpSa8_1yKWK20mIn-mvrb4rIGPCaymnr8Mo,855442
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_semver_index_js.3f6754ac5116d47de76b.js,sha256=uYUAFa0GVVYcoVvYcw9cvNeF3OdZidQJVIrUE3UtRMQ,84610
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_semver_index_js.3f6754ac5116d47de76b.js.map,sha256=C0IxSaFYPj6nwxyvYGf9LLdBTFTKk93pPYEikFa1YQc,87246
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_vscode-diff_dist_index_js.ea55f1f9346638aafbcf.js,sha256=5gfmIE7bAaJYJZdTp5WuyTW6dKdU545i3X4LTSmiEcM,223110
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_vscode-diff_dist_index_js.ea55f1f9346638aafbcf.js.map,sha256=azV-DYUA9Yp99BcesR3Q0DzRIiD64xVRhE4lDbvK3Sg,245869
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/themes/mito_ai/index.css,sha256=kDe4GLY_3mJu5AM_zaKZAb1zXMcB-lInLpz2OK4vbuQ,25660
-mito_ai-0.1.57.data/data/share/jupyter/labextensions/mito_ai/themes/mito_ai/index.js,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mito_ai-0.1.57.dist-info/METADATA,sha256=gdGCRPmWVL3ethNCEf17x5effXsBv-XYnsaesu08Q8o,7756
-mito_ai-0.1.57.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-mito_ai-0.1.57.dist-info/entry_points.txt,sha256=OGsjpFBUq1QsuAlA3r2PeZXRwnnIWvzfgVhUqmx4qe4,79
-mito_ai-0.1.57.dist-info/licenses/LICENSE,sha256=O2F2Pp4Q1SmfzgYnl8krdrXZOaEo7Chjhk7OTYuGlDw,115
-mito_ai-0.1.57.dist-info/RECORD,,
+mito_ai-0.1.59.data/data/etc/jupyter/jupyter_server_config.d/mito_ai.json,sha256=jnrJTWHAtiNWoD0APwyPSP0v4oXBO9aAqGwzTCq7ij8,82
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/build_log.json,sha256=_k_tr22MR58B08_--UspEugzoyCOZCOGSsEpiXwJcwc,24285
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/package.json,sha256=HeU6qXAuGa95slNbtLxeZhbmne2_0MWcLxU1wu9kB_c,7342
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/schemas/mito_ai/package.json.orig,sha256=vIGKv8KXtKpT678COclNlHlxGhTiiv-DE9pdf4Qvigo,7200
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/schemas/mito_ai/toolbar-buttons.json,sha256=ec8DpF_QTgroq8BgLhmo1G2ByPPF-_tCnG3qRVq4qRE,894
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.44c109c7be36fb884d25.js,sha256=a_OhcAOqy8Nz_w4e8gUUZK5oYhxG02fldBUB7lZ5SYo,1691341
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/lib_index_js.44c109c7be36fb884d25.js.map,sha256=lYWmd0wSYzfAnR-MenQXnOkNFKM8ypnVg3oo1kKRc_M,1746362
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/node_modules_process_browser_js.4b128e94d31a81ebd209.js,sha256=5YLXH9YrEcyuTYXT9Ko_VJhj-CEJX9J2K9ILSlypLt8,5831
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/node_modules_process_browser_js.4b128e94d31a81ebd209.js.map,sha256=mUU_J-LI5MzCUorOfESVGeaop5hs344g-agqSgEkTak,6760
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/remoteEntry.f7decebaf69618541e0f.js,sha256=3s0WyNASl6-rVmtd_-gyZ1inVvVN5C8eoAPHaV1JuaY,37928
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/remoteEntry.f7decebaf69618541e0f.js.map,sha256=cwToi0-VeX56c0xvnV-_kp56u6hyJlx2GaGC8Bz2N7s,36639
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/style.js,sha256=ymGkha4YxU1j3e7KeoKfqYpcsPp1u3RRuHzMaiUnOhw,150
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/style_index_js.f5d476ac514294615881.js,sha256=I5Tbj3I3be-EAs9zsatymPNQMwvt-OkXztMu9poDrto,34327
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/style_index_js.f5d476ac514294615881.js.map,sha256=QGA8jQzVFE2hVOKLeyhFlcZ_CC6JkgntOGzXpDXMP3Y,29252
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_apis_signOut_mjs-node_module-75790d.688c25857e7b81b1740f.js,sha256=yetgbLQTWpV9FiAdJhwM5jTWzAsTbqYhOckJCwDGwVw,33226
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_apis_signOut_mjs-node_module-75790d.688c25857e7b81b1740f.js.map,sha256=E2rwLG1ARY-9kexmFn88i3ADA-cellPzEZfjZTWegEE,19070
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_tokenProvider_tokenProvider_-72f1c8.a917210f057fcfe224ad.js,sha256=7qZt4qZgqoFq-T_fNAz0Z8cVPKQOHTCrcdOZFZazA8g,307327
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_auth_dist_esm_providers_cognito_tokenProvider_tokenProvider_-72f1c8.a917210f057fcfe224ad.js.map,sha256=TdWZQcuq4PCOo6dtkZNsG3mfSfQP0HsnFh_3ANvGDNs,244867
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_dist_esm_index_mjs.6bac1a8c4cc93f15f6b7.js,sha256=LTdWh-QP_Jfd7B1AD1ctH99ud2vuVy9a_F_Dj9qbLwc,57349
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_dist_esm_index_mjs.6bac1a8c4cc93f15f6b7.js.map,sha256=dBmkthEcsYp8b1aZ-SHek5VkKnfR0gegHxpxNWmQshk,39713
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_ui-react_dist_esm_index_mjs.4fcecd65bef9e9847609.js,sha256=HFbycQUh9rIqkJ1lWyKcuk5F8OfiB9-Ah_lnxDUdOk8,2858564
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_aws-amplify_ui-react_dist_esm_index_mjs.4fcecd65bef9e9847609.js.map,sha256=5pW1Y1B6SSe6wS4209P-_LQlWDtBS9QsTrhHSVaDm4Y,2274714
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_react-dom_client_js-node_modules_aws-amplify_ui-react_dist_styles_css.b43d4249e4d3dac9ad7b.js,sha256=TXjBjPWd-496FSCl0JcUYIFhxGHByLSu_Po_gb8CasU,743178
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_react-dom_client_js-node_modules_aws-amplify_ui-react_dist_styles_css.b43d4249e4d3dac9ad7b.js.map,sha256=h9TcYmXGZpSa8_1yKWK20mIn-mvrb4rIGPCaymnr8Mo,855442
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_semver_index_js.3f6754ac5116d47de76b.js,sha256=uYUAFa0GVVYcoVvYcw9cvNeF3OdZidQJVIrUE3UtRMQ,84610
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_semver_index_js.3f6754ac5116d47de76b.js.map,sha256=C0IxSaFYPj6nwxyvYGf9LLdBTFTKk93pPYEikFa1YQc,87246
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_vscode-diff_dist_index_js.ea55f1f9346638aafbcf.js,sha256=5gfmIE7bAaJYJZdTp5WuyTW6dKdU545i3X4LTSmiEcM,223110
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/static/vendors-node_modules_vscode-diff_dist_index_js.ea55f1f9346638aafbcf.js.map,sha256=azV-DYUA9Yp99BcesR3Q0DzRIiD64xVRhE4lDbvK3Sg,245869
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/themes/mito_ai/index.css,sha256=FsbUSRxu-Hb5pxE5GAg1ElnBsRSELMvlRJLgFtBgnAg,27500
+mito_ai-0.1.59.data/data/share/jupyter/labextensions/mito_ai/themes/mito_ai/index.js,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mito_ai-0.1.59.dist-info/METADATA,sha256=yBF3xw899m5PZBrNAJASKvuZE0SVBmNFPontlSgg2rg,7779
+mito_ai-0.1.59.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mito_ai-0.1.59.dist-info/entry_points.txt,sha256=OGsjpFBUq1QsuAlA3r2PeZXRwnnIWvzfgVhUqmx4qe4,79
+mito_ai-0.1.59.dist-info/licenses/LICENSE,sha256=O2F2Pp4Q1SmfzgYnl8krdrXZOaEo7Chjhk7OTYuGlDw,115
+mito_ai-0.1.59.dist-info/RECORD,,

mito_ai/completions/providers.py

@@ -1,284 +0,0 @@
-# Copyright (c) Saga Inc.
-# Distributed under the terms of the GNU Affero General Public License v3.0 License.
-
-from __future__ import annotations
-import asyncio
-from typing import Any, Callable, Dict, List, Optional, Union, cast
-from mito_ai import constants
-from openai.types.chat import ChatCompletionMessageParam
-from traitlets import Instance, Unicode, default, validate
-from traitlets.config import LoggingConfigurable
-from openai.types.chat import ChatCompletionMessageParam
-
-from mito_ai import constants
-from mito_ai.enterprise.utils import is_azure_openai_configured
-from mito_ai.gemini_client import GeminiClient
-from mito_ai.openai_client import OpenAIClient
-from mito_ai.anthropic_client import AnthropicClient
-from mito_ai.logger import get_logger
-from mito_ai.completions.models import (
-    AICapabilities,
-    CompletionError,
-    CompletionItem,
-    CompletionItemError,
-    CompletionReply,
-    CompletionStreamChunk,
-    MessageType,
-    ResponseFormatInfo, CompletionItemError,
-)
-from mito_ai.utils.telemetry_utils import (
-    KEY_TYPE_PARAM,
-    MITO_AI_COMPLETION_ERROR,
-    MITO_AI_COMPLETION_RETRY,
-    MITO_SERVER_KEY,
-    USER_KEY,
-    log,
-    log_ai_completion_error,
-    log_ai_completion_retry,
-    log_ai_completion_success,
-)
-from mito_ai.utils.provider_utils import get_model_provider
-from mito_ai.utils.mito_server_utils import ProviderCompletionException
-
-__all__ = ["OpenAIProvider"]
-
-class OpenAIProvider(LoggingConfigurable):
-    """Provide AI feature through OpenAI services."""
-
-    api_key = Unicode(
-        config=True,
-        allow_none=True,
-        help="OpenAI API key. Default value is read from the OPENAI_API_KEY environment variable.",
-    )
-
-    last_error = Instance(
-        CompletionError,
-        allow_none=True,
-        help="""Last error encountered when using the OpenAI provider.
-
-This attribute is observed by the websocket provider to push the error to the client.""",
-    )
-
-    def __init__(self, **kwargs: Dict[str, Any]) -> None:
-        config = kwargs.get('config', {})
-        if 'api_key' in kwargs:
-            config['OpenAIClient'] = {'api_key': kwargs['api_key']}
-        kwargs['config'] = config
-
-        super().__init__(log=get_logger(), **kwargs)
-        self.last_error = None
-        self._openai_client: Optional[OpenAIClient] = OpenAIClient(**config)
-
-    @property
-    def capabilities(self) -> AICapabilities:
-        """
-        Returns the capabilities of the AI provider.
-        """
-        if constants.CLAUDE_API_KEY and not self.api_key:
-            return AICapabilities(
-                configuration={"model": "<dynamic>"},
-                provider="Claude",
-            )
-        if constants.GEMINI_API_KEY and not self.api_key:
-            return AICapabilities(
-                configuration={"model": "<dynamic>"},
-                provider="Gemini",
-            )
-        if self._openai_client:
-            return self._openai_client.capabilities
-        
-        return AICapabilities(
-            configuration={"model": "<dynamic>"},
-            provider="Mito server",
-        )
-
-    @property
-    def key_type(self) -> str:
-        if constants.CLAUDE_API_KEY and not self.api_key:
-            return "claude"
-        if constants.GEMINI_API_KEY and not self.api_key:
-            return "gemini"
-        if self._openai_client:
-            return self._openai_client.key_type
-        return MITO_SERVER_KEY
-
-    async def request_completions(
-        self,
-        message_type: MessageType,
-        messages: List[ChatCompletionMessageParam],
-        model: str,
-        response_format_info: Optional[ResponseFormatInfo] = None,
-        user_input: Optional[str] = None,
-        thread_id: Optional[str] = None,
-        max_retries: int = 3
-    ) -> str:
-        """
-        Request completions from the AI provider.
-        """
-        self.last_error = None
-        completion = None
-        last_message_content = str(messages[-1].get('content', '')) if messages else ""
-        model_type = get_model_provider(model)
-        
-        # Retry loop
-        for attempt in range(max_retries + 1):
-            try:
-                if model_type == "claude":
-                    api_key = constants.CLAUDE_API_KEY
-                    anthropic_client = AnthropicClient(api_key=api_key)
-                    completion = await anthropic_client.request_completions(messages, model, response_format_info, message_type)
-                elif model_type == "gemini":
-                    api_key = constants.GEMINI_API_KEY
-                    gemini_client = GeminiClient(api_key=api_key)
-                    messages_for_gemini = [dict(m) for m in messages]
-                    completion = await gemini_client.request_completions(messages_for_gemini, model, response_format_info, message_type)
-                elif model_type == "openai":
-                    if not self._openai_client:
-                        raise RuntimeError("OpenAI client is not initialized.")
-                    completion = await self._openai_client.request_completions(
-                        message_type=message_type,
-                        messages=messages,
-                        model=model,
-                        response_format_info=response_format_info
-                    )
-                else:
-                    raise ValueError(f"No AI provider configured for model: {model}")
-                
-                # Success! Log and return
-                log_ai_completion_success(
-                    key_type=USER_KEY if self.key_type == "user" else MITO_SERVER_KEY,
-                    message_type=message_type,
-                    last_message_content=last_message_content,
-                    response={"completion": completion},
-                    user_input=user_input or "",
-                    thread_id=thread_id or "",
-                    model=model
-                )
-                return completion # type: ignore
-            
-            except PermissionError as e:
-                # If we hit a free tier limit, then raise an exception right away without retrying.
-                self.log.exception(f"Error during request_completions: {e}")
-                self.last_error = CompletionError.from_exception(e)
-                log_ai_completion_error('user_key' if self.key_type != MITO_SERVER_KEY else 'mito_server_key', thread_id or "", message_type, e)
-                raise
-
-            except BaseException as e:
-                # Check if we should retry (not on the last attempt)
-                if attempt < max_retries:
-                    # Exponential backoff: wait 2^attempt seconds
-                    wait_time = 2 ** attempt
-                    self.log.info(f"Retrying request_completions after {wait_time}s (attempt {attempt + 1}/{max_retries + 1}): {str(e)}")
-                    log_ai_completion_retry('user_key' if self.key_type != MITO_SERVER_KEY else 'mito_server_key', thread_id or "", message_type, e)
-                    await asyncio.sleep(wait_time)
-                    continue
-                else:
-                    # Final failure after all retries - set error state and raise
-                    self.log.exception(f"Error during request_completions after {attempt + 1} attempts: {e}")
-                    self.last_error = CompletionError.from_exception(e)
-                    log_ai_completion_error('user_key' if self.key_type != MITO_SERVER_KEY else 'mito_server_key', thread_id or "", message_type, e)
-                    raise
-        
-        # This should never be reached due to the raise in the except block,
-        # but added to satisfy the linter
-        raise RuntimeError("Unexpected code path in request_completions")
-
-    async def stream_completions(
-        self,
-        message_type: MessageType,
-        messages: List[ChatCompletionMessageParam],
-        model: str,
-        message_id: str,
-        thread_id: str,
-        reply_fn: Callable[[Union[CompletionReply, CompletionStreamChunk]], None],
-        user_input: Optional[str] = None,
-        response_format_info: Optional[ResponseFormatInfo] = None
-    ) -> str:
-        """
-        Stream completions from the AI provider and return the accumulated response.
-        Returns: The accumulated response string.
-        """
-        self.last_error = None
-        accumulated_response = ""
-        last_message_content = str(messages[-1].get('content', '')) if messages else ""
-        model_type = get_model_provider(model)
-        reply_fn(CompletionReply(
-            items=[
-                CompletionItem(content="", isIncomplete=True, token=message_id)
-            ],
-            parent_id=message_id,
-        ))
-
-        try:
-            if model_type == "claude":
-                api_key = constants.CLAUDE_API_KEY
-                anthropic_client = AnthropicClient(api_key=api_key)
-                accumulated_response = await anthropic_client.stream_completions(
-                    messages=messages,
-                    model=model,
-                    message_type=message_type,
-                    message_id=message_id,
-                    reply_fn=reply_fn
-                )
-            elif model_type == "gemini":
-                api_key = constants.GEMINI_API_KEY
-                gemini_client = GeminiClient(api_key=api_key)
-                # TODO: We shouldn't need to do this because the messages should already be dictionaries... 
-                # but if we do have to do some pre-processing, we should do it in the gemini_client instead.
-                messages_for_gemini = [dict(m) for m in messages]
-                accumulated_response = await gemini_client.stream_completions(
-                    messages=messages_for_gemini,
-                    model=model,
-                    message_id=message_id,
-                    reply_fn=reply_fn,
-                    message_type=message_type
-                )
-            elif model_type == "openai":
-                if not self._openai_client:
-                    raise RuntimeError("OpenAI client is not initialized.")
-                accumulated_response = await self._openai_client.stream_completions(
-                    message_type=message_type,
-                    messages=messages,
-                    model=model,
-                    message_id=message_id,
-                    thread_id=thread_id,
-                    reply_fn=reply_fn,
-                    user_input=user_input,
-                    response_format_info=response_format_info
-                )
-            else:
-                raise ValueError(f"No AI provider configured for model: {model}")
-
-            # Log the successful completion
-            log_ai_completion_success(
-                key_type=USER_KEY if self.key_type == "user" else MITO_SERVER_KEY,
-                message_type=message_type,
-                last_message_content=last_message_content,
-                response={"completion": accumulated_response},
-                user_input=user_input or "",
-                thread_id=thread_id,
-                model=model
-            )
-            return accumulated_response
-
-        except BaseException as e:
-            self.log.exception(f"Error during stream_completions: {e}")
-            self.last_error = CompletionError.from_exception(e)
-            log_ai_completion_error('user_key' if self.key_type != MITO_SERVER_KEY else 'mito_server_key', thread_id, message_type, e)
-
-            # Send error message to client before raising
-            reply_fn(CompletionStreamChunk(
-                parent_id=message_id,
-                chunk=CompletionItem(
-                    content="",
-                    isIncomplete=True,
-                    error=CompletionItemError(
-                        message=f"Failed to process completion: {e!r}"
-                    ),
-                    token=message_id,
-                ),
-                done=True,
-                error=CompletionError.from_exception(e),
-            ))
-            raise
-