fix(clustering): add removed fileds for new ai plugins

max_request_body_size field
Kong · Jul 10, 2024 · 2f7d8d8 · 2f7d8d8
1 parent df88c98
commit 2f7d8d8
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 11 deletions.
diff --git a/kong/clustering/compat/removed_fields.lua b/kong/clustering/compat/removed_fields.lua
@@ -167,5 +167,23 @@ return {
       "traces_endpoint",
       "logs_endpoint",
     },
+    ai_proxy = {
+      "max_request_body_size",
+    },
+    ai_prompt_decorator = {
+      "max_request_body_size",
+    },
+    ai_prompt_guard = {
+      "max_request_body_size",
+    },
+    ai_prompt_template = {
+      "max_request_body_size",
+    },
+    ai_request_transformer = {
+      "max_request_body_size",
+    },
+    ai_response_transformer = {
+      "max_request_body_size",
+    },
   },
 }
diff --git a/spec/02-integration/09-hybrid_mode/09-config-compat_spec.lua b/spec/02-integration/09-hybrid_mode/09-config-compat_spec.lua
@@ -547,12 +547,22 @@ describe("CP/DP config compat transformations #" .. strategy, function()
         }
         -- ]]
 
-        local expected_ai_proxy_prior_37 = cycle_aware_deep_copy(ai_proxy)
-        expected_ai_proxy_prior_37.config.response_streaming = nil
-        expected_ai_proxy_prior_37.config.model.options.upstream_path = nil
-        expected_ai_proxy_prior_37.config.route_type = "llm/v1/chat"
+        local expected = cycle_aware_deep_copy(ai_proxy)
 
-        do_assert(uuid(), "3.6.0", expected_ai_proxy_prior_37)
+        expected.config.max_request_body_size = nil
+
+        do_assert(uuid(), "3.7.0", expected)
+
+        local expected = cycle_aware_deep_copy(ai_proxy)
+        expected.config.response_streaming = nil
+        expected.config.model.options.upstream_path = nil
+        expected.config.auth.azure_use_managed_identity = nil
+        expected.config.auth.azure_client_id = nil
+        expected.config.auth.azure_client_secret = nil
+        expected.config.auth.azure_tenant_id = nil
+        expected.config.route_type = "llm/v1/chat"
+
+        do_assert(uuid(), "3.6.0", expected)
 
         -- cleanup
         admin.plugins:remove({ id = ai_proxy.id })
@@ -588,11 +598,19 @@ describe("CP/DP config compat transformations #" .. strategy, function()
         }
         -- ]]
 
-        local expected_ai_request_transformer_prior_37 = cycle_aware_deep_copy(ai_request_transformer)
-        expected_ai_request_transformer_prior_37.config.llm.model.options.upstream_path = nil
+        local expected = cycle_aware_deep_copy(ai_request_transformer)
+
+        expected.config.max_request_body_size = nil
 
-        do_assert(uuid(), "3.6.0", expected_ai_request_transformer_prior_37)
+        do_assert(uuid(), "3.7.0", expected)
 
+        expected.config.llm.model.options.upstream_path = nil
+        expected.config.llm.auth.azure_use_managed_identity = nil
+        expected.config.llm.auth.azure_client_id = nil
+        expected.config.llm.auth.azure_client_secret = nil
+        expected.config.llm.auth.azure_tenant_id = nil
+
+        do_assert(uuid(), "3.6.0", expected)
         -- cleanup
         admin.plugins:remove({ id = ai_request_transformer.id })
       end)
@@ -625,10 +643,20 @@ describe("CP/DP config compat transformations #" .. strategy, function()
         }
         -- ]]
 
-        local expected_ai_response_transformer_prior_37 = cycle_aware_deep_copy(ai_response_transformer)
-        expected_ai_response_transformer_prior_37.config.llm.model.options.upstream_path = nil
+        local expected = cycle_aware_deep_copy(ai_response_transformer)
+
+        expected.config.max_request_body_size = nil
+
+        do_assert(uuid(), "3.7.0", expected)
+
+        local expected = cycle_aware_deep_copy(ai_response_transformer)
+        expected.config.llm.model.options.upstream_path = nil
+        expected.config.llm.auth.azure_use_managed_identity = nil
+        expected.config.llm.auth.azure_client_id = nil
+        expected.config.llm.auth.azure_client_secret = nil
+        expected.config.llm.auth.azure_tenant_id = nil
 
-        do_assert(uuid(), "3.6.0", expected_ai_response_transformer_prior_37)
+        do_assert(uuid(), "3.6.0", expected)
 
         -- cleanup
         admin.plugins:remove({ id = ai_response_transformer.id })