ts: show coordinator program id in front

Author: dsocolobsky

website/backend/src/coordinator.ts

@@ -1,6 +1,12 @@
-export type UniqueRunKey = `${string}-${string}-${string}` & { __uniqueRunKey: true }
+export type UniqueRunKey = `${string}-${string}-${string}` & {
+	__uniqueRunKey: true
+}
 
-export function runKey(programId: string, runId: string, index: number): UniqueRunKey {
+export function runKey(
+	programId: string,
+	runId: string,
+	index: number
+): UniqueRunKey {
 	return `${programId}-${runId}-${index}` as UniqueRunKey
 }
 
@@ -13,13 +19,13 @@ export function getRunFromKey(
 		const [runId, index] = splitAtLastInstance(runKey, '-')
 		return ['', runId, Number.parseInt(index, 10)]
 	}
-	
+
 	// New format: programId-runId-index
 	// Handle case where runId might contain dashes by joining middle parts
 	const programId = parts[0]
 	const index = parts[parts.length - 1]
 	const runId = parts.slice(1, -1).join('-')
-	
+
 	return [programId, runId, Number.parseInt(index, 10)]
 }
 

website/backend/src/dataStore.ts

@@ -23,7 +23,7 @@ export interface ChainDataStore {
 
 export interface CoordinatorDataStore extends ChainDataStore {
 	eventEmitter: EventEmitter<{ update: [UniqueRunKey] }>
-	
+
 	readonly programId: PublicKey
 
 	createRun(

website/backend/src/dataStores/flatFileCoordinator.ts

@@ -263,7 +263,9 @@ export class FlatFileCoordinatorDataStore implements CoordinatorDataStore {
 			})
 		}
 
-		this.#runsMutatedSinceLastSync.add(runKey(this.#programId.toString(), lastRun.runId, index))
+		this.#runsMutatedSinceLastSync.add(
+			runKey(this.#programId.toString(), lastRun.runId, index)
+		)
 	}
 
 	setRunPaused(pubkey: string, paused: boolean, timestamp: ChainTimestamp) {
@@ -278,7 +280,9 @@ export class FlatFileCoordinatorDataStore implements CoordinatorDataStore {
 		lastRun.lastUpdated = timestamp
 		lastRun.pauseTimestamps.push([newPauseState, timestamp])
 
-		this.#runsMutatedSinceLastSync.add(runKey(this.#programId.toString(), lastRun.runId, index))
+		this.#runsMutatedSinceLastSync.add(
+			runKey(this.#programId.toString(), lastRun.runId, index)
+		)
 	}
 
 	witnessRun(
@@ -432,7 +436,9 @@ export class FlatFileCoordinatorDataStore implements CoordinatorDataStore {
 	}
 
 	getRunDataById(runId: string, index: number): RunData | null {
-		const cachedRun = this.#runCache.get(runKey(this.#programId.toString(), runId, index))
+		const cachedRun = this.#runCache.get(
+			runKey(this.#programId.toString(), runId, index)
+		)
 		if (cachedRun) {
 			return cachedRun
 		}
@@ -577,14 +583,18 @@ export class FlatFileCoordinatorDataStore implements CoordinatorDataStore {
 
 		const runData = {
 			info,
+			programId: this.#programId.toString(),
 			state,
 			recentTxs: run.recentTxs,
 			metrics: {
 				summary,
 				history,
 			},
 		}
-		this.#runCache.set(runKey(this.#programId.toString(), runId, index), runData)
+		this.#runCache.set(
+			runKey(this.#programId.toString(), runId, index),
+			runData
+		)
 		return runData
 	}
 }
@@ -671,6 +681,7 @@ function makeRunSummary(
 		size: run.lastState.metadata.num_parameters,
 		trainingStep,
 		type: 'text', // TODO add type / tags? :)
+		programId: this.#programId.toString(),
 	}
 	return summary
 }

website/backend/src/index.ts

@@ -262,7 +262,11 @@ async function main() {
 				return
 			}
 
-			const key = runKey(coordinator.dataStore.programId.toString(), matchingRun.info.id, matchingRun.info.index)
+			const key = runKey(
+				coordinator.dataStore.programId.toString(),
+				matchingRun.info.id,
+				matchingRun.info.index
+			)
 			let listeners = liveRunListeners.get(key)
 			if (!listeners) {
 				listeners = new Set()

website/frontend/src/components/RunSummary.tsx

@@ -64,6 +64,7 @@ export function RunSummaryCard({
 		type,
 		index,
 		isOnlyRunAtThisIndex,
+		programId,
 	},
 }: {
 	info: RunSummary
@@ -80,6 +81,7 @@ export function RunSummaryCard({
 				<RunTitleRow>
 					<RunTitle className={text['display/2xl']}>
 						{name || id} {isOnlyRunAtThisIndex ? '' : `(v${index + 1})`}
+						{programId && ` - ${programId.slice(0, 12)}`}
 					</RunTitle>
 					<StatusChip status={status.type} style="minimal" />
 				</RunTitleRow>

website/frontend/src/routes/runs/$run.$index.tsx

@@ -143,6 +143,7 @@ function RouteComponent() {
 							<span className={text['display/4xl']}>
 								{info.name || info.id}{' '}
 								{info.isOnlyRunAtThisIndex ? '' : `(v${info.index + 1})`}
+								{run.programId && ` - ${run.programId.slice(0, 12)}`}
 							</span>
 							<TitleRightInfo>
 								<StatusChip status={info.status.type} style="minimal" />

website/shared/index.ts

@@ -74,6 +74,7 @@ export interface RunSummary {
 	size: bigint
 	arch: LLMArchitecture
 	type: ModelType
+	programId: string
 }
 
 export type Metrics = {
@@ -109,6 +110,7 @@ export interface TxSummary {
 
 export interface RunData {
 	info: RunSummary
+	programId: string
 	state?: {
 		phase: RunState
 		phaseStartTime: Date