vanna.openai.openai_chat

class OpenAI_Chat(vanna.base.base.VannaBase):

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

@staticmethod

def system_message(message: str) -> dict:

@staticmethod

def user_message(message: str) -> dict:

@staticmethod

def assistant_message(message: str) -> dict:

@staticmethod

def str_to_approx_token_count(string: str) -> int:

@staticmethod

def add_ddl_to_prompt(initial_prompt: str, ddl_list: list[str], max_tokens: int = 14000) -> str:

@staticmethod

def add_documentation_to_prompt( initial_prompt: str, documentation_list: list[str], max_tokens: int = 14000) -> str:

@staticmethod

def add_sql_to_prompt(initial_prompt: str, sql_list: list[str], max_tokens: int = 14000) -> str:

def get_sql_prompt( self, question: str, question_sql_list: list, ddl_list: list, doc_list: list, **kwargs):

def get_followup_questions_prompt( self, question: str, df: pandas.core.frame.DataFrame, question_sql_list: list, ddl_list: list, doc_list: list, **kwargs):

def generate_question(self, sql: str, **kwargs) -> str:

def generate_plotly_code( self, question: str = None, sql: str = None, df_metadata: str = None, **kwargs) -> str:

def submit_prompt(self, prompt, **kwargs) -> str:

Inherited Members

vanna.base.base.VannaBase: config; run_sql_is_set; generate_sql; generate_followup_questions; generate_questions; generate_embedding; get_similar_question_sql; get_related_ddl; get_related_documentation; add_question_sql; add_ddl; add_documentation; get_training_data; remove_training_data; 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