80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
|
"""
|
|||
|
|
RAG Lab endpoints for debugging and experimentation.
|
|||
|
|
[AC-ASA-05] RAG experiment debugging with retrieval results and prompt visualization.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from typing import Annotated, Any, List
|
|||
|
|
|
|||
|
|
from fastapi import APIRouter, Depends
|
|||
|
|
from fastapi.responses import JSONResponse
|
|||
|
|
|
|||
|
|
from app.core.tenant import get_tenant_id
|
|||
|
|
from app.models import ErrorResponse
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
router = APIRouter(prefix="/admin/rag", tags=["RAG Lab"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
@router.post(
|
|||
|
|
"/experiments/run",
|
|||
|
|
operation_id="runRagExperiment",
|
|||
|
|
summary="Run RAG debugging experiment",
|
|||
|
|
description="[AC-ASA-05] Trigger RAG experiment with retrieval and prompt generation.",
|
|||
|
|
responses={
|
|||
|
|
200: {"description": "Experiment results with retrieval and prompt"},
|
|||
|
|
401: {"description": "Unauthorized", "model": ErrorResponse},
|
|||
|
|
403: {"description": "Forbidden", "model": ErrorResponse},
|
|||
|
|
},
|
|||
|
|
)
|
|||
|
|
async def run_rag_experiment(
|
|||
|
|
tenant_id: Annotated[str, Depends(get_tenant_id)],
|
|||
|
|
query: str,
|
|||
|
|
kb_ids: List[str],
|
|||
|
|
params: dict = None,
|
|||
|
|
) -> JSONResponse:
|
|||
|
|
"""
|
|||
|
|
[AC-ASA-05] Run RAG experiment and return retrieval results with final prompt.
|
|||
|
|
"""
|
|||
|
|
logger.info(
|
|||
|
|
f"[AC-ASA-05] Running RAG experiment: tenant={tenant_id}, "
|
|||
|
|
f"query={query}, kb_ids={kb_ids}"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
mock_retrieval_results = [
|
|||
|
|
{
|
|||
|
|
"content": "产品价格根据套餐不同有所差异，基础版每月99元，专业版每月299元。",
|
|||
|
|
"score": 0.92,
|
|||
|
|
"source": "product_manual.pdf",
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"content": "企业版提供定制化服务，请联系销售获取报价。",
|
|||
|
|
"score": 0.85,
|
|||
|
|
"source": "pricing_guide.docx",
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"content": "所有套餐均支持7天无理由退款。",
|
|||
|
|
"score": 0.78,
|
|||
|
|
"source": "faq.pdf",
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
mock_final_prompt = f"""基于以下检索到的信息，回答用户问题：
|
|||
|
|
|
|||
|
|
用户问题：{query}
|
|||
|
|
|
|||
|
|
检索结果：
|
|||
|
|
1. [Score: 0.92] 产品价格根据套餐不同有所差异，基础版每月99元，专业版每月299元。
|
|||
|
|
2. [Score: 0.85] 企业版提供定制化服务，请联系销售获取报价。
|
|||
|
|
3. [Score: 0.78] 所有套餐均支持7天无理由退款。
|
|||
|
|
|
|||
|
|
请基于以上信息生成专业、准确的回答。"""
|
|||
|
|
|
|||
|
|
return JSONResponse(
|
|||
|
|
content={
|
|||
|
|
"retrievalResults": mock_retrieval_results,
|
|||
|
|
"finalPrompt": mock_final_prompt,
|
|||
|
|
}
|
|||
|
|
)
|