@n8n/ai-utilities 0.36.2 → 0.36.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cjs/model-discovery/index.d.ts +2 -1
- package/dist/cjs/model-discovery/index.js +5 -1
- package/dist/cjs/model-discovery/index.js.map +1 -1
- package/dist/cjs/model-discovery/providers/google.d.ts +21 -0
- package/dist/cjs/model-discovery/providers/google.js +32 -8
- package/dist/cjs/model-discovery/providers/google.js.map +1 -1
- package/dist/cjs/model-discovery/types.d.ts +1 -0
- package/dist/cjs/typecheck.tsbuildinfo +1 -1
- package/dist/esm/model-discovery/index.d.ts +2 -1
- package/dist/esm/model-discovery/index.js +5 -1
- package/dist/esm/model-discovery/index.js.map +1 -1
- package/dist/esm/model-discovery/providers/google.d.ts +21 -0
- package/dist/esm/model-discovery/providers/google.js +32 -8
- package/dist/esm/model-discovery/providers/google.js.map +1 -1
- package/dist/esm/model-discovery/types.d.ts +1 -0
- package/dist/esm/typecheck.tsbuildinfo +1 -1
- package/package.json +5 -5
|
@@ -8,7 +8,8 @@ export { listAnthropicModels } from './providers/anthropic';
|
|
|
8
8
|
export { listAzureOpenAiModels, shouldIncludeAzureModel, type AzureOpenAiListModelsOptions, } from './providers/azure';
|
|
9
9
|
export { listCohereModels } from './providers/cohere';
|
|
10
10
|
export { listDeepSeekModels } from './providers/deepseek';
|
|
11
|
-
export { listGoogleModels, shouldIncludeGoogleModel } from './providers/google';
|
|
11
|
+
export { getGoogleModelLabel, listGoogleModels, parseGoogleModels, shouldIncludeGoogleModel, type GoogleModel, } from './providers/google';
|
|
12
|
+
export { pinPrioritizedModels } from '@n8n/api-types';
|
|
12
13
|
export { listGroqModels } from './providers/groq';
|
|
13
14
|
export { listMiniMaxModels } from './providers/minimax';
|
|
14
15
|
export { listMistralModels } from './providers/mistral';
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.ensureUrlPathSuffix = exports.listXaiModels = exports.listVercelModels = exports.listOpenRouterModels = exports.isOpenAiCustomEndpoint = exports.OFFICIAL_OPENAI_HOSTNAMES = exports.shouldIncludeOpenAiModel = exports.listOpenAiModels = exports.listNvidiaModels = exports.listMoonshotAiModels = exports.listMistralModels = exports.listMiniMaxModels = exports.listGroqModels = exports.shouldIncludeGoogleModel = exports.listGoogleModels = exports.listDeepSeekModels = exports.listCohereModels = exports.shouldIncludeAzureModel = exports.listAzureOpenAiModels = exports.listAnthropicModels = exports.shouldIncludeAlibabaModel = exports.listAlibabaModels = exports.classifyChatModelFailure = exports.MODEL_DISCOVERY_PROVIDERS = void 0;
|
|
3
|
+
exports.ensureUrlPathSuffix = exports.listXaiModels = exports.listVercelModels = exports.listOpenRouterModels = exports.isOpenAiCustomEndpoint = exports.OFFICIAL_OPENAI_HOSTNAMES = exports.shouldIncludeOpenAiModel = exports.listOpenAiModels = exports.listNvidiaModels = exports.listMoonshotAiModels = exports.listMistralModels = exports.listMiniMaxModels = exports.listGroqModels = exports.pinPrioritizedModels = exports.shouldIncludeGoogleModel = exports.parseGoogleModels = exports.listGoogleModels = exports.getGoogleModelLabel = exports.listDeepSeekModels = exports.listCohereModels = exports.shouldIncludeAzureModel = exports.listAzureOpenAiModels = exports.listAnthropicModels = exports.shouldIncludeAlibabaModel = exports.listAlibabaModels = exports.classifyChatModelFailure = exports.MODEL_DISCOVERY_PROVIDERS = void 0;
|
|
4
4
|
exports.isModelDiscoveryProvider = isModelDiscoveryProvider;
|
|
5
5
|
exports.listModelsForProvider = listModelsForProvider;
|
|
6
6
|
const alibaba_1 = require("./providers/alibaba");
|
|
@@ -58,8 +58,12 @@ Object.defineProperty(exports, "listCohereModels", { enumerable: true, get: func
|
|
|
58
58
|
var deepseek_2 = require("./providers/deepseek");
|
|
59
59
|
Object.defineProperty(exports, "listDeepSeekModels", { enumerable: true, get: function () { return deepseek_2.listDeepSeekModels; } });
|
|
60
60
|
var google_2 = require("./providers/google");
|
|
61
|
+
Object.defineProperty(exports, "getGoogleModelLabel", { enumerable: true, get: function () { return google_2.getGoogleModelLabel; } });
|
|
61
62
|
Object.defineProperty(exports, "listGoogleModels", { enumerable: true, get: function () { return google_2.listGoogleModels; } });
|
|
63
|
+
Object.defineProperty(exports, "parseGoogleModels", { enumerable: true, get: function () { return google_2.parseGoogleModels; } });
|
|
62
64
|
Object.defineProperty(exports, "shouldIncludeGoogleModel", { enumerable: true, get: function () { return google_2.shouldIncludeGoogleModel; } });
|
|
65
|
+
var api_types_1 = require("@n8n/api-types");
|
|
66
|
+
Object.defineProperty(exports, "pinPrioritizedModels", { enumerable: true, get: function () { return api_types_1.pinPrioritizedModels; } });
|
|
63
67
|
var groq_2 = require("./providers/groq");
|
|
64
68
|
Object.defineProperty(exports, "listGroqModels", { enumerable: true, get: function () { return groq_2.listGroqModels; } });
|
|
65
69
|
var minimax_2 = require("./providers/minimax");
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/model-discovery/index.ts"],"names":[],"mappings":";;;;;AAAA,iDAAwD;AACxD,qDAA4D;AAC5D,+CAAsD;AACtD,mDAA0D;AAC1D,+CAAsD;AACtD,2CAAkD;AAClD,iDAAwD;AACxD,iDAAwD;AACxD,uDAA8D;AAC9D,+CAAsD;AACtD,+CAAsD;AACtD,uDAA8D;AAC9D,+CAAsD;AACtD,yCAAgD;AAcnC,QAAA,yBAAyB,GAAiC;IACtE,OAAO,EAAE,2BAAiB;IAC1B,SAAS,EAAE,+BAAmB;IAC9B,MAAM,EAAE,yBAAgB;IACxB,QAAQ,EAAE,6BAAkB;IAC5B,MAAM,EAAE,yBAAgB;IACxB,IAAI,EAAE,qBAAc;IACpB,OAAO,EAAE,2BAAiB;IAC1B,OAAO,EAAE,2BAAiB;IAC1B,UAAU,EAAE,iCAAoB;IAChC,MAAM,EAAE,yBAAgB;IACxB,MAAM,EAAE,yBAAgB;IACxB,UAAU,EAAE,iCAAoB;IAChC,MAAM,EAAE,yBAAgB;IACxB,GAAG,EAAE,mBAAa;CAClB,CAAC;AAEF,kCAAyC,QAAgB;IACxD,OAAO,QAAQ,IAAI,QAAA,yBAAyB,CAAC;AAC9C,CAAC;AAEM,KAAK,gCACX,QAAgB,EAChB,OAA0B;IAE1B,MAAM,UAAU,GAAG,QAAA,yBAAyB,CAAC,QAAQ,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,sCAAsC,QAAQ,GAAG,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC;AAClC,CAAC;AAED,yDAG6B;AAF5B,6HAAA,wBAAwB,OAAA;AAGzB,+CAAmF;AAA1E,4GAAA,iBAAiB,OAAA;AAAE,oHAAA,yBAAyB,OAAA;AACrD,mDAA4D;AAAnD,gHAAA,mBAAmB,OAAA;AAC5B,2CAI2B;AAH1B,8GAAA,qBAAqB,OAAA;AACrB,gHAAA,uBAAuB,OAAA;AAGxB,6CAAsD;AAA7C,0GAAA,gBAAgB,OAAA;AACzB,iDAA0D;AAAjD,8GAAA,kBAAkB,OAAA;AAC3B,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/model-discovery/index.ts"],"names":[],"mappings":";;;;;AAAA,iDAAwD;AACxD,qDAA4D;AAC5D,+CAAsD;AACtD,mDAA0D;AAC1D,+CAAsD;AACtD,2CAAkD;AAClD,iDAAwD;AACxD,iDAAwD;AACxD,uDAA8D;AAC9D,+CAAsD;AACtD,+CAAsD;AACtD,uDAA8D;AAC9D,+CAAsD;AACtD,yCAAgD;AAcnC,QAAA,yBAAyB,GAAiC;IACtE,OAAO,EAAE,2BAAiB;IAC1B,SAAS,EAAE,+BAAmB;IAC9B,MAAM,EAAE,yBAAgB;IACxB,QAAQ,EAAE,6BAAkB;IAC5B,MAAM,EAAE,yBAAgB;IACxB,IAAI,EAAE,qBAAc;IACpB,OAAO,EAAE,2BAAiB;IAC1B,OAAO,EAAE,2BAAiB;IAC1B,UAAU,EAAE,iCAAoB;IAChC,MAAM,EAAE,yBAAgB;IACxB,MAAM,EAAE,yBAAgB;IACxB,UAAU,EAAE,iCAAoB;IAChC,MAAM,EAAE,yBAAgB;IACxB,GAAG,EAAE,mBAAa;CAClB,CAAC;AAEF,kCAAyC,QAAgB;IACxD,OAAO,QAAQ,IAAI,QAAA,yBAAyB,CAAC;AAC9C,CAAC;AAEM,KAAK,gCACX,QAAgB,EAChB,OAA0B;IAE1B,MAAM,UAAU,GAAG,QAAA,yBAAyB,CAAC,QAAQ,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,sCAAsC,QAAQ,GAAG,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC;AAClC,CAAC;AAED,yDAG6B;AAF5B,6HAAA,wBAAwB,OAAA;AAGzB,+CAAmF;AAA1E,4GAAA,iBAAiB,OAAA;AAAE,oHAAA,yBAAyB,OAAA;AACrD,mDAA4D;AAAnD,gHAAA,mBAAmB,OAAA;AAC5B,2CAI2B;AAH1B,8GAAA,qBAAqB,OAAA;AACrB,gHAAA,uBAAuB,OAAA;AAGxB,6CAAsD;AAA7C,0GAAA,gBAAgB,OAAA;AACzB,iDAA0D;AAAjD,8GAAA,kBAAkB,OAAA;AAC3B,6CAM4B;AAL3B,6GAAA,mBAAmB,OAAA;AACnB,0GAAA,gBAAgB,OAAA;AAChB,2GAAA,iBAAiB,OAAA;AACjB,kHAAA,wBAAwB,OAAA;AAGzB,4CAAsD;AAA7C,iHAAA,oBAAoB,OAAA;AAC7B,yCAAkD;AAAzC,sGAAA,cAAc,OAAA;AACvB,+CAAwD;AAA/C,4GAAA,iBAAiB,OAAA;AAC1B,+CAAwD;AAA/C,4GAAA,iBAAiB,OAAA;AAC1B,qDAA8D;AAArD,kHAAA,oBAAoB,OAAA;AAC7B,6CAAsD;AAA7C,0GAAA,gBAAgB,OAAA;AACzB,6CAK4B;AAJ3B,0GAAA,gBAAgB,OAAA;AAChB,kHAAA,wBAAwB,OAAA;AACxB,mHAAA,yBAAyB,OAAA;AACzB,gHAAA,sBAAsB,OAAA;AAEvB,qDAA8D;AAArD,kHAAA,oBAAoB,OAAA;AAC7B,6CAAsD;AAA7C,0GAAA,gBAAgB,OAAA;AACzB,uCAAgD;AAAvC,oGAAA,aAAa,OAAA;AACtB,qCAAgD;AAAvC,8GAAA,mBAAmB,OAAA"}
|
|
@@ -1,6 +1,27 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
1
2
|
import type { ListModelsFn } from '../types';
|
|
2
3
|
export declare function shouldIncludeGoogleModel(model: {
|
|
3
4
|
name: string;
|
|
4
5
|
supportedGenerationMethods?: unknown;
|
|
5
6
|
}): boolean;
|
|
6
7
|
export declare const listGoogleModels: ListModelsFn;
|
|
8
|
+
declare const googleModelSchema: z.ZodObject<{
|
|
9
|
+
name: z.ZodString;
|
|
10
|
+
displayName: z.ZodCatch<z.ZodOptional<z.ZodString>>;
|
|
11
|
+
priority: z.ZodCatch<z.ZodOptional<z.ZodNumber>>;
|
|
12
|
+
supportedGenerationMethods: z.ZodCatch<z.ZodOptional<z.ZodArray<z.ZodString, "many">>>;
|
|
13
|
+
}, "strip", z.ZodTypeAny, {
|
|
14
|
+
name: string;
|
|
15
|
+
displayName?: string | undefined;
|
|
16
|
+
priority?: number | undefined;
|
|
17
|
+
supportedGenerationMethods?: string[] | undefined;
|
|
18
|
+
}, {
|
|
19
|
+
name: string;
|
|
20
|
+
displayName?: unknown;
|
|
21
|
+
priority?: unknown;
|
|
22
|
+
supportedGenerationMethods?: unknown;
|
|
23
|
+
}>;
|
|
24
|
+
export type GoogleModel = z.infer<typeof googleModelSchema>;
|
|
25
|
+
export declare function parseGoogleModels(data: unknown): GoogleModel[];
|
|
26
|
+
export declare function getGoogleModelLabel(modelId: string, displayName?: string): string;
|
|
27
|
+
export {};
|
|
@@ -2,6 +2,10 @@
|
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.listGoogleModels = void 0;
|
|
4
4
|
exports.shouldIncludeGoogleModel = shouldIncludeGoogleModel;
|
|
5
|
+
exports.parseGoogleModels = parseGoogleModels;
|
|
6
|
+
exports.getGoogleModelLabel = getGoogleModelLabel;
|
|
7
|
+
const api_types_1 = require("@n8n/api-types");
|
|
8
|
+
const zod_1 = require("zod");
|
|
5
9
|
const request_1 = require("../request");
|
|
6
10
|
function shouldIncludeGoogleModel(model) {
|
|
7
11
|
const methods = model.supportedGenerationMethods;
|
|
@@ -13,15 +17,35 @@ function shouldIncludeGoogleModel(model) {
|
|
|
13
17
|
}
|
|
14
18
|
const listGoogleModels = async (options) => {
|
|
15
19
|
const base = (0, request_1.baseUrl)(options, 'https://generativelanguage.googleapis.com');
|
|
16
|
-
const data =
|
|
17
|
-
|
|
18
|
-
.filter(
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
})
|
|
23
|
-
|
|
20
|
+
const data = await (0, request_1.getJson)(`${base}/v1beta/models`, { 'x-goog-api-key': options.apiKey }, options, 'google');
|
|
21
|
+
const models = parseGoogleModels(data)
|
|
22
|
+
.filter(shouldIncludeGoogleModel)
|
|
23
|
+
.map((model) => ({
|
|
24
|
+
id: model.name,
|
|
25
|
+
name: getGoogleModelLabel(model.name, model.displayName),
|
|
26
|
+
...(model.priority !== undefined ? { priority: model.priority } : {}),
|
|
27
|
+
}))
|
|
24
28
|
.sort(request_1.byName);
|
|
29
|
+
return (0, api_types_1.pinPrioritizedModels)(models, (model) => model.priority);
|
|
25
30
|
};
|
|
26
31
|
exports.listGoogleModels = listGoogleModels;
|
|
32
|
+
const googleModelSchema = zod_1.z.object({
|
|
33
|
+
name: zod_1.z.string(),
|
|
34
|
+
displayName: zod_1.z.string().optional().catch(undefined),
|
|
35
|
+
priority: zod_1.z.number().finite().optional().catch(undefined),
|
|
36
|
+
supportedGenerationMethods: zod_1.z.array(zod_1.z.string()).optional().catch(undefined),
|
|
37
|
+
});
|
|
38
|
+
const googleModelsResponseSchema = zod_1.z.object({
|
|
39
|
+
models: zod_1.z.array(zod_1.z.unknown()).optional().catch(undefined),
|
|
40
|
+
});
|
|
41
|
+
function parseGoogleModels(data) {
|
|
42
|
+
const response = googleModelsResponseSchema.safeParse(data);
|
|
43
|
+
return (response.success ? (response.data.models ?? []) : []).flatMap((item) => {
|
|
44
|
+
const model = googleModelSchema.safeParse(item);
|
|
45
|
+
return model.success ? [model.data] : [];
|
|
46
|
+
});
|
|
47
|
+
}
|
|
48
|
+
function getGoogleModelLabel(modelId, displayName) {
|
|
49
|
+
return displayName?.trim() || modelId;
|
|
50
|
+
}
|
|
27
51
|
//# sourceMappingURL=google.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"google.js","sourceRoot":"","sources":["../../../../src/model-discovery/providers/google.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"google.js","sourceRoot":"","sources":["../../../../src/model-discovery/providers/google.ts"],"names":[],"mappings":";;;;;;AAAA,8CAAsD;AACtD,6BAAwB;AAExB,wCAAsD;AAYtD,kCAAyC,KAGxC;IACA,MAAM,OAAO,GAAG,KAAK,CAAC,0BAA0B,CAAC;IACjD,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;IACnF,OAAO,CACN,YAAY;QACZ,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QAC7B,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAC3B,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CACjC,CAAC;AACH,CAAC;AASM,MAAM,gBAAgB,GAAiB,KAAK,EAAE,OAAO,EAAE,EAAE;IAC/D,MAAM,IAAI,GAAG,IAAA,iBAAO,EAAC,OAAO,EAAE,2CAA2C,CAAC,CAAC;IAC3E,MAAM,IAAI,GAAG,MAAM,IAAA,iBAAO,EACzB,GAAG,IAAI,gBAAgB,EACvB,EAAE,gBAAgB,EAAE,OAAO,CAAC,MAAM,EAAE,EACpC,OAAO,EACP,QAAQ,CACR,CAAC;IAEF,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,CAAC;SACpC,MAAM,CAAC,wBAAwB,CAAC;SAChC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAChB,EAAE,EAAE,KAAK,CAAC,IAAI;QACd,IAAI,EAAE,mBAAmB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,CAAC;QACxD,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACrE,CAAC,CAAC;SACF,IAAI,CAAC,gBAAM,CAAC,CAAC;IACf,OAAO,IAAA,gCAAoB,EAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;AAChE,CAAC,CAAC;AAlBW,QAAA,gBAAgB,GAAhB,gBAAgB,CAkB3B;AAGF,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IAClC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;IAChB,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;IACnD,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;IACzD,0BAA0B,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;CAC3E,CAAC,CAAC;AAEH,MAAM,0BAA0B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;CACxD,CAAC,CAAC;AAKH,2BAAkC,IAAa;IAC9C,MAAM,QAAQ,GAAG,0BAA0B,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC5D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC9E,MAAM,KAAK,GAAG,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAChD,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;AACJ,CAAC;AAGD,6BAAoC,OAAe,EAAE,WAAoB;IACxE,OAAO,WAAW,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC;AACvC,CAAC"}
|