From 398496c45b413bdc3689ff99220cb3d916b65c9c Mon Sep 17 00:00:00 2001
From: Ayaan Zaidi <hi@obviy.us>
Date: Sat, 25 Apr 2026 09:02:01 +0530
Subject: [PATCH] test(config): cover agent context token validation

---
 src/config/types.agents.ts                   |  1 -
 src/config/zod-schema.agent-defaults.test.ts | 25 ++++++++++++++++----
 2 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/src/config/types.agents.ts b/src/config/types.agents.ts
index 976eafa422f..aee07358b89 100644
--- a/src/config/types.agents.ts
+++ b/src/config/types.agents.ts
@@ -95,7 +95,6 @@ export type AgentConfig = {
   skillsLimits?: Pick<SkillsLimitsConfig, "maxSkillsPromptChars">;
   /** Optional per-agent overrides for selected context/token-heavy limits. */
   contextLimits?: AgentContextLimitsConfig;
-  /** Optional per-agent override for the model's total context window (in tokens). */
   contextTokens?: number;
   /** Optional per-agent heartbeat overrides. */
   heartbeat?: AgentDefaultsConfig["heartbeat"];
diff --git a/src/config/zod-schema.agent-defaults.test.ts b/src/config/zod-schema.agent-defaults.test.ts
index f0d1ac15690..069c495cb4b 100644
--- a/src/config/zod-schema.agent-defaults.test.ts
+++ b/src/config/zod-schema.agent-defaults.test.ts
@@ -1,4 +1,5 @@
 import { describe, expect, it } from "vitest";
+import { validateConfigObject } from "./validation.js";
 import { AgentDefaultsSchema } from "./zod-schema.agent-defaults.js";
 import { AgentEntrySchema } from "./zod-schema.agent-runtime.js";
 
@@ -117,12 +118,26 @@ describe("agent defaults schema", () => {
     expect(() => AgentEntrySchema.parse({ id: "ops", heartbeat: { timeoutSeconds: 0 } })).toThrow();
   });
 
-  it("accepts per-agent contextTokens override", () => {
-    const agent = AgentEntrySchema.parse({
-      id: "ops",
-      contextTokens: 1_048_576,
+  it("preserves per-agent contextTokens through config validation", () => {
+    const result = validateConfigObject({
+      agents: {
+        list: [
+          {
+            id: "ops",
+            contextTokens: 1_048_576,
+          },
+        ],
+      },
+    });
+
+    expect(result).toMatchObject({
+      ok: true,
+      config: {
+        agents: {
+          list: [{ contextTokens: 1_048_576 }],
+        },
+      },
     });
-    expect(agent.contextTokens).toBe(1_048_576);
   });
 
   it("rejects non-positive contextTokens on agent entries and defaults", () => {