from pydantic import BaseModel
from enum import Enum

class StockAnalyzeReq(BaseModel):
    query: str

class TickerInfo(BaseModel):
    ticker: str
    company_name: str


# enum
# risk_type : aggressive, balanced, conservative
class RiskType(str, Enum):
    AGGRESSIVE = "aggressive"
    BALANCED = "balanced"
    CONSERVATIVE = "conservative"

class RecommendStock(BaseModel):
    tickers: list[str]
    risk_type : RiskType