33 lines
1005 B
Python
33 lines
1005 B
Python
"""工作台到投顾 Agent 的客户数据查询代理。"""
|
||
from __future__ import annotations
|
||
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from model.sys_user import SysUser
|
||
from schemas.advisor import AdvisorDataQueryReq
|
||
from service.advisor.agent_client import get_agent_client
|
||
from service.advisor.permissions import ensure_customer_owned
|
||
|
||
|
||
async def query_customer_data(
|
||
db: AsyncSession,
|
||
user: SysUser,
|
||
*,
|
||
auth_header: str,
|
||
trace_id: str,
|
||
req: AdvisorDataQueryReq,
|
||
) -> dict:
|
||
"""先做工作台归属校验，再透传到 Agent；customer_id 不由 Agent 客户端覆盖。"""
|
||
await ensure_customer_owned(db, user.id, req.customer_id)
|
||
payload = req.model_dump(exclude_none=True)
|
||
result = await get_agent_client().data_query(
|
||
payload,
|
||
auth_header=auth_header,
|
||
trace_id=trace_id,
|
||
)
|
||
data = result["data"] or {}
|
||
data.pop("sql", None)
|
||
if result["warning"]:
|
||
data["warning"] = result["warning"]
|
||
return data
|