GenerateId Added to backend

The generateId is now added to the backend and is also now fixed, so each procedure contains it now on the backend

app/main.py

@@ -391,6 +391,7 @@ class GenerateRequest(BaseModel):
     countryCode: Optional[str] = None
     sectionKey: Optional[str] = None       # e.g. "prompt_ExecutiveSummary"
     returnContent: Optional[bool] = False  # NEW — default to id-only responses 
+    generateId: Optional[str] = None
 
 # ──────────────────────────────────────────────────────────────────────────────
 # LLM plumbing
@@ -930,7 +931,7 @@ def _compose_prompt_vars(
 async def _ensure_business_plan(pool, *, bp_id: Optional[str], summary: str, full_plan: str,
                                 model: str, answers: List[str], plan_type: Optional[str],
                                 user_id: Optional[str], anon_id: Optional[str],
-                                country_code: Optional[str]) -> str:
+                                country_code: Optional[str], generate_id: Optional[str] = None) -> str:
     async with pool.acquire() as conn:
         if bp_id:
             exists = await conn.fetchval('SELECT 1 FROM "BusinessPlan" WHERE id = $1', bp_id)
@@ -939,19 +940,20 @@ async def _ensure_business_plan(pool, *, bp_id: Optional[str], summary: str, ful
                     'UPDATE "BusinessPlan" SET '
                     '"summary"=$1, "model"=$2, "answers"=$3, '
                     '"planType"=COALESCE($4,\'free\'), "countryCode"=$5, '
-                    '"updatedAt"=now(), "userId"=COALESCE($6,"userId") '
-                    'WHERE id=$7',
-                    summary, model, answers, plan_type, country_code, user_id, bp_id
+                    '"generateId"=COALESCE($6,"generateId"), '
+                    '"updatedAt"=now(), "userId"=COALESCE($7,"userId") '
+                    'WHERE id=$8',
+                    summary, model, answers, plan_type, country_code, generate_id, user_id, bp_id
                 )
                 return bp_id
 
         new_id = new_cuid()
         row = await conn.fetchrow(
             'INSERT INTO "BusinessPlan" '
-            '("id","summary","fullPlan","model","answers","planType","userId","anonymousId","countryCode","updatedAt") '
-            'VALUES ($1,$2,$3,$4,$5,COALESCE($6,\'free\'),$7,$8,$9, now()) '
+            '("id","summary","fullPlan","model","answers","planType","userId","anonymousId","countryCode","generateId","updatedAt") '
+            'VALUES ($1,$2,$3,$4,$5,COALESCE($6,\'free\'),$7,$8,$9,$10, now()) '
             'RETURNING id',
-            new_id, summary, "", model, answers, plan_type, user_id, anon_id, country_code  # fullPlan is intentionally ""
+            new_id, summary, "", model, answers, plan_type, user_id, anon_id, country_code, generate_id
         )
         return row["id"]
 
@@ -1077,6 +1079,7 @@ class CreatePlanRequest(BaseModel):
     countryCode: Optional[str] = None
     anonymousId: Optional[str] = None
     businessIdea: Optional[str] = None
+    generateId: Optional[str] = None
 
 @app.post("/plan")
 async def create_plan_shell(request: Request, data: CreatePlanRequest):
@@ -1099,13 +1102,14 @@ async def create_plan_shell(request: Request, data: CreatePlanRequest):
         user_id=user_id,
         anon_id=anon_id,
         country_code=data.countryCode,
+        generate_id=data.generateId,
     )
     await _ensure_sections_row(pool, bp_id)
 
     # Tell subscribers a plan exists
     await _publish(bp_id, "plan.created", {"businessPlanId": bp_id})
 
-    return {"businessPlanId": bp_id}
+    return {"businessPlanId": bp_id, "generateId": data.generateId}
     
 @app.post("/generate/batch")
 async def generate_business_plan_batch(request: Request, data: BatchGenerateRequest):
@@ -1134,6 +1138,7 @@ async def generate_business_plan_batch(request: Request, data: BatchGenerateRequ
             user_id=user_id,
             anon_id=anon_id,
             country_code=data.countryCode,
+            generate_id=data.generateId,
         )
         # logging.info(f"[plan] ensured id={bp_id} linked to user_id={user_id} anon_id={anon_id}")
 
@@ -1225,6 +1230,7 @@ async def generate_business_plan_batch(request: Request, data: BatchGenerateRequ
             user_id=user_id,
             anon_id=anon_id,
             country_code=data.countryCode,
+            generate_id=data.generateId,
         )
 
         await _publish(bp_id, "plan.updated", {
@@ -1363,6 +1369,7 @@ async def generate_business_plan(request: Request, data: GenerateRequest):
                 user_id=user_id,
                 anon_id=anon_id,
                 country_code=data.countryCode,
+                generate_id=data.generateId,
             )
             # logging.info(f"[plan] ensured section id={business_plan_id} linked to user_id={user_id} anon_id={anon_id}")
 
@@ -1427,6 +1434,7 @@ async def generate_business_plan(request: Request, data: GenerateRequest):
             user_id=user_id,
             anon_id=anon_id,
             country_code=data.countryCode,
+            generate_id=data.generateId,
         )
         logging.info(f"[plan] ensured whole plan id={business_plan_id} linked to user_id={user_id} anon_id={anon_id}")
 

app/models.py

@@ -50,6 +50,7 @@ class BatchGenerateRequest(BaseModel):
     countryCode: Optional[str] = None
     businessIdea: Optional[str] = None
     anonymousId: Optional[str] = None
+    generateId: Optional[str] = None
     feedback: Optional[str] = None
     sections: List[SectionJob]
     format: Optional[str] = None
@@ -72,6 +73,7 @@ class GenerateRequest(BaseModel):
     countryCode: Optional[str] = None
     sectionKey: Optional[str] = None       # e.g. "prompt_ExecutiveSummary"
     returnContent: Optional[bool] = False  # NEW — default to id-only responses
+    generateId: Optional[str] = None
 
 class CreatePlanRequest(BaseModel):
     """Request model for creating a new business plan"""
@@ -79,4 +81,5 @@ class CreatePlanRequest(BaseModel):
     planType: Optional[str] = "free"
     countryCode: Optional[str] = None
     anonymousId: Optional[str] = None
-    businessIdea: Optional[str] = None
+    businessIdea: Optional[str] = None
+    generateId: Optional[str] = None