Refactor code structure for improved readability and maintainability

app/schemas/router_schemas.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 from enum import Enum
-from typing import List, Optional
+from typing import Any, List, Optional
 
 from pydantic import BaseModel
 
@@ -89,11 +89,20 @@ class InvestorSchema(BaseModel):
     id: int
     name: str
     description: Optional[str]
+    website: Optional[str] = None
+    headquarters: Optional[str] = None
     aum: int | None
+    aum_as_of_date: str | None = None
+    aum_source_url: str | None = None
     check_size_lower: int | None
     check_size_upper: int | None
     geographic_focus: str | None
-    stage_focus: InvestmentStage
+    investment_thesis: Any = (
+        None  # Flexible JSON field - can be list, dict, or list of dicts
+    )
+    portfolio_highlights: Any = (
+        None  # Flexible JSON field - can be list, dict, or list of dicts
+    )
     number_of_investments: int | None
     created_at: Optional[datetime] = None
     updated_at: Optional[datetime] = None
@@ -131,8 +140,8 @@ class InvestorFundData(BaseModel):
     aum: int | None
     aum_as_of_date: str | None
     aum_source_url: str | None
-    investment_thesis: List[str] | None
-    portfolio_highlights: List[str] | None
+    investment_thesis: Any = None  # Flexible JSON field
+    portfolio_highlights: Any = None  # Flexible JSON field
     number_of_investments: int | None
 
     # Fund fields