当前位置:   article > 正文

【GPT调用】本地使用python调用GPT接口_gpt接口调用

gpt接口调用

环境变量设置

.env文件中配置GPT环境变量

api_key="你的GPT-API-KEY"
url=https://ai-proxy.ksord.com/wps.openai.azure.com/openai/deployments/gpt-4-32k/chat/completions?api-version=2023-09-01-preview
  • 1
  • 2

主调用方法

一般需要先安装load_dotenv库:

pip install load_dotenv

import requests
import os
from dotenv import load_dotenv

# 获取env文件中的环境变量
load_dotenv()

# 环境变量赋值
endpoint = os.getenv("url")
api_key = os.getenv("api_key")

# 封装GPT请求方法
def get_completion(messages, model="gpt-4-32k"):
    headers = {
        "Content-Type": "application/json",
        "api-key": api_key,
    }

    payload = {"model": model, "messages": messages, "temperature": 0.5, "top_p": 1, "frequency_penalty": 0, "presence_penalty": 0, "stop": None}
    response = requests.post(endpoint, headers=headers, json=payload, stream=True)
    try:
    	# 提取GPT返回的结果信息
        return response.json()["choices"][0]["message"]["content"]
    except Exception as e:
    	# 请求报错时,返回错误原因
        print(response.json())
        raise e

# 赋初值,也可设为空List
messages = [{"role": "system", "content": "You are a helpful assistant."}]

# 设置传入message值为input型
# 设置特殊指令break
# 打印返回结果
while True:
    message = input("User: ")
    if message.lower() in ("quit", "exit"):
        break

    messages.append({"role": "user", "content": message})
    anwser = get_completion(messages)

    print(f"Bot: {anwser}")
    messages.append({"role": "assistant", "content": anwser})

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45

执行结果

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/875462
推荐阅读
相关标签
  

闽ICP备14008679号