models.py updated

app/models.py

@@ -32,30 +32,51 @@ class ExportResponse(BaseModel):
 
 class SectionJob(BaseModel):
     """Job model for section generation"""
-    businessPlanId: str
     sectionKey: str
-    businessIdea: str
-    countryCode: Optional[str] = None
-    defaultCountry: Optional[str] = None
+    # The server will load prompt/model defaults from GeneratePrompt.
+    # Client MAY override but is not required to send these.
+    prompt: Optional[str] = None
+    promptVariables: Dict[str, str] = {}
+    model: Optional[str] = None
+    provider: Optional[str] = None
+    temperature: Optional[float] = None
+    maxTokens: Optional[int] = None
 
 class BatchGenerateRequest(BaseModel):
     """Request model for batch generation"""
-    businessPlanId: str
-    businessIdea: str
-    sections: List[str]
+    answers: List[str]
+    planType: Optional[str] = "free"
+    businessPlanId: Optional[str] = None
     countryCode: Optional[str] = None
-    defaultCountry: Optional[str] = None
+    businessIdea: Optional[str] = None
+    anonymousId: Optional[str] = None
+    feedback: Optional[str] = None
+    sections: List[SectionJob]
+    format: Optional[str] = None
+    chartImages: Optional[List[str]] = None
 
 class GenerateRequest(BaseModel):
     """Request model for single section generation"""
-    businessPlanId: str
-    sectionKey: str
-    businessIdea: str
+    answers: List[str]
+    model: str
+    prompt: str
+    promptVariables: Dict[str, str]
+    provider: str
+    temperature: float
+    maxTokens: int
+    # NEW — optional, for persistence & ownership
+    businessPlanId: Optional[str] = None
+    planType: Optional[str] = "free"
+    userId: Optional[str] = None           # will be overridden by verified JWT
+    anonymousId: Optional[str] = None
     countryCode: Optional[str] = None
-    defaultCountry: Optional[str] = None
+    sectionKey: Optional[str] = None       # e.g. "prompt_ExecutiveSummary"
+    returnContent: Optional[bool] = False  # NEW — default to id-only responses
 
 class CreatePlanRequest(BaseModel):
     """Request model for creating a new business plan"""
-    businessIdea: str
+    answers: List[str] = []
+    planType: Optional[str] = "free"
     countryCode: Optional[str] = None
-    defaultCountry: Optional[str] = None
+    anonymousId: Optional[str] = None
+    businessIdea: Optional[str] = None