51 lines
1.0 KiB
Python
51 lines
1.0 KiB
Python
from pydantic import BaseModel
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
|
|
class AnalysisStatus(str, Enum):
|
|
PENDING = "pending"
|
|
IN_PROGRESS = "in_progress"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
class StockSearchRequest(BaseModel):
|
|
query: str
|
|
|
|
class StockSearchResponse(BaseModel):
|
|
market: str
|
|
symbol: str
|
|
company_name: str
|
|
|
|
class AnalysisRequest(BaseModel):
|
|
market: str
|
|
symbol: str
|
|
company_name: str
|
|
model: Optional[str] = None
|
|
data_source: Optional[str] = None
|
|
|
|
class ReportSectionSchema(BaseModel):
|
|
section_name: str
|
|
content: str
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class ReportResponse(BaseModel):
|
|
id: int
|
|
market: str
|
|
symbol: str
|
|
company_name: str
|
|
status: AnalysisStatus
|
|
ai_model: Optional[str] = None
|
|
created_at: datetime
|
|
sections: List[ReportSectionSchema] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
class ConfigUpdateRequest(BaseModel):
|
|
key: str
|
|
value: str
|