gitlab-ai-provider 6.4.0 → 6.4.2

package/dist/index.mjs

@@ -1582,7 +1582,7 @@ var GitLabOpenAILanguageModel = class {
 import WebSocket from "isomorphic-ws";
 
 // src/version.ts
-var VERSION = true ? "6.3.0" : "0.0.0-dev";
+var VERSION = true ? "6.4.1" : "0.0.0-dev";
 
 // src/gitlab-workflow-types.ts
 var WorkflowType = /* @__PURE__ */ ((WorkflowType2) => {
@@ -1594,7 +1594,7 @@ var WS_KEEPALIVE_PING_INTERVAL_MS = 45e3;
 var WS_HEARTBEAT_INTERVAL_MS = 6e4;
 var DEFAULT_WORKFLOW_DEFINITION = "chat" /* CHAT */;
 var DEFAULT_CLIENT_CAPABILITIES = ["shell_command"];
-var CLIENT_VERSION = "1.0";
+var CLIENT_VERSION = "8.51.0";
 var STOP_REASON_USER = "USER_ACTION_TRIGGERED_STOP";
 var AGENT_PRIVILEGES = {
   READ_WRITE_FILES: 1,
@@ -1781,6 +1781,7 @@ var GitLabWorkflowClient = class {
     }
     delete headers["content-type"];
     headers["x-gitlab-client-type"] = "node-websocket";
+    headers["x-gitlab-language-server-version"] = CLIENT_VERSION;
     const parsedUrl = new URL(options.instanceUrl);
     const origin = parsedUrl.origin;
     headers["origin"] = origin;
@@ -3419,7 +3420,8 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
         this.sessionWorkflows.set(sessionKey, {
           workflowId,
           workflowDefinition: effectiveDefinition,
-          agentEmitted: /* @__PURE__ */ new Map()
+          agentEmitted: /* @__PURE__ */ new Map(),
+          toolEntries: /* @__PURE__ */ new Set()
         });
         sess = this.sessionWorkflows.get(sessionKey);
       }
@@ -3439,6 +3441,8 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
       currentAgentMessageId: "",
       activeClient: wsClient,
       processedRequestIDs: /* @__PURE__ */ new Set(),
+      emittedToolEntries: new Set(sess?.toolEntries),
+      emittedToolKeys: /* @__PURE__ */ new Set(),
       sessionKey
     };
     for (const msg of options.prompt) {
@@ -3517,10 +3521,6 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
           const resolvedFlowConfigSchemaVersion = callFlowConfigSchemaVersion ?? this.workflowOptions.flowConfigSchemaVersion;
           if (resolvedFlowConfig) {
             startReq.flowConfig = resolvedFlowConfig;
-          } else if (this.workflowOptions.systemPrompt) {
-            startReq.flowConfig = {
-              prompts: [{ prompt_template: { system: this.workflowOptions.systemPrompt } }]
-            };
           }
           if (resolvedFlowConfigSchemaVersion) {
             startReq.flowConfigSchemaVersion = resolvedFlowConfigSchemaVersion;
@@ -3588,6 +3588,7 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
         const { requestID, data } = event;
         if (ss.processedRequestIDs.has(requestID)) break;
         ss.processedRequestIDs.add(requestID);
+        ss.emittedToolKeys.add(data.name);
         let parsedArgs;
         try {
           JSON.parse(data.args);
@@ -3637,6 +3638,7 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
         if (ss.processedRequestIDs.has(event.requestID)) break;
         ss.processedRequestIDs.add(event.requestID);
         const mapped = mapBuiltinTool(event.toolName, event.data, availableToolNames);
+        ss.emittedToolKeys.add(mapped.toolName);
         const mappedArgs = JSON.stringify(mapped.args);
         if (ss.activeTextBlockId) {
           controller.enqueue({ type: "text-end", id: ss.activeTextBlockId });
@@ -3818,6 +3820,51 @@ var GitLabWorkflowLanguageModel = class _GitLabWorkflowLanguageModel {
     }
     for (let i = 0; i < chatLog.length; i++) {
       const entry = chatLog[i];
+      if (entry.message_type === "tool" && entry.tool_info) {
+        const toolId = entry.message_id || entry.correlation_id || `tool-${i}`;
+        if (ss.emittedToolEntries.has(toolId)) continue;
+        const name = entry.tool_info.name;
+        const args = JSON.stringify(entry.tool_info.args ?? {});
+        if (ss.emittedToolKeys.has(name)) {
+          ss.emittedToolEntries.add(toolId);
+          const sess2 = this.sessionWorkflows.get(ss.sessionKey);
+          if (sess2) sess2.toolEntries.add(toolId);
+          continue;
+        }
+        if (ss.activeTextBlockId) {
+          controller.enqueue({ type: "text-end", id: ss.activeTextBlockId });
+          ss.activeTextBlockId = null;
+        }
+        controller.enqueue({
+          type: "tool-input-start",
+          id: toolId,
+          toolName: name,
+          providerExecuted: true
+        });
+        controller.enqueue({ type: "tool-input-delta", id: toolId, delta: args });
+        controller.enqueue({ type: "tool-input-end", id: toolId });
+        controller.enqueue({
+          type: "tool-call",
+          toolCallId: toolId,
+          toolName: name,
+          input: args,
+          providerExecuted: true,
+          dynamic: true
+        });
+        const response = entry.tool_info.tool_response;
+        const output = typeof response === "string" ? response : response ? response.content : entry.content || "";
+        controller.enqueue({
+          type: "tool-result",
+          toolCallId: toolId,
+          toolName: name,
+          result: { output, title: name, metadata: {} },
+          isError: false
+        });
+        ss.emittedToolEntries.add(toolId);
+        const sess = this.sessionWorkflows.get(ss.sessionKey);
+        if (sess) sess.toolEntries.add(toolId);
+        continue;
+      }
       if (entry.message_type !== "agent") continue;
       const content = entry.content || "";
       const msgId = entry.message_id || `idx-${i}`;