@proteinjs/conversation 1.0.5 → 1.0.7

package/src/fs/package/PackageModule.ts

@@ -1,21 +1,27 @@
 import { Fs } from '@proteinjs/util-node';
 import { ConversationModule, ConversationModuleFactory } from '../../ConversationModule';
 import { Function } from '../../Function';
-import { packageFunctions, searchLibrariesFunction, searchLibrariesFunctionName, searchPackagesFunction, searchPackagesFunctionName } from './PackageFunctions';
+import {
+  packageFunctions,
+  searchLibrariesFunction,
+  searchLibrariesFunctionName,
+  searchPackagesFunction,
+  searchPackagesFunctionName,
+} from './PackageFunctions';
 import path from 'path';
 import { searchFilesFunctionName } from '../keyword_to_files_index/KeywordToFilesIndexFunctions';
 import { readFilesFunctionName } from '../conversation_fs/FsFunctions';
 
 export type Library = {
-  fileName: string,
-  filePath: string,
-  packageName: string,
-}
+  fileName: string;
+  filePath: string;
+  packageName: string;
+};
 
 export type LibraryImport = {
-  importStatements: string[],
-  typescriptDeclaration: string,
-}
+  importStatements: string[];
+  typescriptDeclaration: string;
+};
 
 export class PackageModule implements ConversationModule {
   private repoPath: string;
@@ -39,11 +45,7 @@ export class PackageModule implements ConversationModule {
   }
 
   getFunctions(): Function[] {
-    return [
-      ...packageFunctions,
-      searchPackagesFunction(this),
-      searchLibrariesFunction(this),
-    ];
+    return [...packageFunctions, searchPackagesFunction(this), searchLibrariesFunction(this)];
   }
 
   getMessageModerators() {
@@ -56,12 +58,16 @@ export class PackageModule implements ConversationModule {
    */
   async searchPackages(keyword: string): Promise<string[]> {
     const matchingPackageJsonPaths: string[] = [];
-    const packageJsonFilePaths = await Fs.getFilePathsMatchingGlob(this.repoPath, '**/package.json', ['**/node_modules/**', '**/dist/**']);
+    const packageJsonFilePaths = await Fs.getFilePathsMatchingGlob(this.repoPath, '**/package.json', [
+      '**/node_modules/**',
+      '**/dist/**',
+    ]);
     const packageJsonFileMap = await Fs.readFiles(packageJsonFilePaths);
-    for (let packageJsonFilePath of Object.keys(packageJsonFileMap)) {
+    for (const packageJsonFilePath of Object.keys(packageJsonFileMap)) {
       const packageJson = JSON.parse(packageJsonFileMap[packageJsonFilePath]);
-      if (packageJson.name.toLowerCase().includes(keyword.toLocaleLowerCase()))
+      if (packageJson.name.toLowerCase().includes(keyword.toLocaleLowerCase())) {
         matchingPackageJsonPaths.push(packageJsonFilePath);
+      }
     }
 
     return matchingPackageJsonPaths;
@@ -74,15 +80,21 @@ export class PackageModule implements ConversationModule {
    */
   async searchLibraries(keyword: string): Promise<Library[]> {
     const matchingLibraries: Library[] = [];
-    const packageJsonFilePaths = await Fs.getFilePathsMatchingGlob(this.repoPath, '**/package.json', ['**/node_modules/**', '**/dist/**']);
+    const packageJsonFilePaths = await Fs.getFilePathsMatchingGlob(this.repoPath, '**/package.json', [
+      '**/node_modules/**',
+      '**/dist/**',
+    ]);
     const packageJsonFileMap = await Fs.readFiles(packageJsonFilePaths);
-    for (let packageJsonFilePath of Object.keys(packageJsonFileMap)) {
+    for (const packageJsonFilePath of Object.keys(packageJsonFileMap)) {
       const packageJson = JSON.parse(packageJsonFileMap[packageJsonFilePath]);
       const packageJsonFilePathParts = packageJsonFilePath.split(path.sep);
       packageJsonFilePathParts.pop();
       const packageDirectory = packageJsonFilePathParts.join(path.sep);
-      const srcFilePaths = await Fs.getFilePathsMatchingGlob(path.join(packageDirectory, 'src'), '**/*.ts', ['**/node_modules/**', '**/dist/**']);
-      for (let srcFilePath of srcFilePaths) {
+      const srcFilePaths = await Fs.getFilePathsMatchingGlob(path.join(packageDirectory, 'src'), '**/*.ts', [
+        '**/node_modules/**',
+        '**/dist/**',
+      ]);
+      for (const srcFilePath of srcFilePaths) {
         const fileNameWithExtension = path.basename(srcFilePath);
         if (fileNameWithExtension.includes(keyword)) {
           const fileName = path.basename(srcFilePath, path.extname(srcFilePath));
@@ -103,4 +115,4 @@ export class PackageModuleFactory implements ConversationModuleFactory {
   async createModule(repoPath: string): Promise<PackageModule> {
     return new PackageModule(repoPath);
   }
-}
+}

package/src/history/MessageHistory.ts

@@ -20,7 +20,7 @@ export class MessageHistory {
   }
 
   toString() {
-    return this.messages.map(message => message.content).join('. ');
+    return this.messages.map((message) => message.content).join('. ');
   }
 
   setMessages(messages: ChatCompletionMessageParam[]): MessageHistory {
@@ -36,19 +36,22 @@ export class MessageHistory {
   }
 
   prune() {
-    if (this.params.enforceMessageLimit === false)
+    if (this.params.enforceMessageLimit === false) {
       return;
+    }
 
     let messageCount = 0;
     const messagesToRemoveIndexes: number[] = [];
     for (let i = this.messages.length - 1; i >= 0; i--) {
       const message = this.messages[i];
-      if (message.role == 'system')
+      if (message.role == 'system') {
         continue;
+      }
 
       messageCount++;
-      if (messageCount > this.params.maxMessages)
+      if (messageCount > this.params.maxMessages) {
         messagesToRemoveIndexes.push(i);
+      }
     }
 
     this.messages = this.messages.filter((message, i) => !messagesToRemoveIndexes.includes(i));

package/src/history/MessageModerator.ts

@@ -3,4 +3,4 @@ import { ChatCompletionMessageParam } from 'openai/resources/chat';
 export interface MessageModerator {
   // given a set of messages, modify and return
   observe(messages: ChatCompletionMessageParam[]): ChatCompletionMessageParam[];
-}
+}

package/src/template/ConversationTemplate.ts

@@ -1,12 +1,12 @@
 export type Question = {
-  text: string,
-  optional?: boolean,
-}
+  text: string;
+  optional?: boolean;
+};
 
 export type ConversationTemplate = {
-  name: string,
-  keywords: string[],
-  description: string,
-  questions: Question[],
-  instructions: () => Promise<string[]>,
-}
+  name: string;
+  keywords: string[];
+  description: string;
+  questions: Question[];
+  instructions: () => Promise<string[]>;
+};

package/src/template/ConversationTemplateFunctions.ts

@@ -14,12 +14,12 @@ export const searchConversationTemplatesFunction = (repo: ConversationTemplateMo
             description: 'Search for conversation template names that match this keyword',
           },
         },
-        required: ['keyword']
+        required: ['keyword'],
       },
     },
     call: async (params: { keyword: string }) => repo.searchConversationTemplates(params.keyword),
-  }
-}
+  };
+};
 
 export const getConversationTemplateFunctionName = 'getConversationTemplate';
 export const getConversationTemplateFunction = (repo: ConversationTemplateModule) => {
@@ -35,9 +35,10 @@ export const getConversationTemplateFunction = (repo: ConversationTemplateModule
             description: 'Get the conversation template that has this name',
           },
         },
-        required: ['conversationTemplateName']
+        required: ['conversationTemplateName'],
       },
     },
-    call: async (params: { conversationTemplateName: string }) => await repo.getConversationTemplate(params.conversationTemplateName),
-  }
-}
+    call: async (params: { conversationTemplateName: string }) =>
+      await repo.getConversationTemplate(params.conversationTemplateName),
+  };
+};

package/src/template/ConversationTemplateModule.ts

@@ -2,7 +2,12 @@ import { Logger } from '@proteinjs/util';
 import { ConversationTemplate } from './ConversationTemplate';
 import { createPackageConversationTemplate } from './createPackage/CreatePackageConversationTemplate';
 import { ConversationModule, ConversationModuleFactory } from '../ConversationModule';
-import { getConversationTemplateFunction, getConversationTemplateFunctionName, searchConversationTemplatesFunction, searchConversationTemplatesFunctionName } from './ConversationTemplateFunctions';
+import {
+  getConversationTemplateFunction,
+  getConversationTemplateFunctionName,
+  searchConversationTemplatesFunction,
+  searchConversationTemplatesFunctionName,
+} from './ConversationTemplateFunctions';
 import { createCodeConversationTemplate } from './createCode/CreateCodeConversationTemplate';
 import { createAppTemplate } from './createApp/CreateAppTemplate';
 
@@ -13,9 +18,9 @@ const conversationTemplates: ConversationTemplate[] = [
 ];
 
 export type ConversationTemplateModuleParams = {
-  conversationTemplates: { [conversationTemplateName: string]: ConversationTemplate},
-  conversationTemplateKeywordIndex: { [keyword: string]: string[] /** conversationTemplateNames */ }
-}
+  conversationTemplates: { [conversationTemplateName: string]: ConversationTemplate };
+  conversationTemplateKeywordIndex: { [keyword: string]: string[] /** conversationTemplateNames */ };
+};
 
 export class ConversationTemplateModule implements ConversationModule {
   private logger = new Logger(this.constructor.name);
@@ -35,10 +40,13 @@ export class ConversationTemplateModule implements ConversationModule {
     return conversationNames || [];
   }
 
-  async getConversationTemplate(conversationTemplateName: string): Promise<Omit<ConversationTemplate, 'instructions'> & { instructions: string[] }> {
+  async getConversationTemplate(
+    conversationTemplateName: string
+  ): Promise<Omit<ConversationTemplate, 'instructions'> & { instructions: string[] }> {
     const conversationTemplate = this.params.conversationTemplates[conversationTemplateName];
-    if (!conversationTemplate)
+    if (!conversationTemplate) {
       return {} as any;
+    }
 
     const instructions = await conversationTemplate.instructions();
     return Object.assign(conversationTemplate, { instructions });
@@ -54,10 +62,7 @@ export class ConversationTemplateModule implements ConversationModule {
   }
 
   getFunctions() {
-    return [
-      searchConversationTemplatesFunction(this),
-      getConversationTemplateFunction(this),
-    ];
+    return [searchConversationTemplatesFunction(this), getConversationTemplateFunction(this)];
   }
 
   getMessageModerators() {
@@ -67,12 +72,16 @@ export class ConversationTemplateModule implements ConversationModule {
 
 export class ConversationTemplateModuleFactory implements ConversationModuleFactory {
   async createModule(repoPath: string) {
-    const params: ConversationTemplateModuleParams = { conversationTemplates: {}, conversationTemplateKeywordIndex: {} };
-    for (let conversationTemplate of conversationTemplates) {
+    const params: ConversationTemplateModuleParams = {
+      conversationTemplates: {},
+      conversationTemplateKeywordIndex: {},
+    };
+    for (const conversationTemplate of conversationTemplates) {
       params.conversationTemplates[conversationTemplate.name] = conversationTemplate;
-      for (let keyword of conversationTemplate.keywords) {
-        if (!params.conversationTemplateKeywordIndex[keyword])
+      for (const keyword of conversationTemplate.keywords) {
+        if (!params.conversationTemplateKeywordIndex[keyword]) {
           params.conversationTemplateKeywordIndex[keyword] = [];
+        }
 
         params.conversationTemplateKeywordIndex[keyword].push(conversationTemplate.name);
       }
@@ -80,4 +89,4 @@ export class ConversationTemplateModuleFactory implements ConversationModuleFact
 
     return new ConversationTemplateModule(params);
   }
-}
+}

package/src/template/createApp/CreateAppTemplate.ts

@@ -30,4 +30,4 @@ export const createAppTemplate: ConversationTemplate = {
   description: 'This template will guide you through the process of creating a new app.',
   questions: createAppQuestions,
   instructions: createAppInstructions,
-};
+};

package/src/template/createCode/CreateCodeConversationTemplate.ts

@@ -1,17 +1,15 @@
 import { fileOrDirectoryExistsFunction, readFilesFunctionName } from '../../fs/conversation_fs/FsFunctions';
-import { generateTypescriptDeclarationsFunction, installPackagesFunction, searchLibrariesFunctionName, searchPackagesFunctionName } from '../../fs/package/PackageFunctions';
+import {
+  generateTypescriptDeclarationsFunction,
+  installPackagesFunction,
+  searchLibrariesFunctionName,
+  searchPackagesFunctionName,
+} from '../../fs/package/PackageFunctions';
 import { ConversationTemplate } from '../ConversationTemplate';
 
 export const createCodeConversationTemplate: ConversationTemplate = {
   name: 'Create code',
-  keywords: [
-    'create code',
-    'implement',
-    'write code',
-    'generate code',
-    'write software',
-    'build something',
-  ],
+  keywords: ['create code', 'implement', 'write code', 'generate code', 'write software', 'build something'],
   description: 'User and ai developing together',
   questions: [],
   instructions: async () => {
@@ -37,5 +35,5 @@ export const createCodeConversationTemplate: ConversationTemplate = {
       `3.c. When writing the code to file, if updating an existing file, be sure to read the file first to not blow away existing code. Be sure to preserve comments as well.`,
       `4. Repeat 3. unless the user asks to switch packages or files`,
     ];
-  }
-}
+  },
+};

package/src/template/createPackage/CreatePackageConversationTemplate.ts

@@ -5,14 +5,7 @@ import tConfig from './tsconfig.json';
 
 export const createPackageConversationTemplate: ConversationTemplate = {
   name: 'Create Package',
-  keywords: [
-    'package',
-    'create package',
-    'create',
-    'create a package',
-    'create a new package',
-    `new package`,
-  ],
+  keywords: ['package', 'create package', 'create', 'create a package', 'create a new package', `new package`],
   description: 'Create a npm package',
   questions: [
     { text: `Which folder should the package be created in?` },
@@ -39,4 +32,4 @@ export const createPackageConversationTemplate: ConversationTemplate = {
       `If it's a typescript project, create a tsconfig.json file in the pakage directory with the following content: ${tsConfig}`,
     ];
   },
-}
+};

package/src/template/createPackage/jest.config.js

@@ -4,6 +4,6 @@ module.exports = {
   transform: {
     '^.+\\.tsx?$': 'ts-jest',
   },
-  testMatch: ["**/?(*.)+(spec|test).ts"],
+  testMatch: ['**/?(*.)+(spec|test).ts'],
   setupFiles: ['./test/setup'],
-};
+};

package/test/createKeywordFilesIndex.test.ts

@@ -1,7 +1,7 @@
 import { KeywordToFilesIndexModuleFactory } from '../src/fs/keyword_to_files_index/KeywordToFilesIndexModule';
 
-test('Create keyword-files index', async() => {
+test('Create keyword-files index', async () => {
   // Example usage
-  const index = new KeywordToFilesIndexModuleFactory().createKeywordFilesIndex(`${process.cwd()}`)
+  const index = new KeywordToFilesIndexModuleFactory().createKeywordFilesIndex(`${process.cwd()}`);
   console.log(JSON.stringify(index, null, 2));
-}, 60000)
+}, 60000);

package/test/openai/openai.generateList.test.ts

@@ -1,6 +1,8 @@
-import { OpenAi } from "../../src/OpenAi";
+import { OpenAi } from '../../src/OpenAi';
 
 test('generateList should return an array of numbers, counting to 10', async () => {
   const numbers = ['one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine', 'ten'];
-  expect((await OpenAi.generateList([`Create a list of numbers spelled out, from 1 to 10`])).join(' ')).toBe(numbers.join(' '));
-});
+  expect((await OpenAi.generateList([`Create a list of numbers spelled out, from 1 to 10`])).join(' ')).toBe(
+    numbers.join(' ')
+  );
+});

package/test/openai/openai.parseCodeFromMarkdown.test.ts

@@ -1,11 +1,16 @@
-import { OpenAi } from "../../src/OpenAi";
+import { OpenAi } from '../../src/OpenAi';
 
 const helloWorldCode = "console.log('hello world');";
 const helloWorldWithTicksCode = "console.log('hello ```world');";
 const logXCode = "const x = 'yo';\nconsole.log(x);";
-const testOneBlock = "Sure! I'm a helpful chatbot.\n```typescript\n" + helloWorldCode + "\n```";
-const testOneBlockWithExtraTicks = "Sure! I'm a helpful chatbot.\n```typescript\n" + helloWorldWithTicksCode + "\n```";
-const testTwoBlocks = "Sure! I'm a helpful chatbot.\n```typescript\n" + helloWorldCode + "\n```\n\nI'm still really helpful.\n```typescript\n" + logXCode + "\n```\n\nMore unhelpful chat";
+const testOneBlock = "Sure! I'm a helpful chatbot.\n```typescript\n" + helloWorldCode + '\n```';
+const testOneBlockWithExtraTicks = "Sure! I'm a helpful chatbot.\n```typescript\n" + helloWorldWithTicksCode + '\n```';
+const testTwoBlocks =
+  "Sure! I'm a helpful chatbot.\n```typescript\n" +
+  helloWorldCode +
+  "\n```\n\nI'm still really helpful.\n```typescript\n" +
+  logXCode +
+  '\n```\n\nMore unhelpful chat';
 
 test('parseCodeFromMarkdown should parse 1 block of code', () => {
   expect(OpenAi.parseCodeFromMarkdown(testOneBlock)).toBe(helloWorldCode);
@@ -17,4 +22,4 @@ test('parseCodeFromMarkdown should parse 1 block of code that contains ticks', (
 
 test('parseCodeFromMarkdown should parse 1 block of code', () => {
   expect(OpenAi.parseCodeFromMarkdown(testTwoBlocks)).toBe(helloWorldCode + '\n\n' + logXCode);
-});
+});

package/test/repo/repo.test.ts

@@ -3,7 +3,6 @@ import { RepoFactory } from '../../src/code_template/Repo';
 import { Conversation } from '../../src/Conversation';
 import { Fs } from '@proteinjs/util-node';
 
-
 // test(`Return repo of the source files and the code they depend on`, async () => {
 //   const tableTemplatePath = require.resolve('conversation/src/table.ts');
 //   const repo = await new RepoFactory([tableTemplatePath]).create();
@@ -15,12 +14,12 @@ import { Fs } from '@proteinjs/util-node';
 // }, 60000);
 
 // test(`Interpret all parameters of a function`, async () => {
-  
+
 // }, 60000);
 
 // test(`Gather parameter data from the user`, async () => {
 //   const conversation = new Conversation();
-  
+
 // }, 60000);
 
 // test(`Generate code from templates guided by user input`, async () => {
@@ -30,4 +29,4 @@ import { Fs } from '@proteinjs/util-node';
 test(`Create repo`, async () => {
   // console.log(JSON.stringify((await RepoFactory.createRepo(`${process.cwd()}`)).params, null, 4));
   // console.log(JSON.stringify(await Fs.getFilesInDirectory(`${process.cwd()}`, ['node_modules', 'dist']), null, 4));
-}, 60000)
+}, 60000);

package/tsconfig.json

@@ -1,23 +1,19 @@
 {
   "compilerOptions": {
-      "rootDir": "./",
-      "target": "es5",
-      "module": "commonjs",
-      "declaration": true,
-      "declarationMap": true,
-      "sourceMap": true,
-      "outDir": "./dist/",
-      "strict": true,
-      "noImplicitAny": true,
-      "esModuleInterop": true,
-      "skipLibCheck": true,
-      "forceConsistentCasingInFileNames": true,
-      "resolveJsonModule": true,
-      "typeRoots": [
-          "./node_modules/@types"
-      ],
-      "types": [
-          "node", "jest"
-      ]
+    "rootDir": "./",
+    "target": "es5",
+    "module": "commonjs",
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "outDir": "./dist/",
+    "strict": true,
+    "noImplicitAny": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "typeRoots": ["./node_modules/@types"],
+    "types": ["node", "jest"]
   }
-}
+}