Function TestChatDebugRunDetail

coderd/database/db2sdk/db2sdk_test.go:486–573

(t *testing.T)

}

func TestChatDebugRunDetail(t *testing.T) {
	t.Parallel()

	startedAt := time.Now().UTC().Round(time.Second)
	finishedAt := startedAt.Add(5 * time.Second)
	rootChatID := uuid.New()
	parentChatID := uuid.New()
	modelConfigID := uuid.New()
	triggerMessageID := int64(7)
	historyTipMessageID := int64(11)

	run := database.ChatDebugRun{
		ID:                  uuid.New(),
		ChatID:              uuid.New(),
		RootChatID:          uuid.NullUUID{UUID: rootChatID, Valid: true},
		ParentChatID:        uuid.NullUUID{UUID: parentChatID, Valid: true},
		ModelConfigID:       uuid.NullUUID{UUID: modelConfigID, Valid: true},
		TriggerMessageID:    sql.NullInt64{Int64: triggerMessageID, Valid: true},
		HistoryTipMessageID: sql.NullInt64{Int64: historyTipMessageID, Valid: true},
		Kind:                "chat_turn",
		Status:              "completed",
		Provider:            sql.NullString{String: "openai", Valid: true},
		Model:               sql.NullString{String: "gpt-4o", Valid: true},
		Summary:             json.RawMessage(`{"step_count":2}`),
		StartedAt:           startedAt,
		UpdatedAt:           finishedAt,
		FinishedAt:          sql.NullTime{Time: finishedAt, Valid: true},
	}
	steps := []database.ChatDebugStep{
		{
			ID:                uuid.New(),
			RunID:             run.ID,
			ChatID:            run.ChatID,
			StepNumber:        1,
			Operation:         "stream",
			Status:            "completed",
			NormalizedRequest: json.RawMessage(`{"messages":[]}`),
			Attempts:          json.RawMessage(`[]`),
			Metadata:          json.RawMessage(`{}`),
			StartedAt:         startedAt,
			UpdatedAt:         finishedAt,
		},
		{
			ID:                uuid.New(),
			RunID:             run.ID,
			ChatID:            run.ChatID,
			StepNumber:        2,
			Operation:         "generate",
			Status:            "completed",
			NormalizedRequest: json.RawMessage(`{"messages":[]}`),
			Attempts:          json.RawMessage(`[]`),
			Metadata:          json.RawMessage(`{}`),
			StartedAt:         startedAt,
			UpdatedAt:         finishedAt,
		},
	}

	sdk := db2sdk.ChatDebugRunDetail(run, steps)