Add: support for abort signal in dust-client (#8570)

dust-tt · Nov 9, 2024 · b6bde37 · b6bde37
1 parent 43bbbdc
commit b6bde37
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sdks/js/src/index.ts b/sdks/js/src/index.ts
@@ -77,6 +77,7 @@ type RequestArgsType = {
   query?: URLSearchParams;
   body?: Record<string, unknown>;
   overrideWorkspaceId?: string;
+  signal?: AbortSignal;
 };
 
 export class DustAPI {
@@ -117,6 +118,10 @@ export class DustAPI {
     return this._credentials.workspaceId;
   }
 
+  setWorkspaceId(workspaceId: string) {
+    this._credentials.workspaceId = workspaceId;
+  }
+
   apiUrl(): string {
     if (this._urlOverride) {
       return this._urlOverride;
@@ -186,6 +191,7 @@ export class DustAPI {
       method: args.method,
       headers,
       body: args.body ? JSON.stringify(args.body) : undefined,
+      signal: args.signal,
     });
 
     return res;
@@ -565,9 +571,11 @@ export class DustAPI {
   async streamAgentAnswerEvents({
     conversation,
     userMessageId,
+    signal,
   }: {
     conversation: ConversationPublicType;
     userMessageId: string;
+    signal?: AbortSignal;
   }) {
     // find the agent message with the parentMessageId equal to the user message id
     const agentMessages = conversation.content
@@ -588,6 +596,7 @@ export class DustAPI {
     const res = await this.request({
       method: "GET",
       path: `assistant/conversations/${conversation.sId}/messages/${agentMessage.sId}/events`,
+      signal,
     });
 
     if (res.isErr()) {