diff --git a/packages/shared/src/schemas/agent.ts b/packages/shared/src/schemas/agent.ts
index bcc31f0..6df0ab6 100644
--- a/packages/shared/src/schemas/agent.ts
+++ b/packages/shared/src/schemas/agent.ts
@@ -36,9 +36,9 @@ export const agentTaskSchema = z.object({
   id: z.string(),
   agent_id: z.string(),
   task_type: z.string(),
-  input: z.record(z.unknown()),
+  input: z.record(z.string(), z.unknown()),
   status: z.enum(['pending', 'running', 'completed', 'failed']).default('pending'),
-  output: z.record(z.unknown()).optional(),
+  output: z.record(z.string(), z.unknown()).optional(),
   error_message: z.string().optional(),
   started_at: z.string().datetime().optional(),
   completed_at: z.string().datetime().optional(),
@@ -70,8 +70,8 @@ export const agentSchema = z.object({
   last_active_at: z.string().datetime().optional(),
   domain: z.string(),
   tags: z.array(z.string()).default([]),
-  config: z.record(z.unknown()).optional(),
-  custom_fields: z.record(z.unknown()).optional(),
+  config: z.record(z.string(), z.unknown()).optional(),
+  custom_fields: z.record(z.string(), z.unknown()).optional(),
   created: z.string().datetime(),
   updated: z.string().datetime(),
 });