Refactor to support Ollama

.gitattributes

@@ -0,0 +1 @@
+*.ipynb linguist-detectable=false

.github/workflows/tests.yml

@@ -1,4 +1,4 @@
-name: Integration Test using Debug Server
+name: Basic Integration Tests
 
 on:
   pull_request: {}
@@ -10,14 +10,18 @@ jobs:
   build:
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v3
+    - uses: actions/checkout@v4
     - name: Set up Python 3.10
-      uses: actions/setup-python@v3
+      uses: actions/setup-python@v5
       with:
         python-version: "3.10"
     - name: Install pip
       run: |
         python -m pip install --upgrade pip
         pip install tox
     - name: Run tests
+      env:
+        VANNA_API_KEY: ${{ secrets.VANNA_API_KEY }}
+        OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+        MISTRAL_API_KEY: ${{ secrets.MISTRAL_API_KEY }}
       run: tox -e py310

.gitignore

@@ -2,8 +2,7 @@ build
 **.egg-info
 venv
 .DS_Store
-notebooks/.ipynb_checkpoints
-notebooks/test*.ipynb
+notebooks/*
 tests/__pycache__
 __pycache__/
 .idea
@@ -12,4 +11,7 @@ docs/*.html
 .ipynb_checkpoints/
 .tox/
 notebooks/chroma.sqlite3
-dist
+dist
+.env
+*.sqlite
+htmlcov

README.md

@@ -30,7 +30,7 @@ See the [base class](src/vanna/base/base.py) for more details on how this works
 ## 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.
 
-- [Jupyter Notebook](https://github.com/vanna-ai/vanna/blob/main/notebooks/getting-started.ipynb)
+- [Jupyter Notebook](https://vanna.ai/docs/postgres-openai-vanna-vannadb/)
 - [vanna-ai/vanna-streamlit](https://github.com/vanna-ai/vanna-streamlit)
 - [vanna-ai/vanna-flask](https://github.com/vanna-ai/vanna-flask)
 - [vanna-ai/vanna-slack](https://github.com/vanna-ai/vanna-slack)
@@ -39,7 +39,7 @@ These are some of the user interfaces that we've built using Vanna. You can use
 ## Getting started
 See the [documentation](https://vanna.ai/docs/) for specifics on your desired database, LLM, etc.
 
-If you want to get a feel for how it works after training, you can try this [Colab notebook](https://colab.research.google.com/github/vanna-ai/vanna/blob/main/notebooks/getting-started.ipynb).
+If you want to get a feel for how it works after training, you can try this [Colab notebook](https://vanna.ai/docs/app/).
 
 
 ### Install

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "vanna"
-version = "0.0.38"
+version = "0.1.0"
 authors = [
   { name="Zain Hoda", email="zain@vanna.ai" },
 ]
@@ -30,7 +30,7 @@ postgres = ["psycopg2-binary", "db-dtypes"]
 bigquery = ["google-cloud-bigquery"]
 snowflake = ["snowflake-connector-python"]
 duckdb = ["duckdb"]
-all = ["psycopg2-binary", "db-dtypes", "google-cloud-bigquery", "snowflake-connector-python","duckdb"]
+all = ["psycopg2-binary", "db-dtypes", "google-cloud-bigquery", "snowflake-connector-python", "duckdb", "openai", "mistralai"]
 test = ["tox"]
 chromadb = ["chromadb"]
 openai = ["openai"]