fix: Cleared the staticcheck with XTestTimeoutKey

server/handlers/completion.go

@@ -343,7 +343,7 @@ func (h *CompletionHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	// Create context with timeout header if present
 	ctx := r.Context()
 	if timeoutHeader := r.Header.Get("X-Test-Timeout"); timeoutHeader != "" {
-		ctx = context.WithValue(ctx, "X-Test-Timeout", timeoutHeader)
+		ctx = context.WithValue(ctx, middleware.XTestTimeoutKey, timeoutHeader)
 		r = r.WithContext(ctx)
 	}
 

server/handlers/integration_test.go

@@ -33,7 +33,7 @@ func TestCompletionHandlerIntegration(t *testing.T) {
 	// Create mock LLM
 	mockLLM := mocks.NewMockLLM(func(ctx context.Context, prompt *gollm.Prompt) (string, error) {
 		// If context has timeout header, simulate timeout
-		if ctx.Value("X-Test-Timeout") != nil {
+		if ctx.Value(middleware.XTestTimeoutKey) != nil {
 			// Sleep longer than the timeout
 			time.Sleep(5 * time.Second)
 		}

server/mocks/llm.go

@@ -7,6 +7,7 @@ import (
 	"github.com/teilomillet/gollm"
 	"github.com/teilomillet/gollm/llm"
 	"github.com/teilomillet/gollm/utils"
+	"github.com/teilomillet/hapax/server/middleware"
 )
 
 // MockLLM implements a mock LLM for testing purposes.
@@ -24,27 +25,27 @@ import (
 //	})
 type MockLLM struct {
 	GenerateFunc func(context.Context, *gollm.Prompt) (string, error)
-	DebugFunc   func(string, ...interface{})
-	Provider    string // Provider name for testing
-	Model       string // Model name for testing
+	DebugFunc    func(string, ...interface{})
+	Provider     string // Provider name for testing
+	Model        string // Model name for testing
 }
 
 // NewMockLLM creates a new MockLLM with optional generate function.
 // If generateFunc is nil, Generate will return empty string with no error.
 func NewMockLLM(generateFunc func(context.Context, *gollm.Prompt) (string, error)) *MockLLM {
 	return &MockLLM{
 		GenerateFunc: generateFunc,
-		Provider:    "mock",
-		Model:       "mock-model",
+		Provider:     "mock",
+		Model:        "mock-model",
 	}
 }
 
 // NewMockLLMWithConfig creates a new MockLLM with specific provider and model names
 func NewMockLLMWithConfig(provider, model string, generateFunc func(context.Context, *gollm.Prompt) (string, error)) *MockLLM {
 	return &MockLLM{
 		GenerateFunc: generateFunc,
-		Provider:    provider,
-		Model:       model,
+		Provider:     provider,
+		Model:        model,
 	}
 }
 
@@ -53,7 +54,7 @@ func NewMockLLMWithConfig(provider, model string, generateFunc func(context.Cont
 // The opts parameter is ignored in the mock to simplify testing.
 func (m *MockLLM) Generate(ctx context.Context, prompt *gollm.Prompt, opts ...llm.GenerateOption) (string, error) {
 	// Check for timeout header
-	if ctx.Value("X-Test-Timeout") != nil {
+	if ctx.Value(middleware.XTestTimeoutKey) != nil {
 		// Sleep longer than the timeout
 		time.Sleep(10 * time.Second)
 		return "", context.DeadlineExceeded

server/processing/processor.go

@@ -10,6 +10,7 @@ import (
 
 	"github.com/teilomillet/gollm"
 	"github.com/teilomillet/hapax/config"
+	"github.com/teilomillet/hapax/server/middleware"
 )
 
 // Processor handles request processing and response formatting for LLM interactions.
@@ -25,7 +26,7 @@ import (
 // The Processor is designed to be reusable across different request types
 // while maintaining consistent formatting and error handling.
 type Processor struct {
-	llm           gollm.LLM                    // The LLM instance to use for generation
+	llm           gollm.LLM                     // The LLM instance to use for generation
 	templates     map[string]*template.Template // Compiled templates for request formatting
 	config        *config.ProcessingConfig      // Configuration for processing behavior
 	defaultPrompt string                        // Default system prompt for all requests
@@ -122,7 +123,7 @@ func (p *Processor) ProcessRequest(ctx context.Context, req *Request) (*Response
 
 	// Pass timeout header to LLM context if present
 	if timeoutHeader := ctx.Value("X-Test-Timeout"); timeoutHeader != nil {
-		ctx = context.WithValue(ctx, "X-Test-Timeout", timeoutHeader)
+		ctx = context.WithValue(ctx, middleware.XTestTimeoutKey, timeoutHeader)
 	}
 
 	// Send request to LLM