@askdb/ai-azure 0.1.0-beta.2 → 1.0.0-beta.3

package/dist/index.d.ts

@@ -1,3 +1,3 @@
-import type { AiProviderAdapter } from "@askdb/ai";
+import { type AiProviderAdapter } from "@askdb/ai";
 export declare const azureProvider: AiProviderAdapter;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAA+B,MAAM,WAAW,CAAC;AAGhF,eAAO,MAAM,aAAa,EAAE,iBA4B3B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAIL,KAAK,iBAAiB,EAEvB,MAAM,WAAW,CAAC;AAenB,eAAO,MAAM,aAAa,EAAE,iBAuD3B,CAAC"}

package/dist/index.js

@@ -1,41 +1,73 @@
 import { createAzure } from "@ai-sdk/azure";
-import { defaultEmbeddingSettingsMiddleware, wrapEmbeddingModel } from "ai";
+import { resolveBaseConfig, withEmbeddingProviderOptions, } from "@askdb/ai";
+const ENV_SPEC = {
+    apiKeyVars: ["AZURE_OPENAI_API_KEY", "AZURE_API_KEY"],
+    apiKeySecondaryVars: ["AZURE_OPENAI_API_KEY_SECONDARY", "AZURE_API_KEY_SECONDARY"],
+    modelVars: ["AZURE_OPENAI_DEPLOYMENT", "AZURE_DEPLOYMENT_NAME"],
+    embeddingModelVars: [
+        "AZURE_OPENAI_EMBEDDING_DEPLOYMENT",
+        "AZURE_EMBEDDING_DEPLOYMENT_NAME",
+    ],
+    baseURLVars: ["AZURE_OPENAI_BASE_URL", "AZURE_OPENAI_ENDPOINT", "AZURE_BASE_URL"],
+    defaultModel: "gpt-4o-mini",
+    defaultEmbeddingModel: "text-embedding-3-small",
+};
 export const azureProvider = {
     provider: "azure",
+    aliases: ["azure-openai", "foundry"],
+    configHint: "For Azure / Microsoft Foundry, set ai.provider: \"azure\" and ai.providerConfig.azure.apiKey in askdb.config.*.",
+    resolveConfig(env, options) {
+        const config = resolveBaseConfig("azure", env, ENV_SPEC, options);
+        if (!config)
+            return undefined;
+        const resourceName = env.ASKDB_AI_AZURE_RESOURCE_NAME || env.AZURE_RESOURCE_NAME || undefined;
+        const apiVersion = env.ASKDB_AI_AZURE_API_VERSION ||
+            env.AZURE_OPENAI_API_VERSION ||
+            env.AZURE_API_VERSION ||
+            undefined;
+        if (!config.baseURL && !resourceName) {
+            throw new Error("Azure provider requires ASKDB_AI_AZURE_RESOURCE_NAME (e.g. 'my-foundry') " +
+                "or ASKDB_AI_BASE_URL pointing at the full endpoint.");
+        }
+        const providerOptions = {
+            ...(resourceName ? { resourceName } : {}),
+            ...(apiVersion ? { apiVersion } : {}),
+        };
+        return {
+            ...config,
+            ...(Object.keys(providerOptions).length > 0 ? { providerOptions } : {}),
+        };
+    },
     createLanguageModel(config) {
+        const { resourceName, apiVersion } = azureConnectionOptions(config);
         const azure = createAzure({
             apiKey: config.apiKey,
-            ...(config.resourceName ? { resourceName: config.resourceName } : {}),
+            ...(resourceName ? { resourceName } : {}),
             ...(config.baseURL ? { baseURL: config.baseURL } : {}),
-            ...(config.apiVersion ? { apiVersion: config.apiVersion } : {}),
+            ...(apiVersion ? { apiVersion } : {}),
         });
         return azure(config.model);
     },
     createEmbeddingModel(config, options = {}) {
+        const { resourceName, apiVersion } = azureConnectionOptions(config);
         const azure = createAzure({
             apiKey: config.apiKey,
-            ...(config.resourceName ? { resourceName: config.resourceName } : {}),
+            ...(resourceName ? { resourceName } : {}),
             ...(config.baseURL ? { baseURL: config.baseURL } : {}),
-            ...(config.apiVersion ? { apiVersion: config.apiVersion } : {}),
+            ...(apiVersion ? { apiVersion } : {}),
         });
         const model = azure.embedding(config.model);
-        const providerOptions = azureProviderOptions(options);
-        if (!providerOptions)
-            return model;
-        return wrapEmbeddingModel({
-            model,
-            middleware: defaultEmbeddingSettingsMiddleware({
-                settings: { providerOptions },
-            }),
-        });
+        return withEmbeddingProviderOptions(model, "azure", options);
     },
 };
-function azureProviderOptions(options) {
-    const azure = {};
-    if (options.dimensions !== undefined)
-        azure.dimensions = options.dimensions;
-    if (options.user !== undefined)
-        azure.user = options.user;
-    return Object.keys(azure).length > 0 ? { azure } : undefined;
+function azureConnectionOptions(config) {
+    return {
+        resourceName: readStringOption(config.providerOptions, "resourceName"),
+        apiVersion: readStringOption(config.providerOptions, "apiVersion"),
+    };
+}
+function readStringOption(options, key) {
+    const value = options?.[key];
+    return typeof value === "string" ? value : undefined;
 }
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,kCAAkC,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AAE5E,MAAM,CAAC,MAAM,aAAa,GAAsB;IAC9C,QAAQ,EAAE,OAAO;IACjB,mBAAmB,CAAC,MAAM;QACxB,MAAM,KAAK,GAAG,WAAW,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC,CAAC;QACH,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IACD,oBAAoB,CAAC,MAAM,EAAE,OAAO,GAAG,EAAE;QACvC,MAAM,KAAK,GAAG,WAAW,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrE,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5C,MAAM,eAAe,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACtD,IAAI,CAAC,eAAe;YAAE,OAAO,KAAK,CAAC;QACnC,OAAO,kBAAkB,CAAC;YACxB,KAAK;YACL,UAAU,EAAE,kCAAkC,CAAC;gBAC7C,QAAQ,EAAE,EAAE,eAAe,EAAE;aAC9B,CAAC;SACH,CAAC,CAAC;IACL,CAAC;CACF,CAAC;AAEF,SAAS,oBAAoB,CAC3B,OAAoC;IAEpC,MAAM,KAAK,GAA2C,EAAE,CAAC;IACzD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS;QAAE,KAAK,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IAC5E,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS;QAAE,KAAK,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAC1D,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/D,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EACL,iBAAiB,EACjB,4BAA4B,GAI7B,MAAM,WAAW,CAAC;AAEnB,MAAM,QAAQ,GAAoB;IAChC,UAAU,EAAE,CAAC,sBAAsB,EAAE,eAAe,CAAC;IACrD,mBAAmB,EAAE,CAAC,gCAAgC,EAAE,yBAAyB,CAAC;IAClF,SAAS,EAAE,CAAC,yBAAyB,EAAE,uBAAuB,CAAC;IAC/D,kBAAkB,EAAE;QAClB,mCAAmC;QACnC,iCAAiC;KAClC;IACD,WAAW,EAAE,CAAC,uBAAuB,EAAE,uBAAuB,EAAE,gBAAgB,CAAC;IACjF,YAAY,EAAE,aAAa;IAC3B,qBAAqB,EAAE,wBAAwB;CAChD,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAsB;IAC9C,QAAQ,EAAE,OAAO;IACjB,OAAO,EAAE,CAAC,cAAc,EAAE,SAAS,CAAC;IACpC,UAAU,EACR,iHAAiH;IACnH,aAAa,CAAC,GAAG,EAAE,OAAO;QACxB,MAAM,MAAM,GAAG,iBAAiB,CAAC,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QAE9B,MAAM,YAAY,GAChB,GAAG,CAAC,4BAA4B,IAAI,GAAG,CAAC,mBAAmB,IAAI,SAAS,CAAC;QAC3E,MAAM,UAAU,GACd,GAAG,CAAC,0BAA0B;YAC9B,GAAG,CAAC,wBAAwB;YAC5B,GAAG,CAAC,iBAAiB;YACrB,SAAS,CAAC;QAEZ,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,YAAY,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,2EAA2E;gBACzE,qDAAqD,CACxD,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAG;YACtB,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACtC,CAAC;QAEF,OAAO;YACL,GAAG,MAAM;YACT,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACxE,CAAC;IACJ,CAAC;IACD,mBAAmB,CAAC,MAAM;QACxB,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;QACpE,MAAM,KAAK,GAAG,WAAW,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACtC,CAAC,CAAC;QACH,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IACD,oBAAoB,CAAC,MAAM,EAAE,OAAO,GAAG,EAAE;QACvC,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;QACpE,MAAM,KAAK,GAAG,WAAW,CAAC;YACxB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACtC,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5C,OAAO,4BAA4B,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/D,CAAC;CACF,CAAC;AAEF,SAAS,sBAAsB,CAC7B,MAAgB;IAEhB,OAAO;QACL,YAAY,EAAE,gBAAgB,CAAC,MAAM,CAAC,eAAe,EAAE,cAAc,CAAC;QACtE,UAAU,EAAE,gBAAgB,CAAC,MAAM,CAAC,eAAe,EAAE,YAAY,CAAC;KACnE,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CACvB,OAA4C,EAC5C,GAAW;IAEX,MAAM,KAAK,GAAG,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACvD,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@askdb/ai-azure",
-  "version": "0.1.0-beta.2",
+  "version": "1.0.0-beta.3",
   "description": "AskDB Azure OpenAI provider adapter for @askdb/ai.",
   "keywords": [
     "askdb",
@@ -39,13 +39,17 @@
     "node": ">=20"
   },
   "dependencies": {
-    "@ai-sdk/azure": "^3.0.72",
-    "ai": "^6.0.200",
-    "@askdb/ai": "0.1.0-beta.2"
+    "@ai-sdk/azure": "^3.0.72"
+  },
+  "peerDependencies": {
+    "ai": "^6.0.0",
+    "@askdb/ai": "^0.1.0-beta.3"
   },
   "devDependencies": {
+    "ai": "^6.0.200",
     "typescript": "^5.9.3",
-    "vitest": "^3.2.6"
+    "vitest": "^3.2.6",
+    "@askdb/ai": "0.1.0-beta.3"
   },
   "scripts": {
     "build": "tsc -p tsconfig.build.json",