rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix various errors

Browse Source 
Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-11-12 11:57:42 +01:00
parent fcbfd93f26
commit 944f61da15
4 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/server/api/mcp/mcputils/mcputils.go
View File

@@ -26,18 +26,14 @@ import (
func LoggingMiddleware(logger *log.Logger) func(mcp.MethodHandler) mcp.MethodHandler {
return func(next mcp.MethodHandler) mcp.MethodHandler {
return func(ctx context.Context, method string, req mcp.Request) (mcp.Result, error) {
sessionID := req.GetSession().ID()
logger.InfoCtx(ctx, fmt.Sprintf("mcp %q method started", method),
log.String("method", method),
log.String("session_id", sessionID),
log.Bool("has_params", req.GetParams() != nil),
)
if ctr, ok := req.(*mcp.CallToolRequest); ok {
logger.InfoCtx(ctx, fmt.Sprintf("calling %q tool", ctr.Params.Name),
log.String("tool_name", ctr.Params.Name),
log.String("session_id", sessionID),
)
}
@@ -48,7 +44,6 @@ func LoggingMiddleware(logger *log.Logger) func(mcp.MethodHandler) mcp.MethodHan
if err != nil {
logger.ErrorCtx(ctx, fmt.Sprintf("mcp %q method failed", method),
log.String("method", method),
log.String("session_id", sessionID),
log.Int64("duration_ms", duration.Milliseconds()),
log.Error(err),
)
@@ -56,14 +51,12 @@ func LoggingMiddleware(logger *log.Logger) func(mcp.MethodHandler) mcp.MethodHan
logger.InfoCtx(ctx, fmt.Sprintf("mcp %q method completed", method),
log.String("method", method),
log.String("session_id", sessionID),
log.Int64("duration_ms", duration.Milliseconds()),
log.Bool("has_result", result != nil),
)
if ctr, ok := result.(*mcp.CallToolResult); ok {
logger.InfoCtx(ctx, "tool call result",
log.String("session_id", sessionID),
log.Bool("is_error", ctr.IsError),
)
}