🔊 Improve server side logs

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)

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