Function AIBridgeSessionThreads

coderd/database/db2sdk/db2sdk.go:1220–1318

AIBridgeSessionThreads converts session metadata and thread interceptions into the threads response. It groups interceptions into threads, builds agentic actions from tool usages and model thoughts, and aggregates token usage with metadata.

(
	session database.ListAIBridgeSessionsRow,
	interceptions []database.ListAIBridgeSessionThreadsRow,
	tokenUsages []database.AIBridgeTokenUsage,
	toolUsages []database.AIBridgeToolUsage,
	userPrompts []database.AIBridgeUserPrompt,
	modelThoughts []database.AIBridgeModelThought,
)

// agentic actions from tool usages and model thoughts, and aggregates
// token usage with metadata.
func AIBridgeSessionThreads(
	session database.ListAIBridgeSessionsRow,
	interceptions []database.ListAIBridgeSessionThreadsRow,
	tokenUsages []database.AIBridgeTokenUsage,
	toolUsages []database.AIBridgeToolUsage,
	userPrompts []database.AIBridgeUserPrompt,
	modelThoughts []database.AIBridgeModelThought,
) codersdk.AIBridgeSessionThreadsResponse {
	// Index subresources by interception ID.
	tokensByInterception := make(map[uuid.UUID][]database.AIBridgeTokenUsage, len(interceptions))
	for _, tu := range tokenUsages {
		tokensByInterception[tu.InterceptionID] = append(tokensByInterception[tu.InterceptionID], tu)
	}
	toolsByInterception := make(map[uuid.UUID][]database.AIBridgeToolUsage, len(interceptions))
	for _, tu := range toolUsages {
		toolsByInterception[tu.InterceptionID] = append(toolsByInterception[tu.InterceptionID], tu)
	}
	promptsByInterception := make(map[uuid.UUID][]database.AIBridgeUserPrompt, len(interceptions))
	for _, up := range userPrompts {
		promptsByInterception[up.InterceptionID] = append(promptsByInterception[up.InterceptionID], up)
	}
	thoughtsByInterception := make(map[uuid.UUID][]database.AIBridgeModelThought, len(interceptions))
	for _, mt := range modelThoughts {
		thoughtsByInterception[mt.InterceptionID] = append(thoughtsByInterception[mt.InterceptionID], mt)
	}

	// Group interceptions by thread_id, preserving the order returned by the
	// SQL query.
	interceptionsByThread := make(map[uuid.UUID][]database.AIBridgeInterception, len(interceptions))
	var threadIDs []uuid.UUID
	for _, row := range interceptions {
		if _, ok := interceptionsByThread[row.ThreadID]; !ok {
			threadIDs = append(threadIDs, row.ThreadID)
		}
		interceptionsByThread[row.ThreadID] = append(interceptionsByThread[row.ThreadID], row.AIBridgeInterception)
	}

	// Build threads and track page time bounds.
	threads := make([]codersdk.AIBridgeThread, 0, len(threadIDs))
	var pageStartedAt, pageEndedAt *time.Time
	for _, threadID := range threadIDs {
		intcs := interceptionsByThread[threadID]
		thread := buildAIBridgeThread(threadID, intcs, tokensByInterception, toolsByInterception, promptsByInterception, thoughtsByInterception)
		for _, intc := range intcs {
			if pageStartedAt == nil || intc.StartedAt.Before(*pageStartedAt) {
				t := intc.StartedAt
				pageStartedAt = &t
			}
			if intc.EndedAt.Valid {
				if pageEndedAt == nil || intc.EndedAt.Time.After(*pageEndedAt) {
					t := intc.EndedAt.Time
					pageEndedAt = &t
				}
			}
		}
		threads = append(threads, thread)
	}