From e3f8a3ec397f462afb1b641f0bf9a141a17ab6eb Mon Sep 17 00:00:00 2001
From: Wangchong Zhou <wangchong@konghq.com>
Date: Thu, 8 Aug 2024 19:20:40 +0800
Subject: [PATCH] aa

---
 kong/clustering/compat/removed_fields.lua                | 2 +-
 .../09-hybrid_mode/09-config-compat_spec.lua             | 9 ++++++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/kong/clustering/compat/removed_fields.lua b/kong/clustering/compat/removed_fields.lua
index 4a2d6d89a27f..d37db9a4172d 100644
--- a/kong/clustering/compat/removed_fields.lua
+++ b/kong/clustering/compat/removed_fields.lua
@@ -175,7 +175,7 @@ return {
       "model.options.bedrock",
       "auth.aws_access_key_id",
       "auth.aws_secret_access_key",
-      "model_header_name",
+      "model_name_header",
     },
     ai_prompt_decorator = {
       "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
index 106fee166b9f..d23bd86c5d9d 100644
--- a/spec/02-integration/09-hybrid_mode/09-config-compat_spec.lua
+++ b/spec/02-integration/09-hybrid_mode/09-config-compat_spec.lua
@@ -482,7 +482,7 @@ describe("CP/DP config compat transformations #" .. strategy, function()
     end)
 
     describe("ai plugins supported providers", function()
-      it("[ai-proxy] tries to use unsupported providers on older Kong versions", function()
+      it("[ai-proxy] tries to use unsupported providers on older Kong versions #ttt", function()
         -- [[ 3.8.x ]] --
         local ai_proxy = admin.plugins:insert {
           name = "ai-proxy",
@@ -510,6 +510,7 @@ describe("CP/DP config compat transformations #" .. strategy, function()
               },
             },
             max_request_body_size = 8192,
+            model_name_header = true,
           },
         }
         -- ]]
@@ -518,6 +519,8 @@ describe("CP/DP config compat transformations #" .. strategy, function()
 
         -- max body size
         expected.config.max_request_body_size = nil
+
+        -- model name header
         expected.config.model_name_header = nil
 
         -- gemini fields
@@ -696,6 +699,7 @@ describe("CP/DP config compat transformations #" .. strategy, function()
               },
             },
             max_request_body_size = 8192,
+            model_name_header = true,
           },
         }
         -- ]]
@@ -705,6 +709,9 @@ describe("CP/DP config compat transformations #" .. strategy, function()
         -- max body size
         expected.config.max_request_body_size = nil
 
+        -- model name header
+        expected.config.model_name_header = nil
+
         -- gemini fields
         expected.config.auth.gcp_service_account_json = nil
         expected.config.auth.gcp_use_service_account = nil