通过fu过 Function Calling 查询数据库

from openai import OpenAI

import os

import json

from dotenv import load_dotenv, find_dotenv

_ = load_dotenv(find_dotenv()) # 读取本地 .env 文件,里面定义了 OPENAI_API_KE

client = OpenAI(

api_key=os.getenv("OPENAI_API_KEY"),

base_url=os.getenv("OPENAI_BASE_URL")

)

def get_sql_completion(messages, model="gpt-3.5-turbo-1106"):

response = client.chat.completions.create(

model=model,

messages=messages,

temperature=0,

tools=[{ # 摘自 OpenAI 官方示例 https://github.com/openai/openai-cookbook/blob/main/examples/How_to_call_functions_with_chat_models.ipynb

"type": "function",

"function": {

"name": "ask_database",

"description": "Use this function to answer user questions about business.

Output should be a fully formed SQL query.",

"parameters": {

"type": "object",

"properties": {

"query": {

"type": "string",

"description": f"""

SQL query extracting info to answer the user's question.

SQL should be written using this database schema:

{database_schema_string}

The query should be returned in plain text, not in JSON.

The query should only contain grammars supported by SQLite.

""",

}

},

"required": ["query"],

}

}

}],

)

return response.choices[0].message

描述数据库表结构

database_schema_string = """

CREATE TABLE orders (

id INT PRIMARY KEY NOT NULL, -- 主键,不允许为空

customer_id INT NOT NULL, -- 客户ID,不允许为空

product_id STR NOT NULL, -- 产品ID,不允许为空

price DECIMAL(10,2) NOT NULL, -- 价格,不允许为空

status INT NOT NULL, -- 订单状态,整数类型,不允许为空。0代表待支付,1代表已支付,2代表已退款

create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP, -- 创建时间,默认为当前时间

pay_time TIMESTAMP -- 支付时间,可以为空

);

"""

import sqlite3

创建数据库连接

conn = sqlite3.connect(':memory:')

cursor = conn.cursor()

创建orders表

cursor.execute(database_schema_string)

插入5条明确的模拟记录

mock_data = [

(1, 1001, 'TSHIRT_1', 50.00, 0, '2023-10-12 10:00:00', None),

(2, 1001, 'TSHIRT_2', 75.50, 1, '2023-10-16 11:00:00', '2023-08-16 12:00:00'),

(3, 1002, 'SHOES_X2', 25.25, 2, '2023-10-17 12:30:00', '2023-08-17 13:00:00'),

(4, 1003, 'HAT_Z112', 60.75, 1, '2023-10-20 14:00:00', '2023-08-20 15:00:00'),

(5, 1002, 'WATCH_X001', 90.00, 0, '2023-10-28 16:00:00', None)

]

for record in mock_data:

cursor.execute('''

INSERT INTO orders (id, customer_id, product_id, price, status, create_time, pay_time)

VALUES (?, ?, ?, ?, ?, ?, ?)

''', record)

提交事务

conn.commit()

def ask_database(query):

cursor.execute(query)

records = cursor.fetchall()

return records

prompt = "上个月的销售额"

prompt = "统计每月每件商品的销售额"

prompt = "哪个用户消费最高?消费多少?"

messages = [

{"role": "system", "content": "基于 order 表回答用户问题"},

{"role": "user", "content": prompt}

]

response = get_sql_completion(messages)

if response.content is None:

response.content = ""

messages.append(response)

print("Function Calling")

print(response)

if response.tool_calls is not None:

tool_call = response.tool_calls[0]

if tool_call.function.name == "ask_database":

arguments = tool_call.function.arguments

args = json.loads(arguments)

print("SQL")

print(args["query"])

result = ask_database(args["query"])

print("DB Records")

print(result)

复制代码
    messages.append({
        "tool_call_id": tool_call.id,
        "role": "tool",
        "name": "ask_database",
        "content": str(result)
    })
    response = get_sql_completion(messages)
    print("====最终回复====")
    print(response.content)
相关推荐
小咕聊编程1 天前
【含文档+PPT+源码】基于SpringBoot+Gpt个人健康管理系统
java·gpt·tomcat·毕业设计·hibernate
Mr.Lee jack1 天前
体验GPT-OSS-120B:在PH8平台上探索超大规模语言模型的强大能力
人工智能·gpt·语言模型·openai·deepseek
哪吒编程1 天前
如何用好 Codex?OpenAI 内部最佳实践指南:8 个最佳应用场景
gpt·openai
viperrrrrrrrrr74 天前
GPT系列模型-详解
人工智能·gpt·llm
2401_841495644 天前
预训练基础模型简介
gpt·语言模型·自然语言处理·bert·transformer·大语言模型·预训练
Skrrapper7 天前
GPT‑5 都更新了些什么?
gpt
三天哥9 天前
Sora 2为什么会火?
人工智能·gpt·ai·aigc·agi·ai视频·sora
. . . . .9 天前
GPT的前世今生及发展
人工智能·gpt
丁学文武9 天前
大模型原理与实践:第三章-预训练语言模型详解_第3部分-Decoder-Only(GPT、LLama、GLM)
人工智能·gpt·语言模型·自然语言处理·大模型·llama·glm
带娃的IT创业者9 天前
第2集:技术选型的智慧:Flask vs FastAPI,GLM-4 vs GPT
python·gpt·flask·fastapi·glm·技术选型