feat:修改nl2sql功能

api/routers/advisor_agent.py

@@ -97,6 +97,14 @@ def _not_ready(request: Request):
     return agent_failure(_NOT_READY_CODE, _NOT_READY_MESSAGE, trace_id=_trace_id(request))
 
 
+def _data_query_error_message(exc: QueryServiceError) -> str:
+    """Expose dependency outages without leaking SQL or database details."""
+    message = str(exc)
+    if message.startswith("投顾 Agent 依赖服务不可用"):
+        return message
+    return "客户数据查询失败，请稍后重试"
+
+
 def _advisor_runtime(request: Request):
     app = request.scope.get("app")
     return getattr(getattr(app, "state", None), "advisor_agent_runtime", None)
@@ -251,8 +259,8 @@ async def chat_stream(
 
     payload = None
 
-    # 请求体只传问题时，从问题中解析客户；解析结果仍必须经过投顾关系授权校验。
-    if customer_id is None:
+    # 单客户范围且未传编号时，兼容从问题中解析客户；投顾范围查询不解析客户。
+    if chat_request.scope == "customer" and customer_id is None:
         customer_id, resolve_error = await _resolve_customer_from_query(
             db,
             advisor_id=user.id,
@@ -276,7 +284,19 @@ async def chat_stream(
             payload = None
 
     # 不带客户编号时只提供通用基金问答，不读取客户画像，也不生成个性化草稿。
-    if customer_id is None:
+    if chat_request.scope == "advisor" and inferred_intent != AGENT_INTENT_DATA_QUERY:
+        payload = agent_failure(
+            ERR_CODE_FORBIDDEN_CUSTOMER,
+            "投顾范围查询仅支持客户数据查询",
+            trace_id=trace_id,
+        )
+    elif chat_request.scope == "advisor" and customer_id is not None:
+        payload = agent_failure(
+            ERR_CODE_FORBIDDEN_CUSTOMER,
+            "投顾范围查询不能指定单个客户",
+            trace_id=trace_id,
+        )
+    elif chat_request.scope == "customer" and customer_id is None:
         if inferred_intent in {
             AGENT_INTENT_RECOMMEND,
             AGENT_INTENT_REBALANCE,
@@ -319,10 +339,12 @@ async def chat_stream(
                 headers={"X-Trace-Id": trace_id},
             )
     else:
-        customer_id = chat_request.customer_id
-        relation = await ensure_customer_access(
-            db, advisor_id=user.id, customer_id=int(customer_id)
-        )
+        relation = None
+        if chat_request.scope == "customer":
+            customer_id = chat_request.customer_id
+            relation = await ensure_customer_access(
+                db, advisor_id=user.id, customer_id=int(customer_id)
+            )
         if inferred_intent == AGENT_INTENT_RECOMMEND:
             memories = await _recall_advisor_memories(
                 request,
@@ -382,15 +404,16 @@ async def chat_stream(
                     result = await execute_advisor_data_query(
                         db,
                         advisor_id=user.id,
-                        customer_id=int(customer_id),
+                        customer_id=int(customer_id) if customer_id is not None else None,
+                        scope=chat_request.scope,
                         question=chat_request.query,
                         trace_id=trace_id,
                         llm_client=getattr(_advisor_runtime(request), "llm_client", None),
                     )
-                except QueryServiceError:
+                except QueryServiceError as exc:
                     payload = agent_failure(
                         ERR_CODE_LLM_ERROR,
-                        "客户数据查询失败，请稍后重试",
+                        _data_query_error_message(exc),
                         trace_id=trace_id,
                     )
                 else:
@@ -507,10 +530,10 @@ async def advisor_data_query(
             sort_by=body.sort_by,
             sort_order=body.sort_order,
         )
-    except QueryServiceError:
+    except QueryServiceError as exc:
         return agent_failure(
             ERR_CODE_LLM_ERROR,
-            "客户数据查询失败，请稍后重试",
+            _data_query_error_message(exc),
             trace_id=trace_id,
         )
     result.pop("sql", None)