Refactor investor-related schemas and models; update database configuration and enhance investor processing logic

app/py_schemas.py

@@ -0,0 +1,77 @@
+from pydantic import BaseModel
+from datetime import datetime
+from typing import List, Optional
+from enum import Enum
+
+
+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 CompanySchema(BaseModel):
+    id: int
+    name: str
+    industry: str
+    location: str
+    founded_year: Optional[int]
+    website: Optional[str]
+    created_at: Optional[datetime]
+    updated_at: Optional[datetime]
+
+    class Config:
+        from_attributes = True
+
+
+class InvestorTeamMemberSchema(BaseModel):
+    id: int
+    name: str
+    role: str
+    email: str
+
+    class Config:
+        from_attributes = True
+
+
+class InvestorSchema(BaseModel):
+    id: int
+    name: str
+    description: Optional[str]
+    aum: int
+    check_size_lower: int
+    check_size_upper: int
+    geographic_focus: str
+    stage_focus: InvestmentStage
+    number_of_investments: int
+    created_at: Optional[datetime]
+    updated_at: Optional[datetime]
+
+    class Config:
+        from_attributes = True
+
+
+class InvestorData(BaseModel):
+    """Comprehensive investor data schema for LLM processing"""
+    investor: InvestorSchema
+    portfolio_companies: List[CompanySchema] = []
+    team_members: List[InvestorTeamMemberSchema] = []
+    sectors: List[SectorSchema] = []
+
+    class Config:
+        from_attributes = True
+        
+
+class InvestorList(BaseModel):
+    investors: List[InvestorData]