Impl AddSessionTool

Author: ezynda3

README.md

@@ -626,6 +626,21 @@ userSpecificTool := mcp.NewTool(
     "user_data",
     mcp.WithDescription("Access user-specific data"),
 )
+// You can use AddSessionTool (similar to AddTool)
+err := s.AddSessionTool(
+    advSession.SessionID(),
+    userSpecificTool,
+    func(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+        // This handler is only available to this specific session
+        return mcp.NewToolResultText("User-specific data for " + advSession.SessionID()), nil
+    },
+)
+if err != nil {
+    log.Printf("Failed to add session tool: %v", err)
+}
+
+// Or use AddSessionTools directly with ServerTool
+/*
 err := s.AddSessionTools(
     advSession.SessionID(),
     server.ServerTool{
@@ -639,6 +654,7 @@ err := s.AddSessionTools(
 if err != nil {
     log.Printf("Failed to add session tool: %v", err)
 }
+*/
 
 // Delete session-specific tools when no longer needed
 err = s.DeleteSessionTools(advSession.SessionID(), "user_data")

server/session.go

@@ -199,6 +199,11 @@ func (s *MCPServer) SendNotificationToSpecificClient(
 	}
 }
 
+// AddSessionTool adds a tool for a specific session
+func (s *MCPServer) AddSessionTool(sessionID string, tool mcp.Tool, handler ToolHandlerFunc) error {
+	return s.AddSessionTools(sessionID, ServerTool{Tool: tool, Handler: handler})
+}
+
 // AddSessionTools adds tools for a specific session
 func (s *MCPServer) AddSessionTools(sessionID string, tools ...ServerTool) error {
 	sessionValue, ok := s.sessions.Load(sessionID)

server/session_test.go

@@ -230,6 +230,45 @@ func TestMCPServer_AddSessionTools(t *testing.T) {
 	assert.Contains(t, session.GetSessionTools(), "session-tool")
 }
 
+func TestMCPServer_AddSessionTool(t *testing.T) {
+	server := NewMCPServer("test-server", "1.0.0", WithToolCapabilities(true))
+	ctx := context.Background()
+
+	// Create a session
+	sessionChan := make(chan mcp.JSONRPCNotification, 10)
+	session := &sessionTestClientWithTools{
+		sessionID:           "session-1",
+		notificationChannel: sessionChan,
+		initialized:         true,
+	}
+
+	// Register the session
+	err := server.RegisterSession(ctx, session)
+	require.NoError(t, err)
+
+	// Add session-specific tool using the new helper method
+	err = server.AddSessionTool(
+		session.SessionID(),
+		mcp.NewTool("session-tool-helper"),
+		func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+			return mcp.NewToolResultText("helper result"), nil
+		},
+	)
+	require.NoError(t, err)
+
+	// Check that notification was sent
+	select {
+	case notification := <-sessionChan:
+		assert.Equal(t, "notifications/tools/list_changed", notification.Method)
+	case <-time.After(100 * time.Millisecond):
+		t.Error("Expected notification not received")
+	}
+
+	// Verify tool was added to session
+	assert.Len(t, session.GetSessionTools(), 1)
+	assert.Contains(t, session.GetSessionTools(), "session-tool-helper")
+}
+
 func TestMCPServer_DeleteSessionTools(t *testing.T) {
 	server := NewMCPServer("test-server", "1.0.0", WithToolCapabilities(true))
 	ctx := context.Background()