'commit'

AI/AiService/CallDeepSeekStream.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Body  # 导入 Body
+from fastapi import FastAPI, Query, Body  # 导入 Query 和 Body
 from fastapi.responses import StreamingResponse, PlainTextResponse
 from fastapi.middleware.cors import CORSMiddleware  # 导入 CORS 中间件
 import socket
@@ -59,19 +59,26 @@ 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.02)  # 控制逐字输出的速度
+                await asyncio.sleep(0.05)  # 控制逐字输出的速度
 
 # 根路由，返回提示信息
 @app.get("/")
 def root():
     return PlainTextResponse("Hello ApiStream")
 
-# 流式返回数据（使用 POST 方法）
-@app.post("/stream")
-async def stream_data(course_name: str = Body(..., embed=True)):  # 从请求体中获取 course_name
-    if not course_name:
+# 流式返回数据（支持 GET 和 POST 方法）
+@app.api_route("/stream", methods=["GET", "POST"])
+async def stream_data(
+    course_name: str = Query(None, description="课程名称（GET 方法使用）"),  # 从查询参数中获取 course_name
+    course_name_body: str = Body(None, embed=True, description="课程名称（POST 方法使用）")  # 从请求体中获取 course_name
+):
+    # 检查 course_name 是否为空
+    if not course_name and not course_name_body:
         return PlainTextResponse("请提供课程名称,例：course_name=三角形面积")
 
+    # 优先使用 POST 请求体中的 course_name
+    course_name = course_name_body if course_name_body else course_name
+
     # 返回流式响应
     return StreamingResponse(
         generate_stream(course_name),