Vanna本地版本无训练数据管理函数，如何管理训练数据？

Vanna本地部署时，训练数据默认存储在初始化时指定的数据库（如SQLite、PostgreSQL等）中，核心存储表通常为training_data（命名可能因具体实现略有差异）。以下是具体的检索和删除操作方案：

1. 检索训练数据 #

• 直接查询数据库表：
  若使用默认SQLite，可执行以下SQL语句获取全部训练数据：

  SELECT * FROM training_data;
  

• 自定义扩展方法：
  基于Vanna基础类封装自己的get_training_data方法，方便后续调用：

  from vanna.base import VannaBase
  
  class MyLocalVanna(VannaBase):
      def get_training_data(self, **kwargs):
          # 查询数据库并返回训练数据
          result = self.run_sql("SELECT * FROM training_data")
          return result
  

2. 删除训练数据 #

• 直接执行SQL删除：
  删除单条指定数据（根据数据ID）：

  DELETE FROM training_data WHERE id = '目标训练数据ID';
  

  删除全部训练数据：

  DELETE FROM training_data;
  

• 自定义扩展删除方法：
  在自定义类中添加delete_training_data方法，增加操作灵活性：

  class MyLocalVanna(VannaBase):
      def delete_training_data(self, data_id=None, **kwargs):
          if data_id:
              self.run_sql(f"DELETE FROM training_data WHERE id = '{data_id}'")
          else:
              # 可添加二次确认逻辑避免误删
              self.run_sql("DELETE FROM training_data")
          return "训练数据删除完成"
  

补充提示 #

• 不同本地部署的数据库表结构可能有细微差别，可先查询表结构确认字段：

  -- SQLite查看表结构
  PRAGMA table_info(training_data);
  -- PostgreSQL查看表结构
  SELECT column_name FROM information_schema.columns WHERE table_name = 'training_data';
  

• 所有数据库操作建议先在测试环境验证，避免误操作导致数据丢失。

内容的提问来源于stack exchange，提问作者Zain