Merge pull request #63 from knabben/refactor/job-history

Refactor/job history

Author: knabben

front/app/ui/dashboard/base/ai-troubleshooting.tsx

@@ -53,7 +53,7 @@ export default function AITroubleshooting({
     if (broken.size > 0) {
       setRequest(Array.from(broken).join(', '));
     }
-  }, [conditions, objectType]);
+  }, [conditions, objectType, objectName, objectNamespace]);
 
   return (
     <Grid justify="flex-start" align="flex-start">

webserver/internal/infra/llm/agent.go

@@ -0,0 +1,30 @@
+package llm
+
+import "time"
+
+type Agent struct {
+	ID           string    `json:"id"`
+	Name         string    `json:"name"`
+	Type         string    `json:"type"`
+	Status       string    `json:"status"`
+	Activity     string    `json:"activity"`
+	LastUpdate   time.Time `json:"last_update"`
+	Capabilities []string  `json:"capabilities"`
+}
+
+func (s *ObservationService) initializeAgents() {
+	agents := []*Agent{
+		{
+			ID:           "cluster-agent",
+			Name:         "Cluster Agent",
+			Type:         "analysis",
+			Status:       "active",
+			Activity:     "Monitoring cluster health and analyzing issues",
+			Capabilities: []string{"cluster-analysis", "resource-monitoring", "pattern-detection"},
+			LastUpdate:   time.Now(),
+		},
+	}
+	for _, agent := range agents {
+		s.agents[agent.ID] = agent
+	}
+}

webserver/internal/infra/llm/client.go

@@ -0,0 +1,58 @@
+package llm
+
+import (
+	"github.com/anthropics/anthropic-sdk-go"
+)
+
+// ConversationManager handles message history
+type ConversationManager struct {
+	stopper  int
+	messages []anthropic.MessageParam
+}
+
+// NewConversationManager creates a new conversation manager
+func NewConversationManager(stopper int) *ConversationManager {
+	return &ConversationManager{
+		stopper:  stopper,
+		messages: make([]anthropic.MessageParam, 0),
+	}
+}
+
+// AddUserMessage adds a user message to the conversation history
+func (cm *ConversationManager) AddUserMessage(content string) {
+	message := anthropic.NewUserMessage(anthropic.NewTextBlock(content))
+	cm.messages = append(cm.messages, message)
+}
+
+// AddAssistantMessage adds an assistant message to the conversation history
+func (cm *ConversationManager) AddAssistantMessage(content string) {
+	message := anthropic.NewAssistantMessage(anthropic.NewTextBlock(content))
+	cm.messages = append(cm.messages, message)
+}
+
+// GetConversationHistory returns the current conversation history
+func (cm *ConversationManager) GetConversationHistory() []anthropic.MessageParam {
+	if len(cm.messages) <= cm.stopper {
+		return cm.messages
+	}
+	return cm.messages[:cm.stopper]
+}
+
+// ClearHistory clears the conversation history
+func (cm *ConversationManager) ClearHistory() {
+	cm.messages = make([]anthropic.MessageParam, 0)
+}
+
+// GetHistoryLength returns the number of messages in history
+func (cm *ConversationManager) GetHistoryLength() int {
+	return len(cm.messages)
+}
+
+// TrimHistory removes older messages to stay within token limits
+// Keeps the last 'stopper' messages
+func (cm *ConversationManager) TrimHistory() {
+	if len(cm.messages) <= cm.stopper {
+		return
+	}
+	cm.messages = cm.messages[len(cm.messages)-cm.stopper:]
+}

webserver/internal/infra/llm/conversation.go

@@ -0,0 +1,110 @@
+package llm
+
+import (
+	"testing"
+
+	"github.com/anthropics/anthropic-sdk-go"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestTrimHistory(t *testing.T) {
+	tests := []struct {
+		name          string
+		initialLength int
+		keepCount     int
+		expectedCount int
+	}{
+		{
+			name:          "NoTrimmingNeeded",
+			initialLength: 3,
+			keepCount:     5,
+			expectedCount: 3,
+		},
+		{
+			name:          "ExactKeepCount",
+			initialLength: 5,
+			keepCount:     5,
+			expectedCount: 5,
+		},
+		{
+			name:          "TrimToKeepCount",
+			initialLength: 10,
+			keepCount:     7,
+			expectedCount: 7,
+		},
+		{
+			name:          "KeepNone",
+			initialLength: 5,
+			keepCount:     0,
+			expectedCount: 0,
+		},
+		{
+			name:          "KeepOne",
+			initialLength: 5,
+			keepCount:     1,
+			expectedCount: 1,
+		},
+		{
+			name:          "EmptyHistory",
+			initialLength: 0,
+			keepCount:     5,
+			expectedCount: 0,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			cm := &ConversationManager{
+				stopper:  tt.keepCount,
+				messages: make([]anthropic.MessageParam, tt.initialLength),
+			}
+			cm.TrimHistory()
+			assert.Equal(t, tt.expectedCount, len(cm.messages))
+		})
+	}
+}
+
+func TestGetConversationHistory(t *testing.T) {
+	tests := []struct {
+		name          string
+		initialLength int
+		stopper       int
+		expectLength  int
+	}{
+		{
+			name:          "EmptyHistory",
+			initialLength: 0,
+			stopper:       5,
+			expectLength:  0,
+		},
+		{
+			name:          "UnderStopper",
+			initialLength: 3,
+			stopper:       5,
+			expectLength:  3,
+		},
+		{
+			name:          "EqualToStopper",
+			initialLength: 5,
+			stopper:       5,
+			expectLength:  5,
+		},
+		{
+			name:          "OverStopper",
+			initialLength: 10,
+			stopper:       7,
+			expectLength:  7,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			cm := &ConversationManager{
+				stopper:  tt.stopper,
+				messages: make([]anthropic.MessageParam, tt.initialLength),
+			}
+			history := cm.GetConversationHistory()
+			assert.Equal(t, tt.expectLength, len(history))
+		})
+	}
+}