Update schema.py

schema.py

@@ -7,10 +7,10 @@ class TaskPlan(BaseModel):
     data_key: Optional[str] = None
     filter: Optional[str] = None
     derive: Optional[List[str]] = None
-    joins: Optional[List[Dict]] = None  # [{"right_key":"...", "left_on":"...", "right_on":"...", "how":"left"}]
+    joins: Optional[List[Dict]] = None
     group_by: Optional[List[str]] = None
-    agg: Optional[List[str]] = None     # ["avg(x)", "median(y)", "p90(z)", "count(*)"]
-    pivot: Optional[Dict] = None        # {"index":["a","b"], "columns":"c", "values":"v"}
+    agg: Optional[List[str]] = None
+    pivot: Optional[Dict] = None
     sort_by: Optional[str] = None
     sort_dir: Optional[str] = "desc"
     top: Optional[int] = None
@@ -23,7 +23,8 @@ class TaskPlan(BaseModel):
     @validator("format")
     def _fmt(cls, v):
         allowed = {"table","list","comparison","map","narrative","chart"}
-        if v not in allowed: raise ValueError(f"format must be one of {allowed}")
+        if v not in allowed:
+            raise ValueError(f"format must be one of {allowed}")
         return v
 
 class ScenarioPlan(BaseModel):