vanna 0.7.8__py3-none-any.whl → 2.0.0__py3-none-any.whl

vanna-2.0.0.dist-info/RECORD

@@ -0,0 +1,289 @@
+vanna/__init__.py,sha256=Jg6uQOJtpWZtepT4rAHMYuFkfQ_iyUkFSNvaxGhkP3c,3683
+vanna/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vanna/agents/__init__.py,sha256=RiCuaPogay1WtzhK1Qo_aUfSRuoaKe5IPOo6ZucOn7A,116
+vanna/capabilities/__init__.py,sha256=ncWH2kW6uFzHTgpESTq5aA9QQvyPOM1q6V9d588nTBs,392
+vanna/capabilities/agent_memory/__init__.py,sha256=VQeJ68auH3QB1Vl7pcx0AkNwDpuMfnC6CHZq4U7_IQ8,350
+vanna/capabilities/agent_memory/base.py,sha256=h70xeuf1X816v6CVHKZT_gm2Y_IBaNwlAGJscrArzWs,2988
+vanna/capabilities/agent_memory/models.py,sha256=fUhb30TZK3FHQrDYGfjmnAnZi_rcOiz9kmEd6goE_mU,1125
+vanna/capabilities/file_system/__init__.py,sha256=ng5Jb5vJfyWUqe24nXMyeyt6rbu7GfmjJBPq4Z5Ox8o,267
+vanna/capabilities/file_system/base.py,sha256=V2222mLo8i8QPi_qZ2B5quZ6-LofSoRSK2eo5tTXtFA,1859
+vanna/capabilities/file_system/models.py,sha256=xM8PdHqurcoaGrvUCRVmmSx7a3tm5MTV_QjdIa5tROE,464
+vanna/capabilities/sql_runner/__init__.py,sha256=PCBbHNGhifKGvQy4rSmdQ23b27zpBVFMPO50hcyeX4k,217
+vanna/capabilities/sql_runner/base.py,sha256=iVQvP5oJXj5VniXGOOJ1a4qdbSLFgE1M72ehWa5JvzY,826
+vanna/capabilities/sql_runner/models.py,sha256=j8Hm8tZS7j6Om0xWLVgvoqsIPCQgTgiWDJ9QsxSREQM,261
+vanna/components/__init__.py,sha256=bN9K8JhbpC4UJXEKb3Xt13AxKEuHKYWK6fTNtiEQ-Xk,2057
+vanna/components/base.py,sha256=xlPjxdplwQnaqbyWzW-k9G6PFreA6LufzDDn5UsciK8,341
+vanna/components/rich/__init__.py,sha256=18Nqtw5syndy48WsV9e_nkzbklzRNd8ajakFgzKy2lo,1735
+vanna/components/rich/text.py,sha256=F17jNGMRxa04KbBUnkrMEbXoGAMNYgo4PseptnTrDE4,485
+vanna/components/rich/containers/__init__.py,sha256=E1aB4DGIj4YgYzgTBbdUcfVWFCwH0odLvROqnKbilV0,108
+vanna/components/rich/containers/card.py,sha256=zCizdVa123aNpr3mvCWlDWpgymIKWRCvKHDIaSh9BJo,717
+vanna/components/rich/data/__init__.py,sha256=wS-N82SVx2fUmxSpQXwVwvAki82wOqHuaKkhJLpJHkA,171
+vanna/components/rich/data/chart.py,sha256=-GPzLgx8fG8fKp_keB2kw9Vf3VVw64MevAOLe6JaDQo,655
+vanna/components/rich/data/dataframe.py,sha256=Py35Bp5jyhnQaD497ovWZDU16kJuXlmwIe3nEsIkiY4,3166
+vanna/components/rich/feedback/__init__.py,sha256=ew__nU_mjFj-0Bwz-bYYWi9eVYZ3zRSd93Se3b23kRg,630
+vanna/components/rich/feedback/badge.py,sha256=nUzQeAVhZCjDRr66l7a6B-nSwU2hVTTJnEMbGSD_9G0,514
+vanna/components/rich/feedback/icon_text.py,sha256=mV4qSXyZbZrsNk1gludo0n7y7v_IxKqQLyXktzSI7sE,467
+vanna/components/rich/feedback/log_viewer.py,sha256=epu87Xs7NMcEXLs186nhSMjT7KjNBXkwFMEFch4oj8I,1332
+vanna/components/rich/feedback/notification.py,sha256=IF7hLN-5YU454izRt4NH_vrvSU-zC2Pjn7Y2nHqpgIA,670
+vanna/components/rich/feedback/progress.py,sha256=lwzmJfhv_iPEirjWfF7f4b7Azf11CrpUoku31WSV4RM,1312
+vanna/components/rich/feedback/status_card.py,sha256=E4uFmywJHQzl3FBPs1lz2d1IKZSdqAgSgdfJzqxB2tY,1058
+vanna/components/rich/feedback/status_indicator.py,sha256=I95xlO3_6Zitm2WVTEvxx2AvHbMS7wS9F86zqb2NqXQ,425
+vanna/components/rich/interactive/__init__.py,sha256=9JeaOg3qvvFezJjWX6E4X2bFo7oUGodnIpcu3xt7rv0,495
+vanna/components/rich/interactive/button.py,sha256=X3fN-XBo95X1WYeYNjnVnaiYWVlekn_ATgBpPStVt2Y,2942
+vanna/components/rich/interactive/task_list.py,sha256=ib7tmcj5K9oN9T8m0Vw2mPUYiB5k58SYnKQQO5SGb2I,2046
+vanna/components/rich/interactive/ui_state.py,sha256=1un4IPbHCAhtN-EZd7mHDUgJ62yEKE3r4MlOCIGJ3fk,3214
+vanna/components/rich/specialized/__init__.py,sha256=EsTiEaZpWoCZcGasD9iX_4nnOUAY7hO6z6Yc38KXRT4,111
+vanna/components/rich/specialized/artifact.py,sha256=31P1JoA7NQCQ2XvScPAbDV_lclAG1M4E5IWfI5bVM4E,752
+vanna/components/simple/__init__.py,sha256=lghjfASmm6lOaf6CpYDRDJaBt7GnvsdOB3Y2bSfjP9s,405
+vanna/components/simple/image.py,sha256=W7ThF_7TxTKwKYR2m17HdYEXjPXUMiJjQGC-RZI-2kE,487
+vanna/components/simple/link.py,sha256=j101ViJfgzRHrYVyT0p1_hKpaulcwL-2HL_Zv_hs4uQ,473
+vanna/components/simple/text.py,sha256=PLXzvehs0kkHIBzNiLRMZtRFoKNVMzFQdcQQCrSdNCk,341
+vanna/core/__init__.py,sha256=yPCSW0T6nB-rhJdYi35xl0cycdiN3zzmSA0BrYyxsis,4776
+vanna/core/_compat.py,sha256=wt574dz9SS5B33lKPYYELliXaji8QCj4EWt--jFBIrY,414
+vanna/core/component_manager.py,sha256=81P8SWTbzap8yzMwwFv9giyjF4pAAx727B7iTI8JqwM,11311
+vanna/core/components.py,sha256=8TeODhEocr3417HjYd3_JR8R6qD9SCC2kR7ucJR2LRA,1868
+vanna/core/errors.py,sha256=3OltRAYzHr8Ae6PDhezpXcN7X-4YVTfV-cxiGrDIh44,751
+vanna/core/registry.py,sha256=IqVzGQ1ruPas3Bypn9faymmKo382IhGJ6NdlY0rkK5Y,9469
+vanna/core/rich_component.py,sha256=SdpcTHGihc7TyvDg_7c5kvcGda4qRl_2TDEYKPA5fSc,4868
+vanna/core/simple_component.py,sha256=_qel5gccFVcNikxmxV1Q-TNR44Dxo64MVuMeDMvElwc,820
+vanna/core/validation.py,sha256=hAmzWGWGSMg9FgH0sLE9TAIxb51Gn1c-BhZjz_lv1zk,5484
+vanna/core/agent/__init__.py,sha256=4X7VQj9HRw7n70BCVOZ5M46dUSWixxLOIxLs5eNSzfU,187
+vanna/core/agent/agent.py,sha256=BnM7QOHNWTfbC0pgHADQxNqF_5ju-iZ9FCLfBwaI4mM,61689
+vanna/core/agent/config.py,sha256=m7Oco9fJvblU_O77toCuPG8NFJcPzMOjLYvd81m12lk,4415
+vanna/core/audit/__init__.py,sha256=gMXKstK3F46PI6hqqE0GGFSTIdabF1kKTc7vHCGqrjk,623
+vanna/core/audit/base.py,sha256=Y9WbFnkZXbtIP26y4QNP0xRP3EZCbaUjM6rtJz82beM,9726
+vanna/core/audit/models.py,sha256=bvP1XALtHJKN9KuZn00uVBzsS2aObh_0nOUFL0kOZ50,3614
+vanna/core/enhancer/__init__.py,sha256=JoHaQvbpTQ-H4QgXEvG9f0bezDMq3lZPEmSqqL-2e5Q,392
+vanna/core/enhancer/base.py,sha256=lMO-qGMwTv78ty4vCjAUo3DbQCvhrH1dPZ-PWaKKrEk,3094
+vanna/core/enhancer/default.py,sha256=8LAsM8IoXrt2rj0WGYzWQ951C8LWBM5NgH_Fh20mdB0,3988
+vanna/core/enricher/__init__.py,sha256=1p9FCogRiltX5un2GndnQVwQJZd4rOWchfLIyjgK-r8,254
+vanna/core/enricher/base.py,sha256=vU-mAKZ4umccrkhEj8WPiIStPwlw0xW-vvOYwAW6BhQ,1759
+vanna/core/evaluation/__init__.py,sha256=s3cOgEBkBKT4voBC1atwr43A1OC034rywfWl9KwgYfs,2053
+vanna/core/evaluation/base.py,sha256=6oOKdVtfc7ZLQHeuYbXTa6LNaxeXYhcWhC7djSx1oOc,6041
+vanna/core/evaluation/dataset.py,sha256=QFmJAQkL5I9qLrlZ4YH5WZR0z3w4j86SLDE7H0wQDWs,8074
+vanna/core/evaluation/evaluators.py,sha256=gxUxx3Royoqd7aFxgkjqLISAPUC8UEhyTbC3CqimsYA,12574
+vanna/core/evaluation/report.py,sha256=QNKV8y7mUukW6_ryhJiQR5BmWpXIwxdRZnM95fGcaYI,10626
+vanna/core/evaluation/runner.py,sha256=PR5IjsfrtwWOySpsl-jYcyKDhDsFJ6f0MUw88Z2TwcA,10108
+vanna/core/filter/__init__.py,sha256=weGEiVwT9mkE4qOnxFHhbnlAHIisNGT5gF8DRkLbDGg,259
+vanna/core/filter/base.py,sha256=Lt-LDyQsE7JZ_NFEAo1TuNQIIikkD_IUiyBQ3h_9-N8,2035
+vanna/core/lifecycle/__init__.py,sha256=VUxPl2JbQ0UHsP3W-GS_c7P3DJuFgKwi4M7fJ2WJFp8,233
+vanna/core/lifecycle/base.py,sha256=7v6IwcBChFaAN3pa1mOQhBwopNxKmwExosgahG3sT0o,2292
+vanna/core/llm/__init__.py,sha256=oQIl2pJ_KiI0PMJ-LyxDzyDiipS0XLUcVfe6_5cWqCo,324
+vanna/core/llm/base.py,sha256=1vC29W1vjkcrhY43-mOBXJyuYUbqGsXinmAp8QlZ6fE,1067
+vanna/core/llm/models.py,sha256=ZTNfKEzlZOvo8LlQofPWQJH-cfE9uMWyXnGU_u70Ld8,1962
+vanna/core/middleware/__init__.py,sha256=kpBDPRCGmJgmiFt6BBxogguZSLl3PJigZeQMncsuXMU,233
+vanna/core/middleware/base.py,sha256=dMoZNXIMZ-QgR69VFhQ-LgqqSKxTT5Y1BV6DlLHRLwc,1985
+vanna/core/observability/__init__.py,sha256=gc3YSAZ21UO-N6XvYfj92dgydkliWtYh33_A1jDrgLE,284
+vanna/core/observability/base.py,sha256=oF6uysF7e4nUhCOMjuz8IAcZUag3dSMBGPI_PHod3vo,2557
+vanna/core/observability/models.py,sha256=xNwHg9RUew9ruEe_Xxn6cFbaeV3ZSs1b9YaZ88H_ttg,1638
+vanna/core/recovery/__init__.py,sha256=TYprY6foCqnVyt7QJGxvG7QRPTSSTvmLqxQeNUfHvjU,335
+vanna/core/recovery/base.py,sha256=pSHrdPVDXTCULSPoujmRJHvpD0m-xIWFzrWix5tBQFw,2736
+vanna/core/recovery/models.py,sha256=qIi6gDPKBOQwEV-Sy4WBfunJMqVn-Kmy-diN0TYY6ko,811
+vanna/core/storage/__init__.py,sha256=pw6o4nIufsmm0PTVVQ_5Ix5eDB8ArJRV_astUSi_-LE,278
+vanna/core/storage/base.py,sha256=mn1XWKPwaAHOV9dlJ-zcScQ5lmnhhK_1IRY-HBOOH3A,1269
+vanna/core/storage/models.py,sha256=0GRWDEV68wo5n5iAdtH0fMCRnoIqWny9bf9vrOQLSCg,1556
+vanna/core/system_prompt/__init__.py,sha256=P05Kc4zm3Vx_m2UCZIGCwhbQIMP7_HdWzTEIoO_Nu2U,296
+vanna/core/system_prompt/base.py,sha256=WGVQYOMnPVT0UzsOg-hOYBTuMvGaNfWDWS5iujilZ_M,986
+vanna/core/system_prompt/default.py,sha256=WjMksK4ehoV0l1O3MVryCJgAkW4huS2EFYFz00H60wI,6926
+vanna/core/tool/__init__.py,sha256=am-aNBu9Wm13XvSXzcJR6cqE4mk323W3zeSTMJ_RqdY,342
+vanna/core/tool/base.py,sha256=jCzTLErJouvBrh3l0flPX7UYfYwBGgRytXrLDuUI4hM,1871
+vanna/core/tool/models.py,sha256=I8DVISWgj5zHsUbhNL-VjXM4S2X5xRDVweCURJaP_Co,2790
+vanna/core/user/__init__.py,sha256=f7ZLZoNN93ncmQONvhvS5AeYpcIS8TcaLeCAjP6sx_c,339
+vanna/core/user/base.py,sha256=oX7FwA0B_0ZzdoA59rwUU6SlU0Q9QB2rnjkEGYHgPIM,753
+vanna/core/user/models.py,sha256=ovWF_L3-3MHEuJa0yJw-kaMGLe6tMF707Ff1RYuylSA,742
+vanna/core/user/request_context.py,sha256=GSbFTuPLmV2deazQ0xYtCgjhC1Z3oZO989czR430qD0,2156
+vanna/core/user/resolver.py,sha256=0rOsGH8BRF8_9nl3kVWSuROWmpetzpZLLKPOsqX5mlo,1293
+vanna/core/workflow/__init__.py,sha256=DX1NQRhLexHEv1ACh1EGhKMuRtGH1lFSVK6VcxygOL0,475
+vanna/core/workflow/base.py,sha256=1WkQBJ07uvHbtpr7AlM1fY22tCkNRarXnPfTP2SYNx8,10059
+vanna/core/workflow/default.py,sha256=vb10Fq6khJ_nOFmijifrPtg7hspxX2Ynu511szNKBic,31946
+vanna/examples/__init__.py,sha256=9tIIF49_yIpac3eAHYVo_zhbaZD_rd0JZf7R5T7Mu4g,53
+vanna/examples/__main__.py,sha256=UbXodGJzj4Ci9ocjLo3n7UNRexoLUoNn9MauHmi73ow,1396
+vanna/examples/anthropic_quickstart.py,sha256=EQEOH8aHbRMLDnFr7MUyXQ8PPZTPKIs_7dp7lwjRk6g,2314
+vanna/examples/artifact_example.py,sha256=c11ioDhaz7f8apWKEt7vZEHse8DNXxOoVpR7J2s0y0s,12210
+vanna/examples/claude_sqlite_example.py,sha256=vSc6vGey238hhAi6wLfQsyheJofKqZE_4dRbZwwKL1s,7951
+vanna/examples/coding_agent_example.py,sha256=kq0BOqsrOosMoYnKaVmjcJIJhKyofOUEUWP_WqQj_YY,11015
+vanna/examples/custom_system_prompt_example.py,sha256=xGf1H3w1TBxN0QeybBMqEZjuc-9F7bP3IPXaoqdT7IU,5541
+vanna/examples/default_workflow_handler_example.py,sha256=ccuTojIGrkp6vSti_rdHQkw-fYQPxFKE5YTIvjdqobI,7238
+vanna/examples/email_auth_example.py,sha256=MMUe08cydlzTDWmBd7yO93AHzIsL6S0pQ_8kAkvRda8,11527
+vanna/examples/evaluation_example.py,sha256=VIGSoaZaMUR8sta3rSftsnFuN8OhBl1dFqHjuqYWGGE,8521
+vanna/examples/extensibility_example.py,sha256=V7b0bKJzRSo4WAuV1sHKEEE4BHIsTMiCiC9ChsdWX1U,9315
+vanna/examples/minimal_example.py,sha256=9YXilkAcoMszYd3BDvOeJfRMbYFTOE2mqOnrlVAa9es,1939
+vanna/examples/mock_auth_example.py,sha256=R_0qacsD0OO1bpjrxwPUJAuo6kB1dQt8xh9Fa-6szpA,7543
+vanna/examples/mock_custom_tool.py,sha256=WuvmoQ5Jt5Vny1gpPdwwsQLm9WPurpJV60c684upz08,10598
+vanna/examples/mock_quickstart.py,sha256=u7gAHsu7Wp7lDuBxLMbaIZdsNZ23SHx-8Ee3zgsbK_o,2004
+vanna/examples/mock_quota_example.py,sha256=ZpmyGT-oZ69of3bG-QnnxT8AONkmEGgPe17ieUrnI4I,4689
+vanna/examples/mock_rich_components_demo.py,sha256=D2qfdh7BnUDbW-n5z600qkbJIGuD16PgqX5NK1-6BP4,12673
+vanna/examples/mock_sqlite_example.py,sha256=9LzerrjYJMW86S5XPO8d4UTAX-BbiiW8Lbn2wrujIBU,7360
+vanna/examples/openai_quickstart.py,sha256=2zH1koN4gSmr_gdEr5HMVBpcs8AjKxUgXr6SD2KyL60,2493
+vanna/examples/primitive_components_demo.py,sha256=rSHMS0QvMvRnCnNx1gSXZ0-ZgXbukrbEt27G0EjuU9g,9742
+vanna/examples/quota_lifecycle_example.py,sha256=saDKTlgXeB-5qA1-uSKtKpo8PxDbwC-kFY0yaOh4IjY,4470
+vanna/examples/visualization_example.py,sha256=XKG21xczP7WnApOEQ18lanD8NiVyc0OwlsNUnlAqpu8,8387
+vanna/integrations/__init__.py,sha256=H_yFvdXVmW4xDQyAAJAjIlWNdUvGom15rIIWSGRrp-c,383
+vanna/integrations/anthropic/__init__.py,sha256=cJ5LcHEgfPPoe6z9hJ4xciDhMv_l5jr1_59EFcN5cik,164
+vanna/integrations/anthropic/llm.py,sha256=e99O55KYx_PjD1Sex60LxasmsS5JyUX8tfFN085cK1E,10189
+vanna/integrations/azureopenai/__init__.py,sha256=P-fp5mzll1egAo5Gc901-To6MhPyUK1kV8jRLv8HAUc,175
+vanna/integrations/azureopenai/llm.py,sha256=JLO2G4IyG1teYFjXJAmmOzIHoLFT4wfAl0kJ65r0uJ0,12428
+vanna/integrations/azuresearch/__init__.py,sha256=HlwedRqCF-ZzCKQX9Ou1Jztry3esSb2obZQ5q0O3da0,146
+vanna/integrations/azuresearch/agent_memory.py,sha256=zzGQmPOgrkLy5PA63joDa5EprrNB-MAkmUrL_3qIQXs,13823
+vanna/integrations/bigquery/__init__.py,sha256=Sw-krShPv3nOLQIfVQSflCQWXqIQ-gF_eCuE6GHatNY,108
+vanna/integrations/bigquery/sql_runner.py,sha256=65CcGdX6kxOS5wUGSQ558w4_0oH412tV0f5jVwDczx0,2691
+vanna/integrations/chromadb/__init__.py,sha256=sPumYrf9hMVgEHh6rzXdWu9OrVkOucA5_HkQ4LXfIUY,3642
+vanna/integrations/chromadb/agent_memory.py,sha256=B7cx1vnR1VVrb_gWlxXFVbWXUirHpCSZnTANJejEW20,14683
+vanna/integrations/clickhouse/__init__.py,sha256=Ke7P4HrHtgJZTfCnKv2P8flAS6IY-luZLtJCZmDlCig,114
+vanna/integrations/clickhouse/sql_runner.py,sha256=195z-Uij69AwyPOW3nO6ipjgGfA0oJ1TgxLcsGa7VtE,2350
+vanna/integrations/duckdb/__init__.py,sha256=kQiTv0Z5ozcIpL8FaZpZeYfxgbE6eteqIwjlbaqV4Ig,102
+vanna/integrations/duckdb/sql_runner.py,sha256=YOSp8YjoLBiyQ9RU_SKx8p1K4pIfg7d-ET3JT0siXUc,2088
+vanna/integrations/faiss/__init__.py,sha256=NYCQIJFd-vGocpC9tV6pSvNa7_-1aBiYXBJfimKPgdc,120
+vanna/integrations/faiss/agent_memory.py,sha256=yvv_lzqhUODZXZ8qZJS6u0NM6gdsGAt8fMlYcU3xBOU,14223
+vanna/integrations/google/__init__.py,sha256=NaQhnwHA9u7gY4Ohw7jxldMOueoFT3eiRFQaS6kukmY,159
+vanna/integrations/google/gemini.py,sha256=mOI_9vpyZLRpuxy6aCXbuLgxAtpPkgneBGf8ovJJdxg,13091
+vanna/integrations/hive/__init__.py,sha256=SXCrTNqxCZ-RtJkENp9etw6l_XuPIPULGRUiX6tZgrM,96
+vanna/integrations/hive/sql_runner.py,sha256=sPsUnc8UzA-f7ZjyOWJL6gCn7E3pc4cF3hN7m9ZnBVM,2584
+vanna/integrations/local/__init__.py,sha256=cCVj3exs4SlU4wjdtrFykM3iHEC0LgX3F16Dr0XTPLc,408
+vanna/integrations/local/audit.py,sha256=tF0Q4mCDHGrnePwmAOhjqTis3GyLMbwRm_veFfxvufc,1838
+vanna/integrations/local/file_system.py,sha256=BcJelipK03Tisd46xzG-dgouPuqlKpUjO_Yv2uAmOC0,8476
+vanna/integrations/local/file_system_conversation_store.py,sha256=Whs-DKj1hQ7a6buoZDEQ1G-F78homzu-xstSGEos5oE,9133
+vanna/integrations/local/storage.py,sha256=oP9au6sXoMzxNp7aXtjDI-nrAhbzzkDmgTNpXfIkV0A,2272
+vanna/integrations/local/agent_memory/__init__.py,sha256=CE6fYtwc_yMAIIOp6TIaVn8y1NRNd_ENbufv5Fjbbg4,115
+vanna/integrations/local/agent_memory/in_memory.py,sha256=QJ3bPxUc-GUlQJQUqT2u_53c_E8sCeLtsptBS5yCHLY,10061
+vanna/integrations/marqo/__init__.py,sha256=laMRSUJK-HlG7EBQE-bfuFzjbtcrhUOQ2UPbGjb71To,120
+vanna/integrations/marqo/agent_memory.py,sha256=O0HMm2k4FqLR0TE_6ATLIc7uKcavp3SyRT4D8Yjj9uI,11276
+vanna/integrations/milvus/__init__.py,sha256=qkCrd9I5hzceOWFnTPXNjURWc8S79yvr4RwHAmUrorM,123
+vanna/integrations/milvus/agent_memory.py,sha256=INSdkFZKRVT0VxoEeU8kkQm7DJskNkGQs0NnfSVG7I4,14975
+vanna/integrations/mock/__init__.py,sha256=uAVqLQBJzDIXvq_Xv8L0wMf7ieR7TQuh6mbp9vQ9dAM,145
+vanna/integrations/mock/llm.py,sha256=R0j1UYIvixgHpYnqHpeGrv1i2Pfh5IQ-FlicNsVrB3U,2215
+vanna/integrations/mssql/__init__.py,sha256=h8xqXwTHLHojrhVplkciS8KCOWVR-ZlngIpFC8J9Tuo,114
+vanna/integrations/mssql/sql_runner.py,sha256=ot0zeezuti67YyPMbbcLDMopVk1r0dtsrs7PzvIhjeU,2142
+vanna/integrations/mysql/__init__.py,sha256=ZaBjQEjpRDS2hNZ39TxO7W_INCyt2mIEDouVHvirg2A,99
+vanna/integrations/mysql/sql_runner.py,sha256=aHPDVuqNRt1thsskbLld_Pshb3WMMy-Mec_oKGgA4fc,2534
+vanna/integrations/ollama/__init__.py,sha256=qK7lOPCg-Gg2Vhm8vVDmCcU4uu7OJye_lNFuUPYJc3k,112
+vanna/integrations/ollama/llm.py,sha256=o8ac6jYKT_ghyInpWVWmNEzfw62QCLmMjsmb_guJYPM,8668
+vanna/integrations/openai/__init__.py,sha256=_Ws2xJr9oGkMVuyGw92pI6kEbPWyRfLkcOst9_hWa-4,225
+vanna/integrations/openai/llm.py,sha256=-0fqbzfByQRdS64rxvOq8GA1EdHpwhtsQ0d9N52uBiI,10127
+vanna/integrations/openai/responses.py,sha256=akfjzHZhDpUrOjs5oDVw-jWcWQPQwB-F9_OGGyJC18Q,6609
+vanna/integrations/opensearch/__init__.py,sha256=RwDgxyCDYEZPyrsrMtzdb4unW2PeU8OwRtpTU3OAq2M,135
+vanna/integrations/opensearch/agent_memory.py,sha256=S8wLnp9GUahTmzyArS2jFnsbnkawrQFchTpgHybrxaY,13654
+vanna/integrations/oracle/__init__.py,sha256=X2CkfwVqWvuAMq-5ZqFkylivEoHxw77Z9aFvMm0TylU,102
+vanna/integrations/oracle/sql_runner.py,sha256=Bcl8HalwNVVLX-2rm1BgKPt26j8tDiKQn25P5NMo9GM,2272
+vanna/integrations/pinecone/__init__.py,sha256=OImWYE_gZWAHA6QxDD35kLkfl8pU-cEvLMTEaM6fa38,129
+vanna/integrations/pinecone/agent_memory.py,sha256=unABI8z1u7nYL6YN37HbH4e5yHU-n9SubrE59N9W9js,10759
+vanna/integrations/plotly/__init__.py,sha256=VdCFgwN0xXO0ubchvPLK7y4ARKzbtMHIbeQ0h-XnIdk,134
+vanna/integrations/plotly/chart_generator.py,sha256=BOhyb4A3W_r8zIk2XPe0F2LW3E1dyg3SMkBXoVVSZws,11289
+vanna/integrations/postgres/__init__.py,sha256=GIL3WmZ2ozbqwyaXwNUqgGPSL_7rgyVd_TXxAtBR3QE,158
+vanna/integrations/postgres/sql_runner.py,sha256=Rm_RXaWL4y2Eco0ajhWJ3W7raodlK1FP8nER9ZKYawU,3926
+vanna/integrations/premium/agent_memory/__init__.py,sha256=BRn3Lum6bAfRQcpuS3aeligiTvztWNgyEMVkWXJlAg4,121
+vanna/integrations/premium/agent_memory/premium.py,sha256=CbHm1leR4h8cgotErbTMZFZgU85DLIzwYsCtnteWOIw,5840
+vanna/integrations/presto/__init__.py,sha256=OAwhistqxKtAtC7o3S9t8ko2q7zd-wo3hcNTsNvNGEg,102
+vanna/integrations/presto/sql_runner.py,sha256=fE-VP8cWI1x0PILxb0YodlBGvl2FWDLy7o_z6O0IaeU,3556
+vanna/integrations/qdrant/__init__.py,sha256=1y-5knDI3LHrnDr-SCfBVYIXNXQKTgumaQDZkQ0Hp28,123
+vanna/integrations/qdrant/agent_memory.py,sha256=SvArrYYMonsPqTdJddkCHPXMOM2ks2HZ-dBYQZ3eBc0,15602
+vanna/integrations/snowflake/__init__.py,sha256=7OPjDlyB-v61NYeAos-xAUwj1F-q6evwT9DUw3LUZw0,111
+vanna/integrations/snowflake/sql_runner.py,sha256=qFMWkuMfKt0pCuSut2VVcNvFLxp07RhVT2QTGsp_L7c,5303
+vanna/integrations/sqlite/__init__.py,sha256=1iM-djlwwGBpLiIAV-s5jQ5QZNybrFaW--pEPkp3H4Y,146
+vanna/integrations/sqlite/sql_runner.py,sha256=gw3RCZQBIchxr0-abBaVDzbnCpJ0WiJLykFS5EhoHKY,2118
+vanna/integrations/weaviate/__init__.py,sha256=Op5o82-avvZdpy0jsTgrWv9r7BEl26GwJ8UF-8ykT5Q,129
+vanna/integrations/weaviate/agent_memory.py,sha256=82MM5Z8Y8MkUibZi6dH6tIX4GEO_IU_ZIt4PvppATQI,15027
+vanna/legacy/__init__.py,sha256=Foo3mL7kOKIWNfK06Sbf1ycrEKk7Xk8y83WOK6OG9nk,9256
+vanna/legacy/adapter.py,sha256=558Hgj-MPbtxENwScqPUtmvkA-Y90jgmcMss6T1wAQE,17024
+vanna/legacy/local.py,sha256=U5s8ybCRQhBUizi8I69o3jqOpTeu_6KGYY6DMwZxjG4,313
+vanna/legacy/remote.py,sha256=GgB_kcsvUUwkUNMWgOyQ4fiQbkb_8PJp9HCGp0ShKcc,1948
+vanna/legacy/utils.py,sha256=vyKkUz8kjjIg0KXZuazpL4-_IqHtNw-nMJcdzX2rtqo,2202
+vanna/legacy/ZhipuAI/ZhipuAI_Chat.py,sha256=WtZKUBIwlNH0BGbb4lZbVR7pTWIrn7b4RLIk-7u0SuQ,8725
+vanna/legacy/ZhipuAI/ZhipuAI_embeddings.py,sha256=Irpno97LjEJQ1E2Gb71tIVhFuHdc9qxSTAiVpMpSr84,2790
+vanna/legacy/ZhipuAI/__init__.py,sha256=NlsijtcZp5Tj9jkOe9fNcOQND_QsGgu7otODsCLBPr0,116
+vanna/legacy/advanced/__init__.py,sha256=3qe7S1F918moggpyOvPhy88kH7rjaWNVx1o9woIVsJc,672
+vanna/legacy/anthropic/__init__.py,sha256=85s_2mAyyPxc0T_0JEvYeAkEKWJwkwqoyUwSC5dw9Gk,43
+vanna/legacy/anthropic/anthropic_chat.py,sha256=hO3It0pwGet4LpQvSdJVB2jZRk8j2EnyxBeS7eRJ8rY,2661
+vanna/legacy/azuresearch/__init__.py,sha256=tZfvsrCJESiL3EnxA4PrOc5NoO8MXEzCfHX_hnj8n-c,58
+vanna/legacy/azuresearch/azuresearch_vector.py,sha256=AZnPw9vaLQbQhNDRgbEKexxoU9Q_AgqDcUWDxruEgkQ,10128
+vanna/legacy/base/__init__.py,sha256=Sl-HM1RRYzAZoSqmL1CZQmF3ZF-byYTCFQP3JZ2A5MU,28
+vanna/legacy/base/base.py,sha256=s24-mouCE6Mi75V9bHcNBzplxAz2gHSWbH16D8s7_oI,74806
+vanna/legacy/bedrock/__init__.py,sha256=1b4t_shKF5N1fK1vHd-Ogc7aIp9G0TIOHuMDxudGJVw,47
+vanna/legacy/bedrock/bedrock_converse.py,sha256=gfEOfMT1kvYfcvXeW8UkmNAd_gPP7rtNNWX9Ppg6BBo,2842
+vanna/legacy/chromadb/__init__.py,sha256=-iL0nW_g4uM8nWKMuWnNePfN4nb9uk8P3WzGvezOqRg,50
+vanna/legacy/chromadb/chromadb_vector.py,sha256=ZFTbP3YWOnWhFLiQXlcB2EDJ-tAVkv2FgKhV-ypHzFk,8814
+vanna/legacy/cohere/__init__.py,sha256=ulcq7-lmob2Gi8OCoqrZtkzxpC6LQ5SzAtlRUATgzro,86
+vanna/legacy/cohere/cohere_chat.py,sha256=DRrZunZU-h0MR0fcVJiVAdsof0THfEMwnmz6kpe2AaQ,3533
+vanna/legacy/cohere/cohere_embeddings.py,sha256=CFlCmo1f1hd90BQHmfp9xyhhM9EVILZmKZEtLJvYp_Q,2608
+vanna/legacy/deepseek/__init__.py,sha256=7SVY3DGJcNH7GTk7Uq922QM8yZKu3-5IO33WQ_-bgCM,40
+vanna/legacy/deepseek/deepseek_chat.py,sha256=KdyxDnLBVmOmhtHtCjUjWBkjx_isORb08QGcUd8-tpY,1841
+vanna/legacy/exceptions/__init__.py,sha256=dJ65xxxZh1lqBeg6nz6Tq_r34jLVmjvBvPO9Q6hFaQ8,685
+vanna/legacy/faiss/__init__.py,sha256=CX9cDFwHjuQSSlivVK-YlfQ9jQRROrv_MpWqQKFf7hs,25
+vanna/legacy/faiss/faiss.py,sha256=_v0eDgnsOcs_qgQQJU44GcjcCxowA_68MYf5yJZghMY,8938
+vanna/legacy/flask/__init__.py,sha256=XR6JpCQ_5OWXAA2ooqHfKw6hduxtO5KIbyHBTK2BE84,44664
+vanna/legacy/flask/assets.py,sha256=5RuY87MNEeOOnfSO93tok4Z_XGlwTJCmXAEfEcZFzMc,453463
+vanna/legacy/flask/auth.py,sha256=pIJu4pD6E_P47wGKTRiPdXrATrQSmR-YnET5hT2Z-SY,1247
+vanna/legacy/google/__init__.py,sha256=6D8rDBjKJJm_jpVn9b4Vc2NR-R779ed_bnHhWmxCJXE,92
+vanna/legacy/google/bigquery_vector.py,sha256=HjHSYdtDCCI2IjekuTqpKhBtLKY9Hl3_bWPQpbsaALY,9784
+vanna/legacy/google/gemini_chat.py,sha256=jcxK9XUbOG4jX3qiFZS0u2L-3fmtbkQpLNEhILAtpac,2751
+vanna/legacy/hf/__init__.py,sha256=vD0bIhfLkA1UsvVSF4MAz3Da8aQunkQo3wlDztmMuj0,19
+vanna/legacy/hf/hf.py,sha256=28wRtZ57DmF5ScexBXE4iHbnKRTLHiZhwIs7-sMzhzw,3019
+vanna/legacy/marqo/__init__.py,sha256=GaAWtJ0B-H5rTY607iLCCrLD7T0zMYM5qWIomEB9gLk,37
+vanna/legacy/marqo/marqo.py,sha256=W7WTtzWp4RJjZVy6OaXHqncUBIPdI4Q7qH7BRCxZ1_A,5242
+vanna/legacy/milvus/__init__.py,sha256=VBasJG2eTKbJI6CEand7kPLNBrqYrn0QCAhSYVz814s,46
+vanna/legacy/milvus/milvus_vector.py,sha256=WTNk7H1vRnMW_TTfzVKvzSuWbwTrlMvenNyNmrwSMZI,11642
+vanna/legacy/mistral/__init__.py,sha256=70rTY-69Z2ehkkMj84dNMCukPo6AWdflBGvIB_pztS0,29
+vanna/legacy/mistral/mistral.py,sha256=rcdgmUSQniLkah2VL23VGYRa9WXpOy_dZN4S0kc__V8,1494
+vanna/legacy/mock/__init__.py,sha256=nYR2WfcV5NdwpK3V64QGOWHBGc3ESN9uV68JLS76aRw,97
+vanna/legacy/mock/embedding.py,sha256=ggnP7KuPh6dlqeUFtoN8t0J0P7_yRNtn9rIq6h8g8-w,250
+vanna/legacy/mock/llm.py,sha256=GaKkpt6GNdXI9DK9zUhTgRVmo51waDFKt-ZaBMYf2us,517
+vanna/legacy/mock/vectordb.py,sha256=RXkRr37quh92L79y9aCxouY3PB7gT9ZZoF1zczqeO60,3223
+vanna/legacy/ollama/__init__.py,sha256=4xyu8aHPdnEHg5a-QAMwr5o0ns5wevsp_zkI-ndMO2k,27
+vanna/legacy/ollama/ollama.py,sha256=S6S2ZBB_NBOYiPuUdYFoIRDh8dlCEnveXIb96zlkN08,4122
+vanna/legacy/openai/__init__.py,sha256=tGkeQ7wTIPsando7QhoSHehtoQVdYLwFbKNlSmCmNeQ,86
+vanna/legacy/openai/openai_chat.py,sha256=Y-epFv8KRLBi2aVHssUMaih4FGeg7-UZnbyY7k0Gi0g,4367
+vanna/legacy/openai/openai_embeddings.py,sha256=g4pNh9LVcYP9wOoO8ecaccDFWmCUYMInebfHucAa2Gc,1260
+vanna/legacy/opensearch/__init__.py,sha256=dc9fNtIrOOpkSGp_JKOhGOk26ffyK6W1bm_Cdn9X09I,126
+vanna/legacy/opensearch/opensearch_vector.py,sha256=0BeUbrf80W5wFgNawUSAfYgGZ8fYP0Yg_nswJGq7yhQ,13465
+vanna/legacy/opensearch/opensearch_vector_semantic.py,sha256=rTedjqojDKKrmElGeAbgbC5HGt0mlALeXG1YFtg4W4w,7516
+vanna/legacy/oracle/__init__.py,sha256=lE9IB9nK4wsAQ0KdAqoidMoLH80QBsB1HRbI0GQJh8c,46
+vanna/legacy/oracle/oracle_vector.py,sha256=na85gzo1gqHZ7L_vokf-7zqWepVfsfC3IKn57LhZ-PI,17533
+vanna/legacy/pgvector/__init__.py,sha256=7Wvu9qcNdNvZu26Dn53jhO9YXELm0_YsrwBab4BdgVM,37
+vanna/legacy/pgvector/pgvector.py,sha256=r-bmR87dNc68r-rQvvd22LeOPemYWIuaUauQf7MqJSs,10398
+vanna/legacy/pinecone/__init__.py,sha256=eO5l8aX8vKL6aIUMgAXGPt1jdqKxB_Hic6cmoVAUrD0,90
+vanna/legacy/pinecone/pinecone_vector.py,sha256=DWJ6USFSGfcFQWC4X9viJddtahAUcDmYq8jrzetB1VE,11445
+vanna/legacy/qdrant/__init__.py,sha256=PX_OsDOiPMvwCJ2iGER1drSdQ9AyM8iN5PEBhRb6qqY,73
+vanna/legacy/qdrant/qdrant.py,sha256=bEnjt-kL36nuIcXtt2adN4qpiI67VeJ1r0apDkLiAsw,12483
+vanna/legacy/qianfan/Qianfan_Chat.py,sha256=luz1ceYQBj1SxkqmpmJqVoIeJwTEu-lnerFHSQfD86w,6099
+vanna/legacy/qianfan/Qianfan_embeddings.py,sha256=6DnOGHbAUhjdHlcXejV2v6zaTG9_p0N-AuhCPKj5EVM,1078
+vanna/legacy/qianfan/__init__.py,sha256=QpR43BjZQZcrcDRkyYcYiS-kyqtYmu23AHDzK0Wy1D0,90
+vanna/legacy/qianwen/QianwenAI_chat.py,sha256=lOC7X7UAzjo0YeuKobdxFww-hCzDp05kak76s4zqZLY,4673
+vanna/legacy/qianwen/QianwenAI_embeddings.py,sha256=55cwKpB_N3OVgXkC8uSGQCaIAK8vojz2UnlANtiXWS8,1253
+vanna/legacy/qianwen/__init__.py,sha256=fBl4zQTpvObGRNJV6EVNjIUQ9aKDDYq-zLPsEZrRpwg,98
+vanna/legacy/types/__init__.py,sha256=Qhn_YscKtJh7mFPCyCDLa2K8a4ORLMGVnPpTbv9uB2U,4957
+vanna/legacy/vannadb/__init__.py,sha256=C6UkYocmO6dmzfPKZaWojN0mI5YlZZ9VIbdcquBE58A,48
+vanna/legacy/vannadb/vannadb_vector.py,sha256=fjkxM6Ka-Nk5NF6BrRgdywto2d0mQl67t6YBUETY1N4,15047
+vanna/legacy/vllm/__init__.py,sha256=aNlUkF9tbURdeXAJ8ytuaaF1gYwcG3ny1MfNl_cwQYg,23
+vanna/legacy/vllm/vllm.py,sha256=QJp8KWrWXfrexRuLMYXuOHqP3nTdp4TJGZkQ6dPgwfE,3083
+vanna/legacy/weaviate/__init__.py,sha256=HL6PAl7ePBAkeG8uln-BmM7IUtWohyTPvDfcPzSGSCg,46
+vanna/legacy/weaviate/weaviate_vector.py,sha256=0ojdzetfhnFDvgspObZoWvRyohEhqYS4roTzLLREnD8,7861
+vanna/legacy/xinference/__init__.py,sha256=EFW_sz-BSB2XgmjACOTZmneeIk3I2EiWgue-VVJpnB0,35
+vanna/legacy/xinference/xinference.py,sha256=8HjddUobASQCWLHB380Q4p65LYwpMyBL9b9I5PBEmec,1880
+vanna/servers/__init__.py,sha256=00YPxpFE0k3_RlDmH28IhGIh5wtlZ7EMUFQQctLdnAM,412
+vanna/servers/__main__.py,sha256=1ZzZN4MKGFNv6zLLFKt7Bsh_GG1n8qcvxp8sel5euto,130
+vanna/servers/base/__init__.py,sha256=hfLUdpbQSvph0BzZptRKRxmOJ_HHMws_x7Pb3d4ITmk,407
+vanna/servers/base/chat_handler.py,sha256=Rwk4B3JedG5XZHMp62sMVXqaX3H6Z2b0RofSRbVQ_UE,1777
+vanna/servers/base/models.py,sha256=6x372PTrfmLTmOP2Vw7eV4WHAHMC6VbetFuFJnGyzmM,3767
+vanna/servers/base/rich_chat_handler.py,sha256=O9rhreMNeAj6vhT536_7U2caHCmJYSLvM6mcKECmuWw,5491
+vanna/servers/base/templates.py,sha256=4ymcR7HPu1Ul724DssJyw9XvQw0KCSJqGvvTcFmMeLk,14418
+vanna/servers/cli/__init__.py,sha256=cCZ0emLtpJe1-Z6sC_0JYgpZR31rJNntGAQuhac457k,130
+vanna/servers/cli/server_runner.py,sha256=oIpjWM_n9RrBFIIJUH6B8KUIRtiGCc8qDeN13tLLhIE,7010
+vanna/servers/fastapi/__init__.py,sha256=f20P_lgHgOL0E2Wx7E4aFZff9_SsJFXAAbdHf_xk5Nc,127
+vanna/servers/fastapi/app.py,sha256=Nnb11tMn989-zluiR45NstUJWU3Rzrlr9_ZvVFc_PPA,5518
+vanna/servers/fastapi/routes.py,sha256=Pu-BELcQR5N7IbVRHGxAw1wjzmXxgt_zwErQwzfqt48,6907
+vanna/servers/flask/__init__.py,sha256=9kVqW_P0oQu-4yVkKitJreXBZ6UcJYTxej8IV1oZszU,121
+vanna/servers/flask/app.py,sha256=rSFVtXRdk4713dAcvLug9G6FIlPAnf8Lita4fSSjK7A,4118
+vanna/servers/flask/routes.py,sha256=p6ehSCQoFLsXUOd3ynPKth0TEqZPpvxabQA-BTBrp5w,4763
+vanna/tools/__init__.py,sha256=HDEAWC6Nsx695qnwIFut4o8uwQpiiHMDvcP_AdnJVl4,865
+vanna/tools/agent_memory.py,sha256=SIuCrEdOU7buFilCTX7L8_QWZ4mAWstsKp7s_DlhrYA,12243
+vanna/tools/file_system.py,sha256=wQpyXUUAjV7lcTYSRzXvqrXmYF6MFfGUH4i72AA0suM,29840
+vanna/tools/python.py,sha256=hPMEevMM4CPtLGOtutgMm6jdssK1f8Lq3k_DtvxSQzA,6582
+vanna/tools/run_sql.py,sha256=RGptT_cRfD6FP-3MnTS_pgYahWJbDSb3gv3yZ-5sR3g,6951
+vanna/tools/visualize_data.py,sha256=qgPMTxEAmbtJjb2j-qtAG1ZwtzvBvd8gdkzGMki_YGE,7551
+vanna/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vanna/web_components/__init__.py,sha256=QLzEJU3ersllv4pgZpaKdws3S3u5vDyTyUivawMc4A0,1040
+vanna-2.0.0.dist-info/entry_points.txt,sha256=lWU4rLAVNmrWnY6E66z_2TQW3QHU2Kzhwu2qvC6EBrs,62
+vanna-2.0.0.dist-info/licenses/LICENSE,sha256=VYiPMMDqj9BcxUkAYqrAzJpn5tCFXCrnglfRqS5l9Rk,1065
+vanna-2.0.0.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+vanna-2.0.0.dist-info/METADATA,sha256=QFnzdsOREK8EnDUTKOqruqNRt6nuL1IZBX0vTeI3ysI,16519
+vanna-2.0.0.dist-info/RECORD,,

vanna-2.0.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+vanna=vanna.servers.cli.server_runner:main
+

vanna/bedrock/__init__.py

@@ -1 +0,0 @@
-from .bedrock_converse import Bedrock_Converse

vanna/cohere/__init__.py

@@ -1,2 +0,0 @@
-from .cohere_chat import Cohere_Chat
-from .cohere_embeddings import Cohere_Embeddings 

vanna/faiss/__init__.py

@@ -1 +0,0 @@
-from .faiss import FAISS

vanna/mock/vectordb.py

@@ -1,55 +0,0 @@
-import pandas as pd
-
-from ..base import VannaBase
-
-
-class MockVectorDB(VannaBase):
-    def __init__(self, config=None):
-        pass
-
-    def _get_id(self, value: str, **kwargs) -> str:
-        # Hash the value and return the ID
-        return str(hash(value))
-
-    def add_ddl(self, ddl: str, **kwargs) -> str:
-        return self._get_id(ddl)
-
-    def add_documentation(self, doc: str, **kwargs) -> str:
-        return self._get_id(doc)
-
-    def add_question_sql(self, question: str, sql: str, **kwargs) -> str:
-        return self._get_id(question)
-
-    def get_related_ddl(self, question: str, **kwargs) -> list:
-        return []
-
-    def get_related_documentation(self, question: str, **kwargs) -> list:
-        return []
-
-    def get_similar_question_sql(self, question: str, **kwargs) -> list:
-        return []
-
-    def get_training_data(self, **kwargs) -> pd.DataFrame:
-        return pd.DataFrame({'id': {0: '19546-ddl',
-  1: '91597-sql',
-  2: '133976-sql',
-  3: '59851-doc',
-  4: '73046-sql'},
- 'training_data_type': {0: 'ddl',
-  1: 'sql',
-  2: 'sql',
-  3: 'documentation',
-  4: 'sql'},
- 'question': {0: None,
-  1: 'What are the top selling genres?',
-  2: 'What are the low 7 artists by sales?',
-  3: None,
-  4: 'What is the total sales for each customer?'},
- 'content': {0: 'CREATE TABLE [Invoice]\n(\n    [InvoiceId] INTEGER  NOT NULL,\n    [CustomerId] INTEGER  NOT NULL,\n    [InvoiceDate] DATETIME  NOT NULL,\n    [BillingAddress] NVARCHAR(70),\n    [BillingCity] NVARCHAR(40),\n    [BillingState] NVARCHAR(40),\n    [BillingCountry] NVARCHAR(40),\n    [BillingPostalCode] NVARCHAR(10),\n    [Total] NUMERIC(10,2)  NOT NULL,\n    CONSTRAINT [PK_Invoice] PRIMARY KEY  ([InvoiceId]),\n    FOREIGN KEY ([CustomerId]) REFERENCES [Customer] ([CustomerId]) \n\t\tON DELETE NO ACTION ON UPDATE NO ACTION\n)',
-  1: 'SELECT g.Name AS Genre, SUM(il.Quantity) AS TotalSales\nFROM Genre g\nJOIN Track t ON g.GenreId = t.GenreId\nJOIN InvoiceLine il ON t.TrackId = il.TrackId\nGROUP BY g.GenreId, g.Name\nORDER BY TotalSales DESC;',
-  2: 'SELECT a.ArtistId, a.Name, SUM(il.Quantity) AS TotalSales\nFROM Artist a\nINNER JOIN Album al ON a.ArtistId = al.ArtistId\nINNER JOIN Track t ON al.AlbumId = t.AlbumId\nINNER JOIN InvoiceLine il ON t.TrackId = il.TrackId\nGROUP BY a.ArtistId, a.Name\nORDER BY TotalSales ASC\nLIMIT 7;',
-  3: 'This is a SQLite database. For dates rememeber to use SQLite syntax.',
-  4: 'SELECT c.CustomerId, c.FirstName, c.LastName, SUM(i.Total) AS TotalSales\nFROM Customer c\nJOIN Invoice i ON c.CustomerId = i.CustomerId\nGROUP BY c.CustomerId, c.FirstName, c.LastName;'}})
-
-    def remove_training_data(id: str, **kwargs) -> bool:
-        return True

vanna/ollama/ollama.py

@@ -1,103 +0,0 @@
-import json
-import re
-
-from httpx import Timeout
-
-from ..base import VannaBase
-from ..exceptions import DependencyError
-
-
-class Ollama(VannaBase):
-  def __init__(self, config=None):
-
-    try:
-      ollama = __import__("ollama")
-    except ImportError:
-      raise DependencyError(
-        "You need to install required dependencies to execute this method, run command:"
-        " \npip install ollama"
-      )
-
-    if not config:
-      raise ValueError("config must contain at least Ollama model")
-    if 'model' not in config.keys():
-      raise ValueError("config must contain at least Ollama model")
-    self.host = config.get("ollama_host", "http://localhost:11434")
-    self.model = config["model"]
-    if ":" not in self.model:
-      self.model += ":latest"
-
-    self.ollama_timeout = config.get("ollama_timeout", 240.0)
-
-    self.ollama_client = ollama.Client(self.host, timeout=Timeout(self.ollama_timeout))
-    self.keep_alive = config.get('keep_alive', None)
-    self.ollama_options = config.get('options', {})
-    self.num_ctx = self.ollama_options.get('num_ctx', 2048)
-    self.__pull_model_if_ne(self.ollama_client, self.model)
-
-  @staticmethod
-  def __pull_model_if_ne(ollama_client, model):
-    model_response = ollama_client.list()
-    model_lists = [model_element['model'] for model_element in
-                   model_response.get('models', [])]
-    if model not in model_lists:
-      ollama_client.pull(model)
-
-  def system_message(self, message: str) -> any:
-    return {"role": "system", "content": message}
-
-  def user_message(self, message: str) -> any:
-    return {"role": "user", "content": message}
-
-  def assistant_message(self, message: str) -> any:
-    return {"role": "assistant", "content": message}
-
-  def extract_sql(self, llm_response):
-    """
-    Extracts the first SQL statement after the word 'select', ignoring case,
-    matches until the first semicolon, three backticks, or the end of the string,
-    and removes three backticks if they exist in the extracted string.
-
-    Args:
-    - llm_response (str): The string to search within for an SQL statement.
-
-    Returns:
-    - str: The first SQL statement found, with three backticks removed, or an empty string if no match is found.
-    """
-    # Remove ollama-generated extra characters
-    llm_response = llm_response.replace("\\_", "_")
-    llm_response = llm_response.replace("\\", "")
-
-    # Regular expression to find ```sql' and capture until '```'
-    sql = re.search(r"```sql\n((.|\n)*?)(?=;|\[|```)", llm_response, re.DOTALL)
-    # Regular expression to find 'select, with (ignoring case) and capture until ';', [ (this happens in case of mistral) or end of string
-    select_with = re.search(r'(select|(with.*?as \())(.*?)(?=;|\[|```)',
-                            llm_response,
-                            re.IGNORECASE | re.DOTALL)
-    if sql:
-      self.log(
-        f"Output from LLM: {llm_response} \nExtracted SQL: {sql.group(1)}")
-      return sql.group(1).replace("```", "")
-    elif select_with:
-      self.log(
-        f"Output from LLM: {llm_response} \nExtracted SQL: {select_with.group(0)}")
-      return select_with.group(0)
-    else:
-      return llm_response
-
-  def submit_prompt(self, prompt, **kwargs) -> str:
-    self.log(
-      f"Ollama parameters:\n"
-      f"model={self.model},\n"
-      f"options={self.ollama_options},\n"
-      f"keep_alive={self.keep_alive}")
-    self.log(f"Prompt Content:\n{json.dumps(prompt, ensure_ascii=False)}")
-    response_dict = self.ollama_client.chat(model=self.model,
-                                            messages=prompt,
-                                            stream=False,
-                                            options=self.ollama_options,
-                                            keep_alive=self.keep_alive)
-
-    self.log(f"Ollama Response:\n{str(response_dict)}")
-
-    return response_dict['message']['content']