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)
相关推荐
半梦半醒*1 分钟前
正则表达式
linux·运维·开发语言·正则表达式·centos·运维开发
用户8356290780513 分钟前
用Python高效处理Excel数据:Excel数据读取指南
后端·python
不吃洋葱.21 分钟前
Bean.
java·开发语言
我星期八休息26 分钟前
深入理解跳表(Skip List):原理、实现与应用
开发语言·数据结构·人工智能·python·算法·list
蒋星熠35 分钟前
如何在Anaconda中配置你的CUDA & Pytorch & cuNN环境(2025最新教程)
开发语言·人工智能·pytorch·python·深度学习·机器学习·ai
合作小小程序员小小店1 小时前
机器学习介绍
人工智能·python·机器学习·scikit-learn·安全威胁分析
JavaEdge在掘金1 小时前
掌握Spring IoC容器和Bean作用,轻松实现依赖注入!
python
We....1 小时前
Java分布式编程:RMI机制
java·开发语言·分布式
€8111 小时前
Java入门级教程17——利用Java SPI机制制作验证码、利用Java RMI机制实现分布式登录验证系统
java·开发语言·java spi机制·远程传输数据
2301_815357702 小时前
parameterType和@Param注解的区别
java·开发语言·数据库