feat: add types, context, and model normalization (#23914)

2026-09-24 15:04:27 +08:00 · 2026-04-13 19:59:47 +02:00
parent a414d37165
commit 8382e96a81
16 changed files with 3697 additions and 185 deletions
@@ -0,0 +1,105 @@
+package chatdebug_test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/v2/coderd/x/chatd/chatdebug"
+)
+
+func TestContextWithRunRoundTrip(t *testing.T) {
+	t.Parallel()
+
+	rc := &chatdebug.RunContext{
+		RunID:               uuid.New(),
+		ChatID:              uuid.New(),
+		RootChatID:          uuid.New(),
+		ParentChatID:        uuid.New(),
+		ModelConfigID:       uuid.New(),
+		TriggerMessageID:    11,
+		HistoryTipMessageID: 22,
+		Kind:                chatdebug.KindChatTurn,
+		Provider:            "anthropic",
+		Model:               "claude-sonnet",
+	}
+
+	ctx := chatdebug.ContextWithRun(context.Background(), rc)
+	got, ok := chatdebug.RunFromContext(ctx)
+	require.True(t, ok)
+	require.Same(t, rc, got)
+	require.Equal(t, *rc, *got)
+}
+
+func TestRunFromContextAbsent(t *testing.T) {
+	t.Parallel()
+
+	got, ok := chatdebug.RunFromContext(context.Background())
+	require.False(t, ok)
+	require.Nil(t, got)
+}
+
+func TestContextWithStepRoundTrip(t *testing.T) {
+	t.Parallel()
+
+	sc := &chatdebug.StepContext{
+		StepID:              uuid.New(),
+		RunID:               uuid.New(),
+		ChatID:              uuid.New(),
+		StepNumber:          7,
+		Operation:           chatdebug.OperationStream,
+		HistoryTipMessageID: 33,
+	}
+
+	ctx := chatdebug.ContextWithStep(context.Background(), sc)
+	got, ok := chatdebug.StepFromContext(ctx)
+	require.True(t, ok)
+	require.Same(t, sc, got)
+	require.Equal(t, *sc, *got)
+}
+
+func TestStepFromContextAbsent(t *testing.T) {
+	t.Parallel()
+
+	got, ok := chatdebug.StepFromContext(context.Background())
+	require.False(t, ok)
+	require.Nil(t, got)
+}
+
+func TestContextWithRunAndStep(t *testing.T) {
+	t.Parallel()
+
+	rc := &chatdebug.RunContext{RunID: uuid.New(), ChatID: uuid.New()}
+	sc := &chatdebug.StepContext{StepID: uuid.New(), RunID: rc.RunID, ChatID: rc.ChatID}
+
+	ctx := chatdebug.ContextWithStep(
+		chatdebug.ContextWithRun(context.Background(), rc),
+		sc,
+	)
+
+	gotRun, ok := chatdebug.RunFromContext(ctx)
+	require.True(t, ok)
+	require.Same(t, rc, gotRun)
+
+	gotStep, ok := chatdebug.StepFromContext(ctx)
+	require.True(t, ok)
+	require.Same(t, sc, gotStep)
+}
+
+func TestContextWithRunPanicsOnNil(t *testing.T) {
+	t.Parallel()
+
+	require.Panics(t, func() {
+		_ = chatdebug.ContextWithRun(context.Background(), nil)
+	})
+}
+
+func TestContextWithStepPanicsOnNil(t *testing.T) {
+	t.Parallel()
+
+	require.Panics(t, func() {
+		_ = chatdebug.ContextWithStep(context.Background(), nil)
+	})
+}