如何使用 OpenAI API ChatGPT,通过自然语言生成 SQL 语句(简易版)

编程入门 行业动态 更新时间:2024-10-11 19:20:03

简介

Human Language to SQL Translator 是一个可以通过ChatGPT 可以生成 SQL 语句的网站。


开源地址: https://github/whoiskatrin/sql-translator

这里使用 python 对其进行了复现

代码

"""自然语言生成 SQL """
import openai


def sql_generation(query, table_schema):
    """
    生成 sql
    :param query:  用户输入的 query
    :param table_schema: 数据库表的 schema
    :return: sql 语句
    """
    prompt = """
    Translate this natural language query into SQL without changing the case of the entries given by me:
    
    
    """ + query + """
    
    
    Use this table schema:
    
    """ + table_schema + """
    
    
    SQL Query:"""

    completion = openai.ChatCompletion.create(model="gpt-3.5-turbo",
                                              # api_base="https://xxxxx.xxxxx/v1",   如果有域名代理可以使用
                                              api_key="sk-xxxxxxxxxxxxxxxxxxx",
                                              messages=[
                                                  {"role": "system", "content": prompt},
                                              ], max_tokens=500, temperature=0)
    return completion.choices[0].message.content


if __name__ == '__main__':
    query = "查询年龄大于23岁的男生"
    print("query: " + query)
    sql = sql_generation("查询年龄大于23岁的男生",
                                "CREATE TABLE USER(id int, age int, sex bool, name varchar, update_time date, create_time date)")
    print("SQL: " + sql)

运行结果

总结

主要是通过 提示语让 ChatGPT 明白自己需要参考表结构将用户输入的文字转换为 SQL 语句

可以通过修改 PROMPT 来实现生成不同的数据库语句

更多推荐

如何使用 OpenAI API ChatGPT,通过自然语言生成 SQL 语句(简易版)

本文发布于:2023-10-12 05:04:00,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1483790.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:自然语言   如何使用   语句   简易版   OpenAI

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!