vanna.mistral.mistral

class Mistral(vanna.base.base.VannaBase)

一个辅助类，提供了使用继承创建 ABC 的标准方法。

客户端

模型

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:

vanna.base.base.VannaBase: 配置; run_sql_is_set; 生成后续问题; 生成问题; 生成嵌入; 获取相似问题 SQL; 获取相关 DDL; 获取相关文档; 添加问题 SQL; 添加 DDL; 添加文档; 获取训练数据; 移除训练数据; 连接到 Snowflake; 连接到 SQLite; 连接到 Postgres; 连接到 BigQuery; 运行 SQL; 提问; 训练; 获取通用训练计划; 获取 Snowflake 训练计划; 获取 Plotly 图形