add start stop script

microsoft · Jun 22, 2024 · 2adc07a · 2adc07a
1 parent bf636c7
commit 2adc07a
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/packages/core/src/server/client.ts b/packages/core/src/server/client.ts
@@ -29,6 +29,9 @@ import {
     ContainerStartResponse,
     ContainerStart,
     ContainerRemove,
+    PromptScriptRunOptions,
+    PromptScriptStart,
+    PromptScriptAbort,
 } from "./messages"
 
 export class WebSocketClient
@@ -195,6 +198,29 @@ export class WebSocketClient
         return res.response
     }
 
+    async startScript(
+        script: string,
+        files: string[],
+        options: PromptScriptRunOptions
+    ): Promise<PromptScriptTestRunResponse> {
+        const res = await this.queue<PromptScriptStart>({
+            type: "script.start",
+            script,
+            files,
+            options,
+        })
+        return res.response
+    }
+
+    async abortScript(runId: string, reason?: string): Promise<ResponseStatus> {
+        const res = await this.queue<PromptScriptAbort>({
+            type: "script.abort",
+            runId,
+            reason,
+        })
+        return res.response
+    }
+
     async runTest(
         script: PromptScript,
         options?: PromptScriptTestRunOptions