feat:投顾agent接入nl2sql能力

schemas/advisor_agent.py

@@ -7,6 +7,7 @@ from pydantic import BaseModel, Field
 
 from common.common_const import (
     AGENT_INTENT_DIALOGUE_SCRIPT,
+    AGENT_INTENT_DATA_QUERY,
     AGENT_INTENT_FUND_ANALYSIS,
     AGENT_INTENT_REBALANCE,
     AGENT_INTENT_RECOMMEND,
@@ -38,10 +39,22 @@ class AdvisorChatReq(BaseModel):
         AGENT_INTENT_REBALANCE,
         AGENT_INTENT_FUND_ANALYSIS,
         AGENT_INTENT_DIALOGUE_SCRIPT,
+        AGENT_INTENT_DATA_QUERY,
     ] | None = None
     query: str | None = Field(default=None, max_length=4000)
 
 
+class AdvisorDataQueryReq(BaseModel):
+    customer_id: int = Field(gt=0)
+    question: str = Field(min_length=1, max_length=2000)
+    session_id: str | None = Field(default=None, max_length=64)
+    max_rows: int | None = Field(default=None, gt=0, le=10000)
+    page: int = Field(default=1, ge=1, le=100000)
+    page_size: int = Field(default=100, ge=1, le=10000)
+    sort_by: str | None = Field(default=None, max_length=128)
+    sort_order: Literal["asc", "desc"] = "asc"
+
+
 class AdvisorFundAnalysisReq(BaseModel):
     customer_id: int | None = Field(default=None, gt=0)
     fund_codes: list[str] = Field(min_length=1)