@jambonz/llm 0.2.7 → 0.3.1

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() {
@@ -1585,11 +1597,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" }
           ]
         }
@@ -1607,6 +1630,7 @@ var VertexOpenAIAdapter = class {
   vendor = vertexOpenAIManifest.vendor;
   acceptedAuth = ["vertexServiceAccount"];
   client;
+  googleAuth;
   init(auth, client) {
     if (auth.kind !== "vertexServiceAccount") {
       throw new Error(
@@ -1623,12 +1647,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 +1681,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 +1712,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 +1927,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 +1958,7 @@ var BedrockAdapter = class {
       );
     }
     this.client = new import_client_bedrock_runtime.BedrockRuntimeClient(config);
+    this.controlClientConfig = config;
   }
   async *stream(req) {
     assertValidRequest(req);
@@ -2100,6 +2137,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 +2480,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 +2679,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()");