promptfoo 0.93.3 → 0.94.1

package/dist/test/{providers.scriptCompletion.test.js → providers/scriptCompletion.test.js}

@@ -29,13 +29,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const child_process_1 = require("child_process");
 const crypto_1 = __importDefault(require("crypto"));
 const fs_1 = __importDefault(require("fs"));
-const cacheModule = __importStar(require("../src/cache"));
-const scriptCompletion_1 = require("../src/providers/scriptCompletion");
+const cacheModule = __importStar(require("../../src/cache"));
+const scriptCompletion_1 = require("../../src/providers/scriptCompletion");
 jest.mock('fs');
 jest.mock('crypto');
 jest.mock('child_process');
-jest.mock('../src/cache');
-jest.mock('../src/logger');
+jest.mock('../../src/cache');
+jest.mock('../../src/logger');
 describe('parseScriptParts', () => {
     it('should parse script parts correctly', () => {
         const scriptPath = `node script.js "arg with 'spaces'" 'another arg' simple_arg`;
@@ -187,4 +187,4 @@ describe('ScriptCompletionProvider', () => {
         expect(result.output).toBe(strippedOutput);
     });
 });
-//# sourceMappingURL=providers.scriptCompletion.test.js.map
+//# sourceMappingURL=scriptCompletion.test.js.map

package/dist/test/providers/scriptCompletion.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scriptCompletion.test.js","sourceRoot":"","sources":["../../../test/providers/scriptCompletion.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,iDAAyC;AACzC,oDAA4B;AAC5B,4CAAoB;AACpB,6DAA+C;AAC/C,2EAI8C;AAE9C,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACpB,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAC3B,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAC7B,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAE9B,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,UAAU,GAAG,6DAA6D,CAAC;QACjF,MAAM,MAAM,GAAG,IAAA,mCAAgB,EAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,WAAW,EAAE,mBAAmB,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,UAAU,GAAG,qBAAqB,CAAC;QACzC,MAAM,MAAM,GAAG,IAAA,mCAAgB,EAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,WAAW,GAAG,CAAC,UAAU,EAAE,UAAU,EAAE,gBAAgB,CAAC,CAAC;QAC/D,MAAM,gBAAgB,GAAG,UAAU,CAAC;QACpC,MAAM,gBAAgB,GAAG,UAAU,CAAC;QACpC,MAAM,SAAS,GAAG,OAAO,CAAC;QAC1B,MAAM,SAAS,GAAG,OAAO,CAAC;QAE1B,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;QACnF,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC;YACvC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI;YAClB,WAAW,EAAE,GAAG,EAAE,CAAC,KAAK;YACxB,aAAa,EAAE,GAAG,EAAE,CAAC,KAAK;YAC1B,iBAAiB,EAAE,GAAG,EAAE,CAAC,KAAK;YAC9B,cAAc,EAAE,GAAG,EAAE,CAAC,KAAK;YAC3B,MAAM,EAAE,GAAG,EAAE,CAAC,KAAK;YACnB,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK;SACV,CAAC,CAAC;QACf,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE;YACvD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACxB,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YACD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACxB,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAClC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACQ,CAAC;QAC5B,IAAI;aACD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC;aAC7B,mBAAmB,CAAC,SAAS,CAAC;aAC9B,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,IAAA,gCAAa,EAAC,WAAW,CAAC,CAAC;QAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,WAAW,GAAG,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,IAAA,gCAAa,EAAC,WAAW,CAAC,CAAC;QAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/C,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,QAAkC,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,2CAAwB,CAAC,gBAAgB,CAAC,CAAC;QAC1D,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,SAAS,EAAE,CAAC;QAC9C,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,SAAS,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,UAAU,GAAG,YAAY,CAAC;QAChC,IAAI,CAAC,MAAM,CAAC,wBAAQ,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;YACvE,QAA4F,CAC3F,IAAI,EACJ,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EACvB,EAAE,CACH,CAAC;YACF,OAAO,EAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,SAAS,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,wBAAQ,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;YACxE,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;YAC7C,CAAC;YACD,OAAO,EAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,YAAY,GAAG,EAAE,MAAM,EAAE,eAAe,EAAE,CAAC;QACjD,MAAM,SAAS,GAAG;YAChB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAC9D,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;SACf,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAkB,CAAC,CAAC;QAC1E,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAEhE,0DAA0D;QAC1D,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,EAAc,CAAC,CAAC;QAC7E,IAAI,CAAC,MAAM,CAAC,YAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAE7D,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YAClC,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC;SACrB,CAAC;QAC5B,IAAI,CAAC,MAAM,CAAC,gBAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAE/D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACrC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACxC,+DAA+D,CAChE,CAAC;QACF,MAAM,CAAC,wBAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,YAAY,GAAG,yBAAyB,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,wBAAQ,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;YACxE,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YAC5C,CAAC;YACD,OAAO,EAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,WAAW,GAAG,+BAA+B,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,wBAAQ,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;YACxE,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;YAC/C,CAAC;YACD,OAAO,EAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,UAAU,GAAG,2CAA2C,CAAC;QAC/D,MAAM,cAAc,GAAG,cAAc,CAAC;QACtC,IAAI,CAAC,MAAM,CAAC,wBAAQ,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;YACxE,IAAI,OAAO,QAAQ,KAAK,UAAU,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,CAAC;YACD,OAAO,EAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test/providers/shared.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=shared.test.d.ts.map

package/dist/test/providers/shared.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"shared.test.d.ts","sourceRoot":"","sources":["../../../test/providers/shared.test.ts"],"names":[],"mappings":""}

package/dist/test/{providers.shared.test.js → providers/shared.test.js}

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const envars_1 = require("../src/envars");
-const shared_1 = require("../src/providers/shared");
-jest.mock('../src/envars');
+const envars_1 = require("../../src/envars");
+const shared_1 = require("../../src/providers/shared");
+jest.mock('../../src/envars');
 describe('Shared Provider Functions', () => {
     describe('parseChatPrompt', () => {
         it('should parse YAML prompt', () => {
@@ -89,4 +89,4 @@ describe('Shared Provider Functions', () => {
         });
     });
 });
-//# sourceMappingURL=providers.shared.test.js.map
+//# sourceMappingURL=shared.test.js.map

package/dist/test/providers/shared.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shared.test.js","sourceRoot":"","sources":["../../../test/providers/shared.test.ts"],"names":[],"mappings":";;AAAA,6CAA8C;AAC9C,uDAAyF;AAEzF,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAE9B,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,UAAU,GAAG;;;;;OAKlB,CAAC;YACF,MAAM,MAAM,GAAG,IAAA,wBAAe,EAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;gBAClC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC;gBAChC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;gBAClC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE;aAC5C,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,wBAAe,EAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;gBAClC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,YAAY,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,IAAA,wBAAe,EAAC,uBAAuB,EAAE,YAAY,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,WAAW,GAAG,8BAA8B,CAAC;YACnD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,wBAAe,EAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CACpD,mDAAmD,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACzF,IAAI,CAAC,MAAM,CAAC,mBAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,WAAW,GAAG,sCAAsC,CAAC,CAAC,yBAAyB;YACrF,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,wBAAe,EAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CACpD,mDAAmD,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,IAAI,CAAC,MAAM,CAAC,mBAAU,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;YAC/C,MAAM,WAAW,GAAG,sBAAsB,CAAC;YAC3C,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,wBAAe,EAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CACpD,mDAAmD,CACpD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,IAAA,oBAAW,EAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACvD,MAAM,CAAC,IAAA,oBAAW,EAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACjE,MAAM,CAAC,IAAA,oBAAW,EAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,IAAA,oBAAW,EAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,CAAC,IAAA,oBAAW,EAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;YACvD,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE;SACxD,CAAC;QAEF,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,IAAI,GAAG,IAAA,sBAAa,EAAC,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YAC5D,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,iCAAiC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,IAAI,GAAG,IAAA,sBAAa,EAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YACzE,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,iCAAiC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,IAAI,GAAG,IAAA,sBAAa,EAAC,aAAa,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YACjE,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,IAAA,sBAAa,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YAC7E,MAAM,CAAC,IAAA,sBAAa,EAAC,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,IAAA,sBAAa,EAAC,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YAC5E,MAAM,CAAC,IAAA,sBAAa,EAAC,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC/E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test/providers/togetherai.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=togetherai.test.d.ts.map

package/dist/test/providers/togetherai.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"togetherai.test.d.ts","sourceRoot":"","sources":["../../../test/providers/togetherai.test.ts"],"names":[],"mappings":""}

package/dist/test/providers/togetherai.test.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const openai_1 = require("../../src/providers/openai");
+const togetherai_1 = require("../../src/providers/togetherai");
+jest.mock('../../src/providers/openai');
+describe('createTogetherAiProvider', () => {
+    beforeEach(() => {
+        jest.clearAllMocks();
+    });
+    it('should create a chat completion provider when type is chat', () => {
+        const provider = (0, togetherai_1.createTogetherAiProvider)('togetherai:chat:model-name');
+        expect(provider).toBeInstanceOf(openai_1.OpenAiChatCompletionProvider);
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith('model-name', expect.any(Object));
+    });
+    it('should create a completion provider when type is completion', () => {
+        const provider = (0, togetherai_1.createTogetherAiProvider)('togetherai:completion:model-name');
+        expect(provider).toBeInstanceOf(openai_1.OpenAiCompletionProvider);
+        expect(openai_1.OpenAiCompletionProvider).toHaveBeenCalledWith('model-name', expect.any(Object));
+    });
+    it('should create an embedding provider when type is embedding', () => {
+        const provider = (0, togetherai_1.createTogetherAiProvider)('togetherai:embedding:model-name');
+        expect(provider).toBeInstanceOf(openai_1.OpenAiEmbeddingProvider);
+        expect(openai_1.OpenAiEmbeddingProvider).toHaveBeenCalledWith('model-name', expect.any(Object));
+    });
+    it('should create an embedding provider when type is embeddings', () => {
+        const provider = (0, togetherai_1.createTogetherAiProvider)('togetherai:embeddings:model-name');
+        expect(provider).toBeInstanceOf(openai_1.OpenAiEmbeddingProvider);
+        expect(openai_1.OpenAiEmbeddingProvider).toHaveBeenCalledWith('model-name', expect.any(Object));
+    });
+    it('should default to chat completion provider when no type is specified', () => {
+        const provider = (0, togetherai_1.createTogetherAiProvider)('togetherai:model-name');
+        expect(provider).toBeInstanceOf(openai_1.OpenAiChatCompletionProvider);
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith('model-name', expect.any(Object));
+    });
+    it('should pass correct configuration to the provider', () => {
+        const options = {
+            id: 'custom-id',
+        };
+        (0, togetherai_1.createTogetherAiProvider)('togetherai:chat:model-name', options);
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith('model-name', expect.objectContaining({
+            config: {
+                apiBaseUrl: 'https://api.together.xyz/v1',
+                apiKeyEnvar: 'TOGETHER_API_KEY',
+            },
+            id: 'custom-id',
+        }));
+    });
+    it('should handle model names with colons', () => {
+        (0, togetherai_1.createTogetherAiProvider)('togetherai:chat:org:model:name');
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith('org:model:name', expect.any(Object));
+    });
+});
+//# sourceMappingURL=togetherai.test.js.map

package/dist/test/providers/togetherai.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"togetherai.test.js","sourceRoot":"","sources":["../../../test/providers/togetherai.test.ts"],"names":[],"mappings":";;AAAA,uDAIoC;AACpC,+DAA0E;AAG1E,IAAI,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AAExC,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,QAAQ,GAAG,IAAA,qCAAwB,EAAC,4BAA4B,CAAC,CAAC;QACxE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,qCAA4B,CAAC,CAAC;QAC9D,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,QAAQ,GAAG,IAAA,qCAAwB,EAAC,kCAAkC,CAAC,CAAC;QAC9E,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,iCAAwB,CAAC,CAAC;QAC1D,MAAM,CAAC,iCAAwB,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAC1F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,QAAQ,GAAG,IAAA,qCAAwB,EAAC,iCAAiC,CAAC,CAAC;QAC7E,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gCAAuB,CAAC,CAAC;QACzD,MAAM,CAAC,gCAAuB,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,QAAQ,GAAG,IAAA,qCAAwB,EAAC,kCAAkC,CAAC,CAAC;QAC9E,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gCAAuB,CAAC,CAAC;QACzD,MAAM,CAAC,gCAAuB,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,MAAM,QAAQ,GAAG,IAAA,qCAAwB,EAAC,uBAAuB,CAAC,CAAC;QACnE,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,qCAA4B,CAAC,CAAC;QAC9D,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,OAAO,GAIT;YACF,EAAE,EAAE,WAAW;SAChB,CAAC;QACF,IAAA,qCAAwB,EAAC,4BAA4B,EAAE,OAAO,CAAC,CAAC;QAChE,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CACvD,YAAY,EACZ,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE;gBACN,UAAU,EAAE,6BAA6B;gBACzC,WAAW,EAAE,kBAAkB;aAChC;YACD,EAAE,EAAE,WAAW;SAChB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,IAAA,qCAAwB,EAAC,gCAAgC,CAAC,CAAC;QAC3D,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAClG,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test/providers/vertex.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=vertex.test.d.ts.map

package/dist/test/providers/vertex.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex.test.d.ts","sourceRoot":"","sources":["../../../test/providers/vertex.test.ts"],"names":[],"mappings":""}

package/dist/test/{providers.vertex.test.js → providers/vertex.test.js}

@@ -22,22 +22,26 @@ var __importStar = (this && this.__importStar) || function (mod) {
     __setModuleDefault(result, mod);
     return result;
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-const cache_1 = require("../src/cache");
-const vertex_1 = require("../src/providers/vertex");
-const vertexUtil = __importStar(require("../src/providers/vertexUtil"));
-jest.mock('../src/cache', () => ({
+const cache_1 = require("../../src/cache");
+const logger_1 = __importDefault(require("../../src/logger"));
+const vertex_1 = require("../../src/providers/vertex");
+const vertexUtil = __importStar(require("../../src/providers/vertexUtil"));
+jest.mock('../../src/cache', () => ({
     getCache: jest.fn().mockReturnValue({
         get: jest.fn(),
         set: jest.fn(),
     }),
     isCacheEnabled: jest.fn(),
 }));
-jest.mock('../src/providers/vertexUtil', () => ({
-    ...jest.requireActual('../src/providers/vertexUtil'),
+jest.mock('../../src/providers/vertexUtil', () => ({
+    ...jest.requireActual('../../src/providers/vertexUtil'),
     getGoogleClient: jest.fn(),
 }));
-jest.mock('../src/logger');
+jest.mock('../../src/logger');
 describe('VertexChatProvider.callGeminiApi', () => {
     let provider;
     beforeEach(() => {
@@ -78,9 +82,10 @@ describe('VertexChatProvider.callGeminiApi', () => {
                 },
             ],
         };
+        const mockRequest = jest.fn().mockResolvedValue(mockResponse);
         jest.spyOn(vertexUtil, 'getGoogleClient').mockResolvedValue({
             client: {
-                request: jest.fn().mockResolvedValue(mockResponse),
+                request: mockRequest,
             },
             projectId: 'test-project-id',
         });
@@ -94,6 +99,15 @@ describe('VertexChatProvider.callGeminiApi', () => {
                 completion: 5,
             },
         });
+        expect(vertexUtil.getGoogleClient).toHaveBeenCalledWith();
+        expect(mockRequest).toHaveBeenCalledWith({
+            url: expect.any(String),
+            method: 'POST',
+            data: expect.objectContaining({
+                contents: [{ parts: { text: 'test prompt' }, role: 'user' }],
+            }),
+            timeout: expect.any(Number),
+        });
     });
     it('should return cached response if available', async () => {
         const mockCachedResponse = {
@@ -151,4 +165,91 @@ describe('VertexChatProvider.callGeminiApi', () => {
         });
     });
 });
-//# sourceMappingURL=providers.vertex.test.js.map
+describe('maybeCoerceToGeminiFormat', () => {
+    it('should return unmodified content if it matches GeminiFormat', () => {
+        const input = [
+            {
+                role: 'user',
+                parts: [{ text: 'Hello, Gemini!' }],
+            },
+        ];
+        const result = vertexUtil.maybeCoerceToGeminiFormat(input);
+        expect(result).toEqual({
+            contents: input,
+            coerced: false,
+            systemInstruction: undefined,
+        });
+    });
+    it('should coerce OpenAI chat format to GeminiFormat', () => {
+        const input = [
+            { role: 'user', content: 'Hello' },
+            { role: 'user', content: ', ' },
+            { role: 'user', content: 'Gemini!' },
+        ];
+        const expected = [
+            {
+                role: 'user',
+                parts: [{ text: 'Hello' }],
+            },
+            {
+                role: 'user',
+                parts: [{ text: ', ' }],
+            },
+            {
+                role: 'user',
+                parts: [{ text: 'Gemini!' }],
+            },
+        ];
+        const result = vertexUtil.maybeCoerceToGeminiFormat(input);
+        expect(result).toEqual({
+            contents: expected,
+            coerced: true,
+            systemInstruction: undefined,
+        });
+    });
+    it('should coerce string input to GeminiFormat', () => {
+        const input = 'Hello, Gemini!';
+        const expected = [
+            {
+                parts: [{ text: 'Hello, Gemini!' }],
+            },
+        ];
+        const result = vertexUtil.maybeCoerceToGeminiFormat(input);
+        expect(result).toEqual({
+            contents: expected,
+            coerced: true,
+            systemInstruction: undefined,
+        });
+    });
+    it('should handle system messages and create systemInstruction', () => {
+        const input = [
+            { role: 'system', content: 'You are a helpful assistant.' },
+            { role: 'user', content: 'Hello!' },
+        ];
+        const result = vertexUtil.maybeCoerceToGeminiFormat(input);
+        expect(result).toEqual({
+            contents: [
+                {
+                    role: 'user',
+                    parts: [{ text: 'Hello!' }],
+                },
+            ],
+            coerced: true,
+            systemInstruction: {
+                parts: [{ text: 'You are a helpful assistant.' }],
+            },
+        });
+    });
+    it('should log a warning and return the input for unknown formats', () => {
+        const loggerSpy = jest.spyOn(logger_1.default, 'warn');
+        const input = { unknownFormat: 'test' };
+        const result = vertexUtil.maybeCoerceToGeminiFormat(input);
+        expect(result).toEqual({
+            contents: input,
+            coerced: false,
+            systemInstruction: undefined,
+        });
+        expect(loggerSpy).toHaveBeenCalledWith(`Unknown format for Gemini: ${JSON.stringify(input)}`);
+    });
+});
+//# sourceMappingURL=vertex.test.js.map

package/dist/test/providers/vertex.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vertex.test.js","sourceRoot":"","sources":["../../../test/providers/vertex.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,2CAA2D;AAC3D,8DAAsC;AACtC,uDAAgE;AAChE,2EAA6D;AAE7D,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;QAClC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;QACd,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;KACf,CAAC;IACF,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,gCAAgC,EAAE,GAAG,EAAE,CAAC,CAAC;IACjD,GAAG,IAAI,CAAC,aAAa,CAAC,gCAAgC,CAAC;IACvD,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;AAE9B,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,IAAI,QAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,2BAAkB,CAAC,YAAY,EAAE;YAC9C,MAAM,EAAE;gBACN,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC;gBAChE,YAAY,EAAE,CAAC,IAAI,CAAC;gBACpB,WAAW,EAAE,GAAG;gBAChB,eAAe,EAAE,GAAG;gBACpB,IAAI,EAAE,GAAG;gBACT,IAAI,EAAE,EAAE;aACT;SACF,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,KAAK,EAAE,EAAS;SACjB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,YAAY,GAAG;YACnB,IAAI,EAAE;gBACJ;oBACE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;oBACjE,aAAa,EAAE;wBACb,eAAe,EAAE,EAAE;wBACnB,gBAAgB,EAAE,CAAC;wBACnB,oBAAoB,EAAE,CAAC;qBACxB;iBACF;aACF;SACF,CAAC;QAEF,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAE9D,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC;YAC1D,MAAM,EAAE;gBACN,OAAO,EAAE,WAAW;aACI;YAC1B,SAAS,EAAE,iBAAiB;SAC7B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAE7D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,eAAe;YACvB,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE;gBACT,MAAM,EAAE,CAAC;gBACT,UAAU,EAAE,CAAC;aACd;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAC1D,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;YACvC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;YACvB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC5B,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aAC7D,CAAC;YACF,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAC5B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,kBAAkB,GAAG;YACzB,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,sBAAsB;YAC9B,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE;gBACT,MAAM,EAAE,CAAC;gBACT,UAAU,EAAE,CAAC;aACd;SACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAQ,GAAE,CAAC,GAAG,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAElF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAE7D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,GAAG,kBAAkB;YACrB,UAAU,EAAE;gBACV,GAAG,kBAAkB,CAAC,UAAU;gBAChC,MAAM,EAAE,kBAAkB,CAAC,UAAU,CAAC,KAAK;aAC5C;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACpD,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC;YAC1D,MAAM,EAAE;gBACN,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;aACvB;YAC1B,SAAS,EAAE,iBAAiB;SAC7B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAE7D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,KAAK,EAAE,6CAA6C;SACrD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,YAAY,GAAG;YACnB,IAAI,EAAE;gBACJ;oBACE,KAAK,EAAE;wBACL,IAAI,EAAE,GAAG;wBACT,OAAO,EAAE,aAAa;qBACvB;iBACF;aACF;SACF,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC;YAC1D,MAAM,EAAE;gBACN,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,YAAY,CAAC;aAC1B;YAC1B,SAAS,EAAE,iBAAiB;SAC7B,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAE7D,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,KAAK,EAAE,wBAAwB;SAChC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,KAAK,GAAG;YACZ;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;aACpC;SACF,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE,KAAK;YACd,iBAAiB,EAAE,SAAS;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK,GAAG;YACZ,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE;YAClC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE;SACrC,CAAC;QACF,MAAM,QAAQ,GAAG;YACf;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3B;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aACxB;YACD;gBACE,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;aAC7B;SACF,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE,QAAQ;YAClB,OAAO,EAAE,IAAI;YACb,iBAAiB,EAAE,SAAS;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,gBAAgB,CAAC;QAC/B,MAAM,QAAQ,GAAG;YACf;gBACE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC;aACpC;SACF,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE,QAAQ;YAClB,OAAO,EAAE,IAAI;YACb,iBAAiB,EAAE,SAAS;SAC7B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,KAAK,GAAG;YACZ,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,8BAA8B,EAAE;YAC3D,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE;SACpC,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;iBAC5B;aACF;YACD,OAAO,EAAE,IAAI;YACb,iBAAiB,EAAE;gBACjB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,CAAC;aAClD;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAM,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,UAAU,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE,KAAK;YACd,iBAAiB,EAAE,SAAS;SAC7B,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,8BAA8B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAChG,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test/providers/watsonx.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=watsonx.test.d.ts.map

package/dist/test/providers/watsonx.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"watsonx.test.d.ts","sourceRoot":"","sources":["../../../test/providers/watsonx.test.ts"],"names":[],"mappings":""}

package/dist/test/{providers.watsonx.test.js → providers/watsonx.test.js}

@@ -28,10 +28,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const watsonx_ai_1 = require("@ibm-cloud/watsonx-ai");
 const ibm_cloud_sdk_core_1 = require("ibm-cloud-sdk-core");
-const cache_1 = require("../src/cache");
-const envarsModule = __importStar(require("../src/envars"));
-const logger_1 = __importDefault(require("../src/logger"));
-const watsonx_1 = require("../src/providers/watsonx");
+const cache_1 = require("../../src/cache");
+const envarsModule = __importStar(require("../../src/envars"));
+const logger_1 = __importDefault(require("../../src/logger"));
+const watsonx_1 = require("../../src/providers/watsonx");
 jest.mock('@ibm-cloud/watsonx-ai', () => ({
     WatsonXAI: {
         newInstance: jest.fn(),
@@ -41,16 +41,16 @@ jest.mock('ibm-cloud-sdk-core', () => ({
     IamAuthenticator: jest.fn(),
     BearerTokenAuthenticator: jest.fn(),
 }));
-jest.mock('../src/cache', () => ({
+jest.mock('../../src/cache', () => ({
     getCache: jest.fn(),
     isCacheEnabled: jest.fn(),
 }));
-jest.mock('../src/logger', () => ({
+jest.mock('../../src/logger', () => ({
     debug: jest.fn(),
     error: jest.fn(),
     info: jest.fn(),
 }));
-jest.mock('../src/envars', () => ({
+jest.mock('../../src/envars', () => ({
     getEnvString: jest.fn(),
     getEnvInt: jest.fn(),
 }));
@@ -67,7 +67,7 @@ describe('WatsonXProvider', () => {
         jest.clearAllMocks();
     });
     describe('constructor', () => {
-        it('should initialize with modelName and config', () => {
+        it('should initialize with modelName and config', async () => {
             const mockedWatsonXAIClient = {
                 generateText: jest.fn(),
             };
@@ -75,6 +75,8 @@ describe('WatsonXProvider', () => {
             const provider = new watsonx_1.WatsonXProvider(modelName, { config });
             expect(provider.modelName).toBe(modelName);
             expect(provider.options.config).toEqual(config);
+            // Get client to trigger authentication
+            await provider.getClient();
             expect(logger_1.default.info).toHaveBeenCalledWith('Using IAM Authentication.');
         });
         it('should initialize with default id based on modelName', () => {
@@ -107,13 +109,13 @@ describe('WatsonXProvider', () => {
         });
     });
     describe('getClient', () => {
-        it('should initialize WatsonXAI client with correct parameters', () => {
+        it('should initialize WatsonXAI client with correct parameters', async () => {
             const mockedWatsonXAIClient = {
                 generateText: jest.fn(),
             };
             jest.mocked(watsonx_ai_1.WatsonXAI.newInstance).mockReturnValue(mockedWatsonXAIClient);
             const provider = new watsonx_1.WatsonXProvider(modelName, { config });
-            const client = provider.getClient();
+            const client = await provider.getClient();
             expect(watsonx_ai_1.WatsonXAI.newInstance).toHaveBeenCalledWith({
                 version: '2023-05-29',
                 serviceUrl: 'https://us-south.ml.cloud.ibm.com',
@@ -122,17 +124,16 @@ describe('WatsonXProvider', () => {
             expect(ibm_cloud_sdk_core_1.IamAuthenticator).toHaveBeenCalledWith({ apikey: 'test-api-key' });
             expect(client).toBe(mockedWatsonXAIClient);
         });
-        it('should throw an error if neither API key nor Bearer Token is set', () => {
+        it('should throw an error if neither API key nor Bearer Token is set', async () => {
             jest.spyOn(envarsModule, 'getEnvString').mockReturnValue(undefined);
-            expect(() => {
-                new watsonx_1.WatsonXProvider(modelName, {
-                    config: { ...config, apiKey: undefined, apiBearerToken: undefined },
-                });
-            }).toThrow(/Authentication credentials not provided\. Please set either `WATSONX_AI_APIKEY` for IAM Authentication or `WATSONX_AI_BEARER_TOKEN` for Bearer Token Authentication\./);
+            const provider = new watsonx_1.WatsonXProvider(modelName, {
+                config: { ...config, apiKey: undefined, apiBearerToken: undefined },
+            });
+            await expect(provider.getClient()).rejects.toThrow(/Authentication credentials not provided\. Please set either `WATSONX_AI_APIKEY` for IAM Authentication or `WATSONX_AI_BEARER_TOKEN` for Bearer Token Authentication\./);
         });
     });
     describe('constructor with Bearer Token', () => {
-        it('should use Bearer Token Authentication when apiKey is not provided', () => {
+        it('should use Bearer Token Authentication when apiKey is not provided', async () => {
             const bearerTokenConfig = {
                 ...config,
                 apiKey: undefined,
@@ -143,28 +144,28 @@ describe('WatsonXProvider', () => {
             };
             jest.mocked(watsonx_ai_1.WatsonXAI.newInstance).mockReturnValue(mockedWatsonXAIClient);
             const provider = new watsonx_1.WatsonXProvider(modelName, { config: bearerTokenConfig });
+            await provider.getClient();
             expect(logger_1.default.info).toHaveBeenCalledWith('Using Bearer Token Authentication.');
             expect(watsonx_ai_1.WatsonXAI.newInstance).toHaveBeenCalledWith({
                 version: '2023-05-29',
                 serviceUrl: 'https://us-south.ml.cloud.ibm.com',
                 authenticator: expect.any(ibm_cloud_sdk_core_1.BearerTokenAuthenticator),
             });
-            expect(provider.getClient()).toBe(mockedWatsonXAIClient);
         });
-        it('should prefer API Key Authentication over Bearer Token Authentication when both are provided', () => {
+        it('should prefer API Key Authentication over Bearer Token Authentication when both are provided', async () => {
             const dualAuthConfig = { ...config, apiBearerToken: 'test-bearer-token' };
             const mockedWatsonXAIClient = {
                 generateText: jest.fn(),
             };
             jest.mocked(watsonx_ai_1.WatsonXAI.newInstance).mockReturnValue(mockedWatsonXAIClient);
             const provider = new watsonx_1.WatsonXProvider(modelName, { config: dualAuthConfig });
+            await provider.getClient();
             expect(logger_1.default.info).toHaveBeenCalledWith('Using IAM Authentication.');
             expect(watsonx_ai_1.WatsonXAI.newInstance).toHaveBeenCalledWith({
                 version: '2023-05-29',
                 serviceUrl: 'https://us-south.ml.cloud.ibm.com',
                 authenticator: expect.any(ibm_cloud_sdk_core_1.IamAuthenticator),
             });
-            expect(provider.getClient()).toBe(mockedWatsonXAIClient);
         });
     });
     describe('callApi', () => {
@@ -246,7 +247,7 @@ describe('WatsonXProvider', () => {
             jest.mocked(cache_1.getCache).mockReturnValue(cache);
             jest.mocked(cache_1.isCacheEnabled).mockReturnValue(true);
             const provider = new watsonx_1.WatsonXProvider(modelName, { config });
-            const generateTextSpy = jest.spyOn(provider.getClient(), 'generateText');
+            const generateTextSpy = jest.spyOn(await provider.getClient(), 'generateText');
             const response = await provider.callApi(prompt);
             expect(cache.get).toHaveBeenCalledWith(cacheKey);
             expect(response).toEqual(cachedResponse);
@@ -278,4 +279,4 @@ describe('WatsonXProvider', () => {
         });
     });
 });
-//# sourceMappingURL=providers.watsonx.test.js.map
+//# sourceMappingURL=watsonx.test.js.map

package/dist/test/providers/watsonx.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"watsonx.test.js","sourceRoot":"","sources":["../../../test/providers/watsonx.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,sDAAkD;AAClD,2DAAgF;AAChF,2CAA2D;AAC3D,+DAAiD;AACjD,8DAAsC;AACtC,yDAAkF;AAElF,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,SAAS,EAAE;QACT,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;KACvB;CACF,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,wBAAwB,EAAE,IAAI,CAAC,EAAE,EAAE;CACpC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;IACnB,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;IAChB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;CAChB,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;IACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,SAAS,GAAG,YAAY,CAAC;IAC/B,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,cAAc;QACtB,SAAS,EAAE,iBAAiB;QAC5B,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,EAAE;KACjB,CAAC;IACF,MAAM,MAAM,GAAG,aAAa,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEhD,uCAAuC;YACvC,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE;QAClB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,qBAAqB,SAAS,GAAG,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE1C,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;YAEH,MAAM,CAAC,qCAAgB,CAAC,CAAC,oBAAoB,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,SAAgB,CAAC,CAAC;YAE3E,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE;gBAC9C,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,SAAS,EAAE;aACpE,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAChD,uKAAuK,CACxK,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,iBAAiB,GAAG;gBACxB,GAAG,MAAM;gBACT,MAAM,EAAE,SAAS;gBACjB,cAAc,EAAE,mBAAmB;aACpC,CAAC;YACF,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC,CAAC;YAC/E,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,oCAAoC,CAAC,CAAC;YAC/E,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,6CAAwB,CAAC;aACpD,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8FAA8F,EAAE,KAAK,IAAI,EAAE;YAC5G,MAAM,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,CAAC;YAC1E,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;aACxB,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAC5E,MAAM,QAAQ,CAAC,SAAS,EAAE,CAAC;YAE3B,MAAM,CAAC,gBAAM,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC;gBACjD,OAAO,EAAE,YAAY;gBACrB,UAAU,EAAE,mCAAmC;gBAC/C,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,qCAAgB,CAAC;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;YAChG,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;oBACxC,MAAM,EAAE;wBACN,QAAQ,EAAE,gBAAgB;wBAC1B,aAAa,EAAE,OAAO;wBACtB,UAAU,EAAE,sBAAsB;wBAClC,OAAO,EAAE;4BACP;gCACE,cAAc,EAAE,4BAA4B;gCAC5C,qBAAqB,EAAE,EAAE;gCACzB,iBAAiB,EAAE,CAAC;gCACpB,WAAW,EAAE,YAAY;6BAC1B;yBACF;qBACF;iBACF,CAAC;aACH,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YAEjF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBAC9D,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,UAAU,EAAE;oBACV,cAAc,EAAE,MAAM,CAAC,YAAY;iBACpC;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,4BAA4B;gBACpC,UAAU,EAAE;oBACV,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACpC,WAAW,SAAS,IAAI,IAAA,4BAAkB,EAAC,MAAM,CAAC,IAAI,MAAM,EAAE,EAC9D,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CACzB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,iBAAiB;gBACzB,UAAU,EAAE;oBACV,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,UAAU,EAAE,CAAC;iBACd;gBACD,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE,SAAS;aACpB,CAAC;YAEF,MAAM,QAAQ,GAAG,WAAW,SAAS,IAAI,IAAA,4BAAkB,EAAC,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YAChF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;gBAChE,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAElD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,SAAS,EAAE,EAAE,cAAc,CAAC,CAAC;YAC/E,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACzC,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,qBAAqB,GAAiB;gBAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;aAClE,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,sBAAS,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,qBAA4B,CAAC,CAAC;YACjF,MAAM,KAAK,GAAiB;gBAC1B,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;gBACtC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;gBACf,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;gBAChB,KAAK,EAAE,EAAS;aACjB,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAQ,CAAC,CAAC,eAAe,CAAC,KAAY,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,sBAAc,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,yBAAe,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACvB,KAAK,EAAE,kCAAkC;gBACzC,MAAM,EAAE,EAAE;gBACV,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YACH,MAAM,CAAC,gBAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,2CAA2C,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/test/providers/xai.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=xai.test.d.ts.map

package/dist/test/providers/xai.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"xai.test.d.ts","sourceRoot":"","sources":["../../../test/providers/xai.test.ts"],"names":[],"mappings":""}

package/dist/test/providers/xai.test.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const openai_1 = require("../../src/providers/openai");
+const xai_1 = require("../../src/providers/xai");
+jest.mock('../../src/providers/openai');
+describe('xAI Provider', () => {
+    beforeEach(() => {
+        jest.clearAllMocks();
+    });
+    it('throws an error if no model name is provided', () => {
+        expect(() => (0, xai_1.createXAIProvider)('xai:')).toThrow('Model name is required');
+    });
+    it('creates an xAI provider with specified model', () => {
+        const provider = (0, xai_1.createXAIProvider)('xai:grok-2');
+        expect(provider).toBeInstanceOf(openai_1.OpenAiChatCompletionProvider);
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith('grok-2', expect.any(Object));
+    });
+    it('sets the correct API base URL and API key environment variable', () => {
+        (0, xai_1.createXAIProvider)('xai:grok-2');
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith(expect.any(String), expect.objectContaining({
+            config: expect.objectContaining({
+                apiBaseUrl: 'https://api.x.ai/v1',
+                apiKeyEnvar: 'XAI_API_KEY',
+            }),
+        }));
+    });
+    it('merges provided options with xAI-specific config', () => {
+        const options = {
+            config: {
+                temperature: 0.7,
+                max_tokens: 100,
+            },
+            id: 'custom-id',
+        };
+        (0, xai_1.createXAIProvider)('xai:grok-2', options);
+        expect(openai_1.OpenAiChatCompletionProvider).toHaveBeenCalledWith('grok-2', expect.objectContaining({
+            config: expect.objectContaining({
+                apiBaseUrl: 'https://api.x.ai/v1',
+                apiKeyEnvar: 'XAI_API_KEY',
+                temperature: 0.7,
+                max_tokens: 100,
+            }),
+            id: 'custom-id',
+        }));
+    });
+});
+//# sourceMappingURL=xai.test.js.map

package/dist/test/providers/xai.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"xai.test.js","sourceRoot":"","sources":["../../../test/providers/xai.test.ts"],"names":[],"mappings":";;AAAA,uDAA0E;AAC1E,iDAA4D;AAG5D,IAAI,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AAExC,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,uBAAiB,EAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,IAAA,uBAAiB,EAAC,YAAY,CAAC,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,qCAA4B,CAAC,CAAC;QAC9D,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IAC1F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,IAAA,uBAAiB,EAAC,YAAY,CAAC,CAAC;QAChC,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CACvD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAClB,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC9B,UAAU,EAAE,qBAAqB;gBACjC,WAAW,EAAE,aAAa;aAC3B,CAAC;SACH,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,OAAO,GAAoB;YAC/B,MAAM,EAAE;gBACN,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,GAAG;aAChB;YACD,EAAE,EAAE,WAAW;SAChB,CAAC;QACF,IAAA,uBAAiB,EAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACzC,MAAM,CAAC,qCAA4B,CAAC,CAAC,oBAAoB,CACvD,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC;YACtB,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC9B,UAAU,EAAE,qBAAqB;gBACjC,WAAW,EAAE,aAAa;gBAC1B,WAAW,EAAE,GAAG;gBAChB,UAAU,EAAE,GAAG;aAChB,CAAC;YACF,EAAE,EAAE,WAAW;SAChB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}