调用openapi实现对话机器人

先记录代码

from openai import OpenAI
import os
import gradio as gr

# 接口配置
os.environ["OPENAI_BASE_URL"] = "https://xxx.com/v1"
os.environ["OPENAI_API_KEY"] = "sk-xxxxxxxxxxxxxxx"

client = OpenAI(
    api_key=os.environ.get("OPENAI_API_KEY"),
    base_url=os.environ.get("OPENAI_BASE_URL"),
    timeout=60
)

SYSTEM_PROMPT = """你是一个中国厨师，用中文回答做菜的问题。你的回答需要满足以下要求:
1. 你的回答必须是中文
2. 回答限制在100个字以内"""

class Conversation:
    def __init__(self, prompt, num_of_round):
        self.prompt = prompt
        self.num_of_round = num_of_round
        self.messages = []
        self.messages.append({"role": "system", "content": self.prompt})

    def ask(self, question):
        try:
            self.messages.append({"role": "user", "content": question})
            # 注意：gpt-5.4不是官方合法模型，后续需要你替换PingPong实际可用模型
            response = client.chat.completions.create(
                model="gpt-5.4",
                messages=self.messages,
                temperature=0.5,
                max_tokens=2048,
                top_p=1,
            )
        except Exception as e:
            print(e)
            return str(e)

        message = response.choices[0].message.content
        self.messages.append({"role": "assistant", "content": message})

        if len(self.messages) > self.num_of_round*2 + 1:
            del self.messages[1:3]  # Remove the first round conversation left.
        return message


conv = Conversation(SYSTEM_PROMPT, 10)

# 改造：返回字典格式消息列表，兼容无type参数的Chatbot
def answer(question, history=None):
    if history is None:
        history = []
    # 调用模型
    response = conv.ask(question)
    # 追加本轮问答
    history.append({"role": "user", "content": question})
    history.append({"role": "assistant", "content": response})
    return history, history

with gr.Blocks() as demo:
    # 删掉 type="tuples"，适配你的旧版Gradio
    chatbot = gr.Chatbot(elem_id="chatbot")
    state = gr.State([])

    with gr.Row():
        txt = gr.Textbox(
            show_label=False,
            placeholder="Enter text and press enter",
            container=False
        )

    txt.submit(answer, [txt, state], [chatbot, state])

demo.launch(css="#chatbot{height:300px} .overflow-y-auto{height:500px}")