s/WithTracing/WithMCPServerTracing

Author: jboolean

contrib/mark3labs/mcp-go/README.md

@@ -19,9 +19,9 @@ func main() {
 
     // Do not use with `server.WithHooks(...)`, as this overwrites the tracing hooks. 
     // To add custom hooks alongside tracing, pass them via TracingConfig.Hooks, e.g.:
-    // mcpgotrace.WithTracing(&mcpgotrace.TracingConfig{Hooks: customHooks})
+    // mcpgotrace.WithMCPServerTracing(&mcpgotrace.TracingConfig{Hooks: customHooks})
     srv := server.NewMCPServer("my-server", "1.0.0",
-		mcpgotrace.WithTracing(nil))
+		mcpgotrace.WithMCPServerTracing(nil))
 }
 ```
 

contrib/mark3labs/mcp-go/example_test.go

@@ -19,7 +19,7 @@ func Example() {
 	defer tracer.Stop()
 
 	srv := server.NewMCPServer("my-server", "1.0.0",
-		mcpgotrace.WithTracing(nil))
+		mcpgotrace.WithMCPServerTracing(nil))
 	_ = srv
 }
 
@@ -33,6 +33,6 @@ func Example_withCustomHooks() {
 	})
 
 	srv := server.NewMCPServer("my-server", "1.0.0",
-		mcpgotrace.WithTracing(&mcpgotrace.TracingConfig{Hooks: customHooks}))
+		mcpgotrace.WithMCPServerTracing(&mcpgotrace.TracingConfig{Hooks: customHooks}))
 	_ = srv
 }

contrib/mark3labs/mcp-go/option.go

@@ -19,7 +19,7 @@ type TracingConfig struct {
 	Hooks *server.Hooks
 }
 
-// WithTracing adds Datadog tracing to an MCP server.
+// WithMCPServerTracing adds Datadog tracing to an MCP server.
 // Pass this option to server.NewMCPServer to enable tracing.
 //
 // Do not use with `server.WithHooks(...)`, as this overwrites the hooks.
@@ -29,16 +29,16 @@ type TracingConfig struct {
 //
 //	// Simple usage with only tracing hooks
 //	srv := server.NewMCPServer("my-server", "1.0.0",
-//	    WithTracing(nil))
+//	    WithMCPServerTracing(nil))
 //
 //	// With custom hooks
 //	customHooks := &server.Hooks{}
 //	customHooks.AddBeforeInitialize(func(ctx context.Context, id any, request *mcp.InitializeRequest) {
 //	    // Your custom logic here
 //	})
 //	srv := server.NewMCPServer("my-server", "1.0.0",
-//	    WithTracing(&TracingConfig{Hooks: customHooks}))
-func WithTracing(options *TracingConfig) server.ServerOption {
+//	    WithMCPServerTracing(&TracingConfig{Hooks: customHooks}))
+func WithMCPServerTracing(options *TracingConfig) server.ServerOption {
 	return func(s *server.MCPServer) {
 		if options == nil {
 			options = new(TracingConfig)

contrib/mark3labs/mcp-go/tracing_test.go

@@ -46,7 +46,7 @@ func TestIntegrationSessionInitialize(t *testing.T) {
 	defer tt.Stop()
 
 	srv := server.NewMCPServer("test-server", "1.0.0",
-		WithTracing(nil))
+		WithMCPServerTracing(nil))
 
 	ctx := context.Background()
 	sessionID := "test-session-init"
@@ -109,7 +109,7 @@ func TestIntegrationToolCallSuccess(t *testing.T) {
 	appendTracingHooks(hooks)
 
 	srv := server.NewMCPServer("test-server", "1.0.0",
-		WithTracing(nil))
+		WithMCPServerTracing(nil))
 
 	calcTool := mcp.NewTool("calculator",
 		mcp.WithDescription("A simple calculator"))
@@ -207,7 +207,7 @@ func TestIntegrationToolCallError(t *testing.T) {
 	defer tt.Stop()
 
 	srv := server.NewMCPServer("test-server", "1.0.0",
-		WithTracing(&TracingConfig{}))
+		WithMCPServerTracing(&TracingConfig{}))
 
 	errorTool := mcp.NewTool("error_tool",
 		mcp.WithDescription("A tool that always errors"))
@@ -254,7 +254,7 @@ func TestIntegrationToolCallError(t *testing.T) {
 	assert.Contains(t, toolSpan.Meta, "input")
 }
 
-func TestWithTracingWithCustomHooks(t *testing.T) {
+func TestWithMCPServerTracingWithCustomHooks(t *testing.T) {
 	tt := testTracer(t)
 	defer tt.Stop()
 
@@ -265,7 +265,7 @@ func TestWithTracingWithCustomHooks(t *testing.T) {
 	})
 
 	srv := server.NewMCPServer("test-server", "1.0.0",
-		WithTracing(&TracingConfig{Hooks: customHooks}))
+		WithMCPServerTracing(&TracingConfig{Hooks: customHooks}))
 
 	ctx := context.Background()
 	initRequest := `{"jsonrpc":"2.0","id":1,"method":"initialize","params":{"protocolVersion":"2024-11-05","capabilities":{},"clientInfo":{"name":"test-client","version":"1.0.0"}}}`