promptfoo 0.92.2 → 0.93.0
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/package.json +12 -10
- package/dist/src/app/assets/index-CQmOdU-e.js +783 -0
- package/dist/src/app/assets/{index-DdvQdOby.css → index-V4sLv2Tk.css} +1 -1
- package/dist/src/app/assets/{index.es-ihzvEu35.js → index.es-tEEAGvDQ.js} +1 -1
- package/dist/src/app/assets/{purify.es-DGIRlouP.js → purify.es-a-CayzAK.js} +2 -2
- package/dist/src/app/assets/{sync-BosjlpGJ.js → sync-BSL909CQ.js} +1 -1
- package/dist/src/app/index.html +2 -2
- package/dist/src/assertions.js +1 -1
- package/dist/src/assertions.js.map +1 -1
- package/dist/src/commands/eval.d.ts.map +1 -1
- package/dist/src/commands/eval.js +5 -3
- package/dist/src/commands/eval.js.map +1 -1
- package/dist/src/commands/init.d.ts +5 -0
- package/dist/src/commands/init.d.ts.map +1 -1
- package/dist/src/commands/init.js +149 -6
- package/dist/src/commands/init.js.map +1 -1
- package/dist/src/database/index.d.ts.map +1 -1
- package/dist/src/database/index.js +9 -1
- package/dist/src/database/index.js.map +1 -1
- package/dist/src/database/tables.d.ts +1 -0
- package/dist/src/database/tables.d.ts.map +1 -1
- package/dist/src/envars.d.ts +2 -0
- package/dist/src/envars.d.ts.map +1 -1
- package/dist/src/envars.js.map +1 -1
- package/dist/src/evaluator.d.ts.map +1 -1
- package/dist/src/evaluator.js +3 -1
- package/dist/src/evaluator.js.map +1 -1
- package/dist/src/evaluatorHelpers.d.ts.map +1 -1
- package/dist/src/evaluatorHelpers.js +1 -0
- package/dist/src/evaluatorHelpers.js.map +1 -1
- package/dist/src/index.d.ts +1 -0
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/matchers.d.ts.map +1 -1
- package/dist/src/matchers.js +15 -36
- package/dist/src/matchers.js.map +1 -1
- package/dist/src/models/provider.d.ts +3 -1
- package/dist/src/models/provider.d.ts.map +1 -1
- package/dist/src/models/provider.js +22 -17
- package/dist/src/models/provider.js.map +1 -1
- package/dist/src/prompts/grading.js +2 -2
- package/dist/src/providers/googleShared.d.ts +2 -0
- package/dist/src/providers/googleShared.d.ts.map +1 -0
- package/dist/src/providers/googleShared.js +46 -0
- package/dist/src/providers/googleShared.js.map +1 -0
- package/dist/src/providers/packageParser.d.ts +3 -0
- package/dist/src/providers/packageParser.d.ts.map +1 -0
- package/dist/src/providers/packageParser.js +47 -0
- package/dist/src/providers/packageParser.js.map +1 -0
- package/dist/src/providers/palm.d.ts +0 -1
- package/dist/src/providers/palm.d.ts.map +1 -1
- package/dist/src/providers/palm.js +3 -3
- package/dist/src/providers/palm.js.map +1 -1
- package/dist/src/providers/vertex.d.ts +0 -1
- package/dist/src/providers/vertex.d.ts.map +1 -1
- package/dist/src/providers/vertex.js +2 -34
- package/dist/src/providers/vertex.js.map +1 -1
- package/dist/src/providers/watsonx.d.ts +20 -0
- package/dist/src/providers/watsonx.d.ts.map +1 -0
- package/dist/src/providers/watsonx.js +192 -0
- package/dist/src/providers/watsonx.js.map +1 -0
- package/dist/src/providers.d.ts +2 -0
- package/dist/src/providers.d.ts.map +1 -1
- package/dist/src/providers.js +11 -0
- package/dist/src/providers.js.map +1 -1
- package/dist/src/redteam/commands/generate.d.ts.map +1 -1
- package/dist/src/redteam/commands/generate.js +39 -8
- package/dist/src/redteam/commands/generate.js.map +1 -1
- package/dist/src/redteam/commands/run.d.ts.map +1 -1
- package/dist/src/redteam/commands/run.js +13 -35
- package/dist/src/redteam/commands/run.js.map +1 -1
- package/dist/src/redteam/index.d.ts.map +1 -1
- package/dist/src/redteam/index.js +17 -16
- package/dist/src/redteam/index.js.map +1 -1
- package/dist/src/redteam/plugins/custom.d.ts +15 -4
- package/dist/src/redteam/plugins/custom.d.ts.map +1 -1
- package/dist/src/redteam/plugins/custom.js +37 -3
- package/dist/src/redteam/plugins/custom.js.map +1 -1
- package/dist/src/redteam/plugins/policy.d.ts.map +1 -1
- package/dist/src/redteam/plugins/policy.js +0 -3
- package/dist/src/redteam/plugins/policy.js.map +1 -1
- package/dist/src/redteam/types.d.ts +1 -0
- package/dist/src/redteam/types.d.ts.map +1 -1
- package/dist/src/remoteGrading.d.ts +8 -0
- package/dist/src/remoteGrading.d.ts.map +1 -0
- package/dist/src/remoteGrading.js +35 -0
- package/dist/src/remoteGrading.js.map +1 -0
- package/dist/src/server/server.d.ts.map +1 -1
- package/dist/src/server/server.js +34 -0
- package/dist/src/server/server.js.map +1 -1
- package/dist/src/share.d.ts.map +1 -1
- package/dist/src/share.js +14 -5
- package/dist/src/share.js.map +1 -1
- package/dist/src/types/index.d.ts +8 -0
- package/dist/src/types/index.d.ts.map +1 -1
- package/dist/src/types/index.js +1 -0
- package/dist/src/types/index.js.map +1 -1
- package/dist/src/types/providers.d.ts +2 -0
- package/dist/src/types/providers.d.ts.map +1 -1
- package/dist/src/types/providers.js.map +1 -1
- package/dist/src/util/config/load.d.ts.map +1 -1
- package/dist/src/util/config/load.js +7 -5
- package/dist/src/util/config/load.js.map +1 -1
- package/dist/test/assertions.test.js +14 -14
- package/dist/test/assertions.test.js.map +1 -1
- package/dist/test/commands/init.test.d.ts +2 -0
- package/dist/test/commands/init.test.d.ts.map +1 -0
- package/dist/test/commands/init.test.js +159 -0
- package/dist/test/commands/init.test.js.map +1 -0
- package/dist/test/factories/evalFactory.d.ts +1 -0
- package/dist/test/factories/evalFactory.d.ts.map +1 -1
- package/dist/test/matchers.test.js +75 -0
- package/dist/test/matchers.test.js.map +1 -1
- package/dist/test/providers.packageParser.test.d.ts +2 -0
- package/dist/test/providers.packageParser.test.d.ts.map +1 -0
- package/dist/test/providers.packageParser.test.js +96 -0
- package/dist/test/providers.packageParser.test.js.map +1 -0
- package/dist/test/providers.watsonx.test.d.ts +2 -0
- package/dist/test/providers.watsonx.test.d.ts.map +1 -0
- package/dist/test/providers.watsonx.test.js +211 -0
- package/dist/test/providers.watsonx.test.js.map +1 -0
- package/dist/test/redteam/plugins/custom.test.d.ts +2 -0
- package/dist/test/redteam/plugins/custom.test.d.ts.map +1 -0
- package/dist/test/redteam/plugins/custom.test.js +87 -0
- package/dist/test/redteam/plugins/custom.test.js.map +1 -0
- package/dist/test/util.config.load.test.ts.js +40 -0
- package/dist/test/util.config.load.test.ts.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +12 -10
- package/dist/src/app/assets/index-BpjzEMiv.js +0 -783
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.CHAT_MODELS = void 0;
|
|
4
|
+
exports.CHAT_MODELS = [
|
|
5
|
+
'aqa',
|
|
6
|
+
'chat-bison',
|
|
7
|
+
'chat-bison-32k',
|
|
8
|
+
'chat-bison-32k@001',
|
|
9
|
+
'chat-bison-32k@002',
|
|
10
|
+
'chat-bison@001',
|
|
11
|
+
'chat-bison@002',
|
|
12
|
+
'codechat-bison',
|
|
13
|
+
'codechat-bison-32k',
|
|
14
|
+
'codechat-bison-32k@001',
|
|
15
|
+
'codechat-bison-32k@002',
|
|
16
|
+
'codechat-bison@001',
|
|
17
|
+
'codechat-bison@002',
|
|
18
|
+
'gemini-1.0-pro',
|
|
19
|
+
'gemini-1.0-pro-001',
|
|
20
|
+
'gemini-1.0-pro-002',
|
|
21
|
+
'gemini-1.0-pro-vision',
|
|
22
|
+
'gemini-1.0-pro-vision-001',
|
|
23
|
+
'gemini-1.0-pro-vision-001',
|
|
24
|
+
'gemini-1.5-flash',
|
|
25
|
+
'gemini-1.5-flash-001',
|
|
26
|
+
'gemini-1.5-flash-002',
|
|
27
|
+
'gemini-1.5-flash-8b-exp-0924',
|
|
28
|
+
'gemini-1.5-flash-8b-exp-0827',
|
|
29
|
+
'gemini-1.5-flash-exp-0827',
|
|
30
|
+
'gemini-1.5-flash-preview-0514',
|
|
31
|
+
'gemini-1.5-flash-latest',
|
|
32
|
+
'gemini-1.5-flash-8b-latest',
|
|
33
|
+
'gemini-1.5-flash-8b',
|
|
34
|
+
'gemini-1.5-flash-8b-001',
|
|
35
|
+
'gemini-1.5-pro',
|
|
36
|
+
'gemini-1.5-pro-001',
|
|
37
|
+
'gemini-1.5-pro-002',
|
|
38
|
+
'gemini-1.5-pro-latest',
|
|
39
|
+
'gemini-1.5-pro-preview-0409',
|
|
40
|
+
'gemini-1.5-pro-preview-0514',
|
|
41
|
+
'gemini-1.5-pro-exp-0827',
|
|
42
|
+
'gemini-pro',
|
|
43
|
+
'gemini-pro-vision',
|
|
44
|
+
'gemini-ultra',
|
|
45
|
+
];
|
|
46
|
+
//# sourceMappingURL=googleShared.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"googleShared.js","sourceRoot":"","sources":["../../../src/providers/googleShared.ts"],"names":[],"mappings":";;;AAAa,QAAA,WAAW,GAAG;IACzB,KAAK;IACL,YAAY;IACZ,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,oBAAoB;IACpB,wBAAwB;IACxB,wBAAwB;IACxB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,uBAAuB;IACvB,2BAA2B;IAC3B,2BAA2B;IAC3B,kBAAkB;IAClB,sBAAsB;IACtB,sBAAsB;IACtB,8BAA8B;IAC9B,8BAA8B;IAC9B,2BAA2B;IAC3B,+BAA+B;IAC/B,yBAAyB;IACzB,4BAA4B;IAC5B,qBAAqB;IACrB,yBAAyB;IACzB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,uBAAuB;IACvB,6BAA6B;IAC7B,6BAA6B;IAC7B,yBAAyB;IACzB,YAAY;IACZ,mBAAmB;IACnB,cAAc;CACf,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"packageParser.d.ts","sourceRoot":"","sources":["../../../src/providers/packageParser.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAQ7D,wBAAsB,oBAAoB,CACxC,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,WAAW,CAAC,CAyCtB"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.parsePackageProvider = parsePackageProvider;
|
|
7
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
8
|
+
const dedent_1 = __importDefault(require("dedent"));
|
|
9
|
+
const fs_1 = __importDefault(require("fs"));
|
|
10
|
+
const path_1 = __importDefault(require("path"));
|
|
11
|
+
const esm_1 = require("../esm");
|
|
12
|
+
const logger_1 = __importDefault(require("../logger"));
|
|
13
|
+
function getValue(obj, path) {
|
|
14
|
+
return path.split('.').reduce((acc, key) => {
|
|
15
|
+
return acc && acc[key] !== undefined ? acc[key] : undefined;
|
|
16
|
+
}, obj);
|
|
17
|
+
}
|
|
18
|
+
async function parsePackageProvider(providerPath, basePath, options) {
|
|
19
|
+
const [, packageName, providerName] = providerPath.split(':');
|
|
20
|
+
if (!packageName || !providerName) {
|
|
21
|
+
throw new Error(`Invalid package provider format: ${providerPath}. Expected format: package:packageName:providerName`);
|
|
22
|
+
}
|
|
23
|
+
const modulePath = path_1.default.join(basePath, 'node_modules', packageName);
|
|
24
|
+
const pkgPath = path_1.default.join(modulePath, 'package.json');
|
|
25
|
+
if (!fs_1.default.existsSync(pkgPath)) {
|
|
26
|
+
throw new Error(`Package not found: ${packageName}. Make sure it's installed in ${basePath}`);
|
|
27
|
+
}
|
|
28
|
+
const pkg = JSON.parse(fs_1.default.readFileSync(pkgPath, 'utf8'));
|
|
29
|
+
if (!pkg.main) {
|
|
30
|
+
logger_1.default.error((0, dedent_1.default) `
|
|
31
|
+
Could not determine main entry in package.json: ${chalk_1.default.bold(pkgPath)}.
|
|
32
|
+
|
|
33
|
+
${chalk_1.default.white((0, dedent_1.default) `Please check your package.json and ensure the property 'main' is correctly specified.`)}
|
|
34
|
+
`);
|
|
35
|
+
process.exit(1);
|
|
36
|
+
}
|
|
37
|
+
const module = await (0, esm_1.importModule)(path_1.default.join(modulePath, pkg.main));
|
|
38
|
+
const Provider = getValue(module, providerName);
|
|
39
|
+
if (!Provider) {
|
|
40
|
+
logger_1.default.error((0, dedent_1.default) `
|
|
41
|
+
Could not find provider: ${chalk_1.default.bold(providerName)} in module: ${chalk_1.default.bold(modulePath)}.
|
|
42
|
+
`);
|
|
43
|
+
process.exit(1);
|
|
44
|
+
}
|
|
45
|
+
return new Provider(options);
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=packageParser.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"packageParser.js","sourceRoot":"","sources":["../../../src/providers/packageParser.ts"],"names":[],"mappings":";;;;;AAcA,oDA6CC;AA3DD,kDAA0B;AAC1B,oDAA4B;AAC5B,4CAAoB;AACpB,gDAAwB;AACxB,gCAAsC;AACtC,uDAA+B;AAG/B,SAAS,QAAQ,CAAqC,GAAM,EAAE,IAAO;IACnE,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,GAAQ,EAAE,GAAW,EAAE,EAAE;QACtD,OAAO,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9D,CAAC,EAAE,GAAG,CAAC,CAAC;AACV,CAAC;AAEM,KAAK,UAAU,oBAAoB,CACxC,YAAoB,EACpB,QAAgB,EAChB,OAAwB;IAExB,MAAM,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAE9D,IAAI,CAAC,WAAW,IAAI,CAAC,YAAY,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,oCAAoC,YAAY,qDAAqD,CACtG,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,cAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;IAEtD,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,sBAAsB,WAAW,iCAAiC,QAAQ,EAAE,CAAC,CAAC;IAChG,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAGtD,CAAC;IAEF,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QACd,gBAAM,CAAC,KAAK,CAAC,IAAA,gBAAM,EAAA;wDACiC,eAAK,CAAC,IAAI,CAAC,OAAO,CAAC;;QAEnE,eAAK,CAAC,KAAK,CAAC,IAAA,gBAAM,EAAA,uFAAuF,CAAC;KAC7G,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,IAAA,kBAAY,EAAC,cAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IACnE,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAEhD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,gBAAM,CAAC,KAAK,CAAC,IAAA,gBAAM,EAAA;iCACU,eAAK,CAAC,IAAI,CAAC,YAAY,CAAC,eAAe,eAAK,CAAC,IAAI,CAAC,UAAU,CAAC;KACzF,CAAC,CAAC;QACH,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC;AAC/B,CAAC"}
|
|
@@ -30,7 +30,6 @@ declare class PalmGenericProvider implements ApiProvider {
|
|
|
30
30
|
callApi(prompt: string): Promise<ProviderResponse>;
|
|
31
31
|
}
|
|
32
32
|
export declare class PalmChatProvider extends PalmGenericProvider {
|
|
33
|
-
static CHAT_MODELS: string[];
|
|
34
33
|
constructor(modelName: string, options?: {
|
|
35
34
|
config?: PalmCompletionOptions;
|
|
36
35
|
id?: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"palm.d.ts","sourceRoot":"","sources":["../../../src/providers/palm.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,UAAU,CAAC;
|
|
1
|
+
{"version":3,"file":"palm.d.ts","sourceRoot":"","sources":["../../../src/providers/palm.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,UAAU,CAAC;AAOlG,UAAU,qBAAqB;IAC7B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,cAAc,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC7D,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACvC,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,cAAM,mBAAoB,YAAW,WAAW;IAC9C,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,qBAAqB,CAAC;IAC9B,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,qBAAqB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASnF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,UAAU,IAAI,MAAM,GAAG,SAAS;IAWhC,SAAS,IAAI,MAAM,GAAG,SAAS;IAWzB,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAGzD;AAED,qBAAa,gBAAiB,SAAQ,mBAAmB;gBAErD,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,qBAAqB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAQ7E,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAoElF,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAqF5F"}
|
|
@@ -8,6 +8,7 @@ const cache_1 = require("../cache");
|
|
|
8
8
|
const envars_1 = require("../envars");
|
|
9
9
|
const logger_1 = __importDefault(require("../logger"));
|
|
10
10
|
const util_1 = require("../util");
|
|
11
|
+
const googleShared_1 = require("./googleShared");
|
|
11
12
|
const shared_1 = require("./shared");
|
|
12
13
|
const DEFAULT_API_HOST = 'generativelanguage.googleapis.com';
|
|
13
14
|
class PalmGenericProvider {
|
|
@@ -46,8 +47,8 @@ class PalmGenericProvider {
|
|
|
46
47
|
}
|
|
47
48
|
class PalmChatProvider extends PalmGenericProvider {
|
|
48
49
|
constructor(modelName, options = {}) {
|
|
49
|
-
if (!
|
|
50
|
-
logger_1.default.
|
|
50
|
+
if (!googleShared_1.CHAT_MODELS.includes(modelName)) {
|
|
51
|
+
logger_1.default.debug(`Using unknown Google chat model: ${modelName}`);
|
|
51
52
|
}
|
|
52
53
|
super(modelName, options);
|
|
53
54
|
}
|
|
@@ -167,5 +168,4 @@ class PalmChatProvider extends PalmGenericProvider {
|
|
|
167
168
|
}
|
|
168
169
|
}
|
|
169
170
|
exports.PalmChatProvider = PalmChatProvider;
|
|
170
|
-
PalmChatProvider.CHAT_MODELS = ['chat-bison-001', 'gemini-pro', 'gemini-pro-vision'];
|
|
171
171
|
//# sourceMappingURL=palm.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"palm.js","sourceRoot":"","sources":["../../../src/providers/palm.ts"],"names":[],"mappings":";;;;;;AAAA,oCAA0C;AAC1C,sCAAyC;AACzC,uDAA+B;AAE/B,kCAAwE;AACxE,qCAA+D;AAE/D,MAAM,gBAAgB,GAAG,mCAAmC,CAAC;AAe7D,MAAM,mBAAmB;IAMvB,YACE,SAAiB,EACjB,UAA+E,EAAE;QAEjF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,QAAQ,IAAI,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAED,QAAQ;QACN,OAAO,8BAA8B,IAAI,CAAC,SAAS,GAAG,CAAC;IACzD,CAAC;IAED,UAAU;QACR,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,OAAO;YACnB,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,IAAI,CAAC,GAAG,EAAE,aAAa;YACvB,IAAA,qBAAY,EAAC,iBAAiB,CAAC;YAC/B,IAAA,qBAAY,EAAC,eAAe,CAAC;YAC7B,gBAAgB,CACjB,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM;YAClB,IAAI,CAAC,GAAG,EAAE,cAAc;YACxB,IAAI,CAAC,GAAG,EAAE,YAAY;YACtB,IAAA,qBAAY,EAAC,gBAAgB,CAAC;YAC9B,IAAA,qBAAY,EAAC,cAAc,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,gBAAiB,SAAQ,mBAAmB;
|
|
1
|
+
{"version":3,"file":"palm.js","sourceRoot":"","sources":["../../../src/providers/palm.ts"],"names":[],"mappings":";;;;;;AAAA,oCAA0C;AAC1C,sCAAyC;AACzC,uDAA+B;AAE/B,kCAAwE;AACxE,iDAA6C;AAC7C,qCAA+D;AAE/D,MAAM,gBAAgB,GAAG,mCAAmC,CAAC;AAe7D,MAAM,mBAAmB;IAMvB,YACE,SAAiB,EACjB,UAA+E,EAAE;QAEjF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,QAAQ,IAAI,CAAC,SAAS,EAAE,CAAC;IAClC,CAAC;IAED,QAAQ;QACN,OAAO,8BAA8B,IAAI,CAAC,SAAS,GAAG,CAAC;IACzD,CAAC;IAED,UAAU;QACR,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,OAAO;YACnB,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,IAAI,CAAC,GAAG,EAAE,aAAa;YACvB,IAAA,qBAAY,EAAC,iBAAiB,CAAC;YAC/B,IAAA,qBAAY,EAAC,eAAe,CAAC;YAC7B,gBAAgB,CACjB,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM;YAClB,IAAI,CAAC,GAAG,EAAE,cAAc;YACxB,IAAI,CAAC,GAAG,EAAE,YAAY;YACtB,IAAA,qBAAY,EAAC,gBAAgB,CAAC;YAC9B,IAAA,qBAAY,EAAC,cAAc,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,gBAAiB,SAAQ,mBAAmB;IACvD,YACE,SAAiB,EACjB,UAA+E,EAAE;QAEjF,IAAI,CAAC,0BAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,gBAAM,CAAC,KAAK,CAAC,oCAAoC,SAAS,EAAE,CAAC,CAAC;QAChE,CAAC;QACD,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,gHAAgH,CACjH,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACrD,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC1C,CAAC;QAED,mEAAmE;QACnE,+DAA+D;QAC/D,MAAM,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAChE,MAAM,IAAI,GAAG;YACX,MAAM,EAAE,EAAE,QAAQ,EAAE;YACpB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YAEtB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;YAC1C,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACxC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;SAC7C,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE5D,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACH,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EAC/B,WAAW,IAAI,CAAC,UAAU,EAAE,mBAC1B,IAAI,CAAC,SACP,wBAAwB,IAAI,CAAC,SAAS,EAAE,EAAE,EAC1C;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAAmB,CAAC,CAAC;QACxB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE/D,IAAI,CAAC,IAAI,EAAE,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,OAAO;gBACL,KAAK,EAAE,+CAA+C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC7E,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1C,OAAO;gBACL,MAAM;aACP,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aACrE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,OAA8B;QAC7D,MAAM,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAwB;YAChC,QAAQ;YACR,gBAAgB,EAAE;gBAChB,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;gBACxC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAChC;YACD,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;SAC3C,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC/B,+DAA+D;YAC/D,IAAI,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,CAAC;gBAC1C,MAAM,IAAI,KAAK,CACb,+EAA+E,CAChF,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,IAAA,gCAAyB,EACtC,IAAA,yBAAkB,EAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,OAAO,EAAE,IAAI,CAAC,CAC9D,CAAC;YAEF,IAAI,CAAC,gBAAgB,CAAC,eAAe,GAAG,MAAM,CAAC;YAC/C,IAAI,CAAC,gBAAgB,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAChE,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE5D,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACH,sDAAsD;YACtD,kDAAkD;YAClD,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,IAAA,sBAAc,EAC/B,WAAW,IAAI,CAAC,UAAU,EAAE,kBAC1B,IAAI,CAAC,SACP,wBAAwB,IAAI,CAAC,SAAS,EAAE,EAAE,EAC1C;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,EACD,2BAAkB,CACnB,CAQA,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE/D,IAAI,CAAC,IAAI,EAAE,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,OAAO;gBACL,KAAK,EAAE,+CAA+C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC7E,CAAC;QACJ,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAExE,IAAI,CAAC;YACH,OAAO;gBACL,MAAM,EAAE,KAAK;aACd,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aACrE,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AApKD,4CAoKC"}
|
|
@@ -89,7 +89,6 @@ declare class VertexGenericProvider implements ApiProvider {
|
|
|
89
89
|
callApi(prompt: string): Promise<ProviderResponse>;
|
|
90
90
|
}
|
|
91
91
|
export declare class VertexChatProvider extends VertexGenericProvider {
|
|
92
|
-
static CHAT_MODELS: string[];
|
|
93
92
|
constructor(modelName: string, options?: {
|
|
94
93
|
config?: VertexCompletionOptions;
|
|
95
94
|
id?: string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"vertex.d.ts","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,oBAAoB,EACpB,WAAW,EACX,oBAAoB,EACpB,YAAY,EACZ,gBAAgB,EAChB,yBAAyB,EAE1B,MAAM,UAAU,CAAC;AAWlB,UAAU,IAAI;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,YAAY;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;CAC/B;AAED,UAAU,gBAAgB;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;CAClC;AAED,UAAU,QAAQ;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,IAAI;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,IAAI,CAAC;IAClB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED,UAAU,OAAO;IACf,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC/C,cAAc,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC7D,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,gBAAgB,CAAC,EAAE;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAC/C,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IAEF,UAAU,CAAC,EAAE;QACX,qBAAqB,CAAC,EAAE;YACtB,IAAI,CAAC,EAAE,kBAAkB,GAAG,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;YACpD,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;SACjC,CAAC;KACH,CAAC;IAEF,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAID,cAAM,qBAAsB,YAAW,WAAW;IAChD,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,uBAAuB,CAAC;IAChC,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASrF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,UAAU,IAAI,MAAM,GAAG,SAAS;IAS1B,YAAY;IASlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,SAAS,IAAI,MAAM;IASnB,YAAY,IAAI,MAAM,GAAG,SAAS;IAU5B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAGzD;AAED,qBAAa,kBAAmB,SAAQ,qBAAqB;
|
|
1
|
+
{"version":3,"file":"vertex.d.ts","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,oBAAoB,EACpB,WAAW,EACX,oBAAoB,EACpB,YAAY,EACZ,gBAAgB,EAChB,yBAAyB,EAE1B,MAAM,UAAU,CAAC;AAWlB,UAAU,IAAI;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,YAAY;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;CAC/B;AAED,UAAU,gBAAgB;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;CAClC;AAED,UAAU,QAAQ;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,IAAI;IACZ,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,IAAI,CAAC;IAClB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;IACpC,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED,UAAU,OAAO;IACf,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IAGnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC/C,cAAc,CAAC,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC7D,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,gBAAgB,CAAC,EAAE;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAC/C,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;QACxB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IAEF,UAAU,CAAC,EAAE;QACX,qBAAqB,CAAC,EAAE;YACtB,IAAI,CAAC,EAAE,kBAAkB,GAAG,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;YACpD,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;SACjC,CAAC;KACH,CAAC;IAEF,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAID,cAAM,qBAAsB,YAAW,WAAW;IAChD,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,EAAE,uBAAuB,CAAC;IAChC,GAAG,CAAC,EAAE,YAAY,CAAC;gBAGjB,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IASrF,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,UAAU,IAAI,MAAM,GAAG,SAAS;IAS1B,YAAY;IASlB,SAAS,IAAI,MAAM,GAAG,SAAS;IAI/B,SAAS,IAAI,MAAM;IASnB,YAAY,IAAI,MAAM,GAAG,SAAS;IAU5B,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CAGzD;AAED,qBAAa,kBAAmB,SAAQ,qBAAqB;gBAIzD,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,uBAAuB,CAAC;QAAC,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,YAAY,CAAA;KAAO;IAK/E,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAOlF,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAgLxF,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA2F9D;AAED,qBAAa,uBAAwB,YAAW,oBAAoB;IAClE,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,GAAG,CAAC;IACZ,GAAG,CAAC,EAAE,GAAG,CAAC;gBAEE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAE,GAAQ,EAAE,GAAG,CAAC,EAAE,GAAG;IAM1D,EAAE;IAIF,SAAS,IAAI,MAAM;IAIb,OAAO,IAAI,OAAO,CAAC,gBAAgB,CAAC;IAIpC,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,yBAAyB,CAAC;CA0C1E;AAED,eAAO,MAAM,sBAAsB,oBAA2C,CAAC;AAC/E,eAAO,MAAM,wBAAwB,yBAAoD,CAAC"}
|
|
@@ -60,6 +60,8 @@ class VertexGenericProvider {
|
|
|
60
60
|
}
|
|
61
61
|
}
|
|
62
62
|
class VertexChatProvider extends VertexGenericProvider {
|
|
63
|
+
// TODO(ian): Completion models
|
|
64
|
+
// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/model-versioning#gemini-model-versions
|
|
63
65
|
constructor(modelName, options = {}) {
|
|
64
66
|
super(modelName, options);
|
|
65
67
|
}
|
|
@@ -316,40 +318,6 @@ class VertexChatProvider extends VertexGenericProvider {
|
|
|
316
318
|
}
|
|
317
319
|
}
|
|
318
320
|
exports.VertexChatProvider = VertexChatProvider;
|
|
319
|
-
// TODO(ian): Completion models
|
|
320
|
-
// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/model-versioning#gemini-model-versions
|
|
321
|
-
VertexChatProvider.CHAT_MODELS = [
|
|
322
|
-
'aqa',
|
|
323
|
-
'chat-bison',
|
|
324
|
-
'chat-bison-32k',
|
|
325
|
-
'chat-bison-32k@001',
|
|
326
|
-
'chat-bison-32k@002',
|
|
327
|
-
'chat-bison@001',
|
|
328
|
-
'chat-bison@002',
|
|
329
|
-
'codechat-bison',
|
|
330
|
-
'codechat-bison-32k',
|
|
331
|
-
'codechat-bison-32k@001',
|
|
332
|
-
'codechat-bison-32k@002',
|
|
333
|
-
'codechat-bison@001',
|
|
334
|
-
'codechat-bison@002',
|
|
335
|
-
'gemini-1.0-pro',
|
|
336
|
-
'gemini-1.0-pro-001',
|
|
337
|
-
'gemini-1.0-pro-002',
|
|
338
|
-
'gemini-1.0-pro-vision',
|
|
339
|
-
'gemini-1.0-pro-vision-001',
|
|
340
|
-
'gemini-1.0-pro-vision-001',
|
|
341
|
-
'gemini-1.5-flash',
|
|
342
|
-
'gemini-1.5-flash-001',
|
|
343
|
-
'gemini-1.5-flash-preview-0514',
|
|
344
|
-
'gemini-1.5-pro',
|
|
345
|
-
'gemini-1.5-pro-001',
|
|
346
|
-
'gemini-1.5-pro-latest',
|
|
347
|
-
'gemini-1.5-pro-preview-0409',
|
|
348
|
-
'gemini-1.5-pro-preview-0514',
|
|
349
|
-
'gemini-pro',
|
|
350
|
-
'gemini-pro-vision',
|
|
351
|
-
'gemini-ultra',
|
|
352
|
-
];
|
|
353
321
|
class VertexEmbeddingProvider {
|
|
354
322
|
constructor(modelName, config = {}, env) {
|
|
355
323
|
this.modelName = modelName;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"vertex.js","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAyB;AACzB,oCAAoD;AACpD,2DAAmC;AACnC,sCAAyC;AACzC,uDAA+B;AAU/B,iDAAsD;AACtD,qCAA+D;AAE/D,6CAKsB;AAwEtB,MAAM,KAAK,GAAG,IAAA,cAAK,GAAE,CAAC;AAEtB,MAAM,qBAAqB;IAMzB,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC;IAED,QAAQ;QACN,OAAO,2BAA2B,IAAI,CAAC,SAAS,GAAG,CAAC;IACtD,CAAC;IAED,UAAU;QACR,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,OAAO;YACnB,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,IAAA,qBAAY,EAAC,iBAAiB,CAAC;YAC/B,GAAG,IAAI,CAAC,SAAS,EAAE,4BAA4B,CAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,CACL,CAAC,MAAM,IAAA,4BAAe,GAAE,CAAC,CAAC,SAAS;YACnC,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,iBAAiB;YAC3B,IAAA,qBAAY,EAAC,mBAAmB,CAAC,CAClC,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,cAAc,IAAI,IAAA,qBAAY,EAAC,gBAAgB,CAAC,CAAC;IAC1F,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM;YAClB,IAAI,CAAC,GAAG,EAAE,aAAa;YACvB,IAAA,qBAAY,EAAC,eAAe,CAAC;YAC7B,aAAa,CACd,CAAC;IACJ,CAAC;IAED,YAAY;QACV,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,gBAAgB;YAC1B,IAAA,qBAAY,EAAC,kBAAkB,CAAC;YAChC,QAAQ,CACT,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,kBAAmB,SAAQ,qBAAqB;IAoC3D,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,OAA8B;QAChE,0FAA0F;QAC1F,IAAI,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,MAAM;aACb;SACF,CAAC,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,GAAG,IAAA,sCAAyB,EAAC,QAAQ,CAAC,CAAC;QACnF,IAAI,OAAO,EAAE,CAAC;YACZ,gBAAM,CAAC,KAAK,CAAC,yCAAyC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClF,QAAQ,GAAG,eAAe,CAAC;QAC7B,CAAC;QAED,IAAI,iBAAsC,CAAC;QAC3C,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,cAAc;YACd,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;YACzD,IAAI,iBAAiB,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;gBACvC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;gBACrC,KAAK,MAAM,IAAI,IAAI,iBAAiB,CAAC,KAAK,EAAE,CAAC;oBAC3C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACd,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;oBAC7D,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,MAAM,IAAI,GAAG;YACX,QAAQ;YACR,gBAAgB,EAAE;gBAChB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAChC;YACD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACpD,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,2DAA2D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhG,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,iBAAiB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAEzD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;gBACvC,CAAC;gBACD,gBAAM,CAAC,KAAK,CAAC,8BAA8B,cAAc,EAAE,CAAC,CAAC;gBAC7D,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,4BAAe,GAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,wBAAwB,CAAC;YACzB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC;gBAC/B,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;gBACV,OAAO,EAAE,2BAAkB;aAC5B,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAyB,CAAC;YACrC,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,WAAW,GAAG,GAAU,CAAC;YAC/B,gBAAM,CAAC,KAAK,CACV,+BAA+B,MAAM,CAAC,WAAW,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,GAAG,CAC7F,CAAC;YACF,IACE,WAAW,CAAC,QAAQ;gBACpB,WAAW,CAAC,QAAQ,CAAC,IAAI;gBACzB,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC5B,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAClC,CAAC;gBACD,MAAM,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBACxD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;gBAC/B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;gBACrC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBACnC,OAAO;oBACL,KAAK,EAAE,0BAA0B,MAAM,UAAU,IAAI,iBAAiB,OAAO,EAAE;iBAChF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7D,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,IAA6B,CAAC;YACpD,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;YACtC,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO;oBACL,KAAK,EAAE,SAAS,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE;iBAC/C,CAAC;YACJ,CAAC;YACD,MAAM,gBAAgB,GAAG,IAA4B,CAAC;YACtD,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,KAAK,MAAM,KAAK,IAAI,gBAAgB,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;oBAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAClD,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;wBACnB,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC;oBACtB,CAAC;yBAAM,CAAC;wBACN,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBACjC,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,KAAK,QAAQ,EAAE,CAAC;oBAC9E,IAAI,kBAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;wBAC7B,uEAAuE;wBACvE,OAAO;4BACL,MAAM,EAAE,6CAA6C;yBACtD,CAAC;oBACJ,CAAC;oBACD,OAAO;wBACL,KAAK,EAAE,6CAA6C;qBACrD,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,gBAAgB,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,WAAW,EAAE,CAAC;gBACvE,IAAI,kBAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC7B,uEAAuE;oBACvE,OAAO;wBACL,MAAM,EAAE,+CAA+C,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE;qBAC5F,CAAC;gBACJ,CAAC;gBACD,OAAO;oBACL,KAAK,EAAE,+CAA+C,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE;iBAC3F,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;oBACL,KAAK,EAAE,gCAAgC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;iBAC9D,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,QAAQ,CAAC,aAAa,EAAE,eAAe,IAAI,CAAC;gBACnD,MAAM,EAAE,QAAQ,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC;gBACrD,UAAU,EAAE,QAAQ,CAAC,aAAa,EAAE,oBAAoB,IAAI,CAAC;aAC9D,CAAC;YACF,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE,KAAK;gBACb,MAAM;gBACN,UAAU;aACX,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtD,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,8BAA8B,MAAM,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC3F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,SAAS,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACxC;gBACE,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,MAAM;wBACd,OAAO,EAAE,MAAM;qBAChB;iBACF;aACF;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG;YACX,SAAS;YACT,UAAU,EAAE;gBACV,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBAC1C,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;aACvB;SACF,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,gBAAgB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAExD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;gBACvC,CAAC;gBACD,gBAAM,CAAC,KAAK,CAAC,8BAA8B,cAAc,EAAE,CAAC,CAAC;gBAC7D,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;QACH,CAAC;QAED,IAAI,IAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,4BAAe,GAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,UAAU,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAmB;gBACjD,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI;gBACV,OAAO,EAAE,2BAAkB;aAC5B,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE;aAChD,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnE,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,KAAK,EAAE,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;iBACzD,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAE3D,MAAM,QAAQ,GAAG;gBACf,MAAM;gBACN,MAAM,EAAE,KAAK;aACd,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtD,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aACrE,CAAC;QACJ,CAAC;IACH,CAAC;;AA5TH,gDA6TC;AA5TC,+BAA+B;AAC/B,qGAAqG;AAC9F,8BAAW,GAAG;IACnB,KAAK;IACL,YAAY;IACZ,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,gBAAgB;IAChB,gBAAgB;IAChB,oBAAoB;IACpB,wBAAwB;IACxB,wBAAwB;IACxB,oBAAoB;IACpB,oBAAoB;IACpB,gBAAgB;IAChB,oBAAoB;IACpB,oBAAoB;IACpB,uBAAuB;IACvB,2BAA2B;IAC3B,2BAA2B;IAC3B,kBAAkB;IAClB,sBAAsB;IACtB,+BAA+B;IAC/B,gBAAgB;IAChB,oBAAoB;IACpB,uBAAuB;IACvB,6BAA6B;IAC7B,6BAA6B;IAC7B,YAAY;IACZ,mBAAmB;IACnB,cAAc;CACf,CAAC;AA6RJ,MAAa,uBAAuB;IAKlC,YAAY,SAAiB,EAAE,SAAc,EAAE,EAAE,GAAS;QACxD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,aAAa,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAa;QAClC,qIAAqI;QACrI,MAAM,IAAI,GAAG;YACX,SAAS,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC/B,UAAU,EAAE;gBACV,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,KAAK;aAChD;SACF,CAAC;QAEF,IAAI,IAAS,CAAC;QACd,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,4BAAe,GAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,SAAS,EAAE,0CAA0C,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,6BACtH,IAAI,CAAC,SACP,UAAU,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAM;gBACpC,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,0BAA0B,GAAG,EAAE,CAAC,CAAC;YAC9C,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,mCAAmC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAExE,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;YACxD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC;YACzE,OAAO;gBACL,SAAS;gBACT,UAAU,EAAE;oBACV,KAAK,EAAE,UAAU;iBAClB;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,iDAAiD,GAAG,EAAE,CAAC,CAAC;YACrE,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AAjED,0DAiEC;AAEY,QAAA,sBAAsB,GAAG,IAAI,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;AAClE,QAAA,wBAAwB,GAAG,IAAI,uBAAuB,CAAC,oBAAoB,CAAC,CAAC"}
|
|
1
|
+
{"version":3,"file":"vertex.js","sourceRoot":"","sources":["../../../src/providers/vertex.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAyB;AACzB,oCAAoD;AACpD,2DAAmC;AACnC,sCAAyC;AACzC,uDAA+B;AAU/B,iDAAsD;AACtD,qCAA+D;AAE/D,6CAKsB;AAwEtB,MAAM,KAAK,GAAG,IAAA,cAAK,GAAE,CAAC;AAEtB,MAAM,qBAAqB;IAMzB,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACpC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;IACpC,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC;IAED,QAAQ;QACN,OAAO,2BAA2B,IAAI,CAAC,SAAS,GAAG,CAAC;IACtD,CAAC;IAED,UAAU;QACR,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,OAAO;YACnB,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,IAAA,qBAAY,EAAC,iBAAiB,CAAC;YAC/B,GAAG,IAAI,CAAC,SAAS,EAAE,4BAA4B,CAChD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,OAAO,CACL,CAAC,MAAM,IAAA,4BAAe,GAAE,CAAC,CAAC,SAAS;YACnC,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,iBAAiB;YAC3B,IAAA,qBAAY,EAAC,mBAAmB,CAAC,CAClC,CAAC;IACJ,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,cAAc,IAAI,IAAA,qBAAY,EAAC,gBAAgB,CAAC,CAAC;IAC1F,CAAC;IAED,SAAS;QACP,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,MAAM;YAClB,IAAI,CAAC,GAAG,EAAE,aAAa;YACvB,IAAA,qBAAY,EAAC,eAAe,CAAC;YAC7B,aAAa,CACd,CAAC;IACJ,CAAC;IAED,YAAY;QACV,OAAO,CACL,IAAI,CAAC,MAAM,CAAC,SAAS;YACrB,IAAI,CAAC,GAAG,EAAE,gBAAgB;YAC1B,IAAA,qBAAY,EAAC,kBAAkB,CAAC;YAChC,QAAQ,CACT,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAED,MAAa,kBAAmB,SAAQ,qBAAqB;IAC3D,+BAA+B;IAC/B,qGAAqG;IACrG,YACE,SAAiB,EACjB,UAAiF,EAAE;QAEnF,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,OAA8B;QAC1D,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACtC,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,OAA8B;QAChE,0FAA0F;QAC1F,IAAI,QAAQ,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACrC,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,IAAI,EAAE,MAAM;aACb;SACF,CAAC,CAAC;QACH,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,GAAG,IAAA,sCAAyB,EAAC,QAAQ,CAAC,CAAC;QACnF,IAAI,OAAO,EAAE,CAAC;YACZ,gBAAM,CAAC,KAAK,CAAC,yCAAyC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClF,QAAQ,GAAG,eAAe,CAAC;QAC7B,CAAC;QAED,IAAI,iBAAsC,CAAC;QAC3C,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,cAAc;YACd,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;YACzD,IAAI,iBAAiB,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;gBACvC,MAAM,QAAQ,GAAG,IAAA,6BAAiB,GAAE,CAAC;gBACrC,KAAK,MAAM,IAAI,IAAI,iBAAiB,CAAC,KAAK,EAAE,CAAC;oBAC3C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACd,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;oBAC7D,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,MAAM,IAAI,GAAG;YACX,QAAQ;YACR,gBAAgB,EAAE;gBAChB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAChC;YACD,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACrF,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACpD,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,2DAA2D,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEhG,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,iBAAiB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAEzD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;gBACvC,CAAC;gBACD,gBAAM,CAAC,KAAK,CAAC,8BAA8B,cAAc,EAAE,CAAC,CAAC;gBAC7D,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,4BAAe,GAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,wBAAwB,CAAC;YACzB,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC;gBAC/B,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;gBACV,OAAO,EAAE,2BAAkB;aAC5B,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAyB,CAAC;YACrC,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,WAAW,GAAG,GAAU,CAAC;YAC/B,gBAAM,CAAC,KAAK,CACV,+BAA+B,MAAM,CAAC,WAAW,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,GAAG,CAC7F,CAAC;YACF,IACE,WAAW,CAAC,QAAQ;gBACpB,WAAW,CAAC,QAAQ,CAAC,IAAI;gBACzB,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC5B,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,EAClC,CAAC;gBACD,MAAM,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;gBACxD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;gBAC/B,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;gBACrC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBACnC,OAAO;oBACL,KAAK,EAAE,0BAA0B,MAAM,UAAU,IAAI,iBAAiB,OAAO,EAAE;iBAChF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;aACxC,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7D,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,IAA6B,CAAC;YACpD,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;YACtC,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO;oBACL,KAAK,EAAE,SAAS,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE;iBAC/C,CAAC;YACJ,CAAC;YACD,MAAM,gBAAgB,GAAG,IAA4B,CAAC;YACtD,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,KAAK,MAAM,KAAK,IAAI,gBAAgB,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;oBAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAClD,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;wBACnB,MAAM,IAAI,IAAI,CAAC,IAAI,CAAC;oBACtB,CAAC;yBAAM,CAAC;wBACN,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBACjC,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,KAAK,QAAQ,EAAE,CAAC;oBAC9E,IAAI,kBAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;wBAC7B,uEAAuE;wBACvE,OAAO;4BACL,MAAM,EAAE,6CAA6C;yBACtD,CAAC;oBACJ,CAAC;oBACD,OAAO;wBACL,KAAK,EAAE,6CAA6C;qBACrD,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,gBAAgB,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,WAAW,EAAE,CAAC;gBACvE,IAAI,kBAAQ,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBAC7B,uEAAuE;oBACvE,OAAO;wBACL,MAAM,EAAE,+CAA+C,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE;qBAC5F,CAAC;gBACJ,CAAC;gBACD,OAAO;oBACL,KAAK,EAAE,+CAA+C,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE;iBAC3F,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;oBACL,KAAK,EAAE,gCAAgC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;iBAC9D,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,gBAAgB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC/D,MAAM,UAAU,GAAG;gBACjB,KAAK,EAAE,QAAQ,CAAC,aAAa,EAAE,eAAe,IAAI,CAAC;gBACnD,MAAM,EAAE,QAAQ,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC;gBACrD,UAAU,EAAE,QAAQ,CAAC,aAAa,EAAE,oBAAoB,IAAI,CAAC;aAC9D,CAAC;YACF,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE,KAAK;gBACb,MAAM;gBACN,UAAU;aACX,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtD,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,8BAA8B,MAAM,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aAC3F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,SAAS,GAAG,IAAA,wBAAe,EAAC,MAAM,EAAE;YACxC;gBACE,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,MAAM;wBACd,OAAO,EAAE,MAAM;qBAChB;iBACF;aACF;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG;YACX,SAAS;YACT,UAAU,EAAE;gBACV,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;gBAC5B,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBAC1C,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBACpC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;gBAC5C,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;aACvB;SACF,CAAC;QACF,gBAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAElE,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAQ,GAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,gBAAgB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;QAExD,IAAI,cAAc,CAAC;QACnB,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;YACrB,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,oBAAoB,GAAG,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAC,CAAC;gBAClE,MAAM,UAAU,GAAG,oBAAoB,CAAC,UAAwB,CAAC;gBACjE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC;gBACvC,CAAC;gBACD,gBAAM,CAAC,KAAK,CAAC,8BAA8B,cAAc,EAAE,CAAC,CAAC;gBAC7D,OAAO,EAAE,GAAG,oBAAoB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YACnD,CAAC;QACH,CAAC;QAED,IAAI,IAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,4BAAe,GAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,UAAU,EAAE,gBAAgB,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,eAAe,IAAI,CAAC,YAAY,EAAE,WAC/H,IAAI,CAAC,SACP,UAAU,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAmB;gBACjD,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI;gBACV,OAAO,EAAE,2BAAkB;aAC5B,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,mBAAmB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE;aAChD,CAAC;QACJ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,8BAA8B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnE,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,KAAK,EAAE,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;iBACzD,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAE3D,MAAM,QAAQ,GAAG;gBACf,MAAM;gBACN,MAAM,EAAE,KAAK;aACd,CAAC;YAEF,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;YACtD,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,KAAK,EAAE,uBAAuB,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;aACrE,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AA5RD,gDA4RC;AAED,MAAa,uBAAuB;IAKlC,YAAY,SAAiB,EAAE,SAAc,EAAE,EAAE,GAAS;QACxD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAED,EAAE;QACA,OAAO,UAAU,IAAI,CAAC,SAAS,EAAE,CAAC;IACpC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,aAAa,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,KAAa;QAClC,qIAAqI;QACrI,MAAM,IAAI,GAAG;YACX,SAAS,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC/B,UAAU,EAAE;gBACV,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,KAAK;aAChD;SACF,CAAC;QAEF,IAAI,IAAS,CAAC;QACd,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAA,4BAAe,GAAE,CAAC;YACtD,MAAM,GAAG,GAAG,WAAW,IAAI,CAAC,SAAS,EAAE,0CAA0C,SAAS,cAAc,IAAI,CAAC,SAAS,EAAE,6BACtH,IAAI,CAAC,SACP,UAAU,CAAC;YACX,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,OAAO,CAAM;gBACpC,GAAG;gBACH,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,IAAI,GAAG,GAAG,CAAC,IAAI,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,0BAA0B,GAAG,EAAE,CAAC,CAAC;YAC9C,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,gBAAM,CAAC,KAAK,CAAC,mCAAmC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAExE,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC;YACxD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC;YACzE,OAAO;gBACL,SAAS;gBACT,UAAU,EAAE;oBACV,KAAK,EAAE,UAAU;iBAClB;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,iDAAiD,GAAG,EAAE,CAAC,CAAC;YACrE,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AAjED,0DAiEC;AAEY,QAAA,sBAAsB,GAAG,IAAI,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;AAClE,QAAA,wBAAwB,GAAG,IAAI,uBAAuB,CAAC,oBAAoB,CAAC,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { WatsonXAI as WatsonXAIClient } from '@ibm-cloud/watsonx-ai';
|
|
2
|
+
import type { ApiProvider, EnvOverrides, ProviderResponse } from '../types';
|
|
3
|
+
import type { ProviderOptions } from '../types/providers';
|
|
4
|
+
export declare function generateConfigHash(config: any): string;
|
|
5
|
+
export declare class WatsonXProvider implements ApiProvider {
|
|
6
|
+
modelName: string;
|
|
7
|
+
options: ProviderOptions;
|
|
8
|
+
env?: EnvOverrides;
|
|
9
|
+
apiKey: string;
|
|
10
|
+
client: WatsonXAIClient;
|
|
11
|
+
constructor(modelName: string, options: ProviderOptions);
|
|
12
|
+
id(): string;
|
|
13
|
+
toString(): string;
|
|
14
|
+
getApiKey(): string;
|
|
15
|
+
getProjectId(): string;
|
|
16
|
+
getModelId(): string;
|
|
17
|
+
getClient(): WatsonXAIClient;
|
|
18
|
+
callApi(prompt: string): Promise<ProviderResponse>;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=watsonx.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"watsonx.d.ts","sourceRoot":"","sources":["../../../src/providers/watsonx.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,IAAI,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAS1E,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,gBAAgB,EAAc,MAAM,UAAU,CAAC;AACxF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAmF1D,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,GAAG,GAAG,MAAM,CAGtD;AAED,qBAAa,eAAgB,YAAW,WAAW;IACjD,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,eAAe,CAAC;IACzB,GAAG,CAAC,EAAE,YAAY,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,eAAe,CAAC;gBAEZ,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe;IAoBvD,EAAE,IAAI,MAAM;IAIZ,QAAQ,IAAI,MAAM;IAIlB,SAAS,IAAI,MAAM;IAgBnB,YAAY,IAAI,MAAM;IAgBtB,UAAU,IAAI,MAAM;IAgBpB,SAAS,IAAI,eAAe;IAItB,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA6DzD"}
|
|
@@ -0,0 +1,192 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.WatsonXProvider = void 0;
|
|
7
|
+
exports.generateConfigHash = generateConfigHash;
|
|
8
|
+
const watsonx_ai_1 = require("@ibm-cloud/watsonx-ai");
|
|
9
|
+
const crypto_1 = __importDefault(require("crypto"));
|
|
10
|
+
const ibm_cloud_sdk_core_1 = require("ibm-cloud-sdk-core");
|
|
11
|
+
const tiny_invariant_1 = __importDefault(require("tiny-invariant"));
|
|
12
|
+
const zod_1 = require("zod");
|
|
13
|
+
const cache_1 = require("../cache");
|
|
14
|
+
const envars_1 = require("../envars");
|
|
15
|
+
const logger_1 = __importDefault(require("../logger"));
|
|
16
|
+
const ConfigSchema = zod_1.z.object({
|
|
17
|
+
apiKey: zod_1.z.string().optional(),
|
|
18
|
+
apiKeyEnvar: zod_1.z.string().optional(),
|
|
19
|
+
serviceUrl: zod_1.z.string().optional(),
|
|
20
|
+
version: zod_1.z.string().optional(),
|
|
21
|
+
projectId: zod_1.z.string().optional(),
|
|
22
|
+
modelId: zod_1.z.string().optional(),
|
|
23
|
+
maxNewTokens: zod_1.z.number().optional(),
|
|
24
|
+
});
|
|
25
|
+
const TextGenResponseSchema = zod_1.z.object({
|
|
26
|
+
model_id: zod_1.z.string(),
|
|
27
|
+
model_version: zod_1.z.string(),
|
|
28
|
+
created_at: zod_1.z.string(),
|
|
29
|
+
results: zod_1.z.array(zod_1.z.object({
|
|
30
|
+
generated_text: zod_1.z.string(),
|
|
31
|
+
generated_token_count: zod_1.z.number().optional(),
|
|
32
|
+
input_token_count: zod_1.z.number().optional(),
|
|
33
|
+
stop_reason: zod_1.z.string().optional(),
|
|
34
|
+
})),
|
|
35
|
+
});
|
|
36
|
+
function convertResponse(response) {
|
|
37
|
+
const firstResult = response.results && response.results[0];
|
|
38
|
+
if (!firstResult) {
|
|
39
|
+
throw new Error('No results returned from text generation API.');
|
|
40
|
+
}
|
|
41
|
+
const totalGeneratedTokens = firstResult.generated_token_count || 0;
|
|
42
|
+
const promptTokens = firstResult.input_token_count || 0;
|
|
43
|
+
const completionTokens = totalGeneratedTokens - promptTokens;
|
|
44
|
+
const tokenUsage = {
|
|
45
|
+
total: totalGeneratedTokens,
|
|
46
|
+
prompt: promptTokens,
|
|
47
|
+
completion: completionTokens >= 0 ? completionTokens : totalGeneratedTokens,
|
|
48
|
+
};
|
|
49
|
+
const providerResponse = {
|
|
50
|
+
error: undefined,
|
|
51
|
+
output: firstResult.generated_text || '',
|
|
52
|
+
tokenUsage,
|
|
53
|
+
cost: undefined,
|
|
54
|
+
cached: undefined,
|
|
55
|
+
logProbs: undefined,
|
|
56
|
+
};
|
|
57
|
+
return providerResponse;
|
|
58
|
+
}
|
|
59
|
+
function sortObject(obj) {
|
|
60
|
+
if (obj === null || typeof obj !== 'object') {
|
|
61
|
+
return obj;
|
|
62
|
+
}
|
|
63
|
+
if (Array.isArray(obj)) {
|
|
64
|
+
return obj.map(sortObject);
|
|
65
|
+
}
|
|
66
|
+
const sortedKeys = Object.keys(obj)
|
|
67
|
+
.filter((key) => obj[key] !== undefined)
|
|
68
|
+
.sort();
|
|
69
|
+
const result = {};
|
|
70
|
+
sortedKeys.forEach((key) => {
|
|
71
|
+
result[key] = sortObject(obj[key]);
|
|
72
|
+
});
|
|
73
|
+
return result;
|
|
74
|
+
}
|
|
75
|
+
function generateConfigHash(config) {
|
|
76
|
+
const sortedConfig = sortObject(config);
|
|
77
|
+
return crypto_1.default.createHash('md5').update(JSON.stringify(sortedConfig)).digest('hex');
|
|
78
|
+
}
|
|
79
|
+
class WatsonXProvider {
|
|
80
|
+
constructor(modelName, options) {
|
|
81
|
+
const validationResult = ConfigSchema.safeParse(options.config);
|
|
82
|
+
if (!validationResult.success) {
|
|
83
|
+
const errors = validationResult.error.errors.map((e) => e.message).join(', ');
|
|
84
|
+
throw new Error(`WatsonXProvider requires a valid config. Issues: ${errors}`);
|
|
85
|
+
}
|
|
86
|
+
const { env } = options;
|
|
87
|
+
this.modelName = modelName;
|
|
88
|
+
this.options = options;
|
|
89
|
+
this.env = env;
|
|
90
|
+
this.apiKey = this.getApiKey();
|
|
91
|
+
this.client = watsonx_ai_1.WatsonXAI.newInstance({
|
|
92
|
+
version: this.options.config.version || '2023-05-29',
|
|
93
|
+
serviceUrl: this.options.config.serviceUrl || 'https://us-south.ml.cloud.ibm.com',
|
|
94
|
+
authenticator: new ibm_cloud_sdk_core_1.IamAuthenticator({ apikey: this.apiKey }),
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
id() {
|
|
98
|
+
return `watsonx:${this.modelName}`;
|
|
99
|
+
}
|
|
100
|
+
toString() {
|
|
101
|
+
return `[Watsonx Provider ${this.modelName}]`;
|
|
102
|
+
}
|
|
103
|
+
getApiKey() {
|
|
104
|
+
const apiKey = this.options.config.apiKey ||
|
|
105
|
+
(this.options.config.apiKeyEnvar
|
|
106
|
+
? process.env[this.options.config.apiKeyEnvar] ||
|
|
107
|
+
this.env?.[this.options.config.apiKeyEnvar]
|
|
108
|
+
: undefined) ||
|
|
109
|
+
this.env?.WATSONX_API_KEY ||
|
|
110
|
+
(0, envars_1.getEnvString)('WATSONX_API_KEY');
|
|
111
|
+
(0, tiny_invariant_1.default)(apiKey, 'WatsonX API key is not set. Set the WATSONX_API_KEY environment variable or add `apiKey` to the provider config.');
|
|
112
|
+
return apiKey;
|
|
113
|
+
}
|
|
114
|
+
getProjectId() {
|
|
115
|
+
const projectId = this.options.config.projectId ||
|
|
116
|
+
(this.options.config.projectIdEnvar
|
|
117
|
+
? process.env[this.options.config.projectIdEnvar] ||
|
|
118
|
+
this.env?.[this.options.config.projectIdEnvar]
|
|
119
|
+
: undefined) ||
|
|
120
|
+
this.env?.WATSONX_PROJECT_ID ||
|
|
121
|
+
(0, envars_1.getEnvString)('WATSONX_PROJECT_ID');
|
|
122
|
+
(0, tiny_invariant_1.default)(projectId && projectId.trim() !== '', 'WatsonX project ID is not set. Set the WATSONX_PROJECT_ID environment variable or add `projectId` to the provider config.');
|
|
123
|
+
return projectId;
|
|
124
|
+
}
|
|
125
|
+
getModelId() {
|
|
126
|
+
if (!this.modelName) {
|
|
127
|
+
throw new Error('Model name must be specified.');
|
|
128
|
+
}
|
|
129
|
+
if (this.modelName.includes(':')) {
|
|
130
|
+
const parts = this.modelName.split(':');
|
|
131
|
+
if (parts.length < 2 || !parts[1]) {
|
|
132
|
+
throw new Error(`Unable to extract modelId from modelName: ${this.modelName}`);
|
|
133
|
+
}
|
|
134
|
+
return parts[1];
|
|
135
|
+
}
|
|
136
|
+
const modelId = this.options.config.modelId || this.modelName;
|
|
137
|
+
(0, tiny_invariant_1.default)(modelId, 'Model ID is required for WatsonX API call.');
|
|
138
|
+
return modelId;
|
|
139
|
+
}
|
|
140
|
+
getClient() {
|
|
141
|
+
return this.client;
|
|
142
|
+
}
|
|
143
|
+
async callApi(prompt) {
|
|
144
|
+
const modelId = this.getModelId();
|
|
145
|
+
const projectId = this.getProjectId();
|
|
146
|
+
const client = this.getClient();
|
|
147
|
+
const cache = (0, cache_1.getCache)();
|
|
148
|
+
const configHash = generateConfigHash(this.options.config);
|
|
149
|
+
const cacheKey = `watsonx:${this.modelName}:${configHash}:${prompt}`;
|
|
150
|
+
const cacheEnabled = (0, cache_1.isCacheEnabled)();
|
|
151
|
+
if (cacheEnabled) {
|
|
152
|
+
const cachedResponse = await cache.get(cacheKey);
|
|
153
|
+
if (cachedResponse) {
|
|
154
|
+
logger_1.default.debug(`Watsonx: Returning cached response for prompt "${prompt}" with config "${configHash}": ${cachedResponse}`);
|
|
155
|
+
return JSON.parse(cachedResponse);
|
|
156
|
+
}
|
|
157
|
+
}
|
|
158
|
+
try {
|
|
159
|
+
const textGenRequestParametersModel = {
|
|
160
|
+
max_new_tokens: this.options.config.maxNewTokens || 100,
|
|
161
|
+
};
|
|
162
|
+
const params = {
|
|
163
|
+
input: prompt,
|
|
164
|
+
modelId,
|
|
165
|
+
projectId,
|
|
166
|
+
parameters: textGenRequestParametersModel,
|
|
167
|
+
};
|
|
168
|
+
const apiResponse = await client.generateText(params);
|
|
169
|
+
const parsedResponse = TextGenResponseSchema.safeParse(apiResponse.result);
|
|
170
|
+
if (!parsedResponse.success) {
|
|
171
|
+
logger_1.default.error(`Watsonx: Invalid response structure for response: ${JSON.stringify(apiResponse.result)}, errors: ${JSON.stringify(parsedResponse.error.errors)}`);
|
|
172
|
+
throw new Error(`Invalid API response structure: ${JSON.stringify(parsedResponse.error.errors)}`);
|
|
173
|
+
}
|
|
174
|
+
const textGenResponse = parsedResponse.data;
|
|
175
|
+
const providerResponse = convertResponse(textGenResponse);
|
|
176
|
+
if ((0, cache_1.isCacheEnabled)()) {
|
|
177
|
+
await cache.set(cacheKey, JSON.stringify(providerResponse));
|
|
178
|
+
}
|
|
179
|
+
return providerResponse;
|
|
180
|
+
}
|
|
181
|
+
catch (err) {
|
|
182
|
+
logger_1.default.error(`Watsonx: API call error: ${String(err)}`);
|
|
183
|
+
return {
|
|
184
|
+
error: `API call error: ${String(err)}`,
|
|
185
|
+
output: '',
|
|
186
|
+
tokenUsage: {},
|
|
187
|
+
};
|
|
188
|
+
}
|
|
189
|
+
}
|
|
190
|
+
}
|
|
191
|
+
exports.WatsonXProvider = WatsonXProvider;
|
|
192
|
+
//# sourceMappingURL=watsonx.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"watsonx.js","sourceRoot":"","sources":["../../../src/providers/watsonx.ts"],"names":[],"mappings":";;;;;;AA6FA,gDAGC;AA/FD,sDAAkD;AAClD,oDAA4B;AAC5B,2DAAsD;AACtD,oEAAuC;AACvC,6BAAwB;AACxB,oCAAoD;AACpD,sCAAyC;AACzC,uDAA+B;AAe/B,MAAM,YAAY,GAAG,OAAC,CAAC,MAAM,CAAC;IAC5B,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC7B,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAChC,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACpC,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,OAAC,CAAC,MAAM,CAAC;IACrC,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;IACpB,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE;IACzB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE;IACtB,OAAO,EAAE,OAAC,CAAC,KAAK,CACd,OAAC,CAAC,MAAM,CAAC;QACP,cAAc,EAAE,OAAC,CAAC,MAAM,EAAE;QAC1B,qBAAqB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QAC5C,iBAAiB,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACxC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KACnC,CAAC,CACH;CACF,CAAC,CAAC;AAEH,SAAS,eAAe,CAAC,QAA+C;IACtE,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAE5D,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,oBAAoB,GAAG,WAAW,CAAC,qBAAqB,IAAI,CAAC,CAAC;IACpE,MAAM,YAAY,GAAG,WAAW,CAAC,iBAAiB,IAAI,CAAC,CAAC;IACxD,MAAM,gBAAgB,GAAG,oBAAoB,GAAG,YAAY,CAAC;IAE7D,MAAM,UAAU,GAAwB;QACtC,KAAK,EAAE,oBAAoB;QAC3B,MAAM,EAAE,YAAY;QACpB,UAAU,EAAE,gBAAgB,IAAI,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,oBAAoB;KAC5E,CAAC;IAEF,MAAM,gBAAgB,GAAqB;QACzC,KAAK,EAAE,SAAS;QAChB,MAAM,EAAE,WAAW,CAAC,cAAc,IAAI,EAAE;QACxC,UAAU;QACV,IAAI,EAAE,SAAS;QACf,MAAM,EAAE,SAAS;QACjB,QAAQ,EAAE,SAAS;KACpB,CAAC;IAEF,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,SAAS,UAAU,CAAC,GAAQ;IAC1B,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QAC5C,OAAO,GAAG,CAAC;IACb,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACvB,OAAO,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IAC7B,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;SAChC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC;SACvC,IAAI,EAAE,CAAC;IACV,MAAM,MAAM,GAAQ,EAAE,CAAC;IACvB,UAAU,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;QACzB,MAAM,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAgB,kBAAkB,CAAC,MAAW;IAC5C,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;IACxC,OAAO,gBAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACrF,CAAC;AAED,MAAa,eAAe;IAO1B,YAAY,SAAiB,EAAE,OAAwB;QACrD,MAAM,gBAAgB,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9E,MAAM,IAAI,KAAK,CAAC,oDAAoD,MAAM,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAE/B,IAAI,CAAC,MAAM,GAAG,sBAAS,CAAC,WAAW,CAAC;YAClC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,YAAY;YACpD,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,IAAI,mCAAmC;YACjF,aAAa,EAAE,IAAI,qCAAgB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;SAC7D,CAAC,CAAC;IACL,CAAC;IAED,EAAE;QACA,OAAO,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAED,QAAQ;QACN,OAAO,qBAAqB,IAAI,CAAC,SAAS,GAAG,CAAC;IAChD,CAAC;IAED,SAAS;QACP,MAAM,MAAM,GACV,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM;YAC1B,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW;gBAC9B,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC;oBAC5C,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAiC,CAAC;gBACnE,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,CAAC,GAAG,EAAE,eAAe;YACzB,IAAA,qBAAY,EAAC,iBAAiB,CAAC,CAAC;QAClC,IAAA,wBAAS,EACP,MAAM,EACN,kHAAkH,CACnH,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,YAAY;QACV,MAAM,SAAS,GACb,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS;YAC7B,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc;gBACjC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC;oBAC/C,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAoC,CAAC;gBACtE,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,CAAC,GAAG,EAAE,kBAAkB;YAC5B,IAAA,qBAAY,EAAC,oBAAoB,CAAC,CAAC;QACrC,IAAA,wBAAS,EACP,SAAS,IAAI,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,EACpC,2HAA2H,CAC5H,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,UAAU;QACR,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACxC,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClC,MAAM,IAAI,KAAK,CAAC,6CAA6C,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACjF,CAAC;YACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC;QAC9D,IAAA,wBAAS,EAAC,OAAO,EAAE,4CAA4C,CAAC,CAAC;QACjE,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAEhC,MAAM,KAAK,GAAG,IAAA,gBAAQ,GAAE,CAAC;QACzB,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,WAAW,IAAI,CAAC,SAAS,IAAI,UAAU,IAAI,MAAM,EAAE,CAAC;QACrE,MAAM,YAAY,GAAG,IAAA,sBAAc,GAAE,CAAC;QACtC,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,cAAc,EAAE,CAAC;gBACnB,gBAAM,CAAC,KAAK,CACV,kDAAkD,MAAM,kBAAkB,UAAU,MAAM,cAAc,EAAE,CAC3G,CAAC;gBACF,OAAO,IAAI,CAAC,KAAK,CAAC,cAAwB,CAAqB,CAAC;YAClE,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,MAAM,6BAA6B,GAAkC;gBACnE,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,IAAI,GAAG;aACxD,CAAC;YAEF,MAAM,MAAM,GAAyB;gBACnC,KAAK,EAAE,MAAM;gBACb,OAAO;gBACP,SAAS;gBACT,UAAU,EAAE,6BAA6B;aAC1C,CAAC;YAEF,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YACtD,MAAM,cAAc,GAAG,qBAAqB,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAE3E,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC5B,gBAAM,CAAC,KAAK,CACV,qDAAqD,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAClJ,CAAC;gBACF,MAAM,IAAI,KAAK,CACb,mCAAmC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CACjF,CAAC;YACJ,CAAC;YAED,MAAM,eAAe,GAAG,cAAc,CAAC,IAAI,CAAC;YAC5C,MAAM,gBAAgB,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;YAE1D,IAAI,IAAA,sBAAc,GAAE,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAC9D,CAAC;YAED,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,gBAAM,CAAC,KAAK,CAAC,4BAA4B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAExD,OAAO;gBACL,KAAK,EAAE,mBAAmB,MAAM,CAAC,GAAG,CAAC,EAAE;gBACvC,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,EAAE;aACf,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AApJD,0CAoJC"}
|
package/dist/src/providers.d.ts
CHANGED
|
@@ -23,6 +23,7 @@ import { ReplicateImageProvider, ReplicateModerationProvider, ReplicateProvider
|
|
|
23
23
|
import { ScriptCompletionProvider } from './providers/scriptCompletion';
|
|
24
24
|
import { VertexChatProvider, VertexEmbeddingProvider } from './providers/vertex';
|
|
25
25
|
import { VoyageEmbeddingProvider } from './providers/voyage';
|
|
26
|
+
import { WatsonXProvider } from './providers/watsonx';
|
|
26
27
|
import { WebhookProvider } from './providers/webhook';
|
|
27
28
|
import { WebSocketProvider } from './providers/websocket';
|
|
28
29
|
import type { TestSuiteConfig } from './types';
|
|
@@ -89,6 +90,7 @@ declare const _default: {
|
|
|
89
90
|
WebhookProvider: typeof WebhookProvider;
|
|
90
91
|
WebSocketProvider: typeof WebSocketProvider;
|
|
91
92
|
loadApiProvider: typeof loadApiProvider;
|
|
93
|
+
WatsonXProvider: typeof WatsonXProvider;
|
|
92
94
|
};
|
|
93
95
|
export default _default;
|
|
94
96
|
//# sourceMappingURL=providers.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"providers.d.ts","sourceRoot":"","sources":["../../src/providers.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,0BAA0B,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,2BAA2B,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAC/F,OAAO,EACL,4BAA4B,EAC5B,iCAAiC,EACjC,6BAA6B,EAC7B,4BAA4B,EAC7B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACxE,OAAO,EAAE,4BAA4B,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChG,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,OAAO,EAAE,4BAA4B,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC3F,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EACL,oCAAoC,EACpC,qCAAqC,EACrC,qCAAqC,EACrC,iCAAiC,EACjC,kCAAkC,EACnC,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EACL,yBAAyB,EACzB,mBAAmB,EACnB,wBAAwB,EACzB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,6BAA6B,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC9F,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,EACnB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,4BAA4B,EAC5B,uBAAuB,EACvB,mBAAmB,EACnB,wBAAwB,EACzB,MAAM,oBAAoB,CAAC;
|
|
1
|
+
{"version":3,"file":"providers.d.ts","sourceRoot":"","sources":["../../src/providers.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,0BAA0B,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,2BAA2B,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AAC/F,OAAO,EACL,4BAA4B,EAC5B,iCAAiC,EACjC,6BAA6B,EAC7B,4BAA4B,EAC7B,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AACxE,OAAO,EAAE,4BAA4B,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChG,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,OAAO,EAAE,4BAA4B,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC3F,OAAO,EAAE,0BAA0B,EAAE,MAAM,iBAAiB,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EACL,oCAAoC,EACpC,qCAAqC,EACrC,qCAAqC,EACrC,iCAAiC,EACjC,kCAAkC,EACnC,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAClD,OAAO,EACL,yBAAyB,EACzB,mBAAmB,EACnB,wBAAwB,EACzB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,6BAA6B,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAC9F,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,kBAAkB,EACnB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,uBAAuB,EACvB,wBAAwB,EACxB,4BAA4B,EAC5B,uBAAuB,EACvB,mBAAmB,EACnB,wBAAwB,EACzB,MAAM,oBAAoB,CAAC;AAE5B,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,6BAA6B,EAAE,MAAM,qBAAqB,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EACL,sBAAsB,EACtB,2BAA2B,EAC3B,iBAAiB,EAClB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AACxE,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AACjF,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAK1D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC/C,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACZ,eAAe,EAEhB,MAAM,mBAAmB,CAAC;AAK3B,wBAAsB,eAAe,CACnC,YAAY,EAAE,MAAM,EACpB,OAAO,GAAE;IACP,OAAO,CAAC,EAAE,eAAe,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;CACf,GACL,OAAO,CAAC,WAAW,CAAC,CAgYtB;AAED,wBAAsB,gBAAgB,CACpC,aAAa,EAAE,eAAe,CAAC,WAAW,CAAC,EAC3C,OAAO,GAAE;IACP,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,YAAY,CAAC;CACf,GACL,OAAO,CAAC,WAAW,EAAE,CAAC,CAwCxB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAED,wBAsDE"}
|
package/dist/src/providers.js
CHANGED
|
@@ -56,6 +56,7 @@ const manualInput_1 = require("./providers/manualInput");
|
|
|
56
56
|
const mistral_1 = require("./providers/mistral");
|
|
57
57
|
const ollama_1 = require("./providers/ollama");
|
|
58
58
|
const openai_1 = require("./providers/openai");
|
|
59
|
+
const packageParser_1 = require("./providers/packageParser");
|
|
59
60
|
const palm_1 = require("./providers/palm");
|
|
60
61
|
const portkey_1 = require("./providers/portkey");
|
|
61
62
|
const pythonCompletion_1 = require("./providers/pythonCompletion");
|
|
@@ -63,6 +64,7 @@ const replicate_1 = require("./providers/replicate");
|
|
|
63
64
|
const scriptCompletion_1 = require("./providers/scriptCompletion");
|
|
64
65
|
const vertex_1 = require("./providers/vertex");
|
|
65
66
|
const voyage_1 = require("./providers/voyage");
|
|
67
|
+
const watsonx_1 = require("./providers/watsonx");
|
|
66
68
|
const webhook_1 = require("./providers/webhook");
|
|
67
69
|
const websocket_1 = require("./providers/websocket");
|
|
68
70
|
const crescendo_1 = __importDefault(require("./redteam/providers/crescendo"));
|
|
@@ -216,6 +218,11 @@ async function loadApiProvider(providerPath, context = {}) {
|
|
|
216
218
|
else if (providerPath.startsWith('voyage:')) {
|
|
217
219
|
ret = new voyage_1.VoyageEmbeddingProvider(providerPath.split(':')[1], providerOptions);
|
|
218
220
|
}
|
|
221
|
+
else if (providerPath.startsWith('watsonx:')) {
|
|
222
|
+
const splits = providerPath.split(':');
|
|
223
|
+
const modelName = splits.slice(1).join(':');
|
|
224
|
+
ret = new watsonx_1.WatsonXProvider(modelName, providerOptions);
|
|
225
|
+
}
|
|
219
226
|
else if (providerPath.startsWith('bedrock:')) {
|
|
220
227
|
const splits = providerPath.split(':');
|
|
221
228
|
const modelType = splits[1];
|
|
@@ -446,6 +453,9 @@ async function loadApiProvider(providerPath, context = {}) {
|
|
|
446
453
|
: providerPath.split(':').slice(1).join(':');
|
|
447
454
|
ret = new golangCompletion_1.GolangProvider(scriptPath, providerOptions);
|
|
448
455
|
}
|
|
456
|
+
else if (providerPath.startsWith('package:')) {
|
|
457
|
+
ret = await (0, packageParser_1.parsePackageProvider)(providerPath, basePath || process.cwd(), providerOptions);
|
|
458
|
+
}
|
|
449
459
|
else if ((0, file_1.isJavascriptFile)(providerPath)) {
|
|
450
460
|
if (providerPath.startsWith('file://')) {
|
|
451
461
|
providerPath = providerPath.slice('file://'.length);
|
|
@@ -570,5 +580,6 @@ exports.default = {
|
|
|
570
580
|
WebhookProvider: webhook_1.WebhookProvider,
|
|
571
581
|
WebSocketProvider: websocket_1.WebSocketProvider,
|
|
572
582
|
loadApiProvider,
|
|
583
|
+
WatsonXProvider: watsonx_1.WatsonXProvider,
|
|
573
584
|
};
|
|
574
585
|
//# sourceMappingURL=providers.js.map
|