使用 Gradio 快速构建对话式 AI 应用
代码示例
import os
import gradio as gr
from openai import OpenAI
client = OpenAI(
api_key=os.getenv("OPENAI_API_KEY", "sk-"),
base_url=os.getenv("OPENAI_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
)
MODEL_NAME = os.getenv("MODEL_NAME", "qwen-plus")
SYSTEM_PROMPT = "你是一个友好、专业、谨慎的中文智能助手。"
def chat_stream(message, history):
messages = [{"role": "system", "content": SYSTEM_PROMPT}]
partial = ""
for item in history:
if item.get("role") in ["user", "assistant"]:
messages.append(item)
messages.append({"role": "user", "content": message})
stream = client.chat.completions.create(
model=MODEL_NAME,
messages=messages,
stream=True,
temperature=0.7,
)
for chunk in stream:
delta = chunk.choices[0].delta.content or ""
partial += delta
yield partial
demo = gr.ChatInterface(
fn=chat_stream,
title="AI 智能助手",
description="基于 Gradio",
)
if __name__ == :
demo.launch(server_name=, server_port=)