当前位置:首页> AI教程> Python调用ChatGPT API接口介绍

Python调用ChatGPT API接口介绍

释放双眼,带上耳机,听听看~!
本文简单介绍如何使用Python调用ChatGPT API接口,实现chat, 生成图片, 识别关键, 改错等功能。

Python 调用ChatGPT API 接口介绍

ChatGPT 介绍

ChatGPT可以实现chat,生成图片,识别关键,改错等等功能,本文简单介绍如何使用python调用ChatGPT API 接口。

Python调用ChatGPT API接口介绍

1. 生成API Key

从openai官网网址:platform.openai.com/account/api…生成我们的API key

Python调用ChatGPT API接口介绍

获得key后我们就可以调用API接口了。

2. 安装openai

使用pip安装openai库,命令: pip install openai。

安装openai库: pip install openai

3. python代码调用API

3.1 主要步骤

# 调用openai api的步骤
# 1. 安装openai库 pip install openai
# 2. 设置openai的api_key 
# 3. 调用openai的api
# 4. 参考文档
# https://platform.openai.com/docs/api-reference/completions/create
# https://platform.openai.com/docs/api-reference/authentication
# https://platform.openai.com/docs/api-reference/completions/create
# https://platform.openai.com/docs/libraries/community-libraries 

3.2 代码示例

API可以实现chat,生成图片,识别关键,改错等等功能。
下面是主要示例代码。
注意:openai.api_key = “sk-xxxFQ” #要更换成自已的API KEY


#!/usr/local/bin/python3.8
# -*- coding: utf8 -*-


# 调用openai api的步骤
# 1. 安装openai库 pip install openai
# 2. 设置openai的api_key 
# 3. 调用openai的api
# 4. 参考文档
# https://platform.openai.com/docs/api-reference/completions/create
# https://platform.openai.com/docs/api-reference/authentication
# https://platform.openai.com/docs/api-reference/completions/create
# https://platform.openai.com/docs/libraries/community-libraries 


import os
import openai
import json
# 1. 准备好请求的url
#openai.organization = "YOUR_ORG_ID" # 
#openai.api_key = os.getenv("OPENAI_API_KEY")
openai.api_key = "sk-xxxFQ" #要更换成自已的API KEY


# 查看可以使用的模型列表
def get_model_list():
    models= openai.Model.list()
    print(models)

# 生成文本示例
def generate_text(prompt):
    completions = openai.Completion.create(
        engine="text-davinci-002",
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )
    message = completions.choices[0].text
    return message.strip()

# 调用openai 画图示例
def generate_image(prompt):
    response = openai.Image.create(
        prompt = prompt,
        n=1,
        size="512x512"
    )
    image_url = response['data'][0]['url']
    return image_url

# 调用openai 问答示例
def chat(prompt):
    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "user", "content":prompt}
    ]
)
    answer = response.choices[0].message.content
    return answer

# 调用openai 改正错词输出正确句子
def correct():
    prompt="改正错词输出正确句子:nn我在京东电商平台买了苹果耳几和华为体脂称"  #建议prompt: 改正错词输出正确句子:nn input_sentence
    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "user", "content":prompt}
    ]
)
    answer = response.choices[0].message.content
    return answer

# 调用openai 识别关键词
def keyword():
    prompt="对下面内容识别2个关键词,每个词字数不超过3个字:nn齐选汽车挂件车内挂饰车载后视镜吊坠高档实心黄铜玉石出入平安保男女 红流苏-玉髓平安扣"  #建议prompt: 对下面内容识别n个关键词,每个词字数不超过m个字:nn input data
    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "user", "content":prompt}
    ]
)
    answer = response.choices[0].message.content
    return answer

# 抽取文本向量 (Embedding)
def embedding():
    content = '苹果手机'
    response = openai.Embedding.create(
    model="text-embedding-ada-002",
    input=content
)
    answer = response.data[0].embedding
    return answer


def api_test():
    # 测试chat 
    # prompt = "人口最多的国家?"
    # response = chat(prompt)
    # print(response)
# 
    # 测试generate_text
    # prompt = "Hello, how are you today?"
    # response = generate_text(prompt)
    # print(response) 

    # 测试generate_image
    #prompt = "a delicious dessert"
    #response = generate_image(prompt)
    #print(response)

    # 测试correct
    # response = correct()
    # print(response) #输出结果: 我在京东电商平台买了苹果耳机和华为体脂秤。

    # 测试keyword
    #response = keyword()
    #print(response) #输出结果: 挂件、平安扣

    # 测试embedding
    result = embedding()
    print(len(result))
    print(result)

if __name__ == '__main__':
    api_test()

4. flask实现chat效果的示例

github.com/openai/open…

下载git代码:

git clone https://github.com/openai/openai-quickstart-python.git
cd openai-quickstart-python
cp .env.example .env

python -m venv venv
. venv/bin/activate
pip install -r requirements.txt
flask run

运行效果:

Python调用ChatGPT API接口介绍

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

轨迹相似度衡量任务及传统方法与深度表示学习方法对比

2023-12-18 20:36:14

AI教程

如何优化GPT终端用户体验?

2023-12-18 20:47:14

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索