feat(llm): recreate http request header and body when balancer retry

Kong · Dec 11, 2024 · 3e7c563 · 3e7c563
1 parent ea6b35f
commit 3e7c563
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kong/llm/plugin/base.lua b/kong/llm/plugin/base.lua
@@ -1,5 +1,6 @@
 local deflate_gzip = require("kong.tools.gzip").deflate_gzip
 local ai_plugin_ctx = require("kong.llm.plugin.ctx")
+local balancer = require("ngx.balancer")
 
 local get_global_ctx, _ = ai_plugin_ctx.get_global_accessors("_base")
 
@@ -93,6 +94,7 @@ end
 
 function MetaPlugin:retry(sub_plugin, conf)
   run_stage(STAGES.REQ_TRANSFORMATION, sub_plugin, conf)
+  balancer.recreate_request()
 end
 
 function MetaPlugin:rewrite(sub_plugin, conf)