refactor: follow new ai-script

offline-ai · Aug 9, 2024 · 50127cb · 50127cb
1 parent 9ac0630
commit 50127cb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/lib/run-script.ts b/src/lib/run-script.ts
@@ -95,7 +95,7 @@ export async function runScript(filename: string, options: IRunScriptOptions) {
 
   let script
   try {
-    script = AIScriptEx.loadFile(filename, {chatsDir: options.chatsDir})
+    script = await AIScriptEx.loadFile(filename, {chatsDir: options.chatsDir})
   } catch(err) {
     console.error('Load script error:',err)
     process.exit(1)
@@ -128,11 +128,11 @@ export async function runScript(filename: string, options: IRunScriptOptions) {
   }
 
   const interrupted = async () => {
-    if (runtime.isAborted()) {
+    if (runtime.isToolAborted()) {
       await saveChatHistory()
       process.exit(0)
     } else {
-      runtime.abort()
+      runtime.abortTool()
     }
   }
   process.on('SIGINT', interrupted)
@@ -147,7 +147,7 @@ export async function runScript(filename: string, options: IRunScriptOptions) {
       let s = llmResult.content
 
       if (quit) {
-        runtime.abort('quit')
+        runtime.abortTool('quit')
         process.emit('SIGINT')
       }
       if (count !== retryCount) {