Somewhat okay refactoring

main.go

@@ -16,6 +16,7 @@ import (
 	"nannyagentv2/internal/logging"
 	"nannyagentv2/internal/metrics"
 	"nannyagentv2/internal/types"
+	"nannyagentv2/internal/websocket"
 )
 
 const Version = "v2.0.0"
@@ -162,7 +163,7 @@ func main() {
 	applicationAgent.model = "tensorzero::function_name::diagnose_and_heal_application"
 
 	// Start WebSocket client for backend communications and investigations
-	wsClient := NewWebSocketClient(applicationAgent, authManager)
+	wsClient := websocket.NewWebSocketClient(applicationAgent, authManager)
 	go func() {
 		if err := wsClient.Start(); err != nil {
 			logging.Error("WebSocket client error: %v", err)