python调用chatgpt

简单写了一下关于文本生成接口的调用,其余更多的调用方法可在官网查看

javascript 复制代码
import os
from dotenv import load_dotenv, find_dotenv
from openai import OpenAI
import httpx


def gpt_config():
    # 为了安全起见,将key写到当前项目根目录下的.env文件中
    # find_dotenv() 寻找并定位 .env 文件的路径
    # load_dotenv() 读取该 .env 文件,并将其中的环境变量加载到当前的运行环境中
    _ = load_dotenv(find_dotenv())
    api_key = os.environ.get("OPENAI_API_KEY")
    # 使用httpx设置代理
    proxy = ''
    proxies = {'http://': proxy, 'https://': proxy}
    http_client = httpx.Client(proxies=proxies, verify=True)
    openai_client = OpenAI(api_key=api_key, http_client=http_client)
    return openai_client


def gpt_response(client, prompt, user_input, temperature, max_tokens):
    """
    :param client:gpt客户端
    :param prompt:system的prompt
    :param user_input:用户输入问题
    :param temperature:回答的随机性,取值范围0-1
    :param max_tokens:模型输出的最大token 数。prompt和模型回复结果的总token 数
    :return:
    """
    response = client.chat.completions.create(
        model="gpt-4o", # 模型名称
        messages=[
            {"role": "system", "content": prompt},
            {"role": "user", "content": user_input}
        ],
        temperature=temperature,
        max_tokens=max_tokens
    )
    assistant_response = response.choices[0].message.content
    return assistant_response


if __name__ == '__main__':
    prompt = '你是一个乐于助人的人工智能小助手'
    user_input = '太阳有多大'
    temperature = 0
    max_tokens = 2048
    client = gpt_config()
    res = gpt_response(client, prompt, user_input, temperature, max_tokens)
    print(res)
相关推荐
万粉变现经纪人16 小时前
如何解决 pip install -r requirements.txt 私有索引未设为 trusted-host 导致拒绝 问题
开发语言·python·scrapy·flask·beautifulsoup·pandas·pip
qq_4798754316 小时前
C++ std::Set<std::pair>
开发语言·c++
查士丁尼·绵17 小时前
笔试-九宫格三阶积幻方
python·九宫格·三阶积幻方
云知谷19 小时前
【C++基本功】C++适合做什么,哪些领域适合哪些领域不适合?
c语言·开发语言·c++·人工智能·团队开发
l1t20 小时前
DeepSeek辅助利用搬移底层xml实现快速编辑xlsx文件的python程序
xml·开发语言·python·xlsx
大飞记Python20 小时前
部门管理|“编辑部门”功能实现(Django5零基础Web平台)
前端·数据库·python·django
C_Liu_21 小时前
C++:list
开发语言·c++
my rainy days21 小时前
C++:友元
开发语言·c++·算法
小梁努力敲代码21 小时前
java数据结构--List的介绍
java·开发语言·数据结构
查士丁尼·绵1 天前
笔试-羊狼过河
python