袁聪的第一次提交,包含nl2sql,行情数据,场外申购
This commit is contained in:
@@ -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)
|
||||
Reference in New Issue
Block a user