diff --git a/server/routes/flow_framework_routes_service.ts b/server/routes/flow_framework_routes_service.ts
index 2e03e9b0..f8151bff 100644
--- a/server/routes/flow_framework_routes_service.ts
+++ b/server/routes/flow_framework_routes_service.ts
@@ -157,6 +157,9 @@ export function registerFlowFrameworkRoutes(
           reprovision: schema.boolean(),
         }),
         body: schema.any(),
+        query: schema.object({
+          data_source_version: schema.maybe(schema.string()),
+        }),
       },
     },
     flowFrameworkRoutesService.updateWorkflow
@@ -188,6 +191,9 @@ export function registerFlowFrameworkRoutes(
         params: schema.object({
           workflow_id: schema.string(),
         }),
+        query: schema.object({
+          data_source_version: schema.maybe(schema.string()),
+        }),
       },
     },
     flowFrameworkRoutesService.provisionWorkflow
diff --git a/server/routes/opensearch_routes_service.ts b/server/routes/opensearch_routes_service.ts
index 6af9cf64..dbabf50d 100644
--- a/server/routes/opensearch_routes_service.ts
+++ b/server/routes/opensearch_routes_service.ts
@@ -124,6 +124,7 @@ export function registerOpenSearchRoutes(
         body: schema.any(),
         query: schema.object({
           verbose: schema.boolean(),
+          data_source_version: schema.maybe(schema.string()),
         }),
       },
     },
@@ -157,6 +158,7 @@ export function registerOpenSearchRoutes(
         body: schema.any(),
         query: schema.object({
           verbose: schema.boolean(),
+          data_source_version: schema.maybe(schema.string()),
         }),
       },
     },