feat(examples): replace process exit with reader close

Signed-off-by: Tomas Dvorak <toomas2d@gmail.com>
i-am-bee · Oct 11, 2024 · 65b7da4 · 65b7da4
1 parent fdfca98
commit 65b7da4
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/docs/llms.md b/docs/llms.md
@@ -49,11 +49,10 @@ const llm = new WatsonXLLM({
 });
 
 const reader = createConsoleReader();
-
 const prompt = await reader.prompt();
 const response = await llm.generate(prompt);
 reader.write(`LLM 🤖 (text) : `, response.getTextContent());
-process.exit(0);
+reader.close();
 ```
 
 _Source: [examples/llms/text.ts](/examples/llms/text.ts)_
@@ -209,7 +208,6 @@ const response = await driver.generate(
   ],
 );
 console.info(response);
-process.exit(0);
 ```
 
 _Source: [examples/llms/structured.ts](/examples/llms/structured.ts)_

diff --git a/examples/agents/bee.ts b/examples/agents/bee.ts
@@ -119,5 +119,5 @@ try {
 } catch (error) {
   logger.error(FrameworkError.ensure(error).dump());
 } finally {
-  process.exit(0);
+  reader.close();
 }
diff --git a/examples/agents/bee_advanced.ts b/examples/agents/bee_advanced.ts
@@ -125,5 +125,5 @@ try {
 } catch (error) {
   logger.error(FrameworkError.ensure(error).dump());
 } finally {
-  process.exit(0);
+  reader.close();
 }
diff --git a/examples/helpers/io.ts b/examples/helpers/io.ts
@@ -33,6 +33,10 @@ export function createConsoleReader({
       }
       process.exit(0);
     },
+    close() {
+      stdin.pause();
+      rl.close();
+    },
     async *[Symbol.asyncIterator]() {
       if (!isActive) {
         return;

diff --git a/examples/llms/structured.ts b/examples/llms/structured.ts
@@ -27,4 +27,3 @@ const response = await driver.generate(
   ],
 );
 console.info(response);
-process.exit(0);
diff --git a/examples/llms/text.ts b/examples/llms/text.ts
@@ -13,8 +13,7 @@ const llm = new WatsonXLLM({
 });
 
 const reader = createConsoleReader();
-
 const prompt = await reader.prompt();
 const response = await llm.generate(prompt);
 reader.write(`LLM 🤖 (text) : `, response.getTextContent());
-process.exit(0);
+reader.close();