袁聪的第一次提交,包含nl2sql,行情数据,场外申购

This commit is contained in:
2026-09-10 09:23:22 +08:00
parent 46fc976b24
commit 5907fcd6d2
49 changed files with 6801 additions and 33 deletions
+53
View File
@@ -0,0 +1,53 @@
from typing import Any
from pydantic import BaseModel, ConfigDict, Field, field_validator
class FinancialNL2SQLInput(BaseModel):
"""金融 NL2SQL 公共工具入参;身份和权限只信任 RequestContext。"""
model_config = ConfigDict(extra="forbid", frozen=True)
question: str = Field(min_length=1, max_length=500)
confirmation: str | None = Field(default=None, max_length=200)
dry_run: bool = False
limit: int = Field(default=50, ge=1, le=200)
@field_validator("question")
@classmethod
def question_must_not_be_blank(cls, value: str) -> str:
if not value.strip():
raise ValueError("question must not be blank")
return value.strip()
class FinancialQueryPlan(BaseModel):
model_config = ConfigDict(extra="forbid", frozen=True)
intent: str
domains: tuple[str, ...]
tables: tuple[str, ...]
metrics: tuple[str, ...] = ()
dimensions: tuple[str, ...] = ()
filters: tuple[dict[str, Any], ...] = ()
time_mode: str = "none"
time_column: str | None = None
start: str | None = None
end: str | None = None
limit: int = Field(default=50, ge=1, le=200)
confidence: float = Field(default=0.0, ge=0, le=1)
needs_confirmation: bool = False
confirmation_question: str | None = None
unsupported_reason: str | None = None
class FinancialNL2SQLResult(BaseModel):
model_config = ConfigDict(extra="forbid", frozen=True)
status: str
message: str
data: dict[str, Any] = Field(default_factory=dict)
query_plan: dict[str, Any] = Field(default_factory=dict)
sql: str | None = None
parameters: dict[str, Any] = Field(default_factory=dict)
audit: dict[str, Any] = Field(default_factory=dict)