@webiny/api-core 6.4.8 → 6.4.9

package/exports/api.d.ts

@@ -1,5 +1,4 @@
-export { AiSdk, AiSdkFactory, AiConnectionFactory, Ai } from "../features/ai/index.js";
-export type { IAiConnection, IAiConnectionInline } from "../features/ai/index.js";
+export { AiSdk, AiSdkFactory, AiConnectionFactory, Ai, AiModelRegistry } from "../features/ai/index.js";
 export { Logger } from "../features/logger/index.js";
 export { Encryption } from "../features/encryption/index.js";
 export { BuildParam, BuildParams } from "../features/buildParams/index.js";

package/exports/api.js

@@ -1,4 +1,4 @@
-export { Ai, AiConnectionFactory, AiSdk, AiSdkFactory } from "../features/ai/index.js";
+export { Ai, AiConnectionFactory, AiModelRegistry, AiSdk, AiSdkFactory } from "../features/ai/index.js";
 export { Logger } from "../features/logger/index.js";
 export { Encryption } from "../features/encryption/index.js";
 export { BuildParam, BuildParams } from "../features/buildParams/index.js";

package/features/ai/Ai.d.ts

@@ -3,15 +3,17 @@ import { streamText } from "ai";
 import { Ai as AiAbstraction } from "./abstractions.js";
 import { AiSdkFactory } from "./abstractions.js";
 import { AiConnectionFactory } from "./abstractions.js";
+import { AiModelRegistry } from "./abstractions.js";
 import type { AiGenerateTextParams, AiModel } from "./abstractions.js";
 import type { AiStreamTextParams } from "./abstractions.js";
 import type { IAiConnectionInline } from "./abstractions.js";
 declare class AiImpl implements AiAbstraction.Interface {
     private readonly sdkFactories;
     private readonly connectionFactories;
+    private readonly modelRegistry;
     private sdkCache;
     private resolvedConnections;
-    constructor(sdkFactories: AiSdkFactory.Interface[], connectionFactories: AiConnectionFactory.Interface[]);
+    constructor(sdkFactories: AiSdkFactory.Interface[], connectionFactories: AiConnectionFactory.Interface[], modelRegistry: AiModelRegistry.Interface);
     generateText(params: AiGenerateTextParams): ReturnType<typeof generateText>;
     streamText(params: AiStreamTextParams): Promise<ReturnType<typeof streamText>>;
     listModels(): Promise<AiModel[]>;

package/features/ai/Ai.js

@@ -1,18 +1,11 @@
 import { createImplementation } from "@webiny/feature/api";
 import { generateText, streamText } from "ai";
-import { Ai, AiConnectionFactory, AiSdkFactory } from "./abstractions.js";
-function toAiModel(factory, modelId, modelName) {
-    return {
-        providerId: factory.id,
-        providerName: factory.name,
-        modelId,
-        modelName
-    };
-}
+import { Ai, AiConnectionFactory, AiModelRegistry, AiSdkFactory } from "./abstractions.js";
 class AiImpl {
-    constructor(sdkFactories, connectionFactories){
+    constructor(sdkFactories, connectionFactories, modelRegistry){
         this.sdkFactories = sdkFactories;
         this.connectionFactories = connectionFactories;
+        this.modelRegistry = modelRegistry;
         this.sdkCache = new Map();
         this.resolvedConnections = null;
     }
@@ -32,30 +25,34 @@ class AiImpl {
         });
     }
     listModels() {
-        return Promise.resolve(this.sdkFactories.flatMap((factory)=>factory.models.map((m)=>toAiModel(factory, m.id, m.name))));
+        return this.modelRegistry.listModels();
     }
     async listModelsByConnections() {
-        const connections = await this.getConnections();
-        return connections.flatMap((conn)=>{
-            const factory = this.sdkFactories.find((f)=>f.id === conn.sdkName);
-            if (!factory) return [];
-            return factory.models.map((m)=>toAiModel(factory, m.id, m.name));
-        });
+        const [connections, models] = await Promise.all([
+            this.getConnections(),
+            this.modelRegistry.listModels()
+        ]);
+        const connectedProviderIds = new Set(connections.map((c)=>c.sdkName));
+        return models.filter((m)=>connectedProviderIds.has(m.providerId));
     }
     async listModelsByConnection(connection) {
-        const conn = await this.resolveConnection(void 0, connection);
-        const factory = this.sdkFactories.find((f)=>f.id === conn.sdkName);
-        if (!factory) return [];
-        return factory.models.map((m)=>toAiModel(factory, m.id, m.name));
+        const [conn, models] = await Promise.all([
+            this.resolveConnection(void 0, connection),
+            this.modelRegistry.listModels()
+        ]);
+        return models.filter((m)=>m.providerId === conn.sdkName);
     }
     async resolveLanguageModel(modelId, connection) {
         const slashIndex = modelId.indexOf("/");
         if (-1 === slashIndex) throw new Error(`Invalid model ID "${modelId}". Expected format: "<providerId>/<modelId>" (e.g. "openai/gpt-4o").`);
         const providerId = modelId.slice(0, slashIndex);
-        const modelName = modelId.slice(slashIndex + 1);
+        const rawModelId = modelId.slice(slashIndex + 1);
+        const models = await this.modelRegistry.listModels();
+        const found = models.some((m)=>m.providerId === providerId && m.modelId === rawModelId);
+        if (!found) throw new Error(`Model "${modelId}" is not available. Use listModels() to see available models.`);
         const conn = await this.resolveConnection(providerId, connection);
         const sdk = await this.getSdk(conn);
-        return sdk.languageModel(modelName);
+        return sdk.languageModel(rawModelId);
     }
     async getConnections() {
         if (!this.resolvedConnections) this.resolvedConnections = await Promise.all(this.connectionFactories.map((f)=>f.execute()));
@@ -108,7 +105,8 @@ const Ai_Ai = createImplementation({
             {
                 multiple: true
             }
-        ]
+        ],
+        AiModelRegistry
     ]
 });
 export { Ai_Ai as Ai };

package/features/ai/Ai.js.map

@@ -1 +1 @@
-{"version":3,"file":"features/ai/Ai.js","sources":["../../../src/features/ai/Ai.ts"],"sourcesContent":["import { createImplementation } from \"@webiny/feature/api\";\nimport { generateText } from \"ai\";\nimport { streamText } from \"ai\";\nimport { Ai as AiAbstraction } from \"./abstractions.js\";\nimport { AiSdkFactory } from \"./abstractions.js\";\nimport { AiConnectionFactory } from \"./abstractions.js\";\nimport type { AiGenerateTextParams, AiModel, IAiSdkFactory } from \"./abstractions.js\";\nimport type { AiStreamTextParams } from \"./abstractions.js\";\nimport type { IAiSdk } from \"./abstractions.js\";\nimport type { IAiConnection } from \"./abstractions.js\";\nimport type { IAiConnectionInline } from \"./abstractions.js\";\nimport type { LanguageModel } from \"ai\";\n\nfunction toAiModel(factory: IAiSdkFactory, modelId: string, modelName: string): AiModel {\n    return {\n        providerId: factory.id,\n        providerName: factory.name,\n        modelId,\n        modelName\n    };\n}\n\nclass AiImpl implements AiAbstraction.Interface {\n    private sdkCache = new Map<string, IAiSdk>();\n    private resolvedConnections: IAiConnection[] | null = null;\n\n    constructor(\n        private readonly sdkFactories: AiSdkFactory.Interface[],\n        private readonly connectionFactories: AiConnectionFactory.Interface[]\n    ) {}\n\n    generateText(params: AiGenerateTextParams): ReturnType<typeof generateText> {\n        const { model, connection, ...rest } = params;\n        return this.resolveLanguageModel(model, connection).then(resolvedModel => {\n            // Cast required: spreading the discriminated Prompt union loses its narrowing.\n            return generateText({ model: resolvedModel, ...rest } as Parameters<\n                typeof generateText\n            >[0]);\n        });\n    }\n\n    async streamText(params: AiStreamTextParams): Promise<ReturnType<typeof streamText>> {\n        const { model, connection, ...rest } = params;\n        const resolvedModel = await this.resolveLanguageModel(model, connection);\n        // Cast required: spreading the discriminated Prompt union loses its narrowing.\n        return streamText({ model: resolvedModel, ...rest } as Parameters<typeof streamText>[0]);\n    }\n\n    listModels(): Promise<AiModel[]> {\n        return Promise.resolve(\n            this.sdkFactories.flatMap(factory =>\n                factory.models.map(m => toAiModel(factory, m.id, m.name))\n            )\n        );\n    }\n\n    async listModelsByConnections(): Promise<AiModel[]> {\n        const connections = await this.getConnections();\n        return connections.flatMap(conn => {\n            const factory = this.sdkFactories.find(f => f.id === conn.sdkName);\n            if (!factory) {\n                return [];\n            }\n            return factory.models.map(m => toAiModel(factory, m.id, m.name));\n        });\n    }\n\n    async listModelsByConnection(connection: string | IAiConnectionInline): Promise<AiModel[]> {\n        const conn = await this.resolveConnection(undefined, connection);\n        const factory = this.sdkFactories.find(f => f.id === conn.sdkName);\n        if (!factory) {\n            return [];\n        }\n        return factory.models.map(m => toAiModel(factory, m.id, m.name));\n    }\n\n    private async resolveLanguageModel(\n        modelId: string,\n        connection?: string | IAiConnectionInline\n    ): Promise<LanguageModel> {\n        const slashIndex = modelId.indexOf(\"/\");\n        if (slashIndex === -1) {\n            throw new Error(\n                `Invalid model ID \"${modelId}\". Expected format: \"<providerId>/<modelId>\" (e.g. \"openai/gpt-4o\").`\n            );\n        }\n\n        const providerId = modelId.slice(0, slashIndex);\n        const modelName = modelId.slice(slashIndex + 1);\n\n        const conn = await this.resolveConnection(providerId, connection);\n        const sdk = await this.getSdk(conn);\n        return sdk.languageModel(modelName);\n    }\n\n    private async getConnections(): Promise<IAiConnection[]> {\n        if (!this.resolvedConnections) {\n            this.resolvedConnections = await Promise.all(\n                this.connectionFactories.map(f => f.execute())\n            );\n        }\n        return this.resolvedConnections;\n    }\n\n    private async resolveConnection(\n        providerId: string | undefined,\n        connection?: string | IAiConnectionInline\n    ): Promise<IAiConnectionInline> {\n        if (typeof connection === \"object\") {\n            return connection;\n        }\n\n        const connections = await this.getConnections();\n\n        if (typeof connection === \"string\") {\n            const found = connections.find(c => c.id === connection);\n            if (!found) {\n                const known = connections.map(c => `\"${c.id}\"`).join(\", \");\n                throw new Error(\n                    `Unknown AI connection \"${connection}\". Registered connections: ${known}.`\n                );\n            }\n            return found;\n        }\n\n        const found = connections.find(c => c.sdkName === providerId);\n        if (!found) {\n            const known = connections.map(c => `\"${c.id}\" (${c.sdkName})`).join(\", \");\n            throw new Error(\n                `No AI connection found for provider \"${providerId}\". Registered connections: ${known}.`\n            );\n        }\n        return found;\n    }\n\n    private async getSdk(connection: IAiConnectionInline): Promise<IAiSdk> {\n        const cacheKey =\n            \"id\" in connection\n                ? (connection as IAiConnection).id\n                : `${connection.sdkName}:${connection.apiKey ?? \"__env__\"}`;\n\n        const cached = this.sdkCache.get(cacheKey);\n        if (cached) {\n            return cached;\n        }\n\n        const factory = this.sdkFactories.find(f => f.id === connection.sdkName);\n        if (!factory) {\n            const known = this.sdkFactories.map(f => `\"${f.id}\"`).join(\", \");\n            throw new Error(\n                `No AI SDK factory found for \"${connection.sdkName}\". Registered factories: ${known}.`\n            );\n        }\n\n        const sdk = await factory.execute(connection.apiKey);\n        this.sdkCache.set(cacheKey, sdk);\n        return sdk;\n    }\n}\n\nexport const Ai = createImplementation({\n    abstraction: AiAbstraction,\n    implementation: AiImpl,\n    dependencies: [\n        [AiSdkFactory, { multiple: true }],\n        [AiConnectionFactory, { multiple: true }]\n    ]\n});\n"],"names":["toAiModel","factory","modelId","modelName","AiImpl","sdkFactories","connectionFactories","Map","params","model","connection","rest","resolvedModel","generateText","streamText","Promise","m","connections","conn","f","undefined","slashIndex","Error","providerId","sdk","found","c","known","cacheKey","cached","Ai","createImplementation","AiAbstraction","AiSdkFactory","AiConnectionFactory"],"mappings":";;;AAaA,SAASA,UAAUC,OAAsB,EAAEC,OAAe,EAAEC,SAAiB;IACzE,OAAO;QACH,YAAYF,QAAQ,EAAE;QACtB,cAAcA,QAAQ,IAAI;QAC1BC;QACAC;IACJ;AACJ;AAEA,MAAMC;IAIF,YACqBC,YAAsC,EACtCC,mBAAoD,CACvE;aAFmBD,YAAY,GAAZA;aACAC,mBAAmB,GAAnBA;aALb,QAAQ,GAAG,IAAIC;aACf,mBAAmB,GAA2B;IAKnD;IAEH,aAAaC,MAA4B,EAAmC;QACxE,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGC,MAAM,GAAGH;QACvC,OAAO,IAAI,CAAC,oBAAoB,CAACC,OAAOC,YAAY,IAAI,CAACE,CAAAA,gBAE9CC,aAAa;gBAAE,OAAOD;gBAAe,GAAGD,IAAI;YAAC;IAI5D;IAEA,MAAM,WAAWH,MAA0B,EAA0C;QACjF,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGC,MAAM,GAAGH;QACvC,MAAMI,gBAAgB,MAAM,IAAI,CAAC,oBAAoB,CAACH,OAAOC;QAE7D,OAAOI,WAAW;YAAE,OAAOF;YAAe,GAAGD,IAAI;QAAC;IACtD;IAEA,aAAiC;QAC7B,OAAOI,QAAQ,OAAO,CAClB,IAAI,CAAC,YAAY,CAAC,OAAO,CAACd,CAAAA,UACtBA,QAAQ,MAAM,CAAC,GAAG,CAACe,CAAAA,IAAKhB,UAAUC,SAASe,EAAE,EAAE,EAAEA,EAAE,IAAI;IAGnE;IAEA,MAAM,0BAA8C;QAChD,MAAMC,cAAc,MAAM,IAAI,CAAC,cAAc;QAC7C,OAAOA,YAAY,OAAO,CAACC,CAAAA;YACvB,MAAMjB,UAAU,IAAI,CAAC,YAAY,CAAC,IAAI,CAACkB,CAAAA,IAAKA,EAAE,EAAE,KAAKD,KAAK,OAAO;YACjE,IAAI,CAACjB,SACD,OAAO,EAAE;YAEb,OAAOA,QAAQ,MAAM,CAAC,GAAG,CAACe,CAAAA,IAAKhB,UAAUC,SAASe,EAAE,EAAE,EAAEA,EAAE,IAAI;QAClE;IACJ;IAEA,MAAM,uBAAuBN,UAAwC,EAAsB;QACvF,MAAMQ,OAAO,MAAM,IAAI,CAAC,iBAAiB,CAACE,QAAWV;QACrD,MAAMT,UAAU,IAAI,CAAC,YAAY,CAAC,IAAI,CAACkB,CAAAA,IAAKA,EAAE,EAAE,KAAKD,KAAK,OAAO;QACjE,IAAI,CAACjB,SACD,OAAO,EAAE;QAEb,OAAOA,QAAQ,MAAM,CAAC,GAAG,CAACe,CAAAA,IAAKhB,UAAUC,SAASe,EAAE,EAAE,EAAEA,EAAE,IAAI;IAClE;IAEA,MAAc,qBACVd,OAAe,EACfQ,UAAyC,EACnB;QACtB,MAAMW,aAAanB,QAAQ,OAAO,CAAC;QACnC,IAAImB,AAAe,OAAfA,YACA,MAAM,IAAIC,MACN,CAAC,kBAAkB,EAAEpB,QAAQ,oEAAoE,CAAC;QAI1G,MAAMqB,aAAarB,QAAQ,KAAK,CAAC,GAAGmB;QACpC,MAAMlB,YAAYD,QAAQ,KAAK,CAACmB,aAAa;QAE7C,MAAMH,OAAO,MAAM,IAAI,CAAC,iBAAiB,CAACK,YAAYb;QACtD,MAAMc,MAAM,MAAM,IAAI,CAAC,MAAM,CAACN;QAC9B,OAAOM,IAAI,aAAa,CAACrB;IAC7B;IAEA,MAAc,iBAA2C;QACrD,IAAI,CAAC,IAAI,CAAC,mBAAmB,EACzB,IAAI,CAAC,mBAAmB,GAAG,MAAMY,QAAQ,GAAG,CACxC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAACI,CAAAA,IAAKA,EAAE,OAAO;QAGnD,OAAO,IAAI,CAAC,mBAAmB;IACnC;IAEA,MAAc,kBACVI,UAA8B,EAC9Bb,UAAyC,EACb;QAC5B,IAAI,AAAsB,YAAtB,OAAOA,YACP,OAAOA;QAGX,MAAMO,cAAc,MAAM,IAAI,CAAC,cAAc;QAE7C,IAAI,AAAsB,YAAtB,OAAOP,YAAyB;YAChC,MAAMe,QAAQR,YAAY,IAAI,CAACS,CAAAA,IAAKA,EAAE,EAAE,KAAKhB;YAC7C,IAAI,CAACe,OAAO;gBACR,MAAME,QAAQV,YAAY,GAAG,CAACS,CAAAA,IAAK,CAAC,CAAC,EAAEA,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;gBACrD,MAAM,IAAIJ,MACN,CAAC,uBAAuB,EAAEZ,WAAW,2BAA2B,EAAEiB,MAAM,CAAC,CAAC;YAElF;YACA,OAAOF;QACX;QAEA,MAAMA,QAAQR,YAAY,IAAI,CAACS,CAAAA,IAAKA,EAAE,OAAO,KAAKH;QAClD,IAAI,CAACE,OAAO;YACR,MAAME,QAAQV,YAAY,GAAG,CAACS,CAAAA,IAAK,CAAC,CAAC,EAAEA,EAAE,EAAE,CAAC,GAAG,EAAEA,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YACpE,MAAM,IAAIJ,MACN,CAAC,qCAAqC,EAAEC,WAAW,2BAA2B,EAAEI,MAAM,CAAC,CAAC;QAEhG;QACA,OAAOF;IACX;IAEA,MAAc,OAAOf,UAA+B,EAAmB;QACnE,MAAMkB,WACF,QAAQlB,aACDA,WAA6B,EAAE,GAChC,GAAGA,WAAW,OAAO,CAAC,CAAC,EAAEA,WAAW,MAAM,IAAI,WAAW;QAEnE,MAAMmB,SAAS,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACD;QACjC,IAAIC,QACA,OAAOA;QAGX,MAAM5B,UAAU,IAAI,CAAC,YAAY,CAAC,IAAI,CAACkB,CAAAA,IAAKA,EAAE,EAAE,KAAKT,WAAW,OAAO;QACvE,IAAI,CAACT,SAAS;YACV,MAAM0B,QAAQ,IAAI,CAAC,YAAY,CAAC,GAAG,CAACR,CAAAA,IAAK,CAAC,CAAC,EAAEA,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAC3D,MAAM,IAAIG,MACN,CAAC,6BAA6B,EAAEZ,WAAW,OAAO,CAAC,yBAAyB,EAAEiB,MAAM,CAAC,CAAC;QAE9F;QAEA,MAAMH,MAAM,MAAMvB,QAAQ,OAAO,CAACS,WAAW,MAAM;QACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACkB,UAAUJ;QAC5B,OAAOA;IACX;AACJ;AAEO,MAAMM,QAAKC,qBAAqB;IACnC,aAAaC;IACb,gBAAgB5B;IAChB,cAAc;QACV;YAAC6B;YAAc;gBAAE,UAAU;YAAK;SAAE;QAClC;YAACC;YAAqB;gBAAE,UAAU;YAAK;SAAE;KAC5C;AACL"}
+{"version":3,"file":"features/ai/Ai.js","sources":["../../../src/features/ai/Ai.ts"],"sourcesContent":["import { createImplementation } from \"@webiny/feature/api\";\nimport { generateText } from \"ai\";\nimport { streamText } from \"ai\";\nimport { Ai as AiAbstraction } from \"./abstractions.js\";\nimport { AiSdkFactory } from \"./abstractions.js\";\nimport { AiConnectionFactory } from \"./abstractions.js\";\nimport { AiModelRegistry } from \"./abstractions.js\";\nimport type { AiGenerateTextParams, AiModel } from \"./abstractions.js\";\nimport type { AiStreamTextParams } from \"./abstractions.js\";\nimport type { IAiSdk } from \"./abstractions.js\";\nimport type { IAiConnection } from \"./abstractions.js\";\nimport type { IAiConnectionInline } from \"./abstractions.js\";\nimport type { LanguageModel } from \"ai\";\n\nclass AiImpl implements AiAbstraction.Interface {\n    private sdkCache = new Map<string, IAiSdk>();\n    private resolvedConnections: IAiConnection[] | null = null;\n\n    constructor(\n        private readonly sdkFactories: AiSdkFactory.Interface[],\n        private readonly connectionFactories: AiConnectionFactory.Interface[],\n        private readonly modelRegistry: AiModelRegistry.Interface\n    ) {}\n\n    generateText(params: AiGenerateTextParams): ReturnType<typeof generateText> {\n        const { model, connection, ...rest } = params;\n        return this.resolveLanguageModel(model, connection).then(resolvedModel => {\n            // Cast required: spreading the discriminated Prompt union loses its narrowing.\n            return generateText({ model: resolvedModel, ...rest } as Parameters<\n                typeof generateText\n            >[0]);\n        });\n    }\n\n    async streamText(params: AiStreamTextParams): Promise<ReturnType<typeof streamText>> {\n        const { model, connection, ...rest } = params;\n        const resolvedModel = await this.resolveLanguageModel(model, connection);\n        // Cast required: spreading the discriminated Prompt union loses its narrowing.\n        return streamText({ model: resolvedModel, ...rest } as Parameters<typeof streamText>[0]);\n    }\n\n    listModels(): Promise<AiModel[]> {\n        return this.modelRegistry.listModels();\n    }\n\n    async listModelsByConnections(): Promise<AiModel[]> {\n        const [connections, models] = await Promise.all([\n            this.getConnections(),\n            this.modelRegistry.listModels()\n        ]);\n        const connectedProviderIds = new Set(connections.map(c => c.sdkName));\n        return models.filter(m => connectedProviderIds.has(m.providerId));\n    }\n\n    async listModelsByConnection(connection: string | IAiConnectionInline): Promise<AiModel[]> {\n        const [conn, models] = await Promise.all([\n            this.resolveConnection(undefined, connection),\n            this.modelRegistry.listModels()\n        ]);\n        return models.filter(m => m.providerId === conn.sdkName);\n    }\n\n    private async resolveLanguageModel(\n        modelId: string,\n        connection?: string | IAiConnectionInline\n    ): Promise<LanguageModel> {\n        const slashIndex = modelId.indexOf(\"/\");\n        if (slashIndex === -1) {\n            throw new Error(\n                `Invalid model ID \"${modelId}\". Expected format: \"<providerId>/<modelId>\" (e.g. \"openai/gpt-4o\").`\n            );\n        }\n\n        const providerId = modelId.slice(0, slashIndex);\n        const rawModelId = modelId.slice(slashIndex + 1);\n\n        // Validate the model is in the registry.\n        const models = await this.modelRegistry.listModels();\n        const found = models.some(m => m.providerId === providerId && m.modelId === rawModelId);\n        if (!found) {\n            throw new Error(\n                `Model \"${modelId}\" is not available. Use listModels() to see available models.`\n            );\n        }\n\n        const conn = await this.resolveConnection(providerId, connection);\n        const sdk = await this.getSdk(conn);\n        return sdk.languageModel(rawModelId);\n    }\n\n    private async getConnections(): Promise<IAiConnection[]> {\n        if (!this.resolvedConnections) {\n            this.resolvedConnections = await Promise.all(\n                this.connectionFactories.map(f => f.execute())\n            );\n        }\n        return this.resolvedConnections;\n    }\n\n    private async resolveConnection(\n        providerId: string | undefined,\n        connection?: string | IAiConnectionInline\n    ): Promise<IAiConnectionInline> {\n        if (typeof connection === \"object\") {\n            return connection;\n        }\n\n        const connections = await this.getConnections();\n\n        if (typeof connection === \"string\") {\n            const found = connections.find(c => c.id === connection);\n            if (!found) {\n                const known = connections.map(c => `\"${c.id}\"`).join(\", \");\n                throw new Error(\n                    `Unknown AI connection \"${connection}\". Registered connections: ${known}.`\n                );\n            }\n            return found;\n        }\n\n        const found = connections.find(c => c.sdkName === providerId);\n        if (!found) {\n            const known = connections.map(c => `\"${c.id}\" (${c.sdkName})`).join(\", \");\n            throw new Error(\n                `No AI connection found for provider \"${providerId}\". Registered connections: ${known}.`\n            );\n        }\n        return found;\n    }\n\n    private async getSdk(connection: IAiConnectionInline): Promise<IAiSdk> {\n        const cacheKey =\n            \"id\" in connection\n                ? (connection as IAiConnection).id\n                : `${connection.sdkName}:${connection.apiKey ?? \"__env__\"}`;\n\n        const cached = this.sdkCache.get(cacheKey);\n        if (cached) {\n            return cached;\n        }\n\n        const factory = this.sdkFactories.find(f => f.id === connection.sdkName);\n        if (!factory) {\n            const known = this.sdkFactories.map(f => `\"${f.id}\"`).join(\", \");\n            throw new Error(\n                `No AI SDK factory found for \"${connection.sdkName}\". Registered factories: ${known}.`\n            );\n        }\n\n        const sdk = await factory.execute(connection.apiKey);\n        this.sdkCache.set(cacheKey, sdk);\n        return sdk;\n    }\n}\n\nexport const Ai = createImplementation({\n    abstraction: AiAbstraction,\n    implementation: AiImpl,\n    dependencies: [\n        [AiSdkFactory, { multiple: true }],\n        [AiConnectionFactory, { multiple: true }],\n        AiModelRegistry\n    ]\n});\n"],"names":["AiImpl","sdkFactories","connectionFactories","modelRegistry","Map","params","model","connection","rest","resolvedModel","generateText","streamText","connections","models","Promise","connectedProviderIds","Set","c","m","conn","undefined","modelId","slashIndex","Error","providerId","rawModelId","found","sdk","f","known","cacheKey","cached","factory","Ai","createImplementation","AiAbstraction","AiSdkFactory","AiConnectionFactory","AiModelRegistry"],"mappings":";;;AAcA,MAAMA;IAIF,YACqBC,YAAsC,EACtCC,mBAAoD,EACpDC,aAAwC,CAC3D;aAHmBF,YAAY,GAAZA;aACAC,mBAAmB,GAAnBA;aACAC,aAAa,GAAbA;aANb,QAAQ,GAAG,IAAIC;aACf,mBAAmB,GAA2B;IAMnD;IAEH,aAAaC,MAA4B,EAAmC;QACxE,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGC,MAAM,GAAGH;QACvC,OAAO,IAAI,CAAC,oBAAoB,CAACC,OAAOC,YAAY,IAAI,CAACE,CAAAA,gBAE9CC,aAAa;gBAAE,OAAOD;gBAAe,GAAGD,IAAI;YAAC;IAI5D;IAEA,MAAM,WAAWH,MAA0B,EAA0C;QACjF,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGC,MAAM,GAAGH;QACvC,MAAMI,gBAAgB,MAAM,IAAI,CAAC,oBAAoB,CAACH,OAAOC;QAE7D,OAAOI,WAAW;YAAE,OAAOF;YAAe,GAAGD,IAAI;QAAC;IACtD;IAEA,aAAiC;QAC7B,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU;IACxC;IAEA,MAAM,0BAA8C;QAChD,MAAM,CAACI,aAAaC,OAAO,GAAG,MAAMC,QAAQ,GAAG,CAAC;YAC5C,IAAI,CAAC,cAAc;YACnB,IAAI,CAAC,aAAa,CAAC,UAAU;SAChC;QACD,MAAMC,uBAAuB,IAAIC,IAAIJ,YAAY,GAAG,CAACK,CAAAA,IAAKA,EAAE,OAAO;QACnE,OAAOJ,OAAO,MAAM,CAACK,CAAAA,IAAKH,qBAAqB,GAAG,CAACG,EAAE,UAAU;IACnE;IAEA,MAAM,uBAAuBX,UAAwC,EAAsB;QACvF,MAAM,CAACY,MAAMN,OAAO,GAAG,MAAMC,QAAQ,GAAG,CAAC;YACrC,IAAI,CAAC,iBAAiB,CAACM,QAAWb;YAClC,IAAI,CAAC,aAAa,CAAC,UAAU;SAChC;QACD,OAAOM,OAAO,MAAM,CAACK,CAAAA,IAAKA,EAAE,UAAU,KAAKC,KAAK,OAAO;IAC3D;IAEA,MAAc,qBACVE,OAAe,EACfd,UAAyC,EACnB;QACtB,MAAMe,aAAaD,QAAQ,OAAO,CAAC;QACnC,IAAIC,AAAe,OAAfA,YACA,MAAM,IAAIC,MACN,CAAC,kBAAkB,EAAEF,QAAQ,oEAAoE,CAAC;QAI1G,MAAMG,aAAaH,QAAQ,KAAK,CAAC,GAAGC;QACpC,MAAMG,aAAaJ,QAAQ,KAAK,CAACC,aAAa;QAG9C,MAAMT,SAAS,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU;QAClD,MAAMa,QAAQb,OAAO,IAAI,CAACK,CAAAA,IAAKA,EAAE,UAAU,KAAKM,cAAcN,EAAE,OAAO,KAAKO;QAC5E,IAAI,CAACC,OACD,MAAM,IAAIH,MACN,CAAC,OAAO,EAAEF,QAAQ,6DAA6D,CAAC;QAIxF,MAAMF,OAAO,MAAM,IAAI,CAAC,iBAAiB,CAACK,YAAYjB;QACtD,MAAMoB,MAAM,MAAM,IAAI,CAAC,MAAM,CAACR;QAC9B,OAAOQ,IAAI,aAAa,CAACF;IAC7B;IAEA,MAAc,iBAA2C;QACrD,IAAI,CAAC,IAAI,CAAC,mBAAmB,EACzB,IAAI,CAAC,mBAAmB,GAAG,MAAMX,QAAQ,GAAG,CACxC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAACc,CAAAA,IAAKA,EAAE,OAAO;QAGnD,OAAO,IAAI,CAAC,mBAAmB;IACnC;IAEA,MAAc,kBACVJ,UAA8B,EAC9BjB,UAAyC,EACb;QAC5B,IAAI,AAAsB,YAAtB,OAAOA,YACP,OAAOA;QAGX,MAAMK,cAAc,MAAM,IAAI,CAAC,cAAc;QAE7C,IAAI,AAAsB,YAAtB,OAAOL,YAAyB;YAChC,MAAMmB,QAAQd,YAAY,IAAI,CAACK,CAAAA,IAAKA,EAAE,EAAE,KAAKV;YAC7C,IAAI,CAACmB,OAAO;gBACR,MAAMG,QAAQjB,YAAY,GAAG,CAACK,CAAAA,IAAK,CAAC,CAAC,EAAEA,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;gBACrD,MAAM,IAAIM,MACN,CAAC,uBAAuB,EAAEhB,WAAW,2BAA2B,EAAEsB,MAAM,CAAC,CAAC;YAElF;YACA,OAAOH;QACX;QAEA,MAAMA,QAAQd,YAAY,IAAI,CAACK,CAAAA,IAAKA,EAAE,OAAO,KAAKO;QAClD,IAAI,CAACE,OAAO;YACR,MAAMG,QAAQjB,YAAY,GAAG,CAACK,CAAAA,IAAK,CAAC,CAAC,EAAEA,EAAE,EAAE,CAAC,GAAG,EAAEA,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YACpE,MAAM,IAAIM,MACN,CAAC,qCAAqC,EAAEC,WAAW,2BAA2B,EAAEK,MAAM,CAAC,CAAC;QAEhG;QACA,OAAOH;IACX;IAEA,MAAc,OAAOnB,UAA+B,EAAmB;QACnE,MAAMuB,WACF,QAAQvB,aACDA,WAA6B,EAAE,GAChC,GAAGA,WAAW,OAAO,CAAC,CAAC,EAAEA,WAAW,MAAM,IAAI,WAAW;QAEnE,MAAMwB,SAAS,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACD;QACjC,IAAIC,QACA,OAAOA;QAGX,MAAMC,UAAU,IAAI,CAAC,YAAY,CAAC,IAAI,CAACJ,CAAAA,IAAKA,EAAE,EAAE,KAAKrB,WAAW,OAAO;QACvE,IAAI,CAACyB,SAAS;YACV,MAAMH,QAAQ,IAAI,CAAC,YAAY,CAAC,GAAG,CAACD,CAAAA,IAAK,CAAC,CAAC,EAAEA,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAC3D,MAAM,IAAIL,MACN,CAAC,6BAA6B,EAAEhB,WAAW,OAAO,CAAC,yBAAyB,EAAEsB,MAAM,CAAC,CAAC;QAE9F;QAEA,MAAMF,MAAM,MAAMK,QAAQ,OAAO,CAACzB,WAAW,MAAM;QACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAACuB,UAAUH;QAC5B,OAAOA;IACX;AACJ;AAEO,MAAMM,QAAKC,qBAAqB;IACnC,aAAaC;IACb,gBAAgBnC;IAChB,cAAc;QACV;YAACoC;YAAc;gBAAE,UAAU;YAAK;SAAE;QAClC;YAACC;YAAqB;gBAAE,UAAU;YAAK;SAAE;QACzCC;KACH;AACL"}

package/features/ai/AiModelRegistry.d.ts

@@ -0,0 +1,12 @@
+import { AiModelRegistry as AiModelRegistryAbstraction } from "./abstractions.js";
+import { AiSdkFactory } from "./abstractions.js";
+import type { AiModel } from "./abstractions.js";
+declare class AiModelRegistryImpl implements AiModelRegistryAbstraction.Interface {
+    private readonly sdkFactories;
+    constructor(sdkFactories: AiSdkFactory.Interface[]);
+    listModels(): Promise<AiModel[]>;
+}
+export declare const AiModelRegistry: typeof AiModelRegistryImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").IAiModelRegistry>;
+};
+export {};

package/features/ai/AiModelRegistry.js

@@ -0,0 +1,28 @@
+import { AiModelRegistry, AiSdkFactory } from "./abstractions.js";
+class AiModelRegistryImpl {
+    constructor(sdkFactories){
+        this.sdkFactories = sdkFactories;
+    }
+    async listModels() {
+        return this.sdkFactories.flatMap((factory)=>factory.models.map((m)=>({
+                    providerId: factory.id,
+                    providerName: factory.name,
+                    modelId: m.id,
+                    modelName: m.name
+                })));
+    }
+}
+const AiModelRegistry_AiModelRegistry = AiModelRegistry.createImplementation({
+    implementation: AiModelRegistryImpl,
+    dependencies: [
+        [
+            AiSdkFactory,
+            {
+                multiple: true
+            }
+        ]
+    ]
+});
+export { AiModelRegistry_AiModelRegistry as AiModelRegistry };
+
+//# sourceMappingURL=AiModelRegistry.js.map

package/features/ai/AiModelRegistry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"features/ai/AiModelRegistry.js","sources":["../../../src/features/ai/AiModelRegistry.ts"],"sourcesContent":["import { AiModelRegistry as AiModelRegistryAbstraction } from \"./abstractions.js\";\nimport { AiSdkFactory } from \"./abstractions.js\";\nimport type { AiModel } from \"./abstractions.js\";\n\nclass AiModelRegistryImpl implements AiModelRegistryAbstraction.Interface {\n    constructor(private readonly sdkFactories: AiSdkFactory.Interface[]) {}\n\n    async listModels(): Promise<AiModel[]> {\n        return this.sdkFactories.flatMap(factory =>\n            factory.models.map(m => ({\n                providerId: factory.id,\n                providerName: factory.name,\n                modelId: m.id,\n                modelName: m.name\n            }))\n        );\n    }\n}\n\nexport const AiModelRegistry = AiModelRegistryAbstraction.createImplementation({\n    implementation: AiModelRegistryImpl,\n    dependencies: [[AiSdkFactory, { multiple: true }]]\n});\n"],"names":["AiModelRegistryImpl","sdkFactories","factory","m","AiModelRegistry","AiModelRegistryAbstraction","AiSdkFactory"],"mappings":";AAIA,MAAMA;IACF,YAA6BC,YAAsC,CAAE;aAAxCA,YAAY,GAAZA;IAAyC;IAEtE,MAAM,aAAiC;QACnC,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAACC,CAAAA,UAC7BA,QAAQ,MAAM,CAAC,GAAG,CAACC,CAAAA,IAAM;oBACrB,YAAYD,QAAQ,EAAE;oBACtB,cAAcA,QAAQ,IAAI;oBAC1B,SAASC,EAAE,EAAE;oBACb,WAAWA,EAAE,IAAI;gBACrB;IAER;AACJ;AAEO,MAAMC,kCAAkBC,gBAAAA,oBAA+C,CAAC;IAC3E,gBAAgBL;IAChB,cAAc;QAAC;YAACM;YAAc;gBAAE,UAAU;YAAK;SAAE;KAAC;AACtD"}

package/features/ai/abstractions.d.ts

@@ -70,6 +70,14 @@ export declare namespace Ai {
     type GenerateTextParams = AiGenerateTextParams;
     type StreamTextParams = AiStreamTextParams;
 }
+export interface IAiModelRegistry {
+    listModels(): Promise<AiModel[]>;
+}
+/** Single source of truth for available AI models. Decoratable to restrict the model list. */
+export declare const AiModelRegistry: import("@webiny/di").Abstraction<IAiModelRegistry>;
+export declare namespace AiModelRegistry {
+    type Interface = IAiModelRegistry;
+}
 export interface IAiSdkTool<TInput = any> {
     readonly name: string;
     readonly description: string;

package/features/ai/abstractions.js

@@ -3,8 +3,9 @@ const AiSdk = createAbstraction("AiSdk");
 const AiSdkFactory = createAbstraction("AiSdkFactory");
 const AiConnectionFactory = createAbstraction("AiConnectionFactory");
 const Ai = createAbstraction("Ai");
+const AiModelRegistry = createAbstraction("AiModelRegistry");
 const AiSdkTool = createAbstraction("AiSdkTool");
 const AiSdkTools = createAbstraction("AiSdkTools");
-export { Ai, AiConnectionFactory, AiSdk, AiSdkFactory, AiSdkTool, AiSdkTools };
+export { Ai, AiConnectionFactory, AiModelRegistry, AiSdk, AiSdkFactory, AiSdkTool, AiSdkTools };
 
 //# sourceMappingURL=abstractions.js.map

package/features/ai/abstractions.js.map

@@ -1 +1 @@
-{"version":3,"file":"features/ai/abstractions.js","sources":["../../../src/features/ai/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\nimport type { generateText } from \"ai\";\nimport type { streamText } from \"ai\";\nimport type { LanguageModel } from \"ai\";\nimport type { FlexibleSchema, ToolSet } from \"ai\";\n\n// AiSdk\n\nexport interface IAiSdkModel {\n    id: string; // raw model id, e.g. \"claude-sonnet-4-5\"\n    name: string; // human-friendly name, e.g. \"Claude Sonnet 4.5\"\n}\n\nexport interface IAiSdk {\n    languageModel(modelId: string): LanguageModel;\n}\n\n/** A single AI SDK instance (e.g. OpenAI, Anthropic) that resolves model instances. */\nexport const AiSdk = createAbstraction<IAiSdk>(\"AiSdk\");\n\nexport namespace AiSdk {\n    export type Interface = IAiSdk;\n}\n\n// AiSdkFactory\n\nexport interface IAiSdkFactory {\n    readonly id: string; // machine id, e.g. \"anthropic\"\n    readonly name: string; // human-friendly name, e.g. \"Anthropic\"\n    readonly models: readonly IAiSdkModel[]; // static model list, no API call needed\n    execute(apiKey?: string): Promise<IAiSdk>;\n}\n\n/** Factory that asynchronously initialises an AI SDK. Register one per provider namespace. */\nexport const AiSdkFactory = createAbstraction<IAiSdkFactory>(\"AiSdkFactory\");\n\nexport namespace AiSdkFactory {\n    export type Interface = IAiSdkFactory;\n}\n\n// AiConnection\n\nexport interface IAiConnectionInline {\n    readonly sdkName: string;\n    readonly apiKey?: string;\n}\n\nexport interface IAiConnection extends IAiConnectionInline {\n    readonly id: string;\n}\n\n// AiConnectionFactory\n\nexport interface IAiConnectionFactory {\n    execute(): Promise<IAiConnection>;\n}\n\n/** Factory that asynchronously produces an AiConnection. */\nexport const AiConnectionFactory = createAbstraction<IAiConnectionFactory>(\"AiConnectionFactory\");\n\nexport namespace AiConnectionFactory {\n    export type Interface = IAiConnectionFactory;\n}\n\n// Ai\n\ntype SDKGenerateTextParams = Parameters<typeof generateText>[0];\ntype SDKStreamTextParams = Parameters<typeof streamText>[0];\n\nexport type AiGenerateTextParams = Omit<SDKGenerateTextParams, \"model\"> & {\n    model: string;\n    connection?: string | IAiConnectionInline;\n};\nexport type AiStreamTextParams = Omit<SDKStreamTextParams, \"model\"> & {\n    model: string;\n    connection?: string | IAiConnectionInline;\n};\n\nexport interface AiModel {\n    providerId: string; // e.g. \"anthropic\"\n    providerName: string; // e.g. \"Anthropic\"\n    modelId: string; // e.g. \"claude-sonnet-4-5\"\n    modelName: string; // e.g. \"Claude Sonnet 4.5\"\n}\n\nexport interface IAi {\n    generateText(params: AiGenerateTextParams): ReturnType<typeof generateText>;\n    streamText(params: AiStreamTextParams): Promise<ReturnType<typeof streamText>>;\n    listModels(): Promise<AiModel[]>;\n    listModelsByConnections(): Promise<AiModel[]>;\n    listModelsByConnection(connection: string | IAiConnectionInline): Promise<AiModel[]>;\n}\n\n/** Interact with AI language models using registered providers. */\nexport const Ai = createAbstraction<IAi>(\"Ai\");\n\nexport namespace Ai {\n    export type Interface = IAi;\n    export type GenerateTextParams = AiGenerateTextParams;\n    export type StreamTextParams = AiStreamTextParams;\n}\n\n// AiSdkTool\n\nexport interface IAiSdkTool<TInput = any> {\n    readonly name: string;\n    readonly description: string;\n    readonly inputSchema: FlexibleSchema<TInput>;\n    execute(input: TInput): Promise<unknown>;\n}\n\n/** A single tool that can be provided to AI generateText/streamText calls. */\nexport const AiSdkTool = createAbstraction<IAiSdkTool>(\"AiSdkTool\");\n\nexport namespace AiSdkTool {\n    export type Interface = IAiSdkTool;\n}\n\n// AiSdkTools\n\nexport interface IAiSdkTools {\n    getToolSet(): ToolSet;\n}\n\n/** Collection of AI SDK tools. Returns a ready-to-use ToolSet for generateText/streamText. */\nexport const AiSdkTools = createAbstraction<IAiSdkTools>(\"AiSdkTools\");\n\nexport namespace AiSdkTools {\n    export type Interface = IAiSdkTools;\n}\n"],"names":["AiSdk","createAbstraction","AiSdkFactory","AiConnectionFactory","Ai","AiSdkTool","AiSdkTools"],"mappings":";AAkBO,MAAMA,QAAQC,kBAA0B;AAgBxC,MAAMC,eAAeD,kBAAiC;AAwBtD,MAAME,sBAAsBF,kBAAwC;AAoCpE,MAAMG,KAAKH,kBAAuB;AAkBlC,MAAMI,YAAYJ,kBAA8B;AAahD,MAAMK,aAAaL,kBAA+B"}
+{"version":3,"file":"features/ai/abstractions.js","sources":["../../../src/features/ai/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\nimport type { generateText } from \"ai\";\nimport type { streamText } from \"ai\";\nimport type { LanguageModel } from \"ai\";\nimport type { FlexibleSchema, ToolSet } from \"ai\";\n\n// AiSdk\n\nexport interface IAiSdkModel {\n    id: string; // raw model id, e.g. \"claude-sonnet-4-5\"\n    name: string; // human-friendly name, e.g. \"Claude Sonnet 4.5\"\n}\n\nexport interface IAiSdk {\n    languageModel(modelId: string): LanguageModel;\n}\n\n/** A single AI SDK instance (e.g. OpenAI, Anthropic) that resolves model instances. */\nexport const AiSdk = createAbstraction<IAiSdk>(\"AiSdk\");\n\nexport namespace AiSdk {\n    export type Interface = IAiSdk;\n}\n\n// AiSdkFactory\n\nexport interface IAiSdkFactory {\n    readonly id: string; // machine id, e.g. \"anthropic\"\n    readonly name: string; // human-friendly name, e.g. \"Anthropic\"\n    readonly models: readonly IAiSdkModel[]; // static model list, no API call needed\n    execute(apiKey?: string): Promise<IAiSdk>;\n}\n\n/** Factory that asynchronously initialises an AI SDK. Register one per provider namespace. */\nexport const AiSdkFactory = createAbstraction<IAiSdkFactory>(\"AiSdkFactory\");\n\nexport namespace AiSdkFactory {\n    export type Interface = IAiSdkFactory;\n}\n\n// AiConnection\n\nexport interface IAiConnectionInline {\n    readonly sdkName: string;\n    readonly apiKey?: string;\n}\n\nexport interface IAiConnection extends IAiConnectionInline {\n    readonly id: string;\n}\n\n// AiConnectionFactory\n\nexport interface IAiConnectionFactory {\n    execute(): Promise<IAiConnection>;\n}\n\n/** Factory that asynchronously produces an AiConnection. */\nexport const AiConnectionFactory = createAbstraction<IAiConnectionFactory>(\"AiConnectionFactory\");\n\nexport namespace AiConnectionFactory {\n    export type Interface = IAiConnectionFactory;\n}\n\n// Ai\n\ntype SDKGenerateTextParams = Parameters<typeof generateText>[0];\ntype SDKStreamTextParams = Parameters<typeof streamText>[0];\n\nexport type AiGenerateTextParams = Omit<SDKGenerateTextParams, \"model\"> & {\n    model: string;\n    connection?: string | IAiConnectionInline;\n};\nexport type AiStreamTextParams = Omit<SDKStreamTextParams, \"model\"> & {\n    model: string;\n    connection?: string | IAiConnectionInline;\n};\n\nexport interface AiModel {\n    providerId: string; // e.g. \"anthropic\"\n    providerName: string; // e.g. \"Anthropic\"\n    modelId: string; // e.g. \"claude-sonnet-4-5\"\n    modelName: string; // e.g. \"Claude Sonnet 4.5\"\n}\n\nexport interface IAi {\n    generateText(params: AiGenerateTextParams): ReturnType<typeof generateText>;\n    streamText(params: AiStreamTextParams): Promise<ReturnType<typeof streamText>>;\n    listModels(): Promise<AiModel[]>;\n    listModelsByConnections(): Promise<AiModel[]>;\n    listModelsByConnection(connection: string | IAiConnectionInline): Promise<AiModel[]>;\n}\n\n/** Interact with AI language models using registered providers. */\nexport const Ai = createAbstraction<IAi>(\"Ai\");\n\nexport namespace Ai {\n    export type Interface = IAi;\n    export type GenerateTextParams = AiGenerateTextParams;\n    export type StreamTextParams = AiStreamTextParams;\n}\n\n// AiModelRegistry\n\nexport interface IAiModelRegistry {\n    listModels(): Promise<AiModel[]>;\n}\n\n/** Single source of truth for available AI models. Decoratable to restrict the model list. */\nexport const AiModelRegistry = createAbstraction<IAiModelRegistry>(\"AiModelRegistry\");\n\nexport namespace AiModelRegistry {\n    export type Interface = IAiModelRegistry;\n}\n\n// AiSdkTool\n\nexport interface IAiSdkTool<TInput = any> {\n    readonly name: string;\n    readonly description: string;\n    readonly inputSchema: FlexibleSchema<TInput>;\n    execute(input: TInput): Promise<unknown>;\n}\n\n/** A single tool that can be provided to AI generateText/streamText calls. */\nexport const AiSdkTool = createAbstraction<IAiSdkTool>(\"AiSdkTool\");\n\nexport namespace AiSdkTool {\n    export type Interface = IAiSdkTool;\n}\n\n// AiSdkTools\n\nexport interface IAiSdkTools {\n    getToolSet(): ToolSet;\n}\n\n/** Collection of AI SDK tools. Returns a ready-to-use ToolSet for generateText/streamText. */\nexport const AiSdkTools = createAbstraction<IAiSdkTools>(\"AiSdkTools\");\n\nexport namespace AiSdkTools {\n    export type Interface = IAiSdkTools;\n}\n"],"names":["AiSdk","createAbstraction","AiSdkFactory","AiConnectionFactory","Ai","AiModelRegistry","AiSdkTool","AiSdkTools"],"mappings":";AAkBO,MAAMA,QAAQC,kBAA0B;AAgBxC,MAAMC,eAAeD,kBAAiC;AAwBtD,MAAME,sBAAsBF,kBAAwC;AAoCpE,MAAMG,KAAKH,kBAAuB;AAelC,MAAMI,kBAAkBJ,kBAAoC;AAgB5D,MAAMK,YAAYL,kBAA8B;AAahD,MAAMM,aAAaN,kBAA+B"}

package/features/ai/feature.js

@@ -2,12 +2,14 @@ import { createFeature } from "@webiny/feature/api";
 import { OpenAiSdkFactory } from "./OpenAiSdkFactory.js";
 import { AnthropicSdkFactory } from "./AnthropicSdkFactory.js";
 import { Ai } from "./Ai.js";
+import { AiModelRegistry } from "./AiModelRegistry.js";
 import { AiSdkTools } from "./AiSdkTools.js";
 const AiFeature = createFeature({
     name: "AiFeature",
     register (container) {
         container.register(OpenAiSdkFactory);
         container.register(AnthropicSdkFactory);
+        container.register(AiModelRegistry);
         container.register(Ai);
         container.register(AiSdkTools);
     }

package/features/ai/feature.js.map

@@ -1 +1 @@
-{"version":3,"file":"features/ai/feature.js","sources":["../../../src/features/ai/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { OpenAiSdkFactory } from \"./OpenAiSdkFactory.js\";\nimport { AnthropicSdkFactory } from \"./AnthropicSdkFactory.js\";\nimport { Ai } from \"./Ai.js\";\nimport { AiSdkTools } from \"./AiSdkTools.js\";\n\nexport const AiFeature = createFeature({\n    name: \"AiFeature\",\n    register(container) {\n        container.register(OpenAiSdkFactory);\n        container.register(AnthropicSdkFactory);\n        container.register(Ai);\n        container.register(AiSdkTools);\n    }\n});\n"],"names":["AiFeature","createFeature","container","OpenAiSdkFactory","AnthropicSdkFactory","Ai","AiSdkTools"],"mappings":";;;;;AAMO,MAAMA,YAAYC,cAAc;IACnC,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;QACnBF,UAAU,QAAQ,CAACG;QACnBH,UAAU,QAAQ,CAACI;IACvB;AACJ"}
+{"version":3,"file":"features/ai/feature.js","sources":["../../../src/features/ai/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { OpenAiSdkFactory } from \"./OpenAiSdkFactory.js\";\nimport { AnthropicSdkFactory } from \"./AnthropicSdkFactory.js\";\nimport { Ai } from \"./Ai.js\";\nimport { AiModelRegistry } from \"./AiModelRegistry.js\";\nimport { AiSdkTools } from \"./AiSdkTools.js\";\n\nexport const AiFeature = createFeature({\n    name: \"AiFeature\",\n    register(container) {\n        container.register(OpenAiSdkFactory);\n        container.register(AnthropicSdkFactory);\n        container.register(AiModelRegistry);\n        container.register(Ai);\n        container.register(AiSdkTools);\n    }\n});\n"],"names":["AiFeature","createFeature","container","OpenAiSdkFactory","AnthropicSdkFactory","AiModelRegistry","Ai","AiSdkTools"],"mappings":";;;;;;AAOO,MAAMA,YAAYC,cAAc;IACnC,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;QACnBF,UAAU,QAAQ,CAACG;QACnBH,UAAU,QAAQ,CAACI;QACnBJ,UAAU,QAAQ,CAACK;IACvB;AACJ"}

package/features/ai/index.d.ts

@@ -1,3 +1,3 @@
-export { AiSdk, AiSdkFactory, AiConnectionFactory, Ai, AiSdkTool, AiSdkTools } from "./abstractions.js";
-export type { IAiConnection, IAiConnectionInline, AiModel, IAiSdkModel, IAiSdkTool, IAiSdkTools } from "./abstractions.js";
+export { AiSdk, AiSdkFactory, AiConnectionFactory, Ai, AiModelRegistry, AiSdkTool, AiSdkTools } from "./abstractions.js";
+export type { IAiConnection, IAiConnectionInline, AiModel, IAiModelRegistry, IAiSdkModel, IAiSdkTool, IAiSdkTools } from "./abstractions.js";
 export { AiFeature } from "./feature.js";

package/features/ai/index.js

@@ -1,2 +1,2 @@
-export { Ai, AiConnectionFactory, AiSdk, AiSdkFactory, AiSdkTool, AiSdkTools } from "./abstractions.js";
+export { Ai, AiConnectionFactory, AiModelRegistry, AiSdk, AiSdkFactory, AiSdkTool, AiSdkTools } from "./abstractions.js";
 export { AiFeature } from "./feature.js";

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@webiny/api-core",
-	"version": "6.4.8",
+	"version": "6.4.9",
 	"type": "module",
 	"repository": {
 		"type": "git",
@@ -16,17 +16,17 @@
 	"dependencies": {
 		"@ai-sdk/anthropic": "4.0.10",
 		"@ai-sdk/openai": "4.0.9",
-		"@webiny/api": "6.4.8",
+		"@webiny/api": "6.4.9",
 		"@webiny/di": "1.0.2",
-		"@webiny/error": "6.4.8",
-		"@webiny/feature": "6.4.8",
-		"@webiny/feature-flags": "6.4.8",
-		"@webiny/handler": "6.4.8",
-		"@webiny/handler-db": "6.4.8",
-		"@webiny/handler-graphql": "6.4.8",
-		"@webiny/plugins": "6.4.8",
-		"@webiny/utils": "6.4.8",
-		"@webiny/wcp": "6.4.8",
+		"@webiny/error": "6.4.9",
+		"@webiny/feature": "6.4.9",
+		"@webiny/feature-flags": "6.4.9",
+		"@webiny/handler": "6.4.9",
+		"@webiny/handler-db": "6.4.9",
+		"@webiny/handler-graphql": "6.4.9",
+		"@webiny/plugins": "6.4.9",
+		"@webiny/utils": "6.4.9",
+		"@webiny/wcp": "6.4.9",
 		"ai": "7.0.18",
 		"dataloader": "2.2.3",
 		"jose": "6.2.3",
@@ -39,10 +39,10 @@
 	},
 	"devDependencies": {
 		"@types/jsonwebtoken": "9.0.10",
-		"@webiny/build-tools": "6.4.8",
-		"@webiny/db-dynamodb": "6.4.8",
-		"@webiny/handler-aws": "6.4.8",
-		"@webiny/project-utils": "6.4.8",
+		"@webiny/build-tools": "6.4.9",
+		"@webiny/db-dynamodb": "6.4.9",
+		"@webiny/handler-aws": "6.4.9",
+		"@webiny/project-utils": "6.4.9",
 		"jest-dynalite": "3.6.1",
 		"rimraf": "6.1.3",
 		"typescript": "6.0.3",