Update schema.py

schema.py

@@ -1,49 +1,13 @@
-
-from pydantic import BaseModel, Field, ValidationError
-from typing import List, Optional
-
-class TimelineItem(BaseModel):
-    date: str = Field(..., description="YYYY-MM-DD")
-    event: str
-    source: str
-
-class OpportunityItem(BaseModel):
-    name: str
-    stage: str
-    amount: float
-    closeDate: Optional[str] = None
-    nextStep: Optional[str] = None
+from pydantic import BaseModel
+from typing import Optional
 
 class PreCallSummary(BaseModel):
-    client_overview: str
-    relationship_timeline: List[TimelineItem]
-    open_items: List[str]
-    opportunities: List[OpportunityItem]
-    risks_flags: List[str]
-    recommendations: List[str]
-    pre_call_questions: List[str]
-    sources: List[str]
-
-def ensure_schema(d: dict) -> PreCallSummary:
-    """Validate and coerce a dict to PreCallSummary, filling missing keys with safe defaults."""
-    defaults = {
-        "client_overview": "",
-        "relationship_timeline": [],
-        "open_items": [],
-        "opportunities": [],
-        "risks_flags": [],
-        "recommendations": [],
-        "pre_call_questions": [],
-        "sources": []
-    }
-    merged = {**defaults, **(d or {})}
-    try:
-        return PreCallSummary.model_validate(merged)
-    except ValidationError as e:
-        # Best-effort repair for common issues
-        for fld in ["relationship_timeline", "open_items", "opportunities", "risks_flags", "recommendations", "pre_call_questions", "sources"]:
-            if merged.get(fld) is None:
-                merged[fld] = []
-        if merged.get("client_overview") is None:
-            merged["client_overview"] = ""
-        return PreCallSummary.model_validate(merged)
+    client_overview: Optional[str] = None
+    relationship_timeline: Optional[str] = None
+    open_items: Optional[str] = None
+    opportunities: Optional[str] = None
+    risks_flags: Optional[str] = None
+    recommendations: Optional[str] = None
+    pre_call_questions: Optional[str] = None
+    sources: Optional[str] = None
+    summary: Optional[str] = None