Format

Author: dgellow

internal/client/client.go

@@ -280,7 +280,7 @@ func (c *Client) addToolsToServer(
 	})
 
 	totalTools := 0
-	
+
 	var sessionWithTools server.SessionWithTools
 	var sessionTools map[string]server.ServerTool
 	if session != nil {
@@ -295,7 +295,7 @@ func (c *Client) addToolsToServer(
 			"sessionID": session.SessionID(),
 		})
 	}
-	
+
 	for {
 		tools, err := c.client.ListTools(ctx, toolsRequest)
 		if err != nil {
@@ -333,7 +333,7 @@ func (c *Client) addToolsToServer(
 				} else {
 					handler = c.client.CallTool
 				}
-				
+
 				if sessionTools != nil {
 					sessionTools[tool.Name] = server.ServerTool{
 						Tool:    tool,
@@ -349,7 +349,7 @@ func (c *Client) addToolsToServer(
 		}
 		toolsRequest.Params.Cursor = tools.NextCursor
 	}
-	
+
 	if len(sessionTools) > 0 {
 		sessionWithTools.SetSessionTools(sessionTools)
 		internal.LogInfoWithFields("client", "Registered session-specific tools", map[string]interface{}{

internal/client/client_session_test.go

@@ -395,4 +395,4 @@ func TestAddToMCPServerWithSession_ErrorHandling(t *testing.T) {
 		assert.Contains(t, err.Error(), "failed to list tools")
 		mockClient.AssertExpectations(t)
 	})
-}
+}

internal/client/session_manager_test.go

@@ -7,14 +7,13 @@ import (
 	"testing"
 	"time"
 
-	"github.com/dgellow/mcp-front/internal/testutil"
 	"github.com/dgellow/mcp-front/internal/config"
+	"github.com/dgellow/mcp-front/internal/testutil"
 	"github.com/mark3labs/mcp-go/mcp"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
-
 func TestStdioSessionManager_CreateAndRetrieve(t *testing.T) {
 	mockCreator := func(name string, config *config.MCPClientConfig) (*Client, error) {
 		mockClient := new(testutil.MockMCPClient)