bolade
39 lines
652 B
Python
39 lines
652 B
Python
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]
|