From 89f09ea57cf4e063b00d69ba16d7acac2a101061 Mon Sep 17 00:00:00 2001
From: Dog foot ruler <3510343+dogfootruler-kr@users.noreply.github.com>
Date: Wed, 13 Aug 2025 05:33:23 +0900
Subject: [PATCH] fix(open-api): Add missing FormatQueryPayloadSchema and
 DashboardScreenshotPostSchema to open-api component schemas (#33202)

---
 superset/dashboards/api.py | 2 ++
 superset/sqllab/api.py     | 1 +
 2 files changed, 3 insertions(+)

diff --git a/superset/dashboards/api.py b/superset/dashboards/api.py
index 30c0cb3de2c..6d8fd0da555 100644
--- a/superset/dashboards/api.py
+++ b/superset/dashboards/api.py
@@ -89,6 +89,7 @@ from superset.dashboards.schemas import (
     DashboardNativeFiltersConfigUpdateSchema,
     DashboardPostSchema,
     DashboardPutSchema,
+    DashboardScreenshotPostSchema,
     EmbeddedDashboardConfigSchema,
     EmbeddedDashboardResponseSchema,
     get_delete_ids_schema,
@@ -312,6 +313,7 @@ class DashboardRestApi(BaseSupersetModelRestApi):
         TabsPayloadSchema,
         GetFavStarIdsSchema,
         EmbeddedDashboardResponseSchema,
+        DashboardScreenshotPostSchema,
     )
     apispec_parameter_schemas = {
         "get_delete_ids_schema": get_delete_ids_schema,
diff --git a/superset/sqllab/api.py b/superset/sqllab/api.py
index a8ffd92f24f..906dd72bcaf 100644
--- a/superset/sqllab/api.py
+++ b/superset/sqllab/api.py
@@ -88,6 +88,7 @@ class SqlLabRestApi(BaseSupersetApi):
     openapi_spec_component_schemas = (
         EstimateQueryCostSchema,
         ExecutePayloadSchema,
+        FormatQueryPayloadSchema,
         QueryExecutionResponseSchema,
         SQLLabBootstrapSchema,
     )