82 lines
5.2 KiB
JavaScript
82 lines
5.2 KiB
JavaScript
"use strict";
|
|
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
}
|
|
Object.defineProperty(o, k2, desc);
|
|
}) : (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
o[k2] = m[k];
|
|
}));
|
|
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
}) : function(o, v) {
|
|
o["default"] = v;
|
|
});
|
|
var __importStar = (this && this.__importStar) || (function () {
|
|
var ownKeys = function(o) {
|
|
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
var ar = [];
|
|
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
return ar;
|
|
};
|
|
return ownKeys(o);
|
|
};
|
|
return function (mod) {
|
|
if (mod && mod.__esModule) return mod;
|
|
var result = {};
|
|
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
__setModuleDefault(result, mod);
|
|
return result;
|
|
};
|
|
})();
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.runCompletion = exports.onCompletion = void 0;
|
|
const marked_1 = require("marked");
|
|
const marked_terminal_1 = require("marked-terminal");
|
|
const path = __importStar(require("path"));
|
|
const write_1 = require("@plastichub/fs/write");
|
|
const osr_commons_1 = require("@plastichub/osr-commons");
|
|
const __1 = require("../");
|
|
const script_1 = require("../utils/script");
|
|
const filters_1 = require("../filters");
|
|
const onCompletion = async (result = "", options) => {
|
|
result = (0, filters_1.applyFilters)(result, options.filters || []);
|
|
if (options.dst) {
|
|
const dstPath = path.resolve((0, osr_commons_1.resolve)(options.dst, false, {
|
|
...options.variables,
|
|
MODEL: path.parse(options.model).name,
|
|
ROUTER: options.router,
|
|
}));
|
|
(0, write_1.sync)(dstPath, result);
|
|
__1.logger.debug(`Wrote completion result to ${dstPath}`);
|
|
}
|
|
else {
|
|
marked_1.marked.use((0, marked_terminal_1.markedTerminal)({
|
|
emoji: false,
|
|
}));
|
|
const content = (0, marked_1.marked)(result);
|
|
process.stdout.write(content);
|
|
}
|
|
(0, script_1.dumpAsScript)(options);
|
|
process.exit(0);
|
|
return result;
|
|
};
|
|
exports.onCompletion = onCompletion;
|
|
const runCompletion = async (client, params, options) => {
|
|
if (options.dry) {
|
|
__1.logger.info('Dry run - skipping API call');
|
|
return false;
|
|
}
|
|
const completion = await client.chat.completions.create({
|
|
model: options.model,
|
|
messages: params.messages,
|
|
});
|
|
const result = completion.choices[0].message.content;
|
|
(0, exports.onCompletion)(result, options);
|
|
return result;
|
|
};
|
|
exports.runCompletion = runCompletion;
|
|
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicnVuLWNvbXBsZXRpb24uanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvY29tbWFuZHMvcnVuLWNvbXBsZXRpb24udHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBRUEsbUNBQStCO0FBQy9CLHFEQUFnRDtBQUNoRCwyQ0FBNEI7QUFDNUIsZ0RBQW9EO0FBQ3BELHlEQUFpRDtBQUNqRCwyQkFBNEI7QUFDNUIsNENBQThDO0FBQzlDLHdDQUFpRDtBQUUxQyxNQUFNLFlBQVksR0FBRyxLQUFLLEVBQUUsU0FBYyxFQUFFLEVBQUUsT0FBa0IsRUFBRSxFQUFFO0lBQ3pFLE1BQU0sR0FBRyxJQUFBLHNCQUFZLEVBQUMsTUFBTSxFQUFFLE9BQU8sQ0FBQyxPQUFtQixJQUFJLEVBQUUsQ0FBQyxDQUFBO0lBQ2hFLElBQUksT0FBTyxDQUFDLEdBQUcsRUFBRSxDQUFDO1FBQ2hCLE1BQU0sT0FBTyxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUMsSUFBQSxxQkFBTyxFQUFDLE9BQU8sQ0FBQyxHQUFHLEVBQUUsS0FBSyxFQUFFO1lBQ3ZELEdBQUcsT0FBTyxDQUFDLFNBQVM7WUFDcEIsS0FBSyxFQUFFLElBQUksQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLEtBQUssQ0FBQyxDQUFDLElBQUk7WUFDckMsTUFBTSxFQUFFLE9BQU8sQ0FBQyxNQUFNO1NBQ3ZCLENBQUMsQ0FBQyxDQUFBO1FBQ0gsSUFBQSxZQUFLLEVBQUMsT0FBTyxFQUFFLE1BQU0sQ0FBQyxDQUFBO1FBQ3RCLFVBQU0sQ0FBQyxLQUFLLENBQUMsOEJBQThCLE9BQU8sRUFBRSxDQUFDLENBQUE7SUFDdkQsQ0FBQztTQUFNLENBQUM7UUFDTixlQUFNLENBQUMsR0FBRyxDQUFDLElBQUEsZ0NBQWMsRUFBQztZQUN4QixLQUFLLEVBQUUsS0FBSztTQUNiLENBQUMsQ0FBQyxDQUFBO1FBQ0gsTUFBTSxPQUFPLEdBQVcsSUFBQSxlQUFNLEVBQUMsTUFBTSxDQUFXLENBQUM7UUFDakQsT0FBTyxDQUFDLE1BQU0sQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLENBQUE7SUFDL0IsQ0FBQztJQUNELElBQUEscUJBQVksRUFBQyxPQUFPLENBQUMsQ0FBQTtJQUNyQixPQUFPLENBQUMsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFBO0lBQ2YsT0FBTyxNQUFNLENBQUE7QUFDZixDQUFDLENBQUE7QUFwQlksUUFBQSxZQUFZLGdCQW9CeEI7QUFFTSxNQUFNLGFBQWEsR0FBRyxLQUFLLEVBQUUsTUFBYyxFQUFFLE1BQVcsRUFBRSxPQUFrQixFQUFFLEVBQUU7SUFDckYsSUFBSSxPQUFPLENBQUMsR0FBRyxFQUFFLENBQUM7UUFDaEIsVUFBTSxDQUFDLElBQUksQ0FBQyw2QkFBNkIsQ0FBQyxDQUFBO1FBQzFDLE9BQU8sS0FBSyxDQUFBO0lBQ2QsQ0FBQztJQUNELE1BQU0sVUFBVSxHQUFHLE1BQU0sTUFBTSxDQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsTUFBTSxDQUFDO1FBQ3RELEtBQUssRUFBRSxPQUFPLENBQUMsS0FBSztRQUNwQixRQUFRLEVBQUUsTUFBTSxDQUFDLFFBQVE7S0FDMUIsQ0FBQyxDQUFBO0lBQ0YsTUFBTSxNQUFNLEdBQUcsVUFBVSxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFBO0lBQ3BELElBQUEsb0JBQVksRUFBQyxNQUFNLEVBQUUsT0FBTyxDQUFDLENBQUE7SUFDN0IsT0FBTyxNQUFNLENBQUE7QUFDZixDQUFDLENBQUE7QUFaWSxRQUFBLGFBQWEsaUJBWXpCIn0=
|