@jambonz/llm 0.2.6 → 0.3.0

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { S as ServiceAccountJson, A as AuthSpec, P as PromptRequest, a as AdapterFactory, b as AdapterManifest, C as CreateLlmArgs, L as LlmAdapter } from './types-bq1XMi_T.cjs';
-export { c as ApiKeyAuth, d as AuthKind, e as AuthKindSchema, B as BedrockApiKeyAuth, f as BedrockIamAuth, g as ClientOptions, h as ContentBlock, F as FinishReason, i as FormField, j as FormFieldType, G as GoogleApiKeyAuth, k as GoogleServiceAccountAuth, J as JsonSchema, l as LlmEvent, M as Message, m as ModelCapabilities, n as ModelInfo, R as Role, T as Tool, o as ToolCallEvent, U as Usage, V as VertexServiceAccountAuth } from './types-bq1XMi_T.cjs';
+import { S as ServiceAccountJson, A as AuthSpec, P as PromptRequest, a as AdapterFactory, b as AdapterManifest, C as CreateLlmArgs, L as LlmAdapter } from './types-4d2QrlAe.cjs';
+export { c as ApiKeyAuth, d as AuthKind, e as AuthKindSchema, B as BedrockApiKeyAuth, f as BedrockIamAuth, g as ClientOptions, h as ContentBlock, F as FinishReason, i as FormField, j as FormFieldType, G as GoogleApiKeyAuth, k as GoogleServiceAccountAuth, J as JsonSchema, l as LlmEvent, M as Message, m as ModelCapabilities, n as ModelInfo, R as Role, T as Tool, o as ToolCallEvent, U as Usage, V as VertexServiceAccountAuth } from './types-4d2QrlAe.cjs';
 
 /**
  * The loose "DB-shape" input accepted by `normalizeAuth`.

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { S as ServiceAccountJson, A as AuthSpec, P as PromptRequest, a as AdapterFactory, b as AdapterManifest, C as CreateLlmArgs, L as LlmAdapter } from './types-bq1XMi_T.js';
-export { c as ApiKeyAuth, d as AuthKind, e as AuthKindSchema, B as BedrockApiKeyAuth, f as BedrockIamAuth, g as ClientOptions, h as ContentBlock, F as FinishReason, i as FormField, j as FormFieldType, G as GoogleApiKeyAuth, k as GoogleServiceAccountAuth, J as JsonSchema, l as LlmEvent, M as Message, m as ModelCapabilities, n as ModelInfo, R as Role, T as Tool, o as ToolCallEvent, U as Usage, V as VertexServiceAccountAuth } from './types-bq1XMi_T.js';
+import { S as ServiceAccountJson, A as AuthSpec, P as PromptRequest, a as AdapterFactory, b as AdapterManifest, C as CreateLlmArgs, L as LlmAdapter } from './types-4d2QrlAe.js';
+export { c as ApiKeyAuth, d as AuthKind, e as AuthKindSchema, B as BedrockApiKeyAuth, f as BedrockIamAuth, g as ClientOptions, h as ContentBlock, F as FinishReason, i as FormField, j as FormFieldType, G as GoogleApiKeyAuth, k as GoogleServiceAccountAuth, J as JsonSchema, l as LlmEvent, M as Message, m as ModelCapabilities, n as ModelInfo, R as Role, T as Tool, o as ToolCallEvent, U as Usage, V as VertexServiceAccountAuth } from './types-4d2QrlAe.js';
 
 /**
  * The loose "DB-shape" input accepted by `normalizeAuth`.

package/dist/index.js

@@ -433,6 +433,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);
@@ -859,6 +862,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);
@@ -1285,6 +1291,9 @@ var GoogleAdapter = class {
   async listAvailableModels() {
     return listGeminiModels(this.ensureClient(), googleManifest.knownModels);
   }
+  async testCredential() {
+    await listGeminiModels(this.ensureClient(), googleManifest.knownModels);
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1455,6 +1464,9 @@ var VertexGeminiAdapter = class {
   async listAvailableModels() {
     return listGeminiModels(this.ensureClient(), vertexGeminiManifest.knownModels);
   }
+  async testCredential() {
+    await listGeminiModels(this.ensureClient(), vertexGeminiManifest.knownModels);
+  }
   async warmup() {
   }
   ensureClient() {
@@ -1575,6 +1587,7 @@ var VertexOpenAIAdapter = class {
   vendor = vertexOpenAIManifest.vendor;
   acceptedAuth = ["vertexServiceAccount"];
   client;
+  googleAuth;
   init(auth, client) {
     if (auth.kind !== "vertexServiceAccount") {
       throw new Error(
@@ -1591,12 +1604,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, init) => {
-      const googleClient = await googleAuth.getClient();
+      const googleClient = await this.googleAuth.getClient();
       const token = await googleClient.getAccessToken();
       const headers = new Headers(init?.headers);
       headers.set("Authorization", `Bearer ${token.token ?? ""}`);
@@ -1625,6 +1638,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() {
@@ -1649,6 +1672,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 = [
@@ -1863,6 +1890,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");
@@ -1893,6 +1921,7 @@ var BedrockAdapter = class {
       );
     }
     this.client = new BedrockRuntimeClient(config);
+    this.controlClientConfig = config;
   }
   async *stream(req) {
     assertValidRequest(req);
@@ -2071,6 +2100,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() {