荆门公司做网站,深圳小程序开发费用,山东济南网站推广,杭州小程序公司实力排名vanna多表关联的实验 1.实验说明2.数据库准备3.启动VANNA的程序4.交互 1.实验说明
本次测试vanna实现多表关联的数据查询#xff0c;效果还可以#xff0c;直接要多次交互下。
环境安装参见本人的另一篇文章
本次使用采用的是postgresql作为数据库#xff0c;并将LLM采用… vanna多表关联的实验 1.实验说明2.数据库准备3.启动VANNA的程序4.交互 1.实验说明
本次测试vanna实现多表关联的数据查询效果还可以直接要多次交互下。
环境安装参见本人的另一篇文章
本次使用采用的是postgresql作为数据库并将LLM采用qwen3:8b
其中安装vanna使用
pip install vanna[chromadb,ollama,mysql,postgresql] -i https://pypi.tuna.tsinghua.edu.cn/simple/2.数据库准备
自行安装postgresql数据库可以采用容器化安装方便一些安装后创建数据库demodb
-- 如下在demodb数据库中执行
-- 创建客户表
CREATE TABLE cust_info (cust_id INT PRIMARY KEY, -- 客户ID主键cust_name VARCHAR(50) NOT NULL -- 客户名非空
);
COMMENT ON TABLE cust_info IS 客户基本信息表;
COMMENT ON COLUMN cust_info.cust_id IS 唯一客户标识;
COMMENT ON COLUMN cust_info.cust_name IS 客户全名;-- 创建账户表含外键约束
CREATE TABLE acct_info (acct_id SERIAL PRIMARY KEY, -- 账户ID自增主键cust_id INT NOT NULL, -- 关联客户IDbal NUMERIC(12,2) DEFAULT 0.00, -- 账户余额默认0精确到分FOREIGN KEY (cust_id) REFERENCES cust_info(cust_id) ON DELETE CASCADE
);
COMMENT ON TABLE acct_info IS 客户账户信息表;
COMMENT ON COLUMN acct_info.acct_id IS 唯一账户标识;
COMMENT ON COLUMN acct_info.cust_id IS 关联客户ID外键;
COMMENT ON COLUMN acct_info.bal IS 账户余额单位元;INSERT INTO cust_info (cust_id, cust_name) VALUES
(1, 张明), (2, 李华), (3, 王芳), (4, 刘洋),
(5, 陈思), (6, 赵雷), (7, 周琪), (8, 吴越),
(9, 郑宇), (10, 孙琳);INSERT INTO acct_info (cust_id, bal) VALUES
-- 客户1的6个账户
(1, 15200.50), (1, 8730.00), (1, 42150.75),
(1, 9300.25), (1, 15600.00), (1, 3200.40),
-- 客户2的3个账户
(2, 78000.00), (2, 14500.60), (2, 9200.30),
-- 客户3的8个账户
(3, 12500.00), (3, 36700.50), (3, 8900.25),
(3, 15400.75), (3, 23000.00), (3, 4200.90),
(3, 17600.30), (3, 29500.45),
-- ... 其他客户账户共50条
(10, 45000.00), (10, 12800.20), (10, 7600.80), (10, 21500.35);3.启动VANNA的程序
大部分不改只改了链接ollama的模型和连接数据库的部分vn.train部分可以不要
from vanna.ollama import Ollama
from vanna.chromadb import ChromaDB_VectorStoreclass MyVanna(ChromaDB_VectorStore, Ollama):def __init__(self, configNone):ChromaDB_VectorStore.__init__(self, configconfig)Ollama.__init__(self, configconfig)vn MyVanna(config{model: qwen3:8b,ollama_host:http://192.168.184.1:11434})# vn.connect_to_mysql(host192.168.184.190, dbnamedemodb, userroot, passwordPG_Dev2022a, port3306)vn.connect_to_postgres(host192.168.184.190, dbnamedemodb, userpostgres, passwordPG_Dev2022a, port5432)# The information schema query may need some tweaking depending on your database. This is a good starting point.
df_information_schema vn.run_sql(SELECT * FROM INFORMATION_SCHEMA.COLUMNS)# This will break up the information schema into bite-sized chunks that can be referenced by the LLM
plan vn.get_training_plan_generic(df_information_schema)vn.train(ddl-- 创建客户表
CREATE TABLE cust_info (cust_id INT PRIMARY KEY, -- 客户ID主键cust_name VARCHAR(50) NOT NULL -- 客户名非空
);
COMMENT ON TABLE cust_info IS 客户基本信息表;
COMMENT ON COLUMN cust_info.cust_id IS 唯一客户标识;
COMMENT ON COLUMN cust_info.cust_name IS 客户全名;-- 创建账户表含外键约束
CREATE TABLE acct_info (acct_id SERIAL PRIMARY KEY, -- 账户ID自增主键cust_id INT NOT NULL, -- 关联客户IDbal NUMERIC(12,2) DEFAULT 0.00, -- 账户余额默认0精确到分FOREIGN KEY (cust_id) REFERENCES cust_info(cust_id) ON DELETE CASCADE
);
COMMENT ON TABLE acct_info IS 客户账户信息表;
COMMENT ON COLUMN acct_info.acct_id IS 唯一账户标识;
COMMENT ON COLUMN acct_info.cust_id IS 关联客户ID外键;
COMMENT ON COLUMN acct_info.bal IS 账户余额单位元;
)from vanna.flask import VannaFlaskApp
app VannaFlaskApp(vn)app.run()4.交互 问题按照客户维度汇总账户余额倒序排列展示的字段是客户ID,客户名称余额排名号。注意余额是数值没有找到账户那么余额就是0 说明 实际上多次交互才摸清这个玩意如果没有提示说余额是数值排序中没关联上的bal展示成null显然不太符合通常的情况。 多表关联最重要的是将表间关系作为训练数据给模型说明清楚了不然模型猜肯定会有偏差。
