58 lines
1.3 KiB
Python
58 lines
1.3 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]
|
|
|
|
|
|
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
|