@jambonz/llm 0.2.7 → 0.3.1

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() {
@@ -1546,11 +1558,22 @@ var vertexOpenAIManifest = {
           type: "select",
           required: true,
           default: "us-central1",
+          // Same region set as vertex-gemini plus us-east5 (where Google
+          // hosts Llama partner models). Partner-model availability varies
+          // per region; Google's docs for each model are the source of truth.
           options: [
             { value: "us-central1", label: "us-central1" },
+            { value: "us-east1", label: "us-east1" },
             { value: "us-east4", label: "us-east4" },
+            { value: "us-east5", label: "us-east5" },
+            { value: "us-west1", label: "us-west1" },
+            { value: "us-west4", label: "us-west4" },
             { value: "europe-west1", label: "europe-west1" },
+            { value: "europe-west2", label: "europe-west2" },
+            { value: "europe-west3", label: "europe-west3" },
             { value: "europe-west4", label: "europe-west4" },
+            { value: "asia-east1", label: "asia-east1" },
+            { value: "asia-northeast1", label: "asia-northeast1" },
             { value: "asia-southeast1", label: "asia-southeast1" }
           ]
         }
@@ -1568,6 +1591,7 @@ var VertexOpenAIAdapter = class {
   vendor = vertexOpenAIManifest.vendor;
   acceptedAuth = ["vertexServiceAccount"];
   client;
+  googleAuth;
   init(auth, client) {
     if (auth.kind !== "vertexServiceAccount") {
       throw new Error(
@@ -1584,12 +1608,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 +1642,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 +1676,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 +1894,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 +1925,7 @@ var BedrockAdapter = class {
       );
     }
     this.client = new BedrockRuntimeClient(config);
+    this.controlClientConfig = config;
   }
   async *stream(req) {
     assertValidRequest(req);
@@ -2064,6 +2104,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 +2447,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 +2646,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()");