vanna 0.4.2__py3-none-any.whl → 0.5.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (13) hide show
  1. vanna/base/base.py +150 -41
  2. vanna/flask/__init__.py +50 -9
  3. vanna/flask/assets.py +16 -16
  4. vanna/hf/__init__.py +1 -0
  5. vanna/hf/hf.py +79 -0
  6. vanna/ollama/ollama.py +94 -69
  7. vanna/opensearch/__init__.py +1 -0
  8. vanna/opensearch/opensearch_vector.py +289 -0
  9. vanna/vllm/__init__.py +1 -0
  10. vanna/vllm/vllm.py +76 -0
  11. {vanna-0.4.2.dist-info → vanna-0.5.0.dist-info}/METADATA +17 -1
  12. {vanna-0.4.2.dist-info → vanna-0.5.0.dist-info}/RECORD +13 -7
  13. {vanna-0.4.2.dist-info → vanna-0.5.0.dist-info}/WHEEL +0 -0
{vanna-0.4.2.dist-info → vanna-0.5.0.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vanna
3
- Version: 0.4.2
3
+ Version: 0.5.0
4
4
  Summary: Generate SQL queries from natural language
5
5
  Author-email: Zain Hoda <zain@vanna.ai>
6
6
  Requires-Python: >=3.9
@@ -15,6 +15,7 @@ Requires-Dist: pandas
15
15
  Requires-Dist: sqlparse
16
16
  Requires-Dist: kaleido
17
17
  Requires-Dist: flask
18
+ Requires-Dist: flask-sock
18
19
  Requires-Dist: sqlalchemy
19
20
  Requires-Dist: psycopg2-binary ; extra == "all"
20
21
  Requires-Dist: db-dtypes ; extra == "all"
@@ -32,6 +33,11 @@ Requires-Dist: google-generativeai ; extra == "all"
32
33
  Requires-Dist: google-cloud-aiplatform ; extra == "all"
33
34
  Requires-Dist: qdrant-client ; extra == "all"
34
35
  Requires-Dist: fastembed ; extra == "all"
36
+ Requires-Dist: ollama ; extra == "all"
37
+ Requires-Dist: httpx ; extra == "all"
38
+ Requires-Dist: opensearch-py ; extra == "all"
39
+ Requires-Dist: opensearch-dsl ; extra == "all"
40
+ Requires-Dist: transformers ; extra == "all"
35
41
  Requires-Dist: anthropic ; extra == "anthropic"
36
42
  Requires-Dist: google-cloud-bigquery ; extra == "bigquery"
37
43
  Requires-Dist: chromadb ; extra == "chromadb"
@@ -39,15 +45,21 @@ Requires-Dist: duckdb ; extra == "duckdb"
39
45
  Requires-Dist: google-generativeai ; extra == "gemini"
40
46
  Requires-Dist: google-generativeai ; extra == "google"
41
47
  Requires-Dist: google-cloud-aiplatform ; extra == "google"
48
+ Requires-Dist: transformers ; extra == "hf"
42
49
  Requires-Dist: marqo ; extra == "marqo"
43
50
  Requires-Dist: mistralai ; extra == "mistralai"
44
51
  Requires-Dist: PyMySQL ; extra == "mysql"
52
+ Requires-Dist: ollama ; extra == "ollama"
53
+ Requires-Dist: httpx ; extra == "ollama"
45
54
  Requires-Dist: openai ; extra == "openai"
55
+ Requires-Dist: opensearch-py ; extra == "opensearch"
56
+ Requires-Dist: opensearch-dsl ; extra == "opensearch"
46
57
  Requires-Dist: psycopg2-binary ; extra == "postgres"
47
58
  Requires-Dist: db-dtypes ; extra == "postgres"
48
59
  Requires-Dist: qdrant-client ; extra == "qdrant"
49
60
  Requires-Dist: snowflake-connector-python ; extra == "snowflake"
50
61
  Requires-Dist: tox ; extra == "test"
62
+ Requires-Dist: vllm ; extra == "vllm"
51
63
  Requires-Dist: zhipuai ; extra == "zhipuai"
52
64
  Project-URL: Bug Tracker, https://github.com/vanna-ai/vanna/issues
53
65
  Project-URL: Homepage, https://github.com/vanna-ai/vanna
@@ -58,14 +70,18 @@ Provides-Extra: chromadb
58
70
  Provides-Extra: duckdb
59
71
  Provides-Extra: gemini
60
72
  Provides-Extra: google
73
+ Provides-Extra: hf
61
74
  Provides-Extra: marqo
62
75
  Provides-Extra: mistralai
63
76
  Provides-Extra: mysql
77
+ Provides-Extra: ollama
64
78
  Provides-Extra: openai
79
+ Provides-Extra: opensearch
65
80
  Provides-Extra: postgres
66
81
  Provides-Extra: qdrant
67
82
  Provides-Extra: snowflake
68
83
  Provides-Extra: test
84
+ Provides-Extra: vllm
69
85
  Provides-Extra: zhipuai
70
86
 
71
87
 
{vanna-0.4.2.dist-info → vanna-0.5.0.dist-info}/RECORD RENAMED
@@ -8,29 +8,35 @@ vanna/ZhipuAI/__init__.py,sha256=NlsijtcZp5Tj9jkOe9fNcOQND_QsGgu7otODsCLBPr0,116
8
8
  vanna/anthropic/__init__.py,sha256=85s_2mAyyPxc0T_0JEvYeAkEKWJwkwqoyUwSC5dw9Gk,43
9
9
  vanna/anthropic/anthropic_chat.py,sha256=Wk0o-NMW1uvR2fhSWxrR_2FqNh-dLprNG4uuVqpqAkY,2615
10
10
  vanna/base/__init__.py,sha256=Sl-HM1RRYzAZoSqmL1CZQmF3ZF-byYTCFQP3JZ2A5MU,28
11
- vanna/base/base.py,sha256=_2vANGAcUe6IrsEhZyFnE6FdO8NMLylfyGlRI4XujWE,58143
11
+ vanna/base/base.py,sha256=IPtPPA1uUI_yblvFRo5wnxVk2_LvWg4I067yIMVjbPE,62126
12
12
  vanna/chromadb/__init__.py,sha256=-iL0nW_g4uM8nWKMuWnNePfN4nb9uk8P3WzGvezOqRg,50
13
13
  vanna/chromadb/chromadb_vector.py,sha256=eKyPck99Y6Jt-BNWojvxLG-zvAERzLSm-3zY-bKXvaA,8792
14
14
  vanna/exceptions/__init__.py,sha256=dJ65xxxZh1lqBeg6nz6Tq_r34jLVmjvBvPO9Q6hFaQ8,685
15
- vanna/flask/__init__.py,sha256=5Du2oK5s-VSLicRPvxMAL1-Gh_jdX847FjJOf5AVapo,23721
16
- vanna/flask/assets.py,sha256=ZESgn0-XrJl4_YV69Lu7Dr7-Y9Eql7xqb7PsoMzrofw,183889
15
+ vanna/flask/__init__.py,sha256=2t9DbgL2-Anm05ESmZ1WAYMd1VSkOUsTf_-exVCebwQ,25192
16
+ vanna/flask/assets.py,sha256=7wMnTS-UPZkNL8SSTrd4U0amuMgeOIIj94WGzBQW1fw,187711
17
17
  vanna/flask/auth.py,sha256=UpKxh7W5cd43W0LGch0VqhncKwB78L6dtOQkl1JY5T0,1246
18
18
  vanna/google/__init__.py,sha256=M-dCxCZcKL4bTQyMLj6r6VRs65YNX9Tl2aoPCuqGm-8,41
19
19
  vanna/google/gemini_chat.py,sha256=ps3A-afFbCo3HeFTLL_nMoQO1PsGvRUUPRUppbMcDew,1584
20
+ vanna/hf/__init__.py,sha256=vD0bIhfLkA1UsvVSF4MAz3Da8aQunkQo3wlDztmMuj0,19
21
+ vanna/hf/hf.py,sha256=v1v6sZnbj5xcrjgmvLP_ytS9NM7E5d0GyMfXXtr6BMU,2703
20
22
  vanna/marqo/__init__.py,sha256=GaAWtJ0B-H5rTY607iLCCrLD7T0zMYM5qWIomEB9gLk,37
21
23
  vanna/marqo/marqo.py,sha256=W7WTtzWp4RJjZVy6OaXHqncUBIPdI4Q7qH7BRCxZ1_A,5242
22
24
  vanna/mistral/__init__.py,sha256=70rTY-69Z2ehkkMj84dNMCukPo6AWdflBGvIB_pztS0,29
23
25
  vanna/mistral/mistral.py,sha256=DAEqAT9SzC91rfMM_S3SuzBZ34MrKHw9qAj6EP2MGVk,1508
24
26
  vanna/ollama/__init__.py,sha256=4xyu8aHPdnEHg5a-QAMwr5o0ns5wevsp_zkI-ndMO2k,27
25
- vanna/ollama/ollama.py,sha256=jfW9VQHAcmzDeo4jF3HJjOMYwAWmptknKqEJaQ0MTno,2418
27
+ vanna/ollama/ollama.py,sha256=Li8quIt8EHSwQGVarEiGHHM3e7NTwrVe_CGJg6yM3bQ,3790
26
28
  vanna/openai/__init__.py,sha256=tGkeQ7wTIPsando7QhoSHehtoQVdYLwFbKNlSmCmNeQ,86
27
29
  vanna/openai/openai_chat.py,sha256=lm-hUsQxu6Q1t06A2csC037zI4VkMk0wFbQ-_Lj74Wg,4764
28
30
  vanna/openai/openai_embeddings.py,sha256=g4pNh9LVcYP9wOoO8ecaccDFWmCUYMInebfHucAa2Gc,1260
31
+ vanna/opensearch/__init__.py,sha256=0unDevWOTs7o8S79TOHUKF1mSiuQbBUVm-7k9jV5WW4,54
32
+ vanna/opensearch/opensearch_vector.py,sha256=T9KH9Lw2DsGzvXL2cbW6UKindoCqlOK___1prPY1RbY,9129
29
33
  vanna/qdrant/__init__.py,sha256=PX_OsDOiPMvwCJ2iGER1drSdQ9AyM8iN5PEBhRb6qqY,73
30
34
  vanna/qdrant/qdrant.py,sha256=XlesB0UniR0LmiXvQ8Ct7o19EdCZPuvUgvyzYSlp94c,11940
31
35
  vanna/types/__init__.py,sha256=Qhn_YscKtJh7mFPCyCDLa2K8a4ORLMGVnPpTbv9uB2U,4957
32
36
  vanna/vannadb/__init__.py,sha256=C6UkYocmO6dmzfPKZaWojN0mI5YlZZ9VIbdcquBE58A,48
33
37
  vanna/vannadb/vannadb_vector.py,sha256=9YwTO3Lh5owWQE7KPMBqLp2EkiGV0RC1sEYhslzJzgI,6168
34
- vanna-0.4.2.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
35
- vanna-0.4.2.dist-info/METADATA,sha256=czAA6dMiYf7ifLS1PApteW64LKQT5jWGNsG9Ygdfz28,10512
36
- vanna-0.4.2.dist-info/RECORD,,
38
+ vanna/vllm/__init__.py,sha256=aNlUkF9tbURdeXAJ8ytuaaF1gYwcG3ny1MfNl_cwQYg,23
39
+ vanna/vllm/vllm.py,sha256=QerC3xF5eNzE_nGBDl6YrPYF4WYnjf0hHxxlDWdKX-0,2427
40
+ vanna-0.5.0.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
41
+ vanna-0.5.0.dist-info/METADATA,sha256=yeg66zu1Sdssn7t9Ma8uVhscSbwNcRLQnTxd_4ZYBRI,11115
42
+ vanna-0.5.0.dist-info/RECORD,,
{vanna-0.4.2.dist-info → vanna-0.5.0.dist-info}/WHEEL RENAMED
File without changes