Chatbox调用Fastapi服务实践

原创

已于 2025-07-07 15:33:40 修改 · 291 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#fastapi #python #前端

于 2025-07-07 11:29:18 首次发布

1. 部署fastapi服务

fastapi_chatbox.py

from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import StreamingResponse
import uvicorn
import signal
import traceback
import sys
import json
import time

app = FastAPI()

# 允许跨域请求（如果Chatbox和FastAPI不在同一个域名下）
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 添加你需要的 /v1/chat/completions 路由
@app.post("/v1/chat/completions")
async def chat_completions(request: Request):
    data = await request.json()
    print("\n|> chat_completions Received chat completion request:\n", data)
    
    # 检查是否是流式请求
    stream = data.get("stream", False)
    
    if stream:
        # 流式响应
        async def generate():
            yield "data: {}\n\n".format(json.dumps({
                "id": "chatcmpl-123",
                "object": "chat.completion.chunk",