from typing import List

from pydantic import BaseModel


class Investor(BaseModel):
    name: str
    aum: int
    check_size: str
    sector_focus: str
    stage_focus: str
    region: str
    investment_thesis: str
    investor_description: str


class InvestorList(BaseModel):
    investor_list: List[Investor]


class QueryResponse(BaseModel):
    name: str
    aum: int
    check_size: str
    sector_focus: str
    stage_focus: str
    region: str
    investment_thesis: str
    investor_description: str
    reason: str


class QueryRequest(BaseModel):
    question: str


class QueryResponseList(BaseModel):
    responses: List[QueryResponse]