demo seed change

services/inventory/app/api/ml_insights.py

@@ -319,3 +319,89 @@ async def ml_insights_health():
             "POST /ml/insights/optimize-safety-stock"
         ]
     }
+
+
+# ================================================================
+# INTERNAL ENDPOINTS (for demo-session service)
+# ================================================================
+
+from fastapi import Request
+
+# Create a separate router for internal endpoints to avoid the tenant prefix
+internal_router = APIRouter(
+    tags=["ML Insights - Internal"]
+)
+
+
+@internal_router.post("/api/v1/tenants/{tenant_id}/inventory/internal/ml/generate-safety-stock-insights")
+async def generate_safety_stock_insights_internal(
+    tenant_id: str,
+    request: Request,
+    db: AsyncSession = Depends(get_db)
+):
+    """
+    Internal endpoint to trigger safety stock insights generation for demo sessions.
+
+    This endpoint is called by the demo-session service after cloning data.
+    It uses the same ML logic as the public endpoint but with optimized defaults.
+
+    Security: Protected by X-Internal-Service header check.
+
+    Args:
+        tenant_id: The tenant UUID
+        request: FastAPI request object
+        db: Database session
+
+    Returns:
+        {
+            "insights_posted": int,
+            "tenant_id": str,
+            "status": str
+        }
+    """
+    # Verify internal service header
+    if not request or request.headers.get("X-Internal-Service") not in ["demo-session", "internal"]:
+        logger.warning("Unauthorized internal API call", tenant_id=tenant_id)
+        raise HTTPException(
+            status_code=403,
+            detail="This endpoint is for internal service use only"
+        )
+
+    logger.info("Internal safety stock insights generation triggered", tenant_id=tenant_id)
+
+    try:
+        # Use the existing safety stock optimization logic with sensible defaults
+        request_data = SafetyStockOptimizationRequest(
+            product_ids=None,  # Analyze all products
+            lookback_days=90,  # 3 months of history
+            min_history_days=30  # Minimum 30 days required
+        )
+
+        # Call the existing safety stock optimization endpoint logic
+        result = await trigger_safety_stock_optimization(
+            tenant_id=tenant_id,
+            request_data=request_data,
+            db=db
+        )
+
+        # Return simplified response for internal use
+        return {
+            "insights_posted": result.total_insights_posted,
+            "tenant_id": tenant_id,
+            "status": "success" if result.success else "failed",
+            "message": result.message,
+            "products_optimized": result.products_optimized,
+            "total_cost_savings": result.total_cost_savings
+        }
+
+    except Exception as e:
+        logger.error(
+            "Internal safety stock insights generation failed",
+            tenant_id=tenant_id,
+            error=str(e),
+            exc_info=True
+        )
+        raise HTTPException(
+            status_code=500,
+            detail=f"Internal safety stock insights generation failed: {str(e)}"
+        )