vanna 0.3.0__tar.gz → 0.3.1__tar.gz

{vanna-0.3.0 → vanna-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vanna
-Version: 0.3.0
+Version: 0.3.1
 Summary: Generate SQL queries from natural language
 Author-email: Zain Hoda <zain@vanna.ai>
 Requires-Python: >=3.9

{vanna-0.3.0 → vanna-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "vanna"
-version = "0.3.0"
+version = "0.3.1"
 authors = [
   { name="Zain Hoda", email="zain@vanna.ai" },
 ]

{vanna-0.3.0 → vanna-0.3.1}/src/vanna/ZhipuAI/ZhipuAI_Chat.py

@@ -1,11 +1,11 @@
 import re
 from typing import List
+
 import pandas as pd
 from zhipuai import ZhipuAI
+
 from ..base import VannaBase
-import re
-from typing import List
-import pandas as pd
+
 
 class ZhipuAI_Chat(VannaBase):
     def __init__(self, config=None):

vanna-0.3.1/src/vanna/ZhipuAI/__init__.py

@@ -0,0 +1,2 @@
+from .ZhipuAI_Chat import ZhipuAI_Chat
+from .ZhipuAI_embeddings import ZhipuAI_Embeddings, ZhipuAIEmbeddingFunction

vanna-0.3.1/src/vanna/anthropic/__init__.py

@@ -0,0 +1 @@
+from .anthropic_chat import Anthropic_Chat

vanna-0.3.1/src/vanna/chromadb/__init__.py

@@ -0,0 +1 @@
+from .chromadb_vector import ChromaDB_VectorStore

vanna-0.3.1/src/vanna/marqo/__init__.py

@@ -0,0 +1 @@
+from .marqo import Marqo_VectorStore

{vanna-0.3.0 → vanna-0.3.1}/src/vanna/marqo/marqo.py

@@ -1,6 +1,4 @@
-import json
 import uuid
-from abc import abstractmethod
 
 import marqo
 import pandas as pd

vanna-0.3.1/src/vanna/mistral/__init__.py

@@ -0,0 +1 @@
+from .mistral import Mistral

vanna-0.3.1/src/vanna/ollama/__init__.py

@@ -0,0 +1 @@
+from .ollama import Ollama

{vanna-0.3.0 → vanna-0.3.1}/src/vanna/ollama/ollama.py

@@ -1,4 +1,3 @@
-import json
 import re
 
 import requests

vanna-0.3.1/src/vanna/openai/__init__.py

@@ -0,0 +1,2 @@
+from .openai_chat import OpenAI_Chat
+from .openai_embeddings import OpenAI_Embeddings

vanna-0.3.1/src/vanna/vannadb/__init__.py

@@ -0,0 +1 @@
+from .vannadb_vector import VannaDB_VectorStore