vanna.mistral.mistral

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

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

client
model
def generate_plotly_code( self, question: str = None, sql: str = None, df_metadata: str = None, **kwargs) -> str:
def generate_question(self, sql: str, **kwargs) -> str:
def get_followup_questions_prompt( self, question: str, question_sql_list: list, ddl_list: list, doc_list: list, **kwargs):
@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 generate_sql(self, question: str, **kwargs) -> str:
def submit_prompt(self, prompt, **kwargs) -> str: