Spaces:
Sleeping
Sleeping
| import assert from "node:assert/strict"; | |
| import test from "node:test"; | |
| import { | |
| assertCredentialsAtCallTime, | |
| missingApiKeyErrorPayload, | |
| resolveCredentials, | |
| } from "./resolveCredentials"; | |
| import { MissingApiKeyError } from "./errors"; | |
| import { resetInferenceManifestCache } from "./config"; | |
| test.beforeEach(() => { | |
| resetInferenceManifestCache(); | |
| delete process.env.INFERENCE_MANIFEST_PATH; | |
| delete process.env.OPENAI_API_KEY; | |
| delete process.env.GEMINI_API_KEY; | |
| }); | |
| test("user openai_api_key is used when env unset", () => { | |
| const key = resolveCredentials("proxy-openai", { | |
| openai: "user-openai-key", | |
| }); | |
| assert.equal(key, "user-openai-key"); | |
| }); | |
| test("env key is used when user key is null", () => { | |
| process.env.GEMINI_API_KEY = "env-gemini-key"; | |
| const key = resolveCredentials("google-genai", { | |
| gemini: null, | |
| }); | |
| assert.equal(key, "env-gemini-key"); | |
| }); | |
| test("missing both user and env credentials returns null at call time", () => { | |
| const key = resolveCredentials("proxy-openai", { openai: null }); | |
| assert.equal(key, null); | |
| }); | |
| test("assertCredentialsAtCallTime throws missing_api_key error", () => { | |
| assert.throws( | |
| () => assertCredentialsAtCallTime("google-genai", { gemini: null }), | |
| MissingApiKeyError, | |
| ); | |
| try { | |
| assertCredentialsAtCallTime("google-genai", { gemini: null }); | |
| } catch (error) { | |
| assert.equal((error as MissingApiKeyError).code, "missing_api_key"); | |
| assert.deepEqual( | |
| missingApiKeyErrorPayload("google-genai"), | |
| { error: "missing_api_key", provider: "google-genai" }, | |
| ); | |
| } | |
| }); | |