import { WithTimestamps } from '@n8n/db';
import { Column, Entity, Index, PrimaryColumn } from '@n8n/typeorm';

@Entity({ name: 'instance_ai_run_snapshots' })
@Index(['threadId', 'messageGroupId'])
@Index(['threadId', 'createdAt'])
export class InstanceAiRunSnapshot extends WithTimestamps {
	@PrimaryColumn('uuid')
	threadId: string;

	@PrimaryColumn({ type: 'varchar', length: 36 })
	runId: string;

	@Column({ type: 'varchar', length: 36, nullable: true })
	messageGroupId: string | null;

	@Column({ type: 'simple-json', nullable: true })
	runIds: string[] | null;

	@Column({ type: 'text' })
	tree: string;

	@Column({ type: 'varchar', length: 64, nullable: true })
	traceId: string | null;

	@Column({ type: 'varchar', length: 64, nullable: true })
	spanId: string | null;

	@Column({ type: 'varchar', length: 36, nullable: true })
	langsmithRunId: string | null;

	@Column({ type: 'varchar', length: 36, nullable: true })
	langsmithTraceId: string | null;
}