nvidia-nat-vanna 1.4.0b2__py3-none-any.whl

nvidia_nat_vanna-1.4.0b2.dist-info/METADATA

@@ -0,0 +1,149 @@
+Metadata-Version: 2.4
+Name: nvidia-nat-vanna
+Version: 1.4.0b2
+Summary: Vanna-based Text-to-SQL integration for NeMo Agent Toolkit with Databricks support
+Keywords: ai,agents,text2sql,vanna,sql,database
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: <3.14,>=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: databricks-sql-connector~=4.1.4
+Requires-Dist: databricks-sqlalchemy~=2.0.8
+Requires-Dist: langchain-core~=0.3.76
+Requires-Dist: nvidia-nat[langchain]~=1.4
+Requires-Dist: pandas~=2.0
+Requires-Dist: pymilvus[model]~=2.5.11
+Requires-Dist: sqlglot~=26.33.0
+Requires-Dist: vanna[chromadb]~=0.7.9
+
+<!--
+SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+-->
+
+# NVIDIA NeMo Agent Toolkit Vanna
+
+Vanna-based Text-to-SQL integration for NeMo Agent toolkit.
+
+## Overview
+
+This package provides production-ready text-to-SQL capabilities using the Vanna framework with Databricks support.
+
+## Features
+
+- **AI-Powered SQL Generation**: Convert natural language to SQL using LLMs
+- **Databricks Support**: Optimized for Databricks SQL warehouses
+- **Vector-Based Similarity Search**: Milvus integration for few-shot learning
+- **Streaming Support**: Real-time progress updates
+- **Query Execution**: Optional database execution with formatted results
+- **Highly Configurable**: Customizable prompts, examples, and connections
+
+## Quick Start
+
+Install the package:
+
+```bash
+pip install nvidia-nat-vanna
+```
+
+Create a workflow configuration:
+
+```yaml
+functions:
+  text2sql:
+    _type: text2sql
+    llm_name: my_llm
+    embedder_name: my_embedder
+    milvus_retriever: my_retriever
+    database_type: databricks
+    connection_url: "${CONNECTION_URL}"
+    execute_sql: false
+
+  execute_db_query:
+    _type: execute_db_query
+    database_type: databricks
+    connection_url: "${CONNECTION_URL}"
+    max_rows: 100
+
+llms:
+  my_llm:
+    _type: nim
+    model_name: meta/llama-3.1-70b-instruct
+    api_key: "${NVIDIA_API_KEY}"
+
+embedders:
+  my_embedder:
+    _type: nim
+    model_name: nvidia/llama-3.2-nv-embedqa-1b-v2
+    api_key: "${NVIDIA_API_KEY}"
+
+retrievers:
+  my_retriever:
+    _type: milvus_retriever
+    uri: "${MILVUS_URI}"
+    connection_args:
+      user: "developer"
+      password: "${MILVUS_PASSWORD}"
+      db_name: "default"
+    embedding_model: my_embedder
+    content_field: text
+    use_async_client: true
+
+workflow:
+  _type: rewoo_agent
+  tool_names: [text2sql, execute_db_query]
+  llm_name: my_llm
+```
+
+Run the workflow:
+
+```bash
+nat run --config config.yml --input "How many customers do we have?"
+```
+
+## Components
+
+### `text2sql` Function
+
+Generates SQL queries from natural language using:
+- Few-shot learning with similar examples
+- DDL (schema) information
+- Custom documentation
+- LLM-powered query generation
+
+### `execute_db_query` Function
+
+Executes SQL queries and returns formatted results:
+- Databricks SQL execution
+- Result limiting and pagination
+- Structured output format
+- SQLAlchemy Object Relational Mapper (ORM)-based connection
+
+## Use Cases
+
+- **Business Intelligence**: Enable non-technical users to query data
+- **Data Exploration**: Rapid prototyping and analysis
+- **Conversational Analytics**: Multi-turn Q&A about your data
+- **SQL Assistance**: Help analysts write complex queries
+
+## Documentation
+
+Full documentation: <https://docs.nvidia.com/nemo/agent-toolkit/latest/>
+
+## License
+
+Part of NVIDIA NeMo Agent toolkit. See repository for license details.

nvidia_nat_vanna-1.4.0b2.dist-info/RECORD

@@ -0,0 +1,13 @@
+nat/meta/pypi.md,sha256=LQqwDtLXfOYbWvBiVrwe-Q-74Z3db5j6LDK6cFoXnqU,3400
+nat/plugins/vanna/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
+nat/plugins/vanna/db_utils.py,sha256=xfctoDUraWFU_RaYFs_U0juxIPsgvZR_6nyAml80WGY,9862
+nat/plugins/vanna/execute_db_query.py,sha256=Yi9rTndkUdG7uy6N4lHzn5emht5g1duPIS79TkHfaTQ,9430
+nat/plugins/vanna/register.py,sha256=mO4kBfFoYohIztU6lZ3uI9uktfwnEuO6g2fAnbqcCmY,840
+nat/plugins/vanna/text2sql.py,sha256=fPNvuI1BNpHdT64_HnfjaNhY3AuYHYp1h6n2fFZPH8s,11183
+nat/plugins/vanna/training_db_schema.py,sha256=P_tXqSnW7ePrEv4cwNr61g28OZ61Lp3UQAZ4Qfk0G-Q,3142
+nat/plugins/vanna/vanna_utils.py,sha256=c7OSXYhhSnSnS3tlfyFnIpbi-MpwWIpW687-PXH3FLo,32977
+nvidia_nat_vanna-1.4.0b2.dist-info/METADATA,sha256=jroNqSlhZ9Bf5A58hhdT8sqvP1oK512fT8C8iaV4of8,4176
+nvidia_nat_vanna-1.4.0b2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_vanna-1.4.0b2.dist-info/entry_points.txt,sha256=R-SYyO1XwbN43yqGE5jM4uGhvkJeUmCWIxSl_LIi_rw,56
+nvidia_nat_vanna-1.4.0b2.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_vanna-1.4.0b2.dist-info/RECORD,,

nvidia_nat_vanna-1.4.0b2.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

nvidia_nat_vanna-1.4.0b2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[nat.components]
+nat_vanna = nat.plugins.vanna.register

nvidia_nat_vanna-1.4.0b2.dist-info/top_level.txt

@@ -0,0 +1 @@
+nat