|
|
|
@ -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(
|
|
|
|
|