diff --git a/dsRag/Start.py b/dsRag/Start.py index 2fa66ac7..61c45016 100644 --- a/dsRag/Start.py +++ b/dsRag/Start.py @@ -4,7 +4,7 @@ from logging.handlers import RotatingFileHandler import jieba # 导入 jieba 分词库 import uvicorn -from fastapi import FastAPI, Request +from fastapi import FastAPI, Request, Body from openai import OpenAI from sse_starlette.sse import EventSourceResponse from starlette.staticfiles import StaticFiles @@ -178,8 +178,7 @@ async def generate_stream(client, milvus_pool, collection_manager, query): http://10.10.21.22:8000/api/rag?query=小学数学中有哪些模型 """ @app.post("/api/rag") -@app.get("/api/rag") -async def rag_stream(query: str, request: Request): +async def rag_stream(request: Request, query: str = Body(...)): """RAG+DeepSeek流式接口""" return EventSourceResponse( generate_stream(