bolade
102 lines
2.1 KiB
Python
102 lines
2.1 KiB
Python
from datetime import datetime
|
|
from enum import Enum
|
|
from typing import List, Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class InvestmentStage(str, Enum):
|
|
SEED = "SEED"
|
|
SERIES_A = "SERIES_A"
|
|
SERIES_B = "SERIES_B"
|
|
SERIES_C = "SERIES_C"
|
|
GROWTH = "GROWTH"
|
|
LATE_STAGE = "LATE_STAGE"
|
|
|
|
|
|
class SectorSchema(BaseModel):
|
|
id: int
|
|
name: str
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class InvestorMemberSchema(BaseModel):
|
|
id: int
|
|
name: str
|
|
role: str | None
|
|
email: str | None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class CompanyMemberSchema(BaseModel):
|
|
id: int
|
|
name: Optional[str]
|
|
linkedin: Optional[str]
|
|
role: Optional[str]
|
|
company_id: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class CompanySchema(BaseModel):
|
|
id: int
|
|
name: str
|
|
industry: str | None
|
|
location: str | None
|
|
description: Optional[str]
|
|
founded_year: Optional[int]
|
|
website: Optional[str]
|
|
created_at: Optional[datetime] = None
|
|
updated_at: Optional[datetime] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class InvestorSchema(BaseModel):
|
|
id: int
|
|
name: str
|
|
description: Optional[str]
|
|
aum: int | None
|
|
check_size_lower: int | None
|
|
check_size_upper: int | None
|
|
geographic_focus: str | None
|
|
stage_focus: InvestmentStage
|
|
number_of_investments: int | None
|
|
created_at: Optional[datetime] = None
|
|
updated_at: Optional[datetime] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class InvestorData(BaseModel):
|
|
"""Comprehensive investor data schema for LLM processing"""
|
|
|
|
investor: InvestorSchema
|
|
portfolio_companies: List[CompanySchema]
|
|
team_members: List[InvestorMemberSchema]
|
|
sectors: List[SectorSchema]
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class CompanyData(BaseModel): # Renamed from CompaniesData for consistency
|
|
company: CompanySchema
|
|
sectors: List[SectorSchema]
|
|
members: List[CompanyMemberSchema]
|
|
investors: List[InvestorSchema]
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class InvestorList(BaseModel):
|
|
investors: List[InvestorData]
|