@jambonz/llm 0.2.6 → 0.3.0

package/dist/test-kit/index.cjs

@@ -465,6 +465,9 @@ var OpenAIAdapter = class {
   async listAvailableModels() {
     return listOpenAICompatibleModels(this.ensureClient(), openAIManifest.knownModels);
   }
+  async testCredential() {
+    await this.ensureClient().models.list();
+  }
   async warmup() {
     const client = this.ensureClient();
     await client.models.list().catch(() => void 0);
@@ -891,6 +894,9 @@ var AnthropicAdapter = class {
     }
     return models;
   }
+  async testCredential() {
+    await this.ensureClient().models.list();
+  }
   async warmup() {
     const client = this.ensureClient();
     await client.models.list().catch(() => void 0);
@@ -1317,6 +1323,9 @@ var GoogleAdapter = class {
   async listAvailableModels() {
     return listGeminiModels(this.ensureClient(), googleManifest.knownModels);
   }
+  async testCredential() {
+    await listGeminiModels(this.ensureClient(), googleManifest.knownModels);
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1487,6 +1496,9 @@ var VertexGeminiAdapter = class {
   async listAvailableModels() {
     return listGeminiModels(this.ensureClient(), vertexGeminiManifest.knownModels);
   }
+  async testCredential() {
+    await listGeminiModels(this.ensureClient(), vertexGeminiManifest.knownModels);
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1607,6 +1619,7 @@ var VertexOpenAIAdapter = class {
   vendor = vertexOpenAIManifest.vendor;
   acceptedAuth = ["vertexServiceAccount"];
   client;
+  googleAuth;
   init(auth, client) {
     if (auth.kind !== "vertexServiceAccount") {
       throw new Error(
@@ -1623,12 +1636,12 @@ var VertexOpenAIAdapter = class {
       throw new Error("VertexOpenAIAdapter: location is required");
     }
     const baseURL = `https://${auth.location}-aiplatform.googleapis.com/v1beta1/projects/${auth.projectId}/locations/${auth.location}/endpoints/openapi`;
-    const googleAuth = new import_google_auth_library.GoogleAuth({
+    this.googleAuth = new import_google_auth_library.GoogleAuth({
       credentials: auth.credentials,
       scopes: ["https://www.googleapis.com/auth/cloud-platform"]
     });
     const authedFetch = async (input, init2) => {
-      const googleClient = await googleAuth.getClient();
+      const googleClient = await this.googleAuth.getClient();
       const token = await googleClient.getAccessToken();
       const headers = new Headers(init2?.headers);
       headers.set("Authorization", `Bearer ${token.token ?? ""}`);
@@ -1657,6 +1670,16 @@ var VertexOpenAIAdapter = class {
   async listAvailableModels() {
     return [...vertexOpenAIManifest.knownModels];
   }
+  async testCredential() {
+    if (!this.googleAuth) {
+      throw new Error("VertexOpenAIAdapter: init() must be called before testCredential()");
+    }
+    const googleClient = await this.googleAuth.getClient();
+    const token = await googleClient.getAccessToken();
+    if (!token.token) {
+      throw new Error("VertexOpenAIAdapter: service account did not mint an access token");
+    }
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1678,6 +1701,7 @@ var vertexOpenAIFactory = {
 
 // src/adapters/bedrock/adapter.ts
 var import_client_bedrock_runtime = require("@aws-sdk/client-bedrock-runtime");
+var import_client_bedrock = require("@aws-sdk/client-bedrock");
 
 // src/adapters/bedrock/manifest.ts
 var BEDROCK_KNOWN_MODELS = [
@@ -1892,6 +1916,7 @@ var BedrockAdapter = class {
   vendor = bedrockManifest.vendor;
   acceptedAuth = ["bedrockApiKey", "bedrockIam"];
   client;
+  controlClientConfig;
   init(auth, client) {
     if (!auth.region) {
       throw new Error("BedrockAdapter: region is required");
@@ -1922,6 +1947,7 @@ var BedrockAdapter = class {
       );
     }
     this.client = new import_client_bedrock_runtime.BedrockRuntimeClient(config);
+    this.controlClientConfig = config;
   }
   async *stream(req) {
     assertValidRequest(req);
@@ -2100,6 +2126,17 @@ var BedrockAdapter = class {
   async listAvailableModels() {
     return [...bedrockManifest.knownModels];
   }
+  async testCredential() {
+    if (!this.controlClientConfig) {
+      throw new Error("BedrockAdapter: init() must be called before testCredential()");
+    }
+    const control = new import_client_bedrock.BedrockClient(this.controlClientConfig);
+    try {
+      await control.send(new import_client_bedrock.ListFoundationModelsCommand({}));
+    } finally {
+      control.destroy();
+    }
+  }
   async warmup() {
   }
   ensureClient() {
@@ -2432,6 +2469,11 @@ function runContractTests(harness) {
         (0, import_vitest.expect)(typeof m.capabilities.tools).toBe("boolean");
       }
     });
+    (0, import_vitest.it)("[15a] testCredential resolves on a valid auth probe", async () => {
+      await harness.mockScenario("list-models");
+      const adapter = await init(harness);
+      await (0, import_vitest.expect)(adapter.testCredential()).resolves.not.toThrow();
+    });
     if (harness.nonToolCapableModel) {
       (0, import_vitest.it)("[16] non-tool-capable model with tools declared either fails fast or runs without tool calls", async () => {
         await harness.mockScenario("simple-stream");
@@ -2626,6 +2668,9 @@ var FakeAdapter = class {
     this.ensureInitialized();
     return fakeManifest.knownModels;
   }
+  async testCredential() {
+    this.ensureInitialized();
+  }
   ensureInitialized() {
     if (!this.initialized) {
       throw new Error("FakeAdapter: init() must be called before stream()/listAvailableModels()");