深度学习-143-Text2SQL之基于langchain的少量样本提示词模板FewShotPromptTemplate的应用实战(二)
文章目录
- 1 基本组件
-
- 1.1 大模型
- 1.2 数据库Chinook
-
- 1.2.1 创建并载入数据
- 1.2.2 SQLDatabase
- 1.2.3 获取建表语句
- 2 应用步骤
-
- 2.1 翻译表名
-
- 2.1.1 方式一(使用大模型)
- 2.1.2 方式二(自定义)
- 2.2 根据描述查找使用的表
-
- 2.2.1 提示词模板
- 2.2.2 采用结构化输出
- 2.2.3 获取相应的建表语句
- 2.3 生成SQL语句
-
- 2.3.1 提示词模板
- 2.3.2 采用结构化输出
- 2.4 执行SQL语句
- 3 整体代码
- 4 参考附录
1 基本组件
1.1 大模型
from langchain_ollama import ChatOllama
import os
os.environ['OLLAMA_HOST'] = '127.0.0.1'
os.environ['OLLAMA_PORT'] = '11434'
# 获取环境变量值,提供默认值以防环境变量未设置
host