diff --git a/src/db/repositories/agent-repository.ts b/src/db/repositories/agent-repository.ts
index daa0a91..97e026e 100644
--- a/src/db/repositories/agent-repository.ts
+++ b/src/db/repositories/agent-repository.ts
@@ -6,6 +6,7 @@
  */
 
 import type { Agent } from '../schema.js';
+import type { AgentMode } from '../../agent/types.js';
 
 /**
  * Agent status values.
@@ -22,6 +23,7 @@ export interface CreateAgentData {
   taskId?: string | null;
   sessionId?: string | null;
   status?: AgentStatus;
+  mode?: AgentMode; // Defaults to 'execute' if not provided
 }
 
 /**
diff --git a/src/db/repositories/drizzle/agent.ts b/src/db/repositories/drizzle/agent.ts
index d398bc1..7f040f8 100644
--- a/src/db/repositories/drizzle/agent.ts
+++ b/src/db/repositories/drizzle/agent.ts
@@ -34,6 +34,7 @@ export class DrizzleAgentRepository implements AgentRepository {
       sessionId: data.sessionId ?? null,
       worktreeId: data.worktreeId,
       status: data.status ?? 'idle',
+      mode: data.mode ?? 'execute',
       createdAt: now,
       updatedAt: now,
     });