gitlab-ai-provider 5.3.1 → 5.3.3

package/dist/index.mjs

@@ -1509,7 +1509,7 @@ var GitLabOpenAILanguageModel = class {
 import WebSocket from "isomorphic-ws";
 
 // src/version.ts
-var VERSION = true ? "5.3.0" : "0.0.0-dev";
+var VERSION = true ? "5.3.2" : "0.0.0-dev";
 
 // src/gitlab-workflow-types.ts
 var WorkflowType = /* @__PURE__ */ ((WorkflowType2) => {
@@ -1735,6 +1735,8 @@ var GitLabWorkflowClient = class {
         });
       } else if (checkpoint.status === "STOPPED" || checkpoint.status === "CANCELLED") {
         this.emit({ type: "completed" });
+      } else if (checkpoint.status === "TOOL_CALL_APPROVAL_REQUIRED" || checkpoint.status === "PLAN_APPROVAL_REQUIRED") {
+        this.emit({ type: "completed" });
       }
       return;
     }
@@ -2949,6 +2951,17 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
    * alongside `toolExecutor`. Takes precedence over `workflowOptions.onSelectModel`.
    */
   onSelectModel = null;
+  /**
+   * Set the system prompt to override DWS's default.
+   * Sent via `flowConfig.prompts[].prompt_template.system` at stream time.
+   * Can be updated between doStream() calls (e.g., per agent/session).
+   */
+  set systemPrompt(prompt) {
+    this.workflowOptions.systemPrompt = prompt ?? void 0;
+  }
+  get systemPrompt() {
+    return this.workflowOptions.systemPrompt ?? null;
+  }
   get toolExecutor() {
     return this._toolExecutor;
   }
@@ -3317,9 +3330,15 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
           };
           if (this.workflowOptions.flowConfig) {
             startReq.flowConfig = this.workflowOptions.flowConfig;
+          } else if (this.workflowOptions.systemPrompt) {
+            startReq.flowConfig = {
+              prompts: [{ prompt_template: { system: this.workflowOptions.systemPrompt } }]
+            };
           }
           if (this.workflowOptions.flowConfigSchemaVersion) {
             startReq.flowConfigSchemaVersion = this.workflowOptions.flowConfigSchemaVersion;
+          } else if (startReq.flowConfig) {
+            startReq.flowConfigSchemaVersion = "v1";
           }
           wsClient.sendStartRequest(startReq);
           controller.enqueue({
@@ -3576,7 +3595,7 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
     if (!chatLog || !Array.isArray(chatLog) || chatLog.length === 0) {
       return;
     }
-    if (checkpoint.status !== "RUNNING" && checkpoint.status !== "INPUT_REQUIRED" && checkpoint.status !== "FINISHED" && checkpoint.status !== "COMPLETED") {
+    if (checkpoint.status !== "RUNNING" && checkpoint.status !== "INPUT_REQUIRED" && checkpoint.status !== "FINISHED" && checkpoint.status !== "COMPLETED" && checkpoint.status !== "TOOL_CALL_APPROVAL_REQUIRED") {
       return;
     }
     for (let i = 0; i < chatLog.length; i++) {
@@ -3722,7 +3741,8 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
   // ---------------------------------------------------------------------------
   async buildWorkflowMetadata() {
     const metadata = {
-      extended_logging: false
+      extended_logging: false,
+      tool_approval_for_session_enabled: true
     };
     try {
       const workDir = this.workflowOptions.workingDirectory ?? process.cwd();
@@ -3760,13 +3780,16 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
   /**
    * Extract the user's goal (last user message) from the AI SDK prompt.
    */
+  static SYSTEM_REMINDER_RE = /<system-reminder>([\s\S]*?)<\/system-reminder>/g;
   extractGoalFromPrompt(prompt) {
     for (let i = prompt.length - 1; i >= 0; i--) {
       const message = prompt[i];
       if (message.role === "user") {
         const textParts = message.content.filter((part) => part.type === "text").map((part) => part.text);
         if (textParts.length > 0) {
-          return textParts.join("\n");
+          const raw = textParts.join("\n");
+          const cleaned = raw.replace(_GitLabWorkflowLanguageModel.SYSTEM_REMINDER_RE, "").trim();
+          return cleaned || raw;
         }
       }
     }
@@ -3860,6 +3883,23 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
             metadata: JSON.stringify({ role: "assistant" })
           });
         }
+      } else if (message.role === "user") {
+        for (const part of message.content) {
+          if (part.type === "text") {
+            const text = part.text;
+            const matches = text.matchAll(_GitLabWorkflowLanguageModel.SYSTEM_REMINDER_RE);
+            for (const match of matches) {
+              const inner = match[1].trim();
+              if (inner) {
+                context.push({
+                  category: "agent_context",
+                  content: inner,
+                  metadata: JSON.stringify({ source: "system-reminder" })
+                });
+              }
+            }
+          }
+        }
       }
     }
     return context;