Refactor investor-related schemas and models; implement investor CRUD operations and update stage_focus values to uppercase

app/py_schemas.py

@@ -1,16 +1,17 @@
-from pydantic import BaseModel
 from datetime import datetime
-from typing import List, Optional
 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"
+    SEED = "SEED"
+    SERIES_A = "SERIES_A"
+    SERIES_B = "SERIES_B"
+    SERIES_C = "SERIES_C"
+    GROWTH = "GROWTH"
+    LATE_STAGE = "LATE_STAGE"
 
 
 class SectorSchema(BaseModel):
@@ -64,6 +65,7 @@ class InvestorSchema(BaseModel):
 
 class InvestorData(BaseModel):
     """Comprehensive investor data schema for LLM processing"""
+
     investor: InvestorSchema
     portfolio_companies: List[CompanySchema] = []
     team_members: List[InvestorTeamMemberSchema] = []
@@ -71,7 +73,7 @@ class InvestorData(BaseModel):
 
     class Config:
         from_attributes = True
-        
+
 
 class InvestorList(BaseModel):
-    investors: List[InvestorData]
+    investors: List[InvestorData]