gitlab-ai-provider 6.2.0 → 6.3.0

package/dist/index.mjs

@@ -1582,7 +1582,7 @@ var GitLabOpenAILanguageModel = class {
 import WebSocket from "isomorphic-ws";
 
 // src/version.ts
-var VERSION = true ? "6.1.1" : "0.0.0-dev";
+var VERSION = true ? "6.2.1" : "0.0.0-dev";
 
 // src/gitlab-workflow-types.ts
 var WorkflowType = /* @__PURE__ */ ((WorkflowType2) => {
@@ -2975,14 +2975,12 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
   // Cached detected project path
   detectedProjectPath = null;
   // Workflow ID persisted across turns for multi-turn conversations.
-  // When DWS sends INPUT_REQUIRED, the workflow stays alive server-side.
-  // On the next doStream() call we reuse this ID (skip createWorkflow).
+  // Per-session workflow state. Keyed by opencode sessionID (x-opencode-session header).
+  // Each opencode session gets its own DWS workflow, reused across turns within that session.
+  sessionWorkflows = /* @__PURE__ */ new Map();
+  // Fallback for callers that don't pass x-opencode-session (e.g. direct SDK use).
   currentWorkflowId = null;
-  // Track which workflowDefinition/flowConfig the current workflow was created with.
-  // When the agent changes, reset the workflow so a new one is created correctly.
   currentWorkflowDefinition = null;
-  // Persisted across turns so that cumulative DWS chat logs don't re-emit
-  // messages that were already streamed in a previous doStream() call.
   persistedAgentEmitted = /* @__PURE__ */ new Map();
   // Track all active stream clients so stopWorkflow() can stop them all.
   activeClients = /* @__PURE__ */ new Set();
@@ -3044,6 +3042,11 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
    * Updated when the user chooses "always" in the approval prompt.
    */
   sessionPreapprovedTools = [];
+  /**
+   * The opencode session ID. Set per-stream by the host to key per-session
+   * DWS workflows. Different sessions get different DWS workflows.
+   */
+  sessionID = "";
   /**
    * Set the approval handler callback.
    * Called when DWS requires tool call approval. Host (e.g., opencode) wires this
@@ -3273,9 +3276,14 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
    * Reset the workflow state, forcing a new workflow to be created on the
    * next doStream() call. Call this when starting a new conversation.
    */
-  resetWorkflow() {
-    this.currentWorkflowId = null;
-    this.persistedAgentEmitted.clear();
+  resetWorkflow(sessionKey) {
+    if (sessionKey) {
+      this.sessionWorkflows.delete(sessionKey);
+    } else {
+      this.sessionWorkflows.clear();
+      this.currentWorkflowId = null;
+      this.persistedAgentEmitted.clear();
+    }
   }
   /**
    * Get the current workflow ID (if any).
@@ -3374,12 +3382,21 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
     );
     const projectId = await this.resolveProjectId();
     const effectiveDefinition = callFlowConfig ?? callWorkflowDefinition ?? DEFAULT_WORKFLOW_DEFINITION;
-    if (this.currentWorkflowId && this.currentWorkflowDefinition !== effectiveDefinition) {
+    const sessionKey = this.sessionID;
+    let sess = this.sessionWorkflows.get(sessionKey);
+    if (sess && sess.workflowDefinition !== effectiveDefinition) {
+      this.sessionWorkflows.delete(sessionKey);
+      sess = void 0;
+    }
+    const useLegacy = sessionKey === "";
+    if (useLegacy && this.currentWorkflowId && this.currentWorkflowDefinition !== effectiveDefinition) {
       this.currentWorkflowId = null;
       this.currentWorkflowDefinition = null;
     }
     let workflowId;
-    if (this.currentWorkflowId) {
+    if (sess) {
+      workflowId = sess.workflowId;
+    } else if (useLegacy && this.currentWorkflowId) {
       workflowId = this.currentWorkflowId;
     } else {
       workflowId = await this.tokenClient.createWorkflow(goal, {
@@ -3389,8 +3406,17 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
         agentPrivileges: this.workflowOptions.agentPrivileges,
         aiCatalogItemVersionId: callAiCatalogItemVersionId
       });
-      this.currentWorkflowId = workflowId;
-      this.currentWorkflowDefinition = effectiveDefinition;
+      if (useLegacy) {
+        this.currentWorkflowId = workflowId;
+        this.currentWorkflowDefinition = effectiveDefinition;
+      } else {
+        this.sessionWorkflows.set(sessionKey, {
+          workflowId,
+          workflowDefinition: effectiveDefinition,
+          agentEmitted: /* @__PURE__ */ new Map()
+        });
+        sess = this.sessionWorkflows.get(sessionKey);
+      }
     }
     const wsClient = new GitLabWorkflowClient();
     this.activeClients.add(wsClient);
@@ -3403,10 +3429,11 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
       approvalPending: false,
       deferredClose: null,
       activeTextBlockId: null,
-      agentMessageEmitted: new Map(this.persistedAgentEmitted),
+      agentMessageEmitted: new Map(sess?.agentEmitted ?? this.persistedAgentEmitted),
       currentAgentMessageId: "",
       activeClient: wsClient,
-      processedRequestIDs: /* @__PURE__ */ new Set()
+      processedRequestIDs: /* @__PURE__ */ new Set(),
+      sessionKey
     };
     for (const msg of options.prompt) {
       if (msg.role === "system") {
@@ -3517,7 +3544,13 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
         wsClient.close();
         this.activeClients.delete(wsClient);
         ss.activeClient = null;
-        this.currentWorkflowId = null;
+        if (!ss.streamClosed) {
+          if (ss.sessionKey) {
+            this.sessionWorkflows.delete(ss.sessionKey);
+          } else {
+            this.currentWorkflowId = null;
+          }
+        }
       }
     });
     return {
@@ -3641,7 +3674,7 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
           toolExecutor,
           nextTextId,
           availableToolNames
-        ).catch(() => {
+        ).catch((_err) => {
           ss.approvalPending = false;
           if (ss.deferredClose) {
             const close = ss.deferredClose;
@@ -3667,7 +3700,7 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
           });
           ss.streamClosed = true;
           controller.close();
-          this.cleanupClient(ss);
+          this.cleanupClient(ss, false);
         };
         if (ss.pendingToolCount > 0 || ss.approvalPending) {
           ss.deferredClose = doCompleteClose;
@@ -3725,7 +3758,7 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
             });
             ss.streamClosed = true;
             controller.close();
-            this.cleanupClient(ss);
+            this.cleanupClient(ss, false);
           }
         };
         if (ss.pendingToolCount > 0 || ss.approvalPending) {
@@ -3794,7 +3827,8 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
       });
       ss.streamedOutputChars += delta.length;
       ss.agentMessageEmitted.set(msgId, content.length);
-      this.persistedAgentEmitted.set(msgId, content.length);
+      const target = this.sessionWorkflows.get(ss.sessionKey)?.agentEmitted ?? this.persistedAgentEmitted;
+      target.set(msgId, content.length);
       ss.currentAgentMessageId = msgId;
     }
   }
@@ -3900,8 +3934,12 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
       ss.activeClient = null;
     }
     if (clearWorkflow) {
-      this.currentWorkflowId = null;
-      this.persistedAgentEmitted.clear();
+      if (ss.sessionKey) {
+        this.sessionWorkflows.delete(ss.sessionKey);
+      } else {
+        this.currentWorkflowId = null;
+        this.persistedAgentEmitted.clear();
+      }
     }
   }
   async approveAndResume(ss, tools, startReq, controller, toolExecutor, nextTextId, availableToolNames) {
@@ -3924,9 +3962,14 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
       return;
     }
     ss.approvalPending = false;
+    ss.deferredClose = null;
     this.cleanupClient(ss, false);
     const approval = decision.approved ? { approval: { tool_name: tools[0]?.name, tool_args_json: tools[0]?.args } } : { rejection: { message: decision.message ?? "User rejected" } };
-    const newStartReq = { ...startReq, approval };
+    const newStartReq = {
+      ...startReq,
+      approval,
+      preapproved_tools: decision.approved ? [...startReq.preapproved_tools ?? [], ...tools.map((t) => t.name)] : startReq.preapproved_tools ?? []
+    };
     const newClient = new GitLabWorkflowClient();
     this.activeClients.add(newClient);
     ss.activeClient = newClient;