vanna.chromadb_vector

default_ef = <chromadb.utils.embedding_functions.ONNXMiniLM_L6_V2 object>

class ChromaDB_VectorStore(vanna.base.VannaBase):

Helper class that provides a standard way to create an ABC using inheritance.

chroma_client

documentation_collection

ddl_collection

sql_collection

def generate_embedding(self, data: str, **kwargs) -> list[float]:

def add_question_sql(self, question: str, sql: str, **kwargs):

def add_ddl(self, ddl: str, **kwargs):

def add_documentation(self, doc: str, **kwargs):

def get_similar_question_sql(self, question: str, **kwargs) -> list:

def get_related_ddl(self, question: str, **kwargs) -> list:

def get_related_documentation(self, question: str, **kwargs) -> list:

vanna.base.VannaBase: config; run_sql_is_set; generate_sql_from_question; get_prompt; submit_prompt; generate_question; generate_plotly_code; connect_to_snowflake; run_sql; ask; train; get_training_plan_snowflake; get_plotly_figure