@jambonz/llm 0.3.4 → 0.4.0

package/dist/test-kit/index.cjs

@@ -2258,6 +2258,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);
@@ -2266,6 +2465,7 @@ registerAdapter(googleFactory);
 registerAdapter(vertexGeminiFactory);
 registerAdapter(vertexOpenAIFactory);
 registerAdapter(bedrockFactory);
+registerAdapter(azureOpenAIFactory);
 
 // src/index.ts
 async function createLlm(args) {