import type { MigrationContext, ReversibleMigration } from '../migration-types';

export class DropAgentExecutionWorkingMemory1784000000002 implements ReversibleMigration {
	async up({ schemaBuilder: { dropColumns } }: MigrationContext) {
		await dropColumns('agent_execution', ['workingMemory'], { recreatesOnSqlite: true });
	}

	async down({ schemaBuilder: { addColumns, column } }: MigrationContext) {
		await addColumns('agent_execution', [column('workingMemory').text], {
			recreatesOnSqlite: true,
		});
	}
}