[add] Change to "Body - json" format and pass as parameters

This commit is contained in:
lanceyq
2026-03-10 13:39:24 +08:00
parent 424d2033ea
commit 58aa60ca0e

View File

@@ -1,4 +1,5 @@
from fastapi import APIRouter, Depends, HTTPException, status, Query from fastapi import APIRouter, Depends, HTTPException, status, Query
from pydantic import BaseModel, Field
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from typing import Optional from typing import Optional
from app.core.response_utils import success from app.core.response_utils import success
@@ -476,11 +477,15 @@ async def get_chunk_insight(
return success(data=data, msg="获取成功") return success(data=data, msg="获取成功")
class GenerateRagProfileRequest(BaseModel):
end_user_id: str = Field(..., description="宿主ID")
limit: int = Field(15, description="参与生成的chunk数量上限")
max_tags: int = Field(10, description="最大标签数量")
@router.post("/generate_rag_profile", response_model=ApiResponse) @router.post("/generate_rag_profile", response_model=ApiResponse)
async def generate_rag_profile( async def generate_rag_profile(
end_user_id: str = Query(..., description="宿主ID"), body: GenerateRagProfileRequest,
limit: int = Query(15, description="参与生成的chunk数量上限"),
max_tags: int = Query(10, description="最大标签数量"),
db: Session = Depends(get_db), db: Session = Depends(get_db),
current_user: User = Depends(get_current_user), current_user: User = Depends(get_current_user),
): ):
@@ -488,12 +493,12 @@ async def generate_rag_profile(
生产接口为RAG存储模式的宿主全量重新生成完整画像并持久化到end_user表。 生产接口为RAG存储模式的宿主全量重新生成完整画像并持久化到end_user表。
每次请求都会重新生成,覆盖已有数据。 每次请求都会重新生成,覆盖已有数据。
""" """
api_logger.info(f"用户 {current_user.username} 触发RAG画像生产: end_user_id={end_user_id}") api_logger.info(f"用户 {current_user.username} 触发RAG画像生产: end_user_id={body.end_user_id}")
data = await memory_dashboard_service.generate_rag_profile( data = await memory_dashboard_service.generate_rag_profile(
end_user_id=end_user_id, end_user_id=body.end_user_id,
limit=limit, limit=body.limit,
max_tags=max_tags, max_tags=body.max_tags,
db=db, db=db,
current_user=current_user, current_user=current_user,
) )