Added stop feature

src/helpers/completion.ts

@@ -9,12 +9,13 @@ import { streamToString } from './stream-to-string';
 import './replace-all-polyfill';
 import i18n from './i18n';
 import { stripRegexPatterns } from './strip-regex-patterns';
+import readline from 'readline';
 
 const explainInSecondRequest = true;
 
 function getOpenAi(key: string, apiEndpoint: string) {
   const openAi = new OpenAIApi(
-    new Configuration({ apiKey: key, basePath: apiEndpoint })
+    new Configuration({ apiKey: key, basePath: apiEndpoint }),
   );
   return openAi;
 }
@@ -73,7 +74,7 @@ export async function generateCompletion({
         n: Math.min(number, 10),
         stream: true,
       },
-      { responseType: 'stream' }
+      { responseType: 'stream' },
     );
 
     return completion.data as unknown as IncomingMessage;
@@ -82,15 +83,15 @@ export async function generateCompletion({
 
     if (error.code === 'ENOTFOUND') {
       throw new KnownError(
-        `Error connecting to ${error.request.hostname} (${error.request.syscall}). Are you connected to the internet?`
+        `Error connecting to ${error.request.hostname} (${error.request.syscall}). Are you connected to the internet?`,
       );
     }
 
     const response = error.response;
     let message = response?.data as string | object | IncomingMessage;
     if (response && message instanceof IncomingMessage) {
       message = await streamToString(
-        response.data as unknown as IncomingMessage
+        response.data as unknown as IncomingMessage,
       );
       try {
         // Handle if the message is JSON. It should be but occasionally will
@@ -113,7 +114,7 @@ export async function generateCompletion({
       ` +
           '\n\n' +
           messageString +
-          '\n'
+          '\n',
       );
     } else if (response && message) {
       throw new KnownError(
@@ -122,7 +123,7 @@ export async function generateCompletion({
       ` +
           '\n\n' +
           messageString +
-          '\n'
+          '\n',
       );
     }
 
@@ -187,6 +188,19 @@ export const readData =
   ) =>
   (writer: (data: string) => void): Promise<string> =>
     new Promise(async (resolve) => {
+      const rl = readline.createInterface({
+        input: process.stdin,
+      });
+
+      rl.input.setRawMode(true);
+
+      rl.input.on('keypress', (key) => {
+        if (typeof key === 'string' && key.toLowerCase().includes('q')) {
+          stopWriting = true;
+        }
+      });
+
+      let stopWriting = false;
       let data = '';
       let content = '';
       let dataStart = false;
@@ -199,7 +213,7 @@ export const readData =
         const payloads = chunk.toString().split('\n\n');
 
         for (const payload of payloads) {
-          if (payload.includes('[DONE]')) {
+          if (payload.includes('[DONE]' || stopWriting)) {
             dataStart = false;
             resolve(data);
             return;
@@ -222,7 +236,7 @@ export const readData =
             if (dataStart && content) {
               const contentWithoutExcluded = stripRegexPatterns(
                 content,
-                excluded
+                excluded,
               );
 
               data += contentWithoutExcluded;