models/analysis.py

# models/analysis.py
from pydantic import BaseModel
from typing import List, Optional

class StockAnalysisResponse(BaseModel):
    basic_analysis: str
    # Add fields for more structured analysis if needed

class SwarmAgentOutput(BaseModel):
    role: str
    content: str

class SwarmAnalysisResponse(BaseModel):
    status: str # 'success' or 'error'
    output: Optional[List[SwarmAgentOutput]] = None
    error: Optional[str] = None

class InvestmentRecommendationResponse(BaseModel):
    recommendation: str