From 7534802752bc4e83c61d7a05f98c44a953d89a3b Mon Sep 17 00:00:00 2001
From: Mario Zechner <badlogicgames@gmail.com>
Date: Tue, 13 Jan 2026 18:12:09 +0100
Subject: [PATCH] refactor: add ModelRegistry.getApiKeyForProvider() for
 cleaner abstraction

---
 packages/coding-agent/src/core/model-registry.ts | 7 +++++++
 packages/coding-agent/src/core/sdk.ts            | 2 +-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/packages/coding-agent/src/core/model-registry.ts b/packages/coding-agent/src/core/model-registry.ts
index 43c55f8e..9e2fc094 100644
--- a/packages/coding-agent/src/core/model-registry.ts
+++ b/packages/coding-agent/src/core/model-registry.ts
@@ -375,6 +375,13 @@ export class ModelRegistry {
 		return this.authStorage.getApiKey(model.provider);
 	}
 
+	/**
+	 * Get API key for a provider.
+	 */
+	async getApiKeyForProvider(provider: string): Promise<string | undefined> {
+		return this.authStorage.getApiKey(provider);
+	}
+
 	/**
 	 * Check if a model is using OAuth credentials (subscription).
 	 */
diff --git a/packages/coding-agent/src/core/sdk.ts b/packages/coding-agent/src/core/sdk.ts
index 87f550f4..dbe270bd 100644
--- a/packages/coding-agent/src/core/sdk.ts
+++ b/packages/coding-agent/src/core/sdk.ts
@@ -635,7 +635,7 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 			if (!resolvedProvider) {
 				throw new Error("No model selected");
 			}
-			const key = await modelRegistry.authStorage.getApiKey(resolvedProvider);
+			const key = await modelRegistry.getApiKeyForProvider(resolvedProvider);
 			if (!key) {
 				throw new Error(`No API key found for provider "${resolvedProvider}"`);
 			}