From 19d154897153dc4143b9cf066e0cd3358a20c36d Mon Sep 17 00:00:00 2001
From: StarryNight <wangkai19@corp.netease.com>
Date: Mon, 18 Aug 2025 11:01:35 +0800
Subject: [PATCH] update ai-prompt-decorator to new plugin wrapper api (#2777)

---
 .../wasm-go/extensions/ai-prompt-decorator/main.go   | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/plugins/wasm-go/extensions/ai-prompt-decorator/main.go b/plugins/wasm-go/extensions/ai-prompt-decorator/main.go
index 6299e864f..f9efea368 100644
--- a/plugins/wasm-go/extensions/ai-prompt-decorator/main.go
+++ b/plugins/wasm-go/extensions/ai-prompt-decorator/main.go
@@ -18,9 +18,9 @@ func main() {}
 func init() {
 	wrapper.SetCtx(
 		"ai-prompt-decorator",
-		wrapper.ParseConfigBy(parseConfig),
-		wrapper.ProcessRequestHeadersBy(onHttpRequestHeaders),
-		wrapper.ProcessRequestBodyBy(onHttpRequestBody),
+		wrapper.ParseConfig(parseConfig),
+		wrapper.ProcessRequestHeaders(onHttpRequestHeaders),
+		wrapper.ProcessRequestBody(onHttpRequestBody),
 	)
 }
 
@@ -34,11 +34,11 @@ type AIPromptDecoratorConfig struct {
 	Append  []Message `json:"append"`
 }
 
-func parseConfig(jsonConfig gjson.Result, config *AIPromptDecoratorConfig, log log.Log) error {
+func parseConfig(jsonConfig gjson.Result, config *AIPromptDecoratorConfig) error {
 	return json.Unmarshal([]byte(jsonConfig.Raw), config)
 }
 
-func onHttpRequestHeaders(ctx wrapper.HttpContext, config AIPromptDecoratorConfig, log log.Log) types.Action {
+func onHttpRequestHeaders(ctx wrapper.HttpContext, config AIPromptDecoratorConfig) types.Action {
 	ctx.DisableReroute()
 	proxywasm.RemoveHttpRequestHeader("content-length")
 	return types.ActionContinue
@@ -70,7 +70,7 @@ func decorateGeographicPrompt(entry *Message) (*Message, error) {
 	return entry, nil
 }
 
-func onHttpRequestBody(ctx wrapper.HttpContext, config AIPromptDecoratorConfig, body []byte, log log.Log) types.Action {
+func onHttpRequestBody(ctx wrapper.HttpContext, config AIPromptDecoratorConfig, body []byte) types.Action {
 	messageJson := `{"messages":[]}`
 
 	for _, entry := range config.Prepend {