3475138419
- 后端: Tushare provider 新增 get_employee_number, get_holder_number, get_tax_to_ebt 方法，并在 financial 路由中集成。 - 前端: report 页面新增对应图表展示，并更新相关类型与工具函数。 - 清理: 移除多个过时的测试脚本。 - 文档: 创建 2025-11-04 开发日志并更新用户手册。
74 lines
1.6 KiB
Python
74 lines
1.6 KiB
Python
"""
|
|
Pydantic schemas for financial APIs
|
|
"""
|
|
from typing import Dict, List, Optional
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class PeriodDataPoint(BaseModel):
|
|
period: str
|
|
value: Optional[float]
|
|
|
|
|
|
class StepRecord(BaseModel):
|
|
name: str
|
|
start_ts: str # ISO8601
|
|
end_ts: Optional[str] = None
|
|
duration_ms: Optional[int] = None
|
|
status: str # running|done|error
|
|
error: Optional[str] = None
|
|
|
|
|
|
class FinancialMeta(BaseModel):
|
|
started_at: str # ISO8601
|
|
finished_at: Optional[str] = None
|
|
elapsed_ms: Optional[int] = None
|
|
api_calls_total: int = 0
|
|
api_calls_by_group: Dict[str, int] = {}
|
|
current_action: Optional[str] = None
|
|
steps: List[StepRecord] = []
|
|
|
|
|
|
class BatchFinancialDataResponse(BaseModel):
|
|
ts_code: str
|
|
name: Optional[str] = None
|
|
series: Dict[str, List[PeriodDataPoint]]
|
|
meta: Optional[FinancialMeta] = None
|
|
|
|
|
|
class FinancialConfigResponse(BaseModel):
|
|
api_groups: Dict[str, List[dict]]
|
|
|
|
|
|
class TokenUsage(BaseModel):
|
|
prompt_tokens: int = 0
|
|
completion_tokens: int = 0
|
|
total_tokens: int = 0
|
|
|
|
|
|
class CompanyProfileResponse(BaseModel):
|
|
ts_code: str
|
|
company_name: Optional[str] = None
|
|
content: str
|
|
model: str
|
|
tokens: TokenUsage
|
|
elapsed_ms: int
|
|
success: bool = True
|
|
error: Optional[str] = None
|
|
|
|
|
|
class AnalysisResponse(BaseModel):
|
|
ts_code: str
|
|
company_name: Optional[str] = None
|
|
analysis_type: str
|
|
content: str
|
|
model: str
|
|
tokens: TokenUsage
|
|
elapsed_ms: int
|
|
success: bool = True
|
|
error: Optional[str] = None
|
|
|
|
|
|
class AnalysisConfigResponse(BaseModel):
|
|
analysis_modules: Dict[str, Dict]
|