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

vanna/qdrant/qdrant.py

@@ -79,7 +79,7 @@ class Qdrant_VectorStore(VannaBase):
         self._setup_collections()
 
     def add_question_sql(self, question: str, sql: str, **kwargs) -> str:
-        question_answer = format("Question: {0}\n\nSQL: {1}", question, sql)
+        question_answer = "Question: {0}\n\nSQL: {1}".format(question, sql)
         id = deterministic_uuid(question_answer)
 
         self._client.upsert(
@@ -157,7 +157,7 @@ class Qdrant_VectorStore(VannaBase):
         if ddl_data := self._get_all_points(DDL_COLLECTION_NAME):
             ddl_list = [data.payload["ddl"] for data in ddl_data]
             id_list = [
-                self._format_point_id(data.id, DDL_COLLECTION_NAME) for data in sql_data
+                self._format_point_id(data.id, DDL_COLLECTION_NAME) for data in ddl_data
             ]
 
             df_ddl = pd.DataFrame(
@@ -172,8 +172,6 @@ class Qdrant_VectorStore(VannaBase):
 
             df = pd.concat([df, df_ddl])
 
-        doc_data = self.documentation_collection.get()
-
         if doc_data := self._get_all_points(DOCUMENTATION_COLLECTION_NAME):
             document_list = [data.payload["documentation"] for data in doc_data]
             id_list = [

{vanna-0.5.0.dist-info → vanna-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vanna
-Version: 0.5.0
+Version: 0.5.1
 Summary: Generate SQL queries from natural language
 Author-email: Zain Hoda <zain@vanna.ai>
 Requires-Python: >=3.9
@@ -57,6 +57,7 @@ Requires-Dist: opensearch-dsl ; extra == "opensearch"
 Requires-Dist: psycopg2-binary ; extra == "postgres"
 Requires-Dist: db-dtypes ; extra == "postgres"
 Requires-Dist: qdrant-client ; extra == "qdrant"
+Requires-Dist: fastembed ; extra == "qdrant"
 Requires-Dist: snowflake-connector-python ; extra == "snowflake"
 Requires-Dist: tox ; extra == "test"
 Requires-Dist: vllm ; extra == "vllm"
@@ -111,7 +112,7 @@ Vanna works in two easy steps - train a RAG "model" on your data, and then ask q
 
 If you don't know what RAG is, don't worry -- you don't need to know how this works under the hood to use it. You just need to know that you "train" a model, which stores some metadata and then use it to "ask" questions.
 
-See the [base class](src/vanna/base/base.py) for more details on how this works under the hood.
+See the [base class](https://github.com/vanna-ai/vanna/blob/main/src/vanna/base/base.py) for more details on how this works under the hood.
 
 ## User Interfaces
 These are some of the user interfaces that we've built using Vanna. You can use these as-is or as a starting point for your own custom interface.
@@ -304,7 +305,7 @@ Fine-Tuning
     - Expose to your end users via Slackbot, web app, Streamlit app, or a custom front end.
 
 ## Extending Vanna
-Vanna is designed to connect to any database, LLM, and vector database. There's a [VannaBase](src/vanna/base/base.py) abstract base class that defines some basic functionality. The package provides implementations for use with OpenAI and ChromaDB. You can easily extend Vanna to use your own LLM or vector database. See the [documentation](https://vanna.ai/docs/) for more details.
+Vanna is designed to connect to any database, LLM, and vector database. There's a [VannaBase](https://github.com/vanna-ai/vanna/blob/main/src/vanna/base/base.py) abstract base class that defines some basic functionality. The package provides implementations for use with OpenAI and ChromaDB. You can easily extend Vanna to use your own LLM or vector database. See the [documentation](https://vanna.ai/docs/) for more details.
 
 ## Vanna in 100 Seconds
 

{vanna-0.5.0.dist-info → vanna-0.5.1.dist-info}/RECORD

@@ -31,12 +31,12 @@ vanna/openai/openai_embeddings.py,sha256=g4pNh9LVcYP9wOoO8ecaccDFWmCUYMInebfHucA
 vanna/opensearch/__init__.py,sha256=0unDevWOTs7o8S79TOHUKF1mSiuQbBUVm-7k9jV5WW4,54
 vanna/opensearch/opensearch_vector.py,sha256=T9KH9Lw2DsGzvXL2cbW6UKindoCqlOK___1prPY1RbY,9129
 vanna/qdrant/__init__.py,sha256=PX_OsDOiPMvwCJ2iGER1drSdQ9AyM8iN5PEBhRb6qqY,73
-vanna/qdrant/qdrant.py,sha256=XlesB0UniR0LmiXvQ8Ct7o19EdCZPuvUgvyzYSlp94c,11940
+vanna/qdrant/qdrant.py,sha256=OejFoHB0z0KOvTY6cR_kaXkOChFgE_qjIef4d01PAsY,11883
 vanna/types/__init__.py,sha256=Qhn_YscKtJh7mFPCyCDLa2K8a4ORLMGVnPpTbv9uB2U,4957
 vanna/vannadb/__init__.py,sha256=C6UkYocmO6dmzfPKZaWojN0mI5YlZZ9VIbdcquBE58A,48
 vanna/vannadb/vannadb_vector.py,sha256=9YwTO3Lh5owWQE7KPMBqLp2EkiGV0RC1sEYhslzJzgI,6168
 vanna/vllm/__init__.py,sha256=aNlUkF9tbURdeXAJ8ytuaaF1gYwcG3ny1MfNl_cwQYg,23
 vanna/vllm/vllm.py,sha256=QerC3xF5eNzE_nGBDl6YrPYF4WYnjf0hHxxlDWdKX-0,2427
-vanna-0.5.0.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-vanna-0.5.0.dist-info/METADATA,sha256=yeg66zu1Sdssn7t9Ma8uVhscSbwNcRLQnTxd_4ZYBRI,11115
-vanna-0.5.0.dist-info/RECORD,,
+vanna-0.5.1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+vanna-0.5.1.dist-info/METADATA,sha256=AH1wwjCDLl9P4v9Oq7zWW6Dx13jtpRrRVQEqwQlDWtU,11248
+vanna-0.5.1.dist-info/RECORD,,