|
|
|
@ -1,4 +1,4 @@
|
|
|
|
|
from fastapi import FastAPI
|
|
|
|
|
from fastapi import FastAPI, Body # 导入 Body
|
|
|
|
|
from fastapi.responses import StreamingResponse, PlainTextResponse
|
|
|
|
|
from fastapi.middleware.cors import CORSMiddleware # 导入 CORS 中间件
|
|
|
|
|
import socket
|
|
|
|
@ -59,16 +59,16 @@ async def generate_stream(course_name: str):
|
|
|
|
|
if chunk.choices[0].delta.content:
|
|
|
|
|
for char in chunk.choices[0].delta.content:
|
|
|
|
|
yield char.encode("utf-8")
|
|
|
|
|
await asyncio.sleep(0.05) # 控制逐字输出的速度
|
|
|
|
|
await asyncio.sleep(0.02) # 控制逐字输出的速度
|
|
|
|
|
|
|
|
|
|
# 根路由,返回提示信息
|
|
|
|
|
@app.get("/")
|
|
|
|
|
def root():
|
|
|
|
|
return PlainTextResponse("Hello ApiStream")
|
|
|
|
|
|
|
|
|
|
# 流式返回数据
|
|
|
|
|
@app.get("/stream")
|
|
|
|
|
async def stream_data(course_name: str):
|
|
|
|
|
# 流式返回数据(使用 POST 方法)
|
|
|
|
|
@app.post("/stream")
|
|
|
|
|
async def stream_data(course_name: str = Body(..., embed=True)): # 从请求体中获取 course_name
|
|
|
|
|
if not course_name:
|
|
|
|
|
return PlainTextResponse("请提供课程名称,例:course_name=三角形面积")
|
|
|
|
|
|
|
|
|
|