from typing import Optional

from pydantic import BaseModel


class InsightResponse(BaseModel):
    investment_pattern_analysis: str
    market_position: str
    compatibility_score: Optional[str] = None

    class Config:
        json_schema_extra = {
            "example": {
                "investment_pattern_analysis": "Sequoia has been increasingly active in AI/ML startups (43% increase in last 18 months). Their average investment size has grown 23% year-over-year, indicating confidence in larger rounds. Peak activity in Q2-Q3, suggesting seasonal investment patterns.",
                "market_position": "Top 3 most active VC in enterprise software deals. Strong presence in unicorn companies (47 portfolio unicorns). Consistently leads or co-leads rounds, indicating decision-making influence.",
                "compatibility_score": "0.85",
            }
        }