address review comments

Author: findleyr

mcp/streamable_bench_test.go

@@ -15,6 +15,8 @@ import (
 )
 
 func BenchmarkStreamableServing(b *testing.B) {
+	// This benchmark measures how fast we can handle a single tool on a
+	// streamable server, including tool validation and stream management.
 	customSchemas := map[any]*jsonschema.Schema{
 		Probability(0):  {Type: "number", Minimum: jsonschema.Ptr(0.0), Maximum: jsonschema.Ptr(1.0)},
 		WeatherType(""): {Type: "string", Enum: []any{Sunny, PartlyCloudy, Cloudy, Rainy, Snowy}},

mcp/tool_example_test.go

@@ -189,6 +189,8 @@ func ExampleAddTool_complexSchema() {
 		if err != nil {
 			log.Fatal(err)
 		}
+		// Formatting the entire schemas would be too much output.
+		// Just check that our customizations were effective.
 		fmt.Println("max days:", *t.InputSchema.Properties["days"].Maximum)
 		fmt.Println("max confidence:", *t.OutputSchema.Properties["confidence"].Maximum)
 		fmt.Println("weather types:", t.OutputSchema.Properties["dailyForecast"].Items.Properties["type"].Enum)