vanna.chromadb.chromadb_vector

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

class ChromaDB_VectorStore(vanna.base.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) -> str:

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

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

def get_training_data(self, **kwargs) -> pandas.core.frame.DataFrame:

def remove_training_data(self, id: str, **kwargs) -> bool:

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.base.VannaBase: config; run_sql_is_set; generate_sql; generate_followup_questions; generate_questions; get_sql_prompt; get_followup_questions_prompt; submit_prompt; generate_question; generate_plotly_code; connect_to_snowflake; connect_to_sqlite; connect_to_postgres; connect_to_bigquery; run_sql; ask; train; get_training_plan_generic; get_training_plan_snowflake; get_plotly_figure