75 lines
1.7 KiB
Python
75 lines
1.7 KiB
Python
"""
|
|
Pydantic schemas for financial APIs
|
|
"""
|
|
from typing import Dict, List, Optional
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class YearDataPoint(BaseModel):
|
|
year: str
|
|
value: Optional[float]
|
|
month: Optional[int] = None # 月份信息，用于确定季度
|
|
|
|
|
|
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[YearDataPoint]]
|
|
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]
|