一、获取DeepSeek的API key
1、登录DeepSeek官网
https://www.deepseek.com/
2、创建API key
生成的API key如下:
sk-dbd3e5790bab4e5ea244f2a535af7665
3、实名认证以及充值
这里调用API需要实名认证然后充值才能调用,这里我充值了10元。
4、添加环境变量
在windows本机添加环境变量
setx DEEPSEEK_API_KEY "sk-dbd3e5790bab4e5ea244f2a535af7665"
二、使用AI生成代码
1、提示词如下
借助gradio实现一个页面,有两个输入框 一个框输入表结构,另一个框输入SQL语句 调用deepseek的API来分析语句,API key变量名为:DEEPSEEK_API_KEY,值为:sk-dbd3e5790bab4e5ea244f2a535af7665, 一个输出框输出优化结果,不要连MySQL,一个文件实现,其中deepseek的API调用方式如下:
# Please install OpenAI SDK first: `pip3 install openai`
from openai import OpenAI
client = OpenAI(api_key="<DeepSeek API Key>", base_url="https://api.deepseek.com")
response = client.chat.completions.create(
model="deepseek-chat",
messages=[
{"role": "system", "content": "You are a helpful assistant"},
{"role": "user", "content": "Hello"},
],
stream=False
)
print(response.choices[0].message.content)
2、生成的代码如下
import gradio as gr
from openai import OpenAI
# 设置 DeepSeek API Key
DEEPSEEK_API_KEY = "sk-dbd3e5790bab4e5ea244f2a535af7665"
# 创建 OpenAI 客户端
client = OpenAI(api_key=DEEPSEEK_API_KEY, base_url="https://api.deepseek.com")
def optimize_sql(table_structure, sql_statement):
# 构建向 DeepSeek 发送的消息
content = f"表结构如下:{table_structure}。请优化以下 SQL 语句:{sql_statement}"
try:
# 调用 DeepSeek API
response = client.chat.completions.create(
model="deepseek-chat",
messages=[
{"role": "system",
"content": "You are a helpful assistant that optimizes SQL statements based on the provided table structure."},
{"role": "user", "content": content}
],
stream=False
)
# 提取优化后的 SQL 语句
optimized_sql = response.choices[0].message.content
return optimized_sql
except Exception as e:
return f"发生错误: {str(e)}"
# 创建 Gradio 界面
with gr.Blocks() as demo:
gr.Markdown("### SQL 优化工具")
with gr.Row():
table_structure_input = gr.Textbox(label="表结构", lines=5, placeholder="请输入表结构信息")
sql_input = gr.Textbox(label="SQL 语句", lines=5, placeholder="请输入 SQL 语句")
optimize_button = gr.Button("优化 SQL")
output_text = gr.Textbox(label="优化结果", lines=5)
# 绑定按钮点击事件
optimize_button.click(
fn=optimize_sql,
inputs=[table_structure_input, sql_input],
outputs=output_text
)
if __name__ == "__main__":
demo.launch()
3、安装python
在windows安装python
(1)python官网下载安装包
https://www.python.org/
(2)进行安装
打开cmd,输入
python
4、安装pyCharm
(1)官网下载安装包
https://www.jetbrains.com.cn/pycharm/
(2)进行安装
5、打开PyCharm
6、粘贴代码
7、运行代码
导入依赖,然后运行代码
8、进行测试
http://127.0.0.1:7860
CREATE TABLE Users(id INTEGER PRIMARY KEY,name TEXT,age INTEGER);
SELECT * FROM Users WHERE age > 18;