diff --git a/src/key-management/anthropic/provider.ts b/src/key-management/anthropic/provider.ts index 1afa04f..c1a0a11 100644 --- a/src/key-management/anthropic/provider.ts +++ b/src/key-management/anthropic/provider.ts @@ -140,7 +140,7 @@ export class AnthropicKeyProvider implements KeyProvider { } public disable(key: AnthropicKey) { - const keyFromPool = this.keys.find((k) => k.key === key.key); + const keyFromPool = this.keys.find((k) => k.hash === key.hash); if (!keyFromPool || keyFromPool.isDisabled) return; keyFromPool.isDisabled = true; this.log.warn({ key: key.hash }, "Key disabled"); diff --git a/src/key-management/openai/provider.ts b/src/key-management/openai/provider.ts index 1133adf..7a67e54 100644 --- a/src/key-management/openai/provider.ts +++ b/src/key-management/openai/provider.ts @@ -224,7 +224,7 @@ export class OpenAIKeyProvider implements KeyProvider { /** Disables a key, or does nothing if the key isn't in this pool. */ public disable(key: Key) { - const keyFromPool = this.keys.find((k) => k.key === key.key); + const keyFromPool = this.keys.find((k) => k.hash === key.hash); if (!keyFromPool || keyFromPool.isDisabled) return; keyFromPool.isDisabled = true; this.log.warn({ key: key.hash }, "Key disabled");