fixes

Author: ezynda3

www/docs/pages/servers/advanced.mdx

@@ -13,6 +13,7 @@ package main
 
 import (
     "context"
+    "encoding/json"
     "fmt"
 
     "github.com/mark3labs/mcp-go/mcp"
@@ -120,13 +121,15 @@ func handleCreateUser(ctx context.Context, req mcp.CallToolRequest, input UserCr
         return nil, fmt.Errorf("failed to create user: %w", err)
     }
 
-    return &UserCreateOutput{
+    output := &UserCreateOutput{
         ID:        user.ID,
         Name:      user.Name,
         Email:     user.Email,
         CreatedAt: user.CreatedAt,
         Status:    "created",
-    }, nil
+    }
+
+    return mcp.NewToolResultJSON(output), nil
 }
 ```
 
@@ -302,6 +305,7 @@ func NewLoggingMiddleware(logger *log.Logger) *LoggingMiddleware {
 func (m *LoggingMiddleware) ToolMiddleware(next server.ToolHandlerFunc) server.ToolHandlerFunc {
     return func(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
         start := time.Now()
+        sessionID := server.GetSessionID(ctx)
 
         m.logger.Printf("Tool call started: tool=%s", req.Params.Name)
 

www/docs/pages/servers/prompts.mdx

@@ -89,22 +89,7 @@ func handleCodeReview(ctx context.Context, req mcp.GetPromptRequest) (*mcp.GetPr
         Messages: []mcp.PromptMessage{
             {
                 Role: "user",
-                Content: mcp.NewTextContent(fmt.Sprintf(`Please review the following %s code:
-
-%s
-
-Instructions: %s
-
-Please provide:
-1. Overall assessment
-2. Specific issues found
-3. Suggested improvements
-4. Best practice recommendations
-
-Code:
-```%s
-%s
-````, language, instructions, language, code)),
+                Content: mcp.NewTextContent(fmt.Sprintf("Please review the following %s code:\n\n%s\n\nInstructions: %s\n\nPlease provide:\n1. Overall assessment\n2. Specific issues found\n3. Suggested improvements\n4. Best practice recommendations\n\nCode:\n```%s\n%s\n```", language, instructions, language, code)),
             },
         },
     }, nil

www/docs/pages/servers/resources.mdx

@@ -97,11 +97,13 @@ func handleAppConfig(ctx context.Context, req mcp.ReadResourceRequest) (*mcp.Rea
         return nil, err
     }
 
-    return []mcp.ResourceContents{
-        mcp.TextResourceContents{
-            URI:      req.Params.URI,
-            MIMEType: "application/json",
-            Text:     string(configJSON),
+    return &mcp.ReadResourceResult{
+        Contents: []mcp.ResourceContent{
+            mcp.TextResourceContent{
+                URI:      req.Params.URI,
+                MIMEType: "application/json",
+                Text:     string(configJSON),
+            },
         },
     }, nil
 }
@@ -166,6 +168,16 @@ func extractUserID(uri string) string {
 Expose database records dynamically:
 
 ```go
+import (
+    "context"
+    "database/sql"
+    "encoding/json"
+    "fmt"
+
+    "github.com/mark3labs/mcp-go/mcp"
+    "github.com/mark3labs/mcp-go/server"
+)
+
 // Database table resource
 s.AddResource(
     mcp.NewResource(
@@ -205,14 +217,17 @@ func handleDatabaseRecord(ctx context.Context, req mcp.ReadResourceRequest) (*mc
         return nil, err
     }
 
-    return []mcp.ResourceContents{
-        mcp.TextResourceContents{
-            URI:      req.Params.URI,
-            MIMEType: "application/json",
-            Text:     string(jsonData),
+    return &mcp.ReadResourceResult{
+        Contents: []mcp.ResourceContent{
+            mcp.TextResourceContent{
+                URI:      req.Params.URI,
+                MIMEType: "application/json",
+                Text:     string(jsonData),
+            },
         },
     }, nil
 }
+}
 ```
 
 ### API Resources
@@ -434,14 +449,17 @@ func handleResourceWithTimeout(ctx context.Context, req mcp.ReadResourceRequest)
         return nil, err
     }
 
-    return []mcp.ResourceContents{
-        mcp.TextResourceContents{
-            URI:      req.Params.URI,
-            MIMEType: "application/json",
-            Text:     string(jsonData),
+    return &mcp.ReadResourceResult{
+        Contents: []mcp.ResourceContent{
+            mcp.TextResourceContent{
+                URI:      req.Params.URI,
+                MIMEType: "application/json",
+                Text:     string(jsonData),
+            },
         },
     }, nil
 }
+}
 ```
 
 ## Resource Listing

www/docs/pages/servers/tools.mdx

@@ -257,20 +257,21 @@ func handleCreateFile(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallTo
 
 ```go
 func handleDatabaseQuery(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
-    query, err := req.RequireString("query")
-    if err != nil {
-        return mcp.NewToolResultError(err.Error()), nil
-    }
-    
-    // Handle optional params array
-    var params []interface{}
-    if err := req.BindArguments(&struct {
+    // Define struct to bind both Query and Params
+    var args struct {
         Query  string        `json:"query"`
         Params []interface{} `json:"params"`
-    }{Params: params}); err == nil {
-        // params extracted successfully
     }
 
+    // Bind arguments to the struct
+    if err := req.BindArguments(&args); err != nil {
+        return mcp.NewToolResultError(err.Error()), nil
+    }
+    
+    // Extract values from the bound struct
+    query := args.Query
+    params := args.Params
+    
     // Validate query for security (basic example)
     if !isSelectQuery(query) {
         return mcp.NewToolResultError("only SELECT queries are allowed"), nil
@@ -583,15 +584,6 @@ tool := mcp.NewTool("search_database",
         mcp.Items(mcp.StringSchema()),
     ),
 )
-        },
-        {
-            Description: "Find all products in electronics category",
-            Arguments: map[string]interface{}{
-                "query": "*",
-                "categories": []string{"electronics"},
-            },
-    ),
-)
 
 s.AddTool(tool, handleSearchDatabase)
 ```

www/docs/pages/transports/http.mdx

@@ -281,10 +281,6 @@ func main() {
 }
 
 // Helper functions for the advanced example
-func setupHTTPMiddleware(s *server.MCPServer) {
-    // Placeholder implementation - would add middleware
-}
-
 func addCRUDTools(s *server.MCPServer) {
     // Placeholder implementation - would add CRUD tools
 }

www/docs/pages/transports/stdio.mdx

@@ -188,6 +188,19 @@ func detectMIMEType(path string) string {
 ### Advanced STDIO Server
 
 ```go
+import (
+    "context"
+    "fmt"
+    "log"
+    "os"
+    "os/signal"
+    "syscall"
+    "time"
+
+    "github.com/mark3labs/mcp-go/mcp"
+    "github.com/mark3labs/mcp-go/server"
+)
+
 func main() {
     s := server.NewMCPServer("Advanced CLI Tool", "1.0.0",
         server.WithAllCapabilities(),
@@ -201,18 +214,6 @@ func main() {
             },
         }),
     )
-import (
-    "context"
-    "fmt"
-    "log"
-    "os"
-    "os/signal"
-    "syscall"
-    "time"
-
-    "github.com/mark3labs/mcp-go/mcp"
-    "github.com/mark3labs/mcp-go/server"
-)
 
     // Add comprehensive tools
     addSystemTools(s)