@jambonz/llm 0.2.6 → 0.3.0

package/dist/test-kit/index.d.cts

@@ -1,4 +1,4 @@
-import { a as AdapterFactory, d as AuthKind, A as AuthSpec, L as LlmAdapter, c as ApiKeyAuth, g as ClientOptions, P as PromptRequest, l as LlmEvent, M as Message, o as ToolCallEvent, n as ModelInfo, b as AdapterManifest } from '../types-bq1XMi_T.cjs';
+import { a as AdapterFactory, d as AuthKind, A as AuthSpec, L as LlmAdapter, c as ApiKeyAuth, g as ClientOptions, P as PromptRequest, l as LlmEvent, M as Message, o as ToolCallEvent, n as ModelInfo, b as AdapterManifest } from '../types-4d2QrlAe.cjs';
 
 /**
  * Named scenarios the harness must be able to stage. Each scenario configures
@@ -132,6 +132,7 @@ declare class FakeAdapter implements LlmAdapter<ApiKeyAuth> {
     appendAssistantToolCall(history: Message[], toolCalls: ReadonlyArray<ToolCallEvent>): Message[];
     appendToolResult(history: Message[], toolCallId: string, result: unknown): Message[];
     listAvailableModels(): Promise<ModelInfo[]>;
+    testCredential(): Promise<void>;
     private ensureInitialized;
     private capture;
 }

package/dist/test-kit/index.d.ts

@@ -1,4 +1,4 @@
-import { a as AdapterFactory, d as AuthKind, A as AuthSpec, L as LlmAdapter, c as ApiKeyAuth, g as ClientOptions, P as PromptRequest, l as LlmEvent, M as Message, o as ToolCallEvent, n as ModelInfo, b as AdapterManifest } from '../types-bq1XMi_T.js';
+import { a as AdapterFactory, d as AuthKind, A as AuthSpec, L as LlmAdapter, c as ApiKeyAuth, g as ClientOptions, P as PromptRequest, l as LlmEvent, M as Message, o as ToolCallEvent, n as ModelInfo, b as AdapterManifest } from '../types-4d2QrlAe.js';
 
 /**
  * Named scenarios the harness must be able to stage. Each scenario configures
@@ -132,6 +132,7 @@ declare class FakeAdapter implements LlmAdapter<ApiKeyAuth> {
     appendAssistantToolCall(history: Message[], toolCalls: ReadonlyArray<ToolCallEvent>): Message[];
     appendToolResult(history: Message[], toolCallId: string, result: unknown): Message[];
     listAvailableModels(): Promise<ModelInfo[]>;
+    testCredential(): Promise<void>;
     private ensureInitialized;
     private capture;
 }

package/dist/test-kit/index.js

@@ -426,6 +426,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);
@@ -852,6 +855,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);
@@ -1278,6 +1284,9 @@ var GoogleAdapter = class {
   async listAvailableModels() {
     return listGeminiModels(this.ensureClient(), googleManifest.knownModels);
   }
+  async testCredential() {
+    await listGeminiModels(this.ensureClient(), googleManifest.knownModels);
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1448,6 +1457,9 @@ var VertexGeminiAdapter = class {
   async listAvailableModels() {
     return listGeminiModels(this.ensureClient(), vertexGeminiManifest.knownModels);
   }
+  async testCredential() {
+    await listGeminiModels(this.ensureClient(), vertexGeminiManifest.knownModels);
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1568,6 +1580,7 @@ var VertexOpenAIAdapter = class {
   vendor = vertexOpenAIManifest.vendor;
   acceptedAuth = ["vertexServiceAccount"];
   client;
+  googleAuth;
   init(auth, client) {
     if (auth.kind !== "vertexServiceAccount") {
       throw new Error(
@@ -1584,12 +1597,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 GoogleAuth({
+    this.googleAuth = new 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 ?? ""}`);
@@ -1618,6 +1631,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() {
@@ -1642,6 +1665,10 @@ import {
   BedrockRuntimeClient,
   ConverseStreamCommand
 } from "@aws-sdk/client-bedrock-runtime";
+import {
+  BedrockClient,
+  ListFoundationModelsCommand
+} from "@aws-sdk/client-bedrock";
 
 // src/adapters/bedrock/manifest.ts
 var BEDROCK_KNOWN_MODELS = [
@@ -1856,6 +1883,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");
@@ -1886,6 +1914,7 @@ var BedrockAdapter = class {
       );
     }
     this.client = new BedrockRuntimeClient(config);
+    this.controlClientConfig = config;
   }
   async *stream(req) {
     assertValidRequest(req);
@@ -2064,6 +2093,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 BedrockClient(this.controlClientConfig);
+    try {
+      await control.send(new ListFoundationModelsCommand({}));
+    } finally {
+      control.destroy();
+    }
+  }
   async warmup() {
   }
   ensureClient() {
@@ -2396,6 +2436,11 @@ function runContractTests(harness) {
         expect(typeof m.capabilities.tools).toBe("boolean");
       }
     });
+    it("[15a] testCredential resolves on a valid auth probe", async () => {
+      await harness.mockScenario("list-models");
+      const adapter = await init(harness);
+      await expect(adapter.testCredential()).resolves.not.toThrow();
+    });
     if (harness.nonToolCapableModel) {
       it("[16] non-tool-capable model with tools declared either fails fast or runs without tool calls", async () => {
         await harness.mockScenario("simple-stream");
@@ -2590,6 +2635,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()");