@jambonz/llm 0.3.4 → 0.4.0

package/dist/index.cjs

@@ -36,6 +36,7 @@ __export(src_exports, {
   getManifest: () => getManifest,
   listVendors: () => listVendors,
   normalizeAuth: () => normalizeAuth,
+  pickAuthKind: () => pickAuthKind,
   registerAdapter: () => registerAdapter,
   replaceAdapter: () => replaceAdapter,
   unregisterAdapter: () => unregisterAdapter
@@ -2270,6 +2271,205 @@ var bedrockFactory = {
   create: () => new BedrockAdapter()
 };
 
+// src/adapters/azure-openai/adapter.ts
+var import_openai3 = require("openai");
+
+// src/adapters/azure-openai/manifest.ts
+var AZURE_OPENAI_KNOWN_MODELS = [
+  {
+    id: "gpt-4o",
+    displayName: "GPT-4o (on Azure)",
+    capabilities: {
+      streaming: true,
+      tools: true,
+      vision: true,
+      systemPrompt: true,
+      maxContextTokens: 128e3
+    }
+  },
+  {
+    id: "gpt-4o-mini",
+    displayName: "GPT-4o mini (on Azure)",
+    capabilities: {
+      streaming: true,
+      tools: true,
+      vision: true,
+      systemPrompt: true,
+      maxContextTokens: 128e3
+    }
+  },
+  {
+    id: "gpt-4-turbo",
+    displayName: "GPT-4 Turbo (on Azure)",
+    capabilities: {
+      streaming: true,
+      tools: true,
+      vision: true,
+      systemPrompt: true,
+      maxContextTokens: 128e3
+    }
+  },
+  {
+    // Azure spells it `gpt-35-turbo` (no dot). Keep their spelling for the id
+    // so capability lookups against deployment names that echo the Azure id
+    // hit.
+    id: "gpt-35-turbo",
+    displayName: "GPT-3.5 Turbo (on Azure)",
+    capabilities: {
+      streaming: true,
+      tools: true,
+      vision: false,
+      systemPrompt: true,
+      maxContextTokens: 16385
+    }
+  },
+  {
+    id: "o1-preview",
+    displayName: "o1 preview \u2014 reasoning (on Azure)",
+    capabilities: {
+      streaming: false,
+      tools: false,
+      vision: false,
+      systemPrompt: false,
+      maxContextTokens: 128e3
+    }
+  },
+  {
+    id: "o1-mini",
+    displayName: "o1 mini \u2014 reasoning (on Azure)",
+    capabilities: {
+      streaming: false,
+      tools: false,
+      vision: false,
+      systemPrompt: false,
+      maxContextTokens: 128e3
+    }
+  }
+];
+var azureOpenAIManifest = {
+  vendor: "azure-openai",
+  displayName: "Azure OpenAI",
+  authKinds: [
+    {
+      kind: "azureOpenAIApiKey",
+      displayName: "API Key",
+      fields: [
+        {
+          name: "apiKey",
+          label: "API Key",
+          type: "password",
+          required: true,
+          help: "Azure OpenAI resource key. Find it in the Azure portal under your OpenAI resource \u2192 Keys and Endpoint."
+        },
+        {
+          name: "endpoint",
+          label: "Endpoint",
+          type: "url",
+          required: true,
+          help: "Resource endpoint URL, e.g. https://my-resource.openai.azure.com. Shown in the Azure portal alongside the keys."
+        },
+        {
+          name: "deployment",
+          label: "Deployment Name",
+          type: "text",
+          required: true,
+          help: "The name you chose for the deployment in Azure AI Studio (NOT the underlying model id \u2014 deployments can be named anything)."
+        },
+        {
+          name: "apiVersion",
+          label: "API Version",
+          type: "text",
+          required: true,
+          default: "2024-10-21",
+          help: "Data-plane api-version, e.g. 2024-10-21. Microsoft rolls these frequently; see Azure OpenAI API reference docs for the current stable GA version."
+        }
+      ]
+    }
+  ],
+  knownModels: AZURE_OPENAI_KNOWN_MODELS,
+  // Azure's list-deployments endpoint is on the control plane (ARM) and
+  // requires AAD credentials the data-plane API key doesn't carry. Return
+  // the curated set from the manifest.
+  supportsModelListing: false,
+  docsUrl: "https://learn.microsoft.com/en-us/azure/ai-services/openai/reference"
+};
+
+// src/adapters/azure-openai/adapter.ts
+var AzureOpenAIAdapter = class {
+  vendor = azureOpenAIManifest.vendor;
+  acceptedAuth = ["azureOpenAIApiKey"];
+  client;
+  init(auth, client) {
+    if (auth.kind !== "azureOpenAIApiKey") {
+      throw new Error(
+        `AzureOpenAIAdapter: unsupported auth kind '${auth.kind}'`
+      );
+    }
+    if (!auth.apiKey) {
+      throw new Error("AzureOpenAIAdapter: apiKey is required");
+    }
+    if (!auth.endpoint) {
+      throw new Error("AzureOpenAIAdapter: endpoint is required");
+    }
+    if (!auth.deployment) {
+      throw new Error("AzureOpenAIAdapter: deployment is required");
+    }
+    if (!auth.apiVersion) {
+      throw new Error("AzureOpenAIAdapter: apiVersion is required");
+    }
+    this.client = new import_openai3.AzureOpenAI({
+      apiKey: auth.apiKey,
+      endpoint: auth.endpoint,
+      apiVersion: auth.apiVersion,
+      deployment: auth.deployment,
+      ...client?.timeout !== void 0 ? { timeout: client.timeout } : {},
+      ...client?.maxRetries !== void 0 ? { maxRetries: client.maxRetries } : {}
+    });
+  }
+  stream(req) {
+    return streamFromOpenAI(this.ensureClient(), req, {
+      knownModels: azureOpenAIManifest.knownModels
+    });
+  }
+  appendAssistantToolCall(history, toolCalls) {
+    return appendOpenAIAssistantToolCall(history, toolCalls);
+  }
+  appendToolResult(history, toolCallId, result) {
+    return appendOpenAIToolResult(history, toolCallId, result);
+  }
+  async listAvailableModels() {
+    return [...azureOpenAIManifest.knownModels];
+  }
+  async testCredential() {
+    const client = this.ensureClient();
+    await client.chat.completions.create({
+      // Azure ignores `model` when a deployment is in the URL, but the SDK
+      // still requires it as a parameter.
+      model: "probe",
+      messages: [{ role: "user", content: "ping" }],
+      max_tokens: 1,
+      stream: false
+    });
+  }
+  async warmup() {
+  }
+  ensureClient() {
+    if (!this.client) {
+      throw new Error(
+        "AzureOpenAIAdapter: init() must be called before stream()/listAvailableModels()"
+      );
+    }
+    return this.client;
+  }
+};
+
+// src/adapters/azure-openai/factory.ts
+var azureOpenAIFactory = {
+  vendor: azureOpenAIManifest.vendor,
+  manifest: azureOpenAIManifest,
+  create: () => new AzureOpenAIAdapter()
+};
+
 // src/adapters/index.ts
 registerAdapter(openAIFactory);
 registerAdapter(deepseekFactory);
@@ -2278,6 +2478,7 @@ registerAdapter(googleFactory);
 registerAdapter(vertexGeminiFactory);
 registerAdapter(vertexOpenAIFactory);
 registerAdapter(bedrockFactory);
+registerAdapter(azureOpenAIFactory);
 
 // src/normalize-auth.ts
 function normalizeAuth(vendor, raw) {
@@ -2298,6 +2499,8 @@ function normalizeAuth(vendor, raw) {
       return normalizeVertex(vendor, raw);
     case "bedrock":
       return normalizeBedrock(raw);
+    case "azure-openai":
+      return normalizeAzureOpenAI(raw);
     default:
       throw new Error(
         `normalizeAuth: unknown vendor '${vendor}'. If this is a custom adapter, construct the AuthSpec directly.`
@@ -2377,6 +2580,27 @@ function normalizeBedrock(raw) {
     `normalizeAuth: vendor 'bedrock' requires either (access_key_id + secret_access_key) or api_key`
   );
 }
+function normalizeAzureOpenAI(raw) {
+  if (!raw.api_key) {
+    throw new Error(`normalizeAuth: vendor 'azure-openai' requires 'api_key'`);
+  }
+  if (!raw.endpoint) {
+    throw new Error(`normalizeAuth: vendor 'azure-openai' requires 'endpoint'`);
+  }
+  if (!raw.deployment) {
+    throw new Error(`normalizeAuth: vendor 'azure-openai' requires 'deployment'`);
+  }
+  if (!raw.api_version) {
+    throw new Error(`normalizeAuth: vendor 'azure-openai' requires 'api_version'`);
+  }
+  return {
+    kind: "azureOpenAIApiKey",
+    apiKey: raw.api_key,
+    endpoint: raw.endpoint,
+    deployment: raw.deployment,
+    apiVersion: raw.api_version
+  };
+}
 function parseServiceKey(value) {
   if (typeof value === "string") {
     try {
@@ -2388,6 +2612,51 @@ function parseServiceKey(value) {
   return value;
 }
 
+// src/pick-auth-kind.ts
+function pickAuthKind(vendor, raw) {
+  const factory = getAdapterFactory(vendor);
+  const { authKinds } = factory.manifest;
+  if (authKinds.length === 1) {
+    return authKinds[0];
+  }
+  switch (vendor) {
+    case "bedrock": {
+      const iam = authKinds.find((k) => k.kind === "bedrockIam");
+      const apiKey = authKinds.find((k) => k.kind === "bedrockApiKey");
+      if (raw.access_key_id && raw.secret_access_key) {
+        if (!iam) throw new Error(`pickAuthKind: bedrock manifest missing 'bedrockIam'`);
+        return iam;
+      }
+      if (raw.api_key) {
+        if (!apiKey) throw new Error(`pickAuthKind: bedrock manifest missing 'bedrockApiKey'`);
+        return apiKey;
+      }
+      throw new Error(
+        `pickAuthKind: bedrock credential missing both 'api_key' and ('access_key_id' + 'secret_access_key')`
+      );
+    }
+    case "google": {
+      const svc = authKinds.find((k) => k.kind === "googleServiceAccount");
+      const apiKey = authKinds.find((k) => k.kind === "googleApiKey");
+      if (raw.service_key) {
+        if (!svc) throw new Error(`pickAuthKind: google manifest missing 'googleServiceAccount'`);
+        return svc;
+      }
+      if (raw.api_key) {
+        if (!apiKey) throw new Error(`pickAuthKind: google manifest missing 'googleApiKey'`);
+        return apiKey;
+      }
+      throw new Error(
+        `pickAuthKind: google credential missing both 'api_key' and 'service_key'`
+      );
+    }
+    default:
+      throw new Error(
+        `pickAuthKind: vendor '${vendor}' declares ${authKinds.length} authKinds but has no discriminator rule. Add one in src/pick-auth-kind.ts.`
+      );
+  }
+}
+
 // src/index.ts
 async function createLlm(args) {
   const factory = getAdapterFactory(args.vendor);
@@ -2409,6 +2678,7 @@ async function createLlm(args) {
   getManifest,
   listVendors,
   normalizeAuth,
+  pickAuthKind,
   registerAdapter,
   replaceAdapter,
   unregisterAdapter