From 5c16dcb154edebbf7a6169f08e68650214c9bad8 Mon Sep 17 00:00:00 2001
From: can1357 <me@can.ac>
Date: Tue, 14 Jul 2026 18:02:48 +0200
Subject: [PATCH] fix(auth-broker): preserve config.yaml discovery

---
 packages/ai/src/auth-storage.ts                    |  2 ++
 packages/ai/test/auth-broker-nested-config.test.ts | 10 ++++++++++
 2 files changed, 12 insertions(+)

diff --git a/packages/ai/src/auth-storage.ts b/packages/ai/src/auth-storage.ts
index ec8e0413e..eec406470 100644
--- a/packages/ai/src/auth-storage.ts
+++ b/packages/ai/src/auth-storage.ts
@@ -1933,6 +1933,8 @@ export class AuthStorage {
 			blockScope,
 		});
 		return candidates[0]?.selection ?? fallback;
+	}
+
 	#clearProviderSessionCredentialCache(provider: string): void {
 		try {
 			this.#store.deleteCachePrefix?.(`${SESSION_STICKY_CACHE_PREFIX}${provider}:`);
diff --git a/packages/ai/test/auth-broker-nested-config.test.ts b/packages/ai/test/auth-broker-nested-config.test.ts
index b2f69e1fa..cc3c6fa7c 100644
--- a/packages/ai/test/auth-broker-nested-config.test.ts
+++ b/packages/ai/test/auth-broker-nested-config.test.ts
@@ -62,4 +62,14 @@ describe("auth-broker config.yml key resolution", () => {
 			expect(config).toEqual({ url: "https://nested.example", token: "nested-token" });
 		});
 	});
+	test("config.yaml nested keys resolve broker url and token", async () => {
+		await Bun.write(
+			path.join(agentDir, "config.yaml"),
+			["auth:", "  broker:", "    url: https://broker.example", "    token: yaml-token", ""].join("\n"),
+		);
+		await withEnv(CLEAR_BROKER_ENV, async () => {
+			const config = await resolveAuthBrokerConfig({ agentDir });
+			expect(config).toEqual({ url: "https://broker.example", token: "yaml-token" });
+		});
+	});
 });