From b2e31e49ccc86993dd511a18cdd6ea0e6d3d912d Mon Sep 17 00:00:00 2001
From: OpenWaygate <27287694+OpenWaygate@users.noreply.github.com>
Date: Sat, 21 Feb 2026 22:33:04 +0800
Subject: [PATCH] :loud_sound: Improve server side logs

---
 cmd/agent/agent.go | 6 +++---
 cmd/handler.go     | 2 ++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/cmd/agent/agent.go b/cmd/agent/agent.go
index 1eeef24..757ee0b 100644
--- a/cmd/agent/agent.go
+++ b/cmd/agent/agent.go
@@ -12,7 +12,7 @@ import (
 	"os"
 	"strings"
 
-	rootCmd "github.com/eat-pray-ai/yutu/cmd"
+	"github.com/eat-pray-ai/yutu/cmd"
 	"github.com/modelcontextprotocol/go-sdk/mcp"
 	"github.com/spf13/cobra"
 	"google.golang.org/adk/agent"
@@ -57,7 +57,7 @@ var agentCmd = &cobra.Command{
 }
 
 func init() {
-	rootCmd.RootCmd.AddCommand(agentCmd)
+	cmd.RootCmd.AddCommand(agentCmd)
 	agentCmd.Flags().StringVarP(
 		&launcherArgs, "args", "a", "",
 		"launcher arguments as a single string, e.g. 'console -streaming_mode sse'",
@@ -77,7 +77,7 @@ func launch(ctx context.Context, writer io.Writer, args []string) {
 	}
 
 	clientTransport, serverTransport := mcp.NewInMemoryTransports()
-	_, err = rootCmd.Server.Connect(ctx, serverTransport, nil)
+	_, err = cmd.Server.Connect(ctx, serverTransport, nil)
 	if err != nil {
 		slog.ErrorContext(ctx, "failed to connect to MCP server", "error", err)
 		os.Exit(1)
diff --git a/cmd/handler.go b/cmd/handler.go
index 5b77439..303ba43 100644
--- a/cmd/handler.go
+++ b/cmd/handler.go
@@ -33,6 +33,7 @@ func GenResourceHandler(
 		err := op(req, &writer)
 		if err != nil {
 			logger.ErrorContext(ctx, err.Error(), "uri", req.Params.URI)
+			slog.ErrorContext(ctx, err.Error(), "uri", req.Params.URI)
 			return nil, err
 		}
 
@@ -63,6 +64,7 @@ func GenToolHandler[T any](
 		err := op(input, &writer)
 		if err != nil {
 			logger.ErrorContext(ctx, err.Error(), "input", input)
+			slog.ErrorContext(ctx, err.Error(), "tool", toolName, "input", input)
 			return nil, nil, err
 		}
 		return &mcp.CallToolResult{Content: []mcp.Content{&mcp.TextContent{Text: writer.String()}}}, nil, nil