Merge pull request #36 from CommandAI/fix_aic

double run on bang

Author: rootedbox

.vscode/launch.json

@@ -14,6 +14,32 @@
             ],
             "console": "integratedTerminal"
         },
+        {
+            "type": "node",
+            "request": "launch",
+            "name": "aiChat",
+            "skipFiles": [
+                "<node_internals>/**"
+            ],
+            "program": "${workspaceFolder}/aic.js",
+            "args": [
+                "how much would a woodchuck chuk "
+            ],
+            "console": "integratedTerminal"
+        },
+        {
+            "type": "node",
+            "request": "launch",
+            "name": "aiChat",
+            "skipFiles": [
+                "<node_internals>/**"
+            ],
+            "program": "${workspaceFolder}/aic!.js",
+            "args": [
+                "how much would a woodchuck chuk "
+            ],
+            "console": "integratedTerminal"
+        },
         {
             "type": "node",
             "request": "launch",

ai.js

@@ -147,6 +147,8 @@ async function main(continuePrompt = true) {
   }
 }
 
-main();
+if (import.meta.url === `file://${process.argv[1]}`) {
+  main();
+}
 
 export default main;

aic.js

@@ -37,7 +37,6 @@ async function askQuestion(question, client) {
     spinner.fail(gradient.cristal("Failed to get response."));
     logger.error({ error: error.message }, "Error occurred.");
   }
-  process.exit(0);
 }
 
 async function chatPrompt() {
@@ -72,7 +71,7 @@ async function startConversation(client, initialQuestion) {
   }
 
   console.log(gradient.morning("Goodbye!"));
-  process.exit();
+  process.exit(0);
 }
 
 function shouldExitConversation(question, exitWords) {
@@ -89,12 +88,15 @@ async function main(startConvo = true) {
     } else {
       await askQuestion(command, client);
     }
+    process.exit(0);
   } catch (error) {
     logger.error(`An error occurred: ${error.message}`);
     process.exit(1);
   }
 }
 
-main();
+if (import.meta.url === `file://${process.argv[1]}`) {
+  main();
+}
 
 export default main;

aiq.js

@@ -487,6 +487,8 @@ async function main(prompt = true) {
   process.exit(0);
 }
 
-main();
+if (import.meta.url === `file://${process.argv[1]}`) {
+  main();
+}
 
 export default main;