from typing import Any

from pydantic import BaseModel, Field


class AIAskRequest(BaseModel):
    prompt: str
    context: dict[str, Any] = Field(default_factory=dict)
    actor: str = "api"
    source: str = "api"


class AIAskResponse(BaseModel):
    provider: str
    answer: str
    raw: dict[str, Any] = Field(default_factory=dict)


class DraftPolicyRequest(BaseModel):
    title: str
    policy_type: str
    requirements: list[str]
    actor: str = "api"


class InvestmentResearchRequest(BaseModel):
    symbol_or_topic: str
    risk_preference: str = "balanced"
    actor: str = "api"