Skip to content

Saroj Dhiman / nlp_to_sql_query_generator

Go to a project
Switch branch/tag
vanna_model2.py 827 Bytes
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 
import pandas as pd
import vanna as vn


# Create login and get API key
api_key = vn.get_api_key('saroj.dhiman@ociustechnologies.com') 
vn.set_api_key(api_key)

# Create models
vn.create_model(model="data_warehose", db_type="Snowflake")
vn.create_model(model="marketing_db", db_type="Postgres")
vn.create_model(model="marketing_db", db_type="BigQuery")

models = vn.get_models()

  # Set model
vn.set_model('chinook')

# Connect to database
url= "database.sqlite3"
vn.connect_to_sqlite(url=url)

# Check training data
training_data = vn.get_training_data()
training_data


vn.ask(question='What are the most occured cancer in CancerData?')

# training_data = vn.get_training_data()
# training_data['question'].str.contains('total Stomach records in cancertable?').any()

# vn.ask(question='give the most record of cancer are?')