feat: search scripts can be aborted by ctrl+C to avoid stuck

offline-ai · Aug 14, 2024 · 55548f5 · 55548f5
1 parent 5e02208
commit 55548f5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/README.md b/README.md
@@ -88,7 +88,7 @@ $ npm install -g @offline-ai/cli
 $ ai COMMAND
 running command...
 $ ai (--version)
-@offline-ai/cli/0.1.1 linux-x64 node-v20.14.0
+@offline-ai/cli/0.1.2 linux-x64 node-v20.14.0
 $ ai --help [COMMAND]
 USAGE
   $ ai COMMAND
@@ -318,7 +318,7 @@ EXAMPLES
   $ ai agent publish <agent-name>
 ```
 
-_See code: [src/commands/agent/index.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/agent/index.ts)_
+_See code: [src/commands/agent/index.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/agent/index.ts)_
 
 ## `ai autocomplete [SHELL]`
 
@@ -393,7 +393,7 @@ EXAMPLES
   $ ai brain download <brain-name>
 ```
 
-_See code: [src/commands/brain/index.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/brain/index.ts)_
+_See code: [src/commands/brain/index.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/brain/index.ts)_
 
 ## `ai brain dn [NAME]`
 
@@ -524,7 +524,7 @@ EXAMPLES
   $ ai brain download <brain-name> [-q <QUANT>]
 ```
 
-_See code: [src/commands/brain/download.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/brain/download.ts)_
+_See code: [src/commands/brain/download.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/brain/download.ts)_
 
 ## `ai brain list [NAME]`
 
@@ -557,7 +557,7 @@ ALIASES
   $ ai brain search
 ```
 
-_See code: [src/commands/brain/list.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/brain/list.ts)_
+_See code: [src/commands/brain/list.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/brain/list.ts)_
 
 ## `ai brain refresh`
 
@@ -581,7 +581,7 @@ DESCRIPTION
   refresh brain index from huggingface.co
 ```
 
-_See code: [src/commands/brain/refresh.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/brain/refresh.ts)_
+_See code: [src/commands/brain/refresh.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/brain/refresh.ts)_
 
 ## `ai brain search [NAME]`
 
@@ -649,7 +649,7 @@ EXAMPLES
   }
 ```
 
-_See code: [src/commands/config/index.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/config/index.ts)_
+_See code: [src/commands/config/index.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/config/index.ts)_
 
 ## `ai config save [DATA]`
 
@@ -690,7 +690,7 @@ GLOBAL FLAGS
   --json  Format output as json.
 ```
 
-_See code: [src/commands/config/save.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/config/save.ts)_
+_See code: [src/commands/config/save.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/config/save.ts)_
 
 ## `ai help [COMMAND]`
 
@@ -1052,7 +1052,7 @@ EXAMPLES
   │[info]:Start Script: ...
 ```
 
-_See code: [src/commands/run/index.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/run/index.ts)_
+_See code: [src/commands/run/index.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/run/index.ts)_
 
 ## `ai test`
 
@@ -1100,7 +1100,7 @@ EXAMPLES
   $ ai test -f ./fixture.yaml -l info
 ```
 
-_See code: [src/commands/test/index.ts](https://github.com/offline-ai/cli/blob/v0.1.1/src/commands/test/index.ts)_
+_See code: [src/commands/test/index.ts](https://github.com/offline-ai/cli/blob/v0.1.2/src/commands/test/index.ts)_
 
 ## `ai version`
 

diff --git a/src/lib/run-script.ts b/src/lib/run-script.ts
@@ -12,7 +12,6 @@ import { AIScriptServer, LogLevel, LogLevelMap } from '@isdk/ai-tool-agent'
 import { detectTextLanguage as detectLang } from '@isdk/detect-text-language'
 import { prompt, setHistoryStore, HistoryStore } from './prompt.js'
 // import { initTools } from './init-tools.js'
-import { expandPath } from './load-config.js'
 
 class AIScriptEx extends AIScriptServer {
   $detectLang(text: string) {
@@ -97,8 +96,13 @@ export async function runScript(filename: string, options: IRunScriptOptions) {
   }
 
   let script
+  const aborter = new AbortController()
+  process.on('SIGINT', () => {
+    aborter.abort()}
+  )
+
   try {
-    script = await AIScriptEx.loadFile(filename, {chatsDir: options.chatsDir})
+    script = await AIScriptEx.loadFile(filename, {chatsDir: options.chatsDir}, {ABORT_SEARCH_SCRIPTS_SIGNAL: aborter.signal})
   } catch(err) {
     console.error('Load script error:',err)
     process.exit(1)