Enha: agentclient log

1 files changed, 4 insertions, 6 deletions

diff --git a/agent/webagent.go b/agent/webagent.go
index 0087e8e..ff6cd86 100644
--- a/agent/webagent.go
+++ b/agent/webagent.go
@@ -2,19 +2,17 @@ package agent
 
 import (
 	"fmt"
-	"log/slog"
 )
 
 // WebAgentB is a simple agent that applies formatting functions
 type WebAgentB struct {
 	*AgentClient
 	sysprompt string
-	log       slog.Logger
 }
 
 // NewWebAgentB creates a WebAgentB that uses the given formatting function
-func NewWebAgentB(sysprompt string) *WebAgentB {
-	return &WebAgentB{sysprompt: sysprompt}
+func NewWebAgentB(client *AgentClient, sysprompt string) *WebAgentB {
+	return &WebAgentB{AgentClient: client, sysprompt: sysprompt}
 }
 
 // Process applies the formatting function to raw output
@@ -22,12 +20,12 @@ func (a *WebAgentB) Process(args map[string]string, rawOutput []byte) []byte {
 	msg, err := a.FormMsg(a.sysprompt,
 		fmt.Sprintf("request:\n%+v\ntool response:\n%v", args, string(rawOutput)))
 	if err != nil {
-		a.log.Error("failed to process the request", "error", err)
+		a.Log().Error("failed to process the request", "error", err)
 		return []byte("failed to process the request; err: " + err.Error())
 	}
 	resp, err := a.LLMRequest(msg)
 	if err != nil {
-		a.log.Error("failed to process the request", "error", err)
+		a.Log().Error("failed to process the request", "error", err)
 		return []byte("failed to process the request; err: " + err.Error())
 	}
 	return resp