@rxap/plugin-gpt 19.0.2 → 19.0.3-dev.1

package/CHANGELOG.md

@@ -3,6 +3,18 @@
 All notable changes to this project will be documented in this file.
 See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.
 
+## [19.0.3-dev.1](https://gitlab.com/rxap/packages/compare/@rxap/plugin-gpt@19.0.3-dev.0...@rxap/plugin-gpt@19.0.3-dev.1) (2024-06-20)
+
+### Bug Fixes
+
+- add html code coverage reporters ([73da1d8](https://gitlab.com/rxap/packages/commit/73da1d85274686590952a97e202c713718988caa))
+- update openapi package ([d5afabe](https://gitlab.com/rxap/packages/commit/d5afabec58c65d3178129b826f016e1c593b8846))
+- update to new openai models ([882aab3](https://gitlab.com/rxap/packages/commit/882aab3575ba99c7e271969fa94e2f2bb9361b2c))
+
+## [19.0.3-dev.0](https://gitlab.com/rxap/packages/compare/@rxap/plugin-gpt@19.0.2...@rxap/plugin-gpt@19.0.3-dev.0) (2024-06-18)
+
+**Note:** Version bump only for package @rxap/plugin-gpt
+
 ## [19.0.2](https://gitlab.com/rxap/packages/compare/@rxap/plugin-gpt@19.0.2-dev.5...@rxap/plugin-gpt@19.0.2) (2024-06-18)
 
 **Note:** Version bump only for package @rxap/plugin-gpt

package/package.json

@@ -1,14 +1,14 @@
 {
-  "version": "19.0.2",
+  "version": "19.0.3-dev.1",
   "name": "@rxap/plugin-gpt",
   "license": "GPL-3.0-or-later",
   "dependencies": {
-    "@nx/devkit": "19.1.1",
-    "@rxap/generator-ts-morph": "^1.0.8",
-    "@rxap/ts-morph": "^1.4.0",
-    "@rxap/workspace-utilities": "^19.2.0",
-    "gpt-3-encoder": "1.1.4",
-    "openai": "3.3.0",
+    "@nx/devkit": "19.3.0",
+    "@rxap/ts-morph": "^1.4.1-dev.0",
+    "@rxap/workspace-ts-morph": "*",
+    "@rxap/workspace-utilities": "^19.3.0-dev.0",
+    "gpt-3-encoder": "^1.1.4",
+    "openai": "^4.52.0",
     "ts-morph": "18.0.0",
     "tslib": "2.6.2"
   },
@@ -43,6 +43,6 @@
   },
   "schematics": "./generators.json",
   "type": "commonjs",
-  "gitHead": "d983079aba781cc04989a4ea7948ae5d666dfa24",
+  "gitHead": "3fa5a96cac9ec03017cb91b15f918a0e66db9066",
   "main": "./src/index.js"
 }

package/src/generators/documentation/add-js-doc.d.ts

@@ -0,0 +1,3 @@
+import { JSDocableNode } from 'ts-morph';
+import { DocumentationGeneratorSchema } from './schema';
+export declare function addJsDoc(options: DocumentationGeneratorSchema, node: JSDocableNode, jsDoc: string): void;

package/src/generators/documentation/add-js-doc.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.addJsDoc = void 0;
+const cleanup_js_doc_1 = require("./cleanup-js-doc");
+function addJsDoc(options, node, jsDoc) {
+    if (options.offline) {
+        return;
+    }
+    node.addJsDoc((0, cleanup_js_doc_1.cleanupJsDoc)(jsDoc));
+}
+exports.addJsDoc = addJsDoc;
+//# sourceMappingURL=add-js-doc.js.map

package/src/generators/documentation/add-js-doc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"add-js-doc.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/add-js-doc.ts"],"names":[],"mappings":";;;AACA,qDAAgD;AAGhD,SAAgB,QAAQ,CAAC,OAAqC,EAAE,IAAmB,EAAE,KAAa;IAEhG,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO;IACT,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,IAAA,6BAAY,EAAC,KAAK,CAAC,CAAC,CAAC;AACrC,CAAC;AAPD,4BAOC"}

package/src/generators/documentation/cleanup-js-doc.d.ts

@@ -0,0 +1 @@
+export declare function cleanupJsDoc(jsDoc: string): string;

package/src/generators/documentation/cleanup-js-doc.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cleanupJsDoc = void 0;
+function cleanupJsDoc(jsDoc) {
+    const cleanJsDocArray = jsDoc
+        .split('\n')
+        .map(line => line
+        .trim()
+        .replace(/^\/\*\*/, '')
+        .replace(/^\s?\*\//, '')
+        .replace(/^\s?\*\s?/, '')
+        .replace(/^\s?\*\s```typescript/, ' *')
+        .trim());
+    cleanJsDocArray.pop();
+    return cleanJsDocArray.join('\n');
+}
+exports.cleanupJsDoc = cleanupJsDoc;
+//# sourceMappingURL=cleanup-js-doc.js.map

package/src/generators/documentation/cleanup-js-doc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cleanup-js-doc.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/cleanup-js-doc.ts"],"names":[],"mappings":";;;AAAA,SAAgB,YAAY,CAAC,KAAa;IAExC,MAAM,eAAe,GAAG,KAAK;SAC1B,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI;SACd,IAAI,EAAE;SACN,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;SACxB,OAAO,CAAC,uBAAuB,EAAE,IAAI,CAAC;SACtC,IAAI,EAAE,CACR,CAAC;IAEJ,eAAe,CAAC,GAAG,EAAE,CAAC;IAEtB,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEpC,CAAC;AAjBD,oCAiBC"}

package/src/generators/documentation/const.d.ts

@@ -0,0 +1,3 @@
+export declare const FUNCTION_SYSTEM_PROMPT: string;
+export declare const METHOD_SYSTEM_PROMPT: string;
+export declare const MAX_TIME: number;

package/src/generators/documentation/const.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MAX_TIME = exports.METHOD_SYSTEM_PROMPT = exports.FUNCTION_SYSTEM_PROMPT = void 0;
+const load_system_prompt_1 = require("./load-system-prompt");
+exports.FUNCTION_SYSTEM_PROMPT = (0, load_system_prompt_1.loadSystemPrompt)('function');
+exports.METHOD_SYSTEM_PROMPT = (0, load_system_prompt_1.loadSystemPrompt)('method');
+const CLASS_SYSTEM_PROMPT = (0, load_system_prompt_1.loadSystemPrompt)('class');
+exports.MAX_TIME = 1000 * 60 * 60 * 0.5;
+//# sourceMappingURL=const.js.map

package/src/generators/documentation/const.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"const.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/const.ts"],"names":[],"mappings":";;;AAAA,6DAAwD;AAE3C,QAAA,sBAAsB,GAAG,IAAA,qCAAgB,EAAC,UAAU,CAAC,CAAC;AACtD,QAAA,oBAAoB,GAAG,IAAA,qCAAgB,EAAC,QAAQ,CAAC,CAAC;AAC/D,MAAM,mBAAmB,GAAG,IAAA,qCAAgB,EAAC,OAAO,CAAC,CAAC;AACzC,QAAA,QAAQ,GAAG,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC"}

package/src/generators/documentation/generator.js

@@ -3,177 +3,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.documentationGenerator = void 0;
 const tslib_1 = require("tslib");
 const devkit_1 = require("@nx/devkit");
-const generator_ts_morph_1 = require("@rxap/generator-ts-morph");
-const ts_morph_1 = require("@rxap/ts-morph");
-const workspace_utilities_1 = require("@rxap/workspace-utilities");
-const fs_1 = require("fs");
-const openai_1 = require("openai");
-const path_1 = require("path");
 const process = require("process");
-const simple_prompt_1 = require("./simple-prompt");
-function getOpenAi(param = {}) {
-    if (!process.env.OPENAI_API_KEY) {
-        throw new Error('Can not find OPENAI_API_KEY environment variable');
-    }
-    const configuration = new openai_1.Configuration(Object.assign(Object.assign({}, param), { apiKey: process.env.OPENAI_API_KEY }));
-    return new openai_1.OpenAIApi(configuration);
-}
-function loadSystemPrompt(name) {
-    const filePath = (0, path_1.join)(__dirname, 'system-prompts', `${name}.txt`);
-    if (!(0, fs_1.existsSync)(filePath)) {
-        throw new Error(`Can not find system prompt file '${filePath}'`);
-    }
-    return (0, fs_1.readFileSync)(filePath).toString();
-}
-const FUNCTION_SYSTEM_PROMPT = loadSystemPrompt('function');
-const METHOD_SYSTEM_PROMPT = loadSystemPrompt('method');
-const CLASS_SYSTEM_PROMPT = loadSystemPrompt('class');
-function hasJsDoc(node) {
-    return node.getJsDocs().length > 0;
-}
-function prompt(options, systemPrompt, prompt) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        if (options.offline) {
-            return '';
-        }
-        return (0, simple_prompt_1.SimplePrompt)(systemPrompt, prompt, getOpenAi());
-    });
-}
-function cleanupJsDoc(jsDoc) {
-    const cleanJsDocArray = jsDoc
-        .split('\n')
-        .map(line => line
-        .trim()
-        .replace(/^\/\*\*/, '')
-        .replace(/^\s?\*\//, '')
-        .replace(/^\s?\*\s?/, '')
-        .trim());
-    cleanJsDocArray.pop();
-    return cleanJsDocArray.join('\n');
-}
-function addJsDoc(options, node, jsDoc) {
-    if (options.offline) {
-        return;
-    }
-    node.addJsDoc(cleanupJsDoc(jsDoc));
-}
-function processMethod(options, project, sourceFile, methodDeclaration) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        console.log(`====== Process method: \x1b[36m${methodDeclaration.getName()}\x1b[0m`);
-        if (hasJsDoc(methodDeclaration)) {
-            console.log(`\x1b[33mMethod has already a documentation\x1b[0m`);
-            return false;
-        }
-        const methodText = methodDeclaration.getText();
-        console.log('Method text:');
-        console.log(methodText);
-        const jsDoc = yield prompt(options, METHOD_SYSTEM_PROMPT, methodText);
-        console.log('Method documentation:');
-        console.log(jsDoc);
-        addJsDoc(options, methodDeclaration, jsDoc);
-        return true;
-    });
-}
-function processClass(options, project, sourceFile, classDeclaration) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        console.log(`====== Process class: \x1b[36m${classDeclaration.getName()}\x1b[0m`);
-        let changed = false;
-        for (const md of classDeclaration.getMethods()) {
-            const scope = md.getScope();
-            if (scope === 'public' || scope === undefined) {
-                const hasChanged = yield processMethod(options, project, sourceFile, md);
-                changed = changed || hasChanged;
-            }
-        }
-        return changed;
-    });
-}
-function processFunction(options, project, sourceFile, functionDeclaration) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        console.log(`====== Process function: \x1b[36m${functionDeclaration.getName()}\x1b[0m`);
-        if (hasJsDoc(functionDeclaration)) {
-            console.log(`\x1b[33mFunction has already a documentation\x1b[0m`);
-            return false;
-        }
-        const functionText = functionDeclaration.getText();
-        console.log('Function text:');
-        console.log(functionText);
-        const jsDoc = yield prompt(options, FUNCTION_SYSTEM_PROMPT, functionText);
-        console.log('Function documentation:');
-        console.log(jsDoc);
-        addJsDoc(options, functionDeclaration, jsDoc);
-        return true;
-    });
-}
-function processSourceFile(options, project, sourceFile) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        console.log(`====== Process source file: \x1b[35m${sourceFile.getFilePath()}\x1b[0m`);
-        let changed = false;
-        for (const fd of sourceFile.getFunctions()) {
-            const hasChanged = yield processFunction(options, project, sourceFile, fd);
-            changed = changed || hasChanged;
-        }
-        for (const cd of sourceFile.getClasses()) {
-            const hasChanged = yield processClass(options, project, sourceFile, cd);
-            changed = changed || hasChanged;
-        }
-        return changed;
-    });
-}
-function skipSourceFile(sourceFile, filter) {
-    if (sourceFile.getBaseName() === 'index.ts') {
-        // console.log(`\x1b[33mSkip index file: \x1b[0m${ sourceFile.getFilePath() }`);
-        return true;
-    }
-    if (sourceFile.getBaseName().endsWith('.spec.ts')) {
-        // console.log(`\x1b[33mSkip spec file: \x1b[0m${ sourceFile.getFilePath() }`);
-        return true;
-    }
-    if (sourceFile.getBaseName().endsWith('.stories.ts')) {
-        // console.log(`\x1b[33mSkip stories file: \x1b[0m${ sourceFile.getFilePath() }`);
-        return true;
-    }
-    if (sourceFile.getBaseName().endsWith('.cy.ts')) {
-        // console.log(`\x1b[33mSkip cypress file: \x1b[0m${ sourceFile.getFilePath() }`);
-        return true;
-    }
-    if (filter) {
-        if (!sourceFile.getFilePath().includes(filter)) {
-            // console.log(`\x1b[33mSkip file: \x1b[0m${ sourceFile.getFilePath() }`);
-            return true;
-        }
-    }
-    return false;
-}
-function processProject(options, projectName, tree) {
-    return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        const projectSourceRoot = (0, workspace_utilities_1.GetProjectSourceRoot)(tree, projectName);
-        const project = (0, ts_morph_1.CreateProject)();
-        (0, generator_ts_morph_1.AddDir)(tree, projectSourceRoot, project);
-        const start = Date.now();
-        for (const sourceFile of project.getSourceFiles()) {
-            if (skipSourceFile(sourceFile, options.filter)) {
-                continue;
-            }
-            try {
-                const changed = yield processSourceFile(options, project, sourceFile);
-                if (changed) {
-                    (0, workspace_utilities_1.CoerceFile)(tree, (0, path_1.join)(projectSourceRoot, sourceFile.getFilePath()), sourceFile.getFullText(), true);
-                }
-            }
-            catch (e) {
-                console.error(`\x1b[31mError processing file: \x1b[0m${sourceFile.getFilePath()}`);
-                console.error(e.message);
-                console.error(e.stack);
-                break;
-            }
-            if (Date.now() - start > MAX_TIME) {
-                console.log(`\x1b[33mMax time reached: \x1b[0m${MAX_TIME}`);
-                break;
-            }
-        }
-    });
-}
+const open_ai_instance_1 = require("./open-ai-instance");
+const process_project_1 = require("./process-project");
 function skipProject(project, projectName, options) {
     var _a;
     if ((_a = options.projects) === null || _a === void 0 ? void 0 : _a.length) {
@@ -181,21 +13,31 @@ function skipProject(project, projectName, options) {
     }
     return false;
 }
-const MAX_TIME = 1000 * 60 * 60 * 0.5;
 function documentationGenerator(tree, options) {
     return tslib_1.__awaiter(this, void 0, void 0, function* () {
-        if (options.openaiApiKey) {
-            process.env.OPENAI_API_KEY = options.openaiApiKey;
-        }
-        if (!process.env.OPENAI_API_KEY) {
+        var _a, _b, _c;
+        const openaiApiKey = (_a = options.openaiApiKey) !== null && _a !== void 0 ? _a : process.env.OPENAI_API_KEY;
+        const openaiOrganization = (_b = options.openaiOrgId) !== null && _b !== void 0 ? _b : process.env.OPENAI_ORG_ID;
+        const openaiProjectId = (_c = options.openaiProjectId) !== null && _c !== void 0 ? _c : process.env.OPENAI_PROJECT_ID;
+        if (!openaiApiKey) {
             throw new Error('Can not find OPENAI_API_KEY environment variable');
         }
-        console.log(`\x1b[33mOpenAI API key: \x1b[0m${process.env.OPENAI_API_KEY}`);
+        if (!openaiOrganization) {
+            throw new Error('Can not find OPENAI_ORGANIZATION environment variable');
+        }
+        if (!openaiProjectId) {
+            throw new Error('Can not find OPENAI_PROJECT_ID environment variable');
+        }
+        yield (0, open_ai_instance_1.createOpenApi)({
+            apiKey: openaiApiKey,
+            organization: openaiOrganization,
+            project: openaiProjectId,
+        });
         for (const [projectName, project] of (0, devkit_1.getProjects)(tree).entries()) {
             if (skipProject(project, projectName, options)) {
                 continue;
             }
-            yield processProject(options, projectName, tree);
+            yield (0, process_project_1.processProject)(options, projectName, tree);
         }
     });
 }

package/src/generators/documentation/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/generator.ts"],"names":[],"mappings":";;;;AAAA,uCAIoB;AACpB,iEAAkD;AAClD,6CAA+C;AAC/C,mEAGmC;AACnC,2BAGY;AACZ,mCAIgB;AAChB,+BAA4B;AAC5B,mCAAmC;AAUnC,mDAA+C;AAE/C,SAAS,SAAS,CAAC,QAAiD,EAAE;IAEpE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,sBAAa,iCAClC,KAAK,KACR,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAClC,CAAC;IACH,OAAO,IAAI,kBAAS,CAAC,aAAa,CAAC,CAAC;AAEtC,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,gBAAgB,EAAE,GAAI,IAAK,MAAM,CAAC,CAAC;IACpE,IAAI,CAAC,IAAA,eAAU,EAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,oCAAqC,QAAS,GAAG,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,IAAA,iBAAY,EAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;AAC3C,CAAC;AAED,MAAM,sBAAsB,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAC5D,MAAM,oBAAoB,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;AACxD,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;AAEtD,SAAS,QAAQ,CAAC,IAAmB;IACnC,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;AACrC,CAAC;AAED,SAAe,MAAM,CAAC,OAAO,EAAE,YAAoB,EAAE,MAAc;;QAEjE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,IAAA,4BAAY,EACjB,YAAY,EACZ,MAAM,EACN,SAAS,EAAE,CACZ,CAAC;IACJ,CAAC;CAAA;AAED,SAAS,YAAY,CAAC,KAAa;IAEjC,MAAM,eAAe,GAAG,KAAK;SAC1B,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI;SACd,IAAI,EAAE;SACN,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;SACxB,IAAI,EAAE,CACR,CAAC;IAEJ,eAAe,CAAC,GAAG,EAAE,CAAC;IAEtB,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEpC,CAAC;AAED,SAAS,QAAQ,CAAC,OAAqC,EAAE,IAAmB,EAAE,KAAa;IAEzF,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO;IACT,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;AACrC,CAAC;AAED,SAAe,aAAa,CAC1B,OAAqC,EACrC,OAAgB,EAChB,UAAsB,EACtB,iBAAoC;;QAGpC,OAAO,CAAC,GAAG,CAAC,kCAAmC,iBAAiB,CAAC,OAAO,EAAG,SAAS,CAAC,CAAC;QAEtF,IAAI,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;YACjE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,EAAE,CAAC;QAE/C,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAGxB,MAAM,KAAK,GAAG,MAAM,MAAM,CACxB,OAAO,EACP,oBAAoB,EACpB,UAAU,CACX,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACrC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAEnB,QAAQ,CAAC,OAAO,EAAE,iBAAiB,EAAE,KAAK,CAAC,CAAC;QAE5C,OAAO,IAAI,CAAC;IAEd,CAAC;CAAA;AAED,SAAe,YAAY,CACzB,OAAqC,EACrC,OAAgB,EAChB,UAAsB,EACtB,gBAAkC;;QAGlC,OAAO,CAAC,GAAG,CAAC,iCAAkC,gBAAgB,CAAC,OAAO,EAAG,SAAS,CAAC,CAAC;QAEpF,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,KAAK,MAAM,EAAE,IAAI,gBAAgB,CAAC,UAAU,EAAE,EAAE,CAAC;YAC/C,MAAM,KAAK,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;YAC5B,IAAI,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBAC9C,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;gBACzE,OAAO,GAAG,OAAO,IAAI,UAAU,CAAC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IAEjB,CAAC;CAAA;AAED,SAAe,eAAe,CAC5B,OAAqC,EACrC,OAAgB,EAChB,UAAsB,EACtB,mBAAwC;;QAGxC,OAAO,CAAC,GAAG,CAAC,oCAAqC,mBAAmB,CAAC,OAAO,EAAG,SAAS,CAAC,CAAC;QAE1F,IAAI,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAClC,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;YACnE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,YAAY,GAAG,mBAAmB,CAAC,OAAO,EAAE,CAAC;QAEnD,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC9B,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAE1B,MAAM,KAAK,GAAG,MAAM,MAAM,CACxB,OAAO,EACP,sBAAsB,EACtB,YAAY,CACb,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACvC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAEnB,QAAQ,CAAC,OAAO,EAAE,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAE9C,OAAO,IAAI,CAAC;IAEd,CAAC;CAAA;AAED,SAAe,iBAAiB,CAAC,OAAqC,EAAE,OAAgB,EAAE,UAAsB;;QAE9G,OAAO,CAAC,GAAG,CAAC,uCAAwC,UAAU,CAAC,WAAW,EAAG,SAAS,CAAC,CAAC;QAExF,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,KAAK,MAAM,EAAE,IAAI,UAAU,CAAC,YAAY,EAAE,EAAE,CAAC;YAC3C,MAAM,UAAU,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;YAC3E,OAAO,GAAG,OAAO,IAAI,UAAU,CAAC;QAClC,CAAC;QAED,KAAK,MAAM,EAAE,IAAI,UAAU,CAAC,UAAU,EAAE,EAAE,CAAC;YACzC,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,CAAC,CAAC;YACxE,OAAO,GAAG,OAAO,IAAI,UAAU,CAAC;QAClC,CAAC;QAED,OAAO,OAAO,CAAC;IAEjB,CAAC;CAAA;AAED,SAAS,cAAc,CAAC,UAAsB,EAAE,MAAe;IAC7D,IAAI,UAAU,CAAC,WAAW,EAAE,KAAK,UAAU,EAAE,CAAC;QAC5C,gFAAgF;QAChF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAClD,+EAA+E;QAC/E,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QACrD,kFAAkF;QAClF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,kFAAkF;QAClF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/C,0EAA0E;YAC1E,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAe,cAAc,CAAC,OAAqC,EAAE,WAAmB,EAAE,IAAU;;QAElG,MAAM,iBAAiB,GAAG,IAAA,0CAAoB,EAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAElE,MAAM,OAAO,GAAG,IAAA,wBAAa,GAAE,CAAC;QAEhC,IAAA,2BAAM,EAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAEzC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC;YAClD,IAAI,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/C,SAAS;YACX,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;gBACtE,IAAI,OAAO,EAAE,CAAC;oBACZ,IAAA,gCAAU,EAAC,IAAI,EAAE,IAAA,WAAI,EAAC,iBAAiB,EAAE,UAAU,CAAC,WAAW,EAAE,CAAC,EAAE,UAAU,CAAC,WAAW,EAAE,EAAE,IAAI,CAAC,CAAC;gBACtG,CAAC;YACH,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,OAAO,CAAC,KAAK,CAAC,yCAA0C,UAAU,CAAC,WAAW,EAAG,EAAE,CAAC,CAAC;gBACrF,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACvB,MAAM;YACR,CAAC;YACD,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,QAAQ,EAAE,CAAC;gBAClC,OAAO,CAAC,GAAG,CAAC,oCAAqC,QAAS,EAAE,CAAC,CAAC;gBAC9D,MAAM;YACR,CAAC;QACH,CAAC;IAEH,CAAC;CAAA;AAED,SAAS,WAAW,CAAC,OAA6B,EAAE,WAAmB,EAAE,OAAqC;;IAE5G,IAAI,MAAA,OAAO,CAAC,QAAQ,0CAAE,MAAM,EAAE,CAAC;QAC7B,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,KAAK,CAAC;AAEf,CAAC;AAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC;AAEtC,SAAsB,sBAAsB,CAC1C,IAAU,EACV,OAAqC;;QAGrC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,OAAO,CAAC,YAAY,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,kCAAmC,OAAO,CAAC,GAAG,CAAC,cAAe,EAAE,CAAC,CAAC;QAE9E,KAAK,MAAM,CAAE,WAAW,EAAE,OAAO,CAAE,IAAI,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YAEnE,IAAI,WAAW,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC/C,SAAS;YACX,CAAC;YAED,MAAM,cAAc,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAEnD,CAAC;IAEH,CAAC;CAAA;AAzBD,wDAyBC;AAED,kBAAe,sBAAsB,CAAC"}
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/generator.ts"],"names":[],"mappings":";;;;AAAA,uCAIoB;AACpB,mCAAmC;AACnC,yDAAmD;AACnD,uDAAmD;AAGnD,SAAS,WAAW,CAAC,OAA6B,EAAE,WAAmB,EAAE,OAAqC;;IAE5G,IAAI,MAAA,OAAO,CAAC,QAAQ,0CAAE,MAAM,EAAE,CAAC;QAC7B,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,KAAK,CAAC;AAEf,CAAC;AAED,SAAsB,sBAAsB,CAC1C,IAAU,EACV,OAAqC;;;QAGrC,MAAM,YAAY,GAAG,MAAA,OAAO,CAAC,YAAY,mCAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QACxE,MAAM,kBAAkB,GAAG,MAAA,OAAO,CAAC,WAAW,mCAAI,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;QAC5E,MAAM,eAAe,GAAG,MAAA,OAAO,CAAC,eAAe,mCAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAEjF,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QAED,MAAM,IAAA,gCAAa,EAAC;YAClB,MAAM,EAAE,YAAY;YACpB,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;QAEH,KAAK,MAAM,CAAE,WAAW,EAAE,OAAO,CAAE,IAAI,IAAA,oBAAW,EAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YAEnE,IAAI,WAAW,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC/C,SAAS;YACX,CAAC;YAED,MAAM,IAAA,gCAAc,EAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAEnD,CAAC;IAEH,CAAC;CAAA;AArCD,wDAqCC;AAED,kBAAe,sBAAsB,CAAC"}

package/src/generators/documentation/has-js-doc.d.ts

@@ -0,0 +1,2 @@
+import { JSDocableNode } from 'ts-morph';
+export declare function hasJsDoc(node: JSDocableNode): boolean;

package/src/generators/documentation/has-js-doc.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.hasJsDoc = void 0;
+function hasJsDoc(node) {
+    return node.getJsDocs().length > 0;
+}
+exports.hasJsDoc = hasJsDoc;
+//# sourceMappingURL=has-js-doc.js.map

package/src/generators/documentation/has-js-doc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"has-js-doc.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/has-js-doc.ts"],"names":[],"mappings":";;;AAEA,SAAgB,QAAQ,CAAC,IAAmB;IAC1C,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;AACrC,CAAC;AAFD,4BAEC"}

package/src/generators/documentation/load-system-prompt.d.ts

@@ -0,0 +1 @@
+export declare function loadSystemPrompt(name: string): string;

package/src/generators/documentation/load-system-prompt.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadSystemPrompt = void 0;
+const fs_1 = require("fs");
+const path_1 = require("path");
+function loadSystemPrompt(name) {
+    const filePath = (0, path_1.join)(__dirname, 'system-prompts', `${name}.txt`);
+    if (!(0, fs_1.existsSync)(filePath)) {
+        throw new Error(`Can not find system prompt file '${filePath}'`);
+    }
+    return (0, fs_1.readFileSync)(filePath).toString();
+}
+exports.loadSystemPrompt = loadSystemPrompt;
+//# sourceMappingURL=load-system-prompt.js.map

package/src/generators/documentation/load-system-prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load-system-prompt.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/load-system-prompt.ts"],"names":[],"mappings":";;;AAAA,2BAGY;AACZ,+BAA4B;AAE5B,SAAgB,gBAAgB,CAAC,IAAY;IAC3C,MAAM,QAAQ,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,gBAAgB,EAAE,GAAI,IAAK,MAAM,CAAC,CAAC;IACpE,IAAI,CAAC,IAAA,eAAU,EAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,oCAAqC,QAAS,GAAG,CAAC,CAAC;IACrE,CAAC;IACD,OAAO,IAAA,iBAAY,EAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;AAC3C,CAAC;AAND,4CAMC"}

package/src/generators/documentation/open-ai-instance.d.ts

@@ -0,0 +1,4 @@
+import { OpenAI } from 'openai';
+import { ClientOptions } from 'openai/src';
+export declare function createOpenApi(param: ClientOptions): Promise<OpenAI>;
+export declare function getOpenAi(): OpenAI;

package/src/generators/documentation/open-ai-instance.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getOpenAi = exports.createOpenApi = void 0;
+const tslib_1 = require("tslib");
+const openai_1 = require("openai");
+let OPENAI_INSTANCE = null;
+function createOpenApi(param) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        if (OPENAI_INSTANCE) {
+            return OPENAI_INSTANCE;
+        }
+        OPENAI_INSTANCE = new openai_1.OpenAI(param);
+        yield OPENAI_INSTANCE.models.list()
+            .catch(() => console.error('Error listing models'))
+            .then(models => {
+            if (models) {
+                console.log('Models:', models.data.map(model => model.id));
+            }
+        });
+        return OPENAI_INSTANCE;
+    });
+}
+exports.createOpenApi = createOpenApi;
+function getOpenAi() {
+    if (!OPENAI_INSTANCE) {
+        throw new Error('OpenAI instance is not created');
+    }
+    return OPENAI_INSTANCE;
+}
+exports.getOpenAi = getOpenAi;
+//# sourceMappingURL=open-ai-instance.js.map

package/src/generators/documentation/open-ai-instance.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"open-ai-instance.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/open-ai-instance.ts"],"names":[],"mappings":";;;;AAAA,mCAAgC;AAGhC,IAAI,eAAe,GAAkB,IAAI,CAAC;AAE1C,SAAsB,aAAa,CAAC,KAAoB;;QAEtD,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,eAAe,CAAC;QACzB,CAAC;QAED,eAAe,GAAG,IAAI,eAAM,CAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE;aAChC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;aAClD,IAAI,CAAC,MAAM,CAAC,EAAE;YACb,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC,CAAC;QAEL,OAAO,eAAe,CAAC;IAEzB,CAAC;CAAA;AAlBD,sCAkBC;AAED,SAAgB,SAAS;IACvB,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IACD,OAAO,eAAe,CAAC;AACzB,CAAC;AALD,8BAKC"}

package/src/generators/documentation/process-class.d.ts

@@ -0,0 +1,3 @@
+import { ClassDeclaration } from 'ts-morph';
+import { DocumentationGeneratorSchema } from './schema';
+export declare function processClass(options: DocumentationGeneratorSchema, classDeclaration: ClassDeclaration): Promise<void>;

package/src/generators/documentation/process-class.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.processClass = void 0;
+const tslib_1 = require("tslib");
+const process_method_1 = require("./process-method");
+function processClass(options, classDeclaration) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        console.log(`====== Process class: \x1b[36m${classDeclaration.getName()}\x1b[0m`);
+        for (const methodDefinition of classDeclaration.getMethods()) {
+            yield (0, process_method_1.processMethod)(options, methodDefinition);
+        }
+    });
+}
+exports.processClass = processClass;
+//# sourceMappingURL=process-class.js.map

package/src/generators/documentation/process-class.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process-class.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/process-class.ts"],"names":[],"mappings":";;;;AACA,qDAAiD;AAGjD,SAAsB,YAAY,CAChC,OAAqC,EACrC,gBAAkC;;QAGlC,OAAO,CAAC,GAAG,CAAC,iCAAkC,gBAAgB,CAAC,OAAO,EAAG,SAAS,CAAC,CAAC;QAEpF,KAAK,MAAM,gBAAgB,IAAI,gBAAgB,CAAC,UAAU,EAAE,EAAE,CAAC;YAC7D,MAAM,IAAA,8BAAa,EAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACjD,CAAC;IAEH,CAAC;CAAA;AAXD,oCAWC"}

package/src/generators/documentation/process-function.d.ts

@@ -0,0 +1,3 @@
+import { FunctionDeclaration } from 'ts-morph';
+import { DocumentationGeneratorSchema } from './schema';
+export declare function processFunction(options: DocumentationGeneratorSchema, functionDeclaration: FunctionDeclaration): Promise<void>;

package/src/generators/documentation/process-function.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.processFunction = void 0;
+const tslib_1 = require("tslib");
+const add_js_doc_1 = require("./add-js-doc");
+const const_1 = require("./const");
+const has_js_doc_1 = require("./has-js-doc");
+const prompt_1 = require("./prompt");
+function processFunction(options, functionDeclaration) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        console.log(`====== Process function: \x1b[36m${functionDeclaration.getName()}\x1b[0m`);
+        if ((0, has_js_doc_1.hasJsDoc)(functionDeclaration)) {
+            console.log(`\x1b[33mFunction has already a documentation\x1b[0m`);
+            return;
+        }
+        const functionText = functionDeclaration.getText();
+        try {
+            const jsDoc = yield (0, prompt_1.prompt)(options, const_1.FUNCTION_SYSTEM_PROMPT, functionText);
+            (0, add_js_doc_1.addJsDoc)(options, functionDeclaration, jsDoc);
+        }
+        catch (e) {
+            console.error(`\x1b[31mError processing function: \x1b[0m${functionDeclaration.getName()}`);
+            console.error(e.message);
+            console.error(e.stack);
+            return;
+        }
+    });
+}
+exports.processFunction = processFunction;
+//# sourceMappingURL=process-function.js.map

package/src/generators/documentation/process-function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process-function.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/process-function.ts"],"names":[],"mappings":";;;;AAKA,6CAAwC;AACxC,mCAAiD;AACjD,6CAAwC;AACxC,qCAAkC;AAGlC,SAAsB,eAAe,CACnC,OAAqC,EACrC,mBAAwC;;QAGxC,OAAO,CAAC,GAAG,CAAC,oCAAqC,mBAAmB,CAAC,OAAO,EAAG,SAAS,CAAC,CAAC;QAE1F,IAAI,IAAA,qBAAQ,EAAC,mBAAmB,CAAC,EAAE,CAAC;YAClC,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;YACnE,OAAO;QACT,CAAC;QAED,MAAM,YAAY,GAAG,mBAAmB,CAAC,OAAO,EAAE,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAA,eAAM,EACxB,OAAO,EACP,8BAAsB,EACtB,YAAY,CACb,CAAC;YACF,IAAA,qBAAQ,EAAC,OAAO,EAAE,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,6CAA8C,mBAAmB,CAAC,OAAO,EAAG,EAAE,CAAC,CAAC;YAC9F,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACzB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACvB,OAAO;QACT,CAAC;IAEH,CAAC;CAAA;AA5BD,0CA4BC"}

package/src/generators/documentation/process-method.d.ts

@@ -0,0 +1,3 @@
+import { MethodDeclaration } from 'ts-morph';
+import { DocumentationGeneratorSchema } from './schema';
+export declare function processMethod(options: DocumentationGeneratorSchema, methodDeclaration: MethodDeclaration): Promise<void>;

package/src/generators/documentation/process-method.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.processMethod = void 0;
+const tslib_1 = require("tslib");
+const add_js_doc_1 = require("./add-js-doc");
+const const_1 = require("./const");
+const has_js_doc_1 = require("./has-js-doc");
+const prompt_1 = require("./prompt");
+function processMethod(options, methodDeclaration) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        console.log(`====== Process method: \x1b[36m${methodDeclaration.getName()}\x1b[0m`);
+        if ((0, has_js_doc_1.hasJsDoc)(methodDeclaration)) {
+            console.log(`\x1b[33mMethod has already a documentation\x1b[0m`);
+            return;
+        }
+        const methodText = methodDeclaration.getText();
+        try {
+            const jsDoc = yield (0, prompt_1.prompt)(options, const_1.METHOD_SYSTEM_PROMPT, methodText);
+            (0, add_js_doc_1.addJsDoc)(options, methodDeclaration, jsDoc);
+        }
+        catch (e) {
+            console.error(`\x1b[31mError processing method: \x1b[0m${methodDeclaration.getName()}`);
+            console.error(e.message);
+            console.error(e.stack);
+            return;
+        }
+    });
+}
+exports.processMethod = processMethod;
+//# sourceMappingURL=process-method.js.map

package/src/generators/documentation/process-method.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process-method.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/process-method.ts"],"names":[],"mappings":";;;;AAKA,6CAAwC;AACxC,mCAA+C;AAC/C,6CAAwC;AACxC,qCAAkC;AAGlC,SAAsB,aAAa,CACjC,OAAqC,EACrC,iBAAoC;;QAGpC,OAAO,CAAC,GAAG,CAAC,kCAAmC,iBAAiB,CAAC,OAAO,EAAG,SAAS,CAAC,CAAC;QAEtF,IAAI,IAAA,qBAAQ,EAAC,iBAAiB,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAC;YACjE,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,EAAE,CAAC;QAE/C,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAA,eAAM,EACxB,OAAO,EACP,4BAAoB,EACpB,UAAU,CACX,CAAC;YAEF,IAAA,qBAAQ,EAAC,OAAO,EAAE,iBAAiB,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,2CAA4C,iBAAiB,CAAC,OAAO,EAAG,EAAE,CAAC,CAAC;YAC1F,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YACzB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACvB,OAAO;QACT,CAAC;IAEH,CAAC;CAAA;AA7BD,sCA6BC"}

package/src/generators/documentation/process-project.d.ts

@@ -0,0 +1,3 @@
+import { Tree } from '@nx/devkit';
+import { DocumentationGeneratorSchema } from './schema';
+export declare function processProject(options: DocumentationGeneratorSchema, projectName: string, tree: Tree): Promise<void>;

package/src/generators/documentation/process-project.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.processProject = void 0;
+const tslib_1 = require("tslib");
+const workspace_ts_morph_1 = require("@rxap/workspace-ts-morph");
+const ts_morph_1 = require("@rxap/ts-morph");
+const workspace_utilities_1 = require("@rxap/workspace-utilities");
+const path_1 = require("path");
+const const_1 = require("./const");
+const process_source_file_1 = require("./process-source-file");
+const skip_source_file_1 = require("./skip-source-file");
+function processProject(options, projectName, tree) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        const projectSourceRoot = (0, workspace_utilities_1.GetProjectSourceRoot)(tree, projectName);
+        const project = (0, ts_morph_1.CreateProject)();
+        (0, workspace_ts_morph_1.AddDir)(tree, projectSourceRoot, project);
+        const start = Date.now();
+        for (const sourceFile of project.getSourceFiles()) {
+            if ((0, skip_source_file_1.skipSourceFile)(sourceFile, options.filter)) {
+                continue;
+            }
+            try {
+                yield (0, process_source_file_1.processSourceFile)(options, sourceFile);
+                (0, workspace_utilities_1.CoerceFile)(tree, (0, path_1.join)(projectSourceRoot, sourceFile.getFilePath()), sourceFile.getFullText(), true);
+            }
+            catch (e) {
+                console.error(`\x1b[31mError processing file: \x1b[0m${sourceFile.getFilePath()}`);
+                console.error(e.message);
+                console.error(e.stack);
+                break;
+            }
+            if (Date.now() - start > const_1.MAX_TIME) {
+                console.log(`\x1b[33mMax time reached: \x1b[0m${const_1.MAX_TIME}`);
+                break;
+            }
+        }
+    });
+}
+exports.processProject = processProject;
+//# sourceMappingURL=process-project.js.map

package/src/generators/documentation/process-project.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process-project.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/process-project.ts"],"names":[],"mappings":";;;;AACA,iEAAkD;AAClD,6CAA+C;AAC/C,mEAGmC;AACnC,+BAA4B;AAC5B,mCAAmC;AACnC,+DAA0D;AAE1D,yDAAoD;AAEpD,SAAsB,cAAc,CAAC,OAAqC,EAAE,WAAmB,EAAE,IAAU;;QAEzG,MAAM,iBAAiB,GAAG,IAAA,0CAAoB,EAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAElE,MAAM,OAAO,GAAG,IAAA,wBAAa,GAAE,CAAC;QAEhC,IAAA,2BAAM,EAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;QAEzC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEzB,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC;YAClD,IAAI,IAAA,iCAAc,EAAC,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/C,SAAS;YACX,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,IAAA,uCAAiB,EAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBAC7C,IAAA,gCAAU,EAAC,IAAI,EAAE,IAAA,WAAI,EAAC,iBAAiB,EAAE,UAAU,CAAC,WAAW,EAAE,CAAC,EAAE,UAAU,CAAC,WAAW,EAAE,EAAE,IAAI,CAAC,CAAC;YACtG,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,OAAO,CAAC,KAAK,CAAC,yCAA0C,UAAU,CAAC,WAAW,EAAG,EAAE,CAAC,CAAC;gBACrF,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBACvB,MAAM;YACR,CAAC;YACD,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,gBAAQ,EAAE,CAAC;gBAClC,OAAO,CAAC,GAAG,CAAC,oCAAqC,gBAAS,EAAE,CAAC,CAAC;gBAC9D,MAAM;YACR,CAAC;QACH,CAAC;IAEH,CAAC;CAAA;AA7BD,wCA6BC"}

package/src/generators/documentation/process-source-file.d.ts

@@ -0,0 +1,3 @@
+import { SourceFile } from 'ts-morph';
+import { DocumentationGeneratorSchema } from './schema';
+export declare function processSourceFile(options: DocumentationGeneratorSchema, sourceFile: SourceFile): Promise<void>;

package/src/generators/documentation/process-source-file.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.processSourceFile = void 0;
+const tslib_1 = require("tslib");
+const process_class_1 = require("./process-class");
+const process_function_1 = require("./process-function");
+function processSourceFile(options, sourceFile) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        console.log(`====== Process source file: \x1b[35m${sourceFile.getFilePath()}\x1b[0m`);
+        for (const functionDeclaration of sourceFile.getFunctions()) {
+            yield (0, process_function_1.processFunction)(options, functionDeclaration);
+        }
+        for (const classDeclaration of sourceFile.getClasses()) {
+            yield (0, process_class_1.processClass)(options, classDeclaration);
+        }
+    });
+}
+exports.processSourceFile = processSourceFile;
+//# sourceMappingURL=process-source-file.js.map

package/src/generators/documentation/process-source-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"process-source-file.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/process-source-file.ts"],"names":[],"mappings":";;;;AAIA,mDAA+C;AAC/C,yDAAqD;AAGrD,SAAsB,iBAAiB,CACrC,OAAqC,EACrC,UAAsB;;QAGtB,OAAO,CAAC,GAAG,CAAC,uCAAwC,UAAU,CAAC,WAAW,EAAG,SAAS,CAAC,CAAC;QAExF,KAAK,MAAM,mBAAmB,IAAI,UAAU,CAAC,YAAY,EAAE,EAAE,CAAC;YAC5D,MAAM,IAAA,kCAAe,EAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QACtD,CAAC;QAED,KAAK,MAAM,gBAAgB,IAAI,UAAU,CAAC,UAAU,EAAE,EAAE,CAAC;YACvD,MAAM,IAAA,4BAAY,EAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QAChD,CAAC;IAGH,CAAC;CAAA;AAhBD,8CAgBC"}

package/src/generators/documentation/prompt.d.ts

@@ -0,0 +1 @@
+export declare function prompt(options: any, systemPrompt: string, prompt: string): Promise<string>;

package/src/generators/documentation/prompt.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.prompt = void 0;
+const tslib_1 = require("tslib");
+const open_ai_instance_1 = require("./open-ai-instance");
+const simple_prompt_1 = require("./simple-prompt");
+function prompt(options, systemPrompt, prompt) {
+    return tslib_1.__awaiter(this, void 0, void 0, function* () {
+        if (options.offline) {
+            return '';
+        }
+        return (0, simple_prompt_1.SimplePrompt)(systemPrompt, prompt, (0, open_ai_instance_1.getOpenAi)());
+    });
+}
+exports.prompt = prompt;
+//# sourceMappingURL=prompt.js.map

package/src/generators/documentation/prompt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/prompt.ts"],"names":[],"mappings":";;;;AAAA,yDAA+C;AAC/C,mDAA+C;AAE/C,SAAsB,MAAM,CAAC,OAAO,EAAE,YAAoB,EAAE,MAAc;;QAExE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,IAAA,4BAAY,EACjB,YAAY,EACZ,MAAM,EACN,IAAA,4BAAS,GAAE,CACZ,CAAC;IACJ,CAAC;CAAA;AAXD,wBAWC"}

package/src/generators/documentation/schema.d.ts

@@ -1,5 +1,7 @@
 export interface DocumentationGeneratorSchema {
   openaiApiKey?: string;
+  openaiOrgId?: string;
+  openaiProjectId?: string;
   projects: string[];
   offline: boolean;
   filter: string;

package/src/generators/documentation/schema.json

@@ -23,6 +23,14 @@
     "openaiApiKey": {
       "type": "string",
       "description": "The openai api key"
+    },
+    "openaiOrgId": {
+      "type": "string",
+      "description": "The openai organization id"
+    },
+    "openaiProjectId": {
+      "type": "string",
+      "description": "The openai project id"
     }
   }
 }

package/src/generators/documentation/simple-prompt.d.ts

@@ -1,6 +1,10 @@
-import { OpenAIApi } from 'openai';
-export interface SimplePromtOptions {
+import { OpenAI } from 'openai';
+import { ChatCompletionCreateParamsBase } from 'openai/resources/chat/completions';
+export type Model = 'whisper-1' | 'dall-e-2' | 'gpt-3.5-turbo-16k' | 'tts-1-hd-1106' | 'tts-1-hd' | 'gpt-4-turbo-2024-04-09' | 'gpt-4-0125-preview' | 'gpt-4-turbo-preview' | 'gpt-4-turbo' | 'gpt-3.5-turbo-instruct-0914' | 'gpt-4o' | 'gpt-3.5-turbo-instruct' | 'text-embedding-3-small' | 'tts-1' | 'gpt-4' | 'text-embedding-3-large' | 'gpt-4-1106-preview' | 'babbage-002' | 'gpt-4-0613' | 'gpt-3.5-turbo-0125' | 'tts-1-1106' | 'dall-e-3' | 'text-embedding-ada-002' | 'davinci-002' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-1106' | 'gpt-4o-2024-05-13';
+export interface SimplePromptOptions {
     max_tokens?: number;
-    model?: 'gpt-4' | 'gpt-3.5-turbo' | 'gpt-3.5-turbo-16k';
+    model?: Model;
 }
-export declare function SimplePrompt(systemPrompt: string, prompt: string, openai: OpenAIApi, options?: SimplePromtOptions): Promise<string>;
+export declare function IsAllDefined(options: SimplePromptOptions): options is Required<SimplePromptOptions>;
+export declare function AssertAllDefined(options: SimplePromptOptions): asserts options is Required<SimplePromptOptions>;
+export declare function SimplePrompt(systemPrompt: string, prompt: string, openai: OpenAI, options?: Partial<ChatCompletionCreateParamsBase>): Promise<string>;

package/src/generators/documentation/simple-prompt.js

@@ -1,23 +1,35 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SimplePrompt = void 0;
+exports.SimplePrompt = exports.AssertAllDefined = exports.IsAllDefined = void 0;
 const tslib_1 = require("tslib");
 const gpt_3_encoder_1 = require("gpt-3-encoder");
 const tokenLimits = {
+    'gpt-4-4o': 128000,
+    'gpt-4-turbo': 128000,
     'gpt-4': 8192,
     'gpt-3.5-turbo': 4096,
     'gpt-3.5-turbo-16k': 16384,
 };
+function IsAllDefined(options) {
+    return options.max_tokens !== undefined && options.model !== undefined;
+}
+exports.IsAllDefined = IsAllDefined;
+function AssertAllDefined(options) {
+    if (!IsAllDefined(options)) {
+        throw new Error(`\x1b[31mOptions are not all defined\x1b[0m`);
+    }
+}
+exports.AssertAllDefined = AssertAllDefined;
 function SimplePrompt(systemPrompt_1, prompt_1, openai_1) {
     return tslib_1.__awaiter(this, arguments, void 0, function* (systemPrompt, prompt, openai, options = {
         max_tokens: 1024,
-        model: 'gpt-4',
+        model: 'gpt-4-turbo',
     }) {
         var _a, _b, _c, _d;
         const systemPromptLength = (0, gpt_3_encoder_1.encode)(systemPrompt).length;
         const promptLength = (0, gpt_3_encoder_1.encode)(prompt).length;
         const inputLength = Math.floor((systemPromptLength + promptLength) * 1.1);
-        (_a = options.model) !== null && _a !== void 0 ? _a : (options.model = 'gpt-4');
+        (_a = options.model) !== null && _a !== void 0 ? _a : (options.model = 'gpt-4o');
         (_b = options.max_tokens) !== null && _b !== void 0 ? _b : (options.max_tokens = 1024);
         if (!tokenLimits[options.model]) {
             throw new Error(`\x1b[31mModel '${options.model}' is not supported.\x1b[0m`);
@@ -36,7 +48,10 @@ function SimplePrompt(systemPrompt_1, prompt_1, openai_1) {
         }
         let content;
         try {
-            const response = yield openai.createChatCompletion(Object.assign(Object.assign({}, options), { messages: [
+            const response = yield openai.chat.completions.create({
+                max_tokens: options.max_tokens,
+                model: options.model,
+                messages: [
                     {
                         'role': 'system',
                         'content': systemPrompt,
@@ -45,8 +60,13 @@ function SimplePrompt(systemPrompt_1, prompt_1, openai_1) {
                         'role': 'user',
                         'content': prompt,
                     },
-                ], temperature: 0, top_p: 1, frequency_penalty: 0, presence_penalty: 0 }));
-            content = (_c = response.data.choices[0].message) === null || _c === void 0 ? void 0 : _c.content;
+                ],
+                temperature: 0,
+                top_p: 1,
+                frequency_penalty: 0,
+                presence_penalty: 0,
+            });
+            content = (_c = response.choices[0].message) === null || _c === void 0 ? void 0 : _c.content;
         }
         catch (e) {
             console.log((_d = e.response) === null || _d === void 0 ? void 0 : _d.data);

package/src/generators/documentation/simple-prompt.js.map

@@ -1 +1 @@
-{"version":3,"file":"simple-prompt.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/simple-prompt.ts"],"names":[],"mappings":";;;;AACA,iDAAuC;AAEvC,MAAM,WAAW,GAAG;IAClB,OAAO,EAAE,IAAI;IACb,eAAe,EAAE,IAAI;IACrB,mBAAmB,EAAE,KAAK;CAC3B,CAAC;AAOF,SAAsB,YAAY;iEAChC,YAAoB,EACpB,MAAc,EACd,MAAiB,EACjB,UAA8B;QAC5B,UAAU,EAAE,IAAI;QAChB,KAAK,EAAE,OAAO;KACf;;QAID,MAAM,kBAAkB,GAAG,IAAA,sBAAM,EAAC,YAAY,CAAC,CAAC,MAAM,CAAC;QACvD,MAAM,YAAY,GAAG,IAAA,sBAAM,EAAC,MAAM,CAAC,CAAC,MAAM,CAAC;QAE3C,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,kBAAkB,GAAG,YAAY,CAAC,GAAG,GAAG,CAAC,CAAC;QAE1E,MAAA,OAAO,CAAC,KAAK,oCAAb,OAAO,CAAC,KAAK,GAAK,OAAO,EAAC;QAC1B,MAAA,OAAO,CAAC,UAAU,oCAAlB,OAAO,CAAC,UAAU,GAAK,IAAI,EAAC;QAE5B,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,kBAAmB,OAAO,CAAC,KAAM,4BAA4B,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,WAAW,GAAG,UAAU,GAAG,GAAG,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,yBAA0B,WAAY,wCAAyC,UAAU;gBACzG,GAAI,SAAS,CAAC,CAAC;QACjB,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,IAAI,OAAO,CAAC,UAAU,GAAG,UAAU,GAAG,WAAW,EAAE,CAAC;gBAClD,OAAO,CAAC,GAAG,CAAC,gCAAiC,OAAO,CAAC,UAAW,wCAAyC,UAAU;oBACnH,WAAY,SAAS,CAAC,CAAC;gBACvB,OAAO,CAAC,UAAU,GAAG,UAAU,GAAG,WAAW,CAAC;YAChD,CAAC;QACH,CAAC;QAED,IAAI,OAA2B,CAAC;QAEhC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,oBAAoB,iCAC5C,OAAe,KACnB,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,QAAQ;wBAChB,SAAS,EAAE,YAAY;qBACxB;oBACD;wBACE,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,MAAM;qBAClB;iBACF,EACD,WAAW,EAAE,CAAC,EACd,KAAK,EAAE,CAAC,EACR,iBAAiB,EAAE,CAAC,EACpB,gBAAgB,EAAE,CAAC,IACnB,CAAC;YAEH,OAAO,GAAG,MAAA,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,0CAAE,OAAO,CAAC;QAEtD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,MAAA,CAAC,CAAC,QAAQ,0CAAE,IAAI,CAAC,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,oBAAoB,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,OAAO,CAAC;IAEjB,CAAC;CAAA;AAxED,oCAwEC"}
+{"version":3,"file":"simple-prompt.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/simple-prompt.ts"],"names":[],"mappings":";;;;AAAA,iDAAuC;AAIvC,MAAM,WAAW,GAAG;IAClB,UAAU,EAAE,MAAO;IACnB,aAAa,EAAE,MAAO;IACtB,OAAO,EAAE,IAAI;IACb,eAAe,EAAE,IAAI;IACrB,mBAAmB,EAAE,KAAK;CAC3B,CAAC;AASF,SAAgB,YAAY,CAAC,OAA4B;IACvD,OAAO,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC;AACzE,CAAC;AAFD,oCAEC;AAED,SAAgB,gBAAgB,CAAC,OAA4B;IAC3D,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAJD,4CAIC;AAED,SAAsB,YAAY;iEAChC,YAAoB,EACpB,MAAc,EACd,MAAc,EACd,UAAmD;QACjD,UAAU,EAAE,IAAI;QAChB,KAAK,EAAE,aAAa;KACrB;;QAGD,MAAM,kBAAkB,GAAG,IAAA,sBAAM,EAAC,YAAY,CAAC,CAAC,MAAM,CAAC;QACvD,MAAM,YAAY,GAAG,IAAA,sBAAM,EAAC,MAAM,CAAC,CAAC,MAAM,CAAC;QAE3C,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,kBAAkB,GAAG,YAAY,CAAC,GAAG,GAAG,CAAC,CAAC;QAE1E,MAAA,OAAO,CAAC,KAAK,oCAAb,OAAO,CAAC,KAAK,GAAK,QAAQ,EAAC;QAC3B,MAAA,OAAO,CAAC,UAAU,oCAAlB,OAAO,CAAC,UAAU,GAAK,IAAI,EAAC;QAE5B,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,kBAAmB,OAAO,CAAC,KAAM,4BAA4B,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAE9C,IAAI,WAAW,GAAG,UAAU,GAAG,GAAG,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,yBAA0B,WAAY,wCAAyC,UAAU;gBACzG,GAAI,SAAS,CAAC,CAAC;QACjB,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,IAAI,OAAO,CAAC,UAAU,GAAG,UAAU,GAAG,WAAW,EAAE,CAAC;gBAClD,OAAO,CAAC,GAAG,CAAC,gCAAiC,OAAO,CAAC,UAAW,wCAAyC,UAAU;oBACnH,WAAY,SAAS,CAAC,CAAC;gBACvB,OAAO,CAAC,UAAU,GAAG,UAAU,GAAG,WAAW,CAAC;YAChD,CAAC;QACH,CAAC;QAED,IAAI,OAA2B,CAAC;QAEhC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACpD,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE;oBACR;wBACE,MAAM,EAAE,QAAQ;wBAChB,SAAS,EAAE,YAAY;qBACxB;oBACD;wBACE,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,MAAM;qBAClB;iBACF;gBACD,WAAW,EAAE,CAAC;gBACd,KAAK,EAAE,CAAC;gBACR,iBAAiB,EAAE,CAAC;gBACpB,gBAAgB,EAAE,CAAC;aACpB,CAAC,CAAC;YAEH,OAAO,GAAG,MAAA,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,0CAAE,OAAO,CAAC;QAEjD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,MAAA,CAAC,CAAC,QAAQ,0CAAE,IAAI,CAAC,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,oBAAoB,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,OAAO,CAAC;IAEjB,CAAC;CAAA;AAxED,oCAwEC"}

package/src/generators/documentation/skip-source-file.d.ts

@@ -0,0 +1,2 @@
+import { SourceFile } from 'ts-morph';
+export declare function skipSourceFile(sourceFile: SourceFile, filter?: string): boolean;

package/src/generators/documentation/skip-source-file.js

@@ -0,0 +1,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.skipSourceFile = void 0;
+function skipSourceFile(sourceFile, filter) {
+    if (sourceFile.getBaseName() === 'index.ts') {
+        // console.log(`\x1b[33mSkip index file: \x1b[0m${ sourceFile.getFilePath() }`);
+        return true;
+    }
+    if (sourceFile.getBaseName().endsWith('.spec.ts')) {
+        // console.log(`\x1b[33mSkip spec file: \x1b[0m${ sourceFile.getFilePath() }`);
+        return true;
+    }
+    if (sourceFile.getBaseName().endsWith('.stories.ts')) {
+        // console.log(`\x1b[33mSkip stories file: \x1b[0m${ sourceFile.getFilePath() }`);
+        return true;
+    }
+    if (sourceFile.getBaseName().endsWith('.cy.ts')) {
+        // console.log(`\x1b[33mSkip cypress file: \x1b[0m${ sourceFile.getFilePath() }`);
+        return true;
+    }
+    if (filter) {
+        if (!sourceFile.getFilePath().includes(filter)) {
+            // console.log(`\x1b[33mSkip file: \x1b[0m${ sourceFile.getFilePath() }`);
+            return true;
+        }
+    }
+    return false;
+}
+exports.skipSourceFile = skipSourceFile;
+//# sourceMappingURL=skip-source-file.js.map

package/src/generators/documentation/skip-source-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"skip-source-file.js","sourceRoot":"","sources":["../../../../../../../packages/plugin/gpt/src/generators/documentation/skip-source-file.ts"],"names":[],"mappings":";;;AAEA,SAAgB,cAAc,CAAC,UAAsB,EAAE,MAAe;IACpE,IAAI,UAAU,CAAC,WAAW,EAAE,KAAK,UAAU,EAAE,CAAC;QAC5C,gFAAgF;QAChF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAClD,+EAA+E;QAC/E,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QACrD,kFAAkF;QAClF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,kFAAkF;QAClF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/C,0EAA0E;YAC1E,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAxBD,wCAwBC"}

package/src/generators/documentation/system-prompts/function.txt

@@ -1 +1 @@
-You're a software developer with a particular task at hand: to generate succinct, insightful, and comprehensive documentation for a TypeScript class methods. Your work should be presented in the form of a documentation comment. Please note, the output of your task should solely consist of the documentation comment, excluding any code or non-documentation elements. Remember not to alter or analyze the logic of the class method - your main focus is on producing precise and understandable documentation. The TypeScript class method to be documented is outlined below:
+You're a software developer with a particular task at hand: to generate succinct, insightful, and comprehensive documentation for a TypeScript functions. Your work should be presented in the form of a documentation comment. Please note, the output of your task should solely consist of the documentation comment, excluding any code or non-documentation elements. Remember not to alter the logic of the function - your main focus is on producing precise and understandable documentation. The TypeScript function to be documented is outlined below: