调用openai api

jianbiao1483

已于 2024-08-02 10:34:40 修改

阅读量502

点赞数 1

文章标签： chatgpt

于 2024-07-24 10:15:19 首次发布

本文链接：https://blog.youkuaiyun.com/jianbiao1483/article/details/140655451

版权

from openai import OpenAI
import os
client = OpenAI(
  api_key=os.environ["OPENAI_API_KEY"],
  base_url="https://api.lmtchina.com/v1"
)

completion = client.chat.completions.create(
  model="gpt-3.5-turbo",
  messages=[
    {"role": "system", "content": "You are a poetic assistant, skilled in explaining complex programming concepts with creative flair."},
    {"role": "user", "content": "使用中文回答，Compose a poem that explains the concept of recursion in programming."}
  ]
)

print(completion.choices[0].message)

结果：

更新：

以下兩種方法都可

（1）使用client

from openai import OpenAI
import os
import openai

client=OpenAI()

client = OpenAI(
  api_key=os.environ["OPENAI_API_KEY"],
  base_url="https://api.lmtchina.com/v1"
)

completion = client.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {
            "role": "user",
            "content": "say hello world",
        },
    ],
)
print(completion.choices[0].message.content)

（2）不使用client

import openai
import os
openai.api_key =os.environ["OPENAI_API_KEY"]
openai.base_url = "https://api.lmtchina.com/v1/"   #最後的/不要省略
completion = openai.chat.completions.create(
    model="gpt-3.5-turbo",
    messages=[
        {
            "role": "user",
            "content": "say hello world",
 
        },           
    ],
 )
print(completion.choices[0].message.content)