gitlab-ai-provider 5.3.1 → 5.3.2

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.1" : "0.0.0-dev";
 
 // src/gitlab-workflow-types.ts
 var WorkflowType = /* @__PURE__ */ ((WorkflowType2) => {
@@ -2949,6 +2949,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 +3328,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({
@@ -3760,13 +3777,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 +3880,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;