feat:投顾agent接入nl2sql能力
This commit is contained in:
@@ -1,10 +1,11 @@
|
||||
"""投顾工作台路由聚合(前缀 /api/advisor,在 api/router.py 以 /api 挂载)。"""
|
||||
from fastapi import APIRouter
|
||||
|
||||
from api.advisor import audit, customers, dashboard, diagnosis, drafts, report, todos, visits
|
||||
from api.advisor import audit, customers, dashboard, data_query, diagnosis, drafts, report, todos, visits
|
||||
|
||||
router = APIRouter(prefix="/advisor", tags=["投顾工作台"])
|
||||
router.include_router(dashboard.router)
|
||||
router.include_router(data_query.router)
|
||||
router.include_router(customers.router)
|
||||
router.include_router(diagnosis.router)
|
||||
router.include_router(drafts.router)
|
||||
|
||||
@@ -0,0 +1,31 @@
|
||||
"""投顾工作台客户数据查询代理路由。"""
|
||||
from fastapi import APIRouter, Depends, Request
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from api.advisor._auth import extract_auth
|
||||
from api.deps import require_advisor
|
||||
from config.deps import get_db
|
||||
from model.sys_user import SysUser
|
||||
from schemas.advisor import AdvisorDataQueryReq
|
||||
from service.advisor.data_query import query_customer_data
|
||||
from utils.response import success
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.post("/data-query", summary="查询当前客户数据(代理投顾Agent NL2SQL)")
|
||||
async def data_query(
|
||||
req: AdvisorDataQueryReq,
|
||||
request: Request,
|
||||
user: SysUser = Depends(require_advisor),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
auth, trace_id = extract_auth(request)
|
||||
data = await query_customer_data(
|
||||
db,
|
||||
user,
|
||||
auth_header=auth,
|
||||
trace_id=trace_id,
|
||||
req=req,
|
||||
)
|
||||
return success(data)
|
||||
Reference in New Issue
Block a user