@codebolt/agent 6.1.19 → 6.1.21

package/dist/unified/base/promptContext.js

@@ -1,5 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.isGeneratedUserContextMessage = isGeneratedUserContextMessage;
 exports.syncProcessedMessageWithRuntimeContext = syncProcessedMessageWithRuntimeContext;
 exports.reconcileRuntimePromptContext = reconcileRuntimePromptContext;
 exports.setSystemPrompt = setSystemPrompt;
@@ -10,6 +11,8 @@ exports.appendTranscriptMessages = appendTranscriptMessages;
 exports.prependTranscriptMessages = prependTranscriptMessages;
 exports.replaceTranscriptMessages = replaceTranscriptMessages;
 exports.getTranscriptMessages = getTranscriptMessages;
+exports.getCurrentUserMessage = getCurrentUserMessage;
+exports.updateCurrentUserMessage = updateCurrentUserMessage;
 exports.buildInferenceParams = buildInferenceParams;
 const RUNTIME_CONTEXT_METADATA_KEY = '__codeboltRuntimePromptContext';
 function cloneContent(content) {
@@ -31,6 +34,90 @@ function cloneMessage(message) {
 function cloneMessages(messages) {
     return messages.map((message) => cloneMessage(message));
 }
+function textFromContent(content) {
+    if (typeof content === 'string') {
+        return content;
+    }
+    try {
+        return JSON.stringify(content);
+    }
+    catch {
+        return String(content);
+    }
+}
+function normalizeTextPart(content) {
+    return {
+        type: 'text',
+        text: textFromContent(content),
+    };
+}
+function appendContent(existingContent, additionalContent) {
+    if (typeof existingContent === 'string' && typeof additionalContent === 'string') {
+        return `${existingContent}\n\n${additionalContent}`;
+    }
+    const existingParts = Array.isArray(existingContent)
+        ? [...existingContent]
+        : [normalizeTextPart(existingContent)];
+    const additionalParts = Array.isArray(additionalContent)
+        ? [...additionalContent]
+        : [normalizeTextPart(additionalContent)];
+    return [...existingParts, ...additionalParts];
+}
+function messagesHaveSameContent(firstMessage, secondMessage) {
+    return firstMessage.role === secondMessage.role &&
+        textFromContent(firstMessage.content) === textFromContent(secondMessage.content);
+}
+function getTextContent(message) {
+    if (typeof message.content === 'string') {
+        return message.content.trim();
+    }
+    if (!Array.isArray(message.content)) {
+        return '';
+    }
+    return message.content
+        .map((contentPart) => {
+        if (contentPart &&
+            typeof contentPart === 'object' &&
+            !Array.isArray(contentPart) &&
+            contentPart.type === 'text' &&
+            typeof contentPart.text === 'string') {
+            return contentPart.text;
+        }
+        return '';
+    })
+        .join('\n')
+        .trim();
+}
+function isGeneratedUserContextMessage(message) {
+    if (message.role !== 'user') {
+        return false;
+    }
+    const content = getTextContent(message);
+    if (!content) {
+        return false;
+    }
+    return content.startsWith('<flags>') ||
+        content.startsWith('<system-reminder>') ||
+        content.startsWith('This is the CodeBolt AI Agent. Setting up context for our chat.') ||
+        content.startsWith("I'm currently working in the effective project directory:") ||
+        content.startsWith('The user selected the following capabilities for this turn:') ||
+        content.startsWith('## Assembled Context') ||
+        content.startsWith('## Rule-Based Context');
+}
+function resolveCurrentUserMessageIndex(runtimeContext) {
+    var _a, _b;
+    const currentIndex = runtimeContext.currentUserMessageIndex;
+    if (currentIndex !== undefined &&
+        ((_a = runtimeContext.transcriptMessages[currentIndex]) === null || _a === void 0 ? void 0 : _a.role) === 'user') {
+        return currentIndex;
+    }
+    for (let i = runtimeContext.transcriptMessages.length - 1; i >= 0; i--) {
+        if (((_b = runtimeContext.transcriptMessages[i]) === null || _b === void 0 ? void 0 : _b.role) === 'user') {
+            return i;
+        }
+    }
+    return -1;
+}
 function buildSystemMessages(runtimeContext) {
     const stringSections = [];
     const nonStringSystemMessages = [];
@@ -55,17 +142,22 @@ function buildSystemMessages(runtimeContext) {
     return flattenedSystemMessages;
 }
 function flattenRuntimeMessages(runtimeContext) {
+    const transcriptMessages = cloneMessages(runtimeContext.transcriptMessages);
+    const currentUserMessageIndex = resolveCurrentUserMessageIndex(runtimeContext);
+    if (runtimeContext.currentUserContextMessage && currentUserMessageIndex !== -1) {
+        transcriptMessages.splice(currentUserMessageIndex, 0, cloneMessage(runtimeContext.currentUserContextMessage));
+    }
     return [
         ...buildSystemMessages(runtimeContext),
         ...cloneMessages(runtimeContext.userContextMessages),
-        ...cloneMessages(runtimeContext.transcriptMessages),
+        ...transcriptMessages,
     ];
 }
 function inferRuntimeContext(prompt) {
     var _a;
-    const messages = (_a = prompt.message.messages) !== null && _a !== void 0 ? _a : [];
+    const messages = (_a = prompt.message.input) !== null && _a !== void 0 ? _a : [];
     const systemMessages = messages.filter((message) => message.role === 'system');
-    const transcriptMessages = messages.filter((message) => message.role !== 'system');
+    const transcriptMessages = messages.filter((message) => message.role !== 'system' && !isGeneratedUserContextMessage(message));
     let systemPrompt;
     const systemContextMessages = [];
     if (systemMessages.length > 0) {
@@ -83,6 +175,8 @@ function inferRuntimeContext(prompt) {
         systemContextMessages,
         userContextMessages: [],
         transcriptMessages: cloneMessages(transcriptMessages),
+        currentUserContextMessage: undefined,
+        currentUserMessageIndex: undefined,
     };
 }
 function getRuntimePromptContext(prompt) {
@@ -97,6 +191,12 @@ function getRuntimePromptContext(prompt) {
         systemContextMessages: cloneMessages(runtimeContext.systemContextMessages),
         userContextMessages: cloneMessages(runtimeContext.userContextMessages),
         transcriptMessages: cloneMessages(runtimeContext.transcriptMessages),
+        currentUserContextMessage: runtimeContext.currentUserContextMessage
+            ? cloneMessage(runtimeContext.currentUserContextMessage)
+            : undefined,
+        currentUserMessageIndex: typeof runtimeContext.currentUserMessageIndex === 'number'
+            ? runtimeContext.currentUserMessageIndex
+            : undefined,
     };
 }
 function ensureRuntimePromptContext(prompt) {
@@ -108,7 +208,8 @@ function withRuntimePromptContext(prompt, runtimeContext) {
         ...prompt,
         message: {
             ...prompt.message,
-            messages: flattenRuntimeMessages(runtimeContext),
+            formatVersion: 'codebolt.llm.v2',
+            input: flattenRuntimeMessages(runtimeContext),
         },
         metadata: {
             ...prompt.metadata,
@@ -117,6 +218,10 @@ function withRuntimePromptContext(prompt, runtimeContext) {
                 systemContextMessages: cloneMessages(runtimeContext.systemContextMessages),
                 userContextMessages: cloneMessages(runtimeContext.userContextMessages),
                 transcriptMessages: cloneMessages(runtimeContext.transcriptMessages),
+                currentUserContextMessage: runtimeContext.currentUserContextMessage
+                    ? cloneMessage(runtimeContext.currentUserContextMessage)
+                    : undefined,
+                currentUserMessageIndex: runtimeContext.currentUserMessageIndex,
             },
         },
     };
@@ -130,14 +235,26 @@ function reconcileRuntimePromptContext(prompt) {
     if (!runtimeContext) {
         return syncProcessedMessageWithRuntimeContext(prompt);
     }
-    const flattenedMessages = (_a = prompt.message.messages) !== null && _a !== void 0 ? _a : [];
-    const transcriptStartIndex = (runtimeContext.systemPrompt ? 1 : 0) +
-        runtimeContext.systemContextMessages.length +
+    const flattenedMessages = (_a = prompt.message.input) !== null && _a !== void 0 ? _a : [];
+    const transcriptStartIndex = buildSystemMessages(runtimeContext).length +
         runtimeContext.userContextMessages.length;
     const normalizedTranscriptStartIndex = Math.min(transcriptStartIndex, flattenedMessages.length);
+    const candidateTranscriptMessages = cloneMessages(flattenedMessages.slice(normalizedTranscriptStartIndex));
+    const possibleCurrentUserContextMessageIndex = runtimeContext.currentUserMessageIndex;
+    const possibleCurrentUserContextMessage = possibleCurrentUserContextMessageIndex === undefined
+        ? undefined
+        : candidateTranscriptMessages[possibleCurrentUserContextMessageIndex];
+    if (runtimeContext.currentUserContextMessage &&
+        possibleCurrentUserContextMessageIndex !== undefined &&
+        possibleCurrentUserContextMessage &&
+        messagesHaveSameContent(possibleCurrentUserContextMessage, runtimeContext.currentUserContextMessage)) {
+        candidateTranscriptMessages.splice(possibleCurrentUserContextMessageIndex, 1);
+    }
     return withRuntimePromptContext(prompt, {
         ...runtimeContext,
-        transcriptMessages: cloneMessages(flattenedMessages.slice(normalizedTranscriptStartIndex)),
+        transcriptMessages: candidateTranscriptMessages,
+        currentUserContextMessage: runtimeContext.currentUserContextMessage,
+        currentUserMessageIndex: runtimeContext.currentUserMessageIndex,
     });
 }
 function setSystemPrompt(prompt, systemPrompt) {
@@ -159,19 +276,44 @@ function appendSystemContextMessage(prompt, message) {
 }
 function appendUserContextMessage(prompt, message) {
     const runtimeContext = ensureRuntimePromptContext(prompt);
+    const currentUserMessageIndex = resolveCurrentUserMessageIndex(runtimeContext);
+    if (currentUserMessageIndex === -1) {
+        return withRuntimePromptContext(prompt, {
+            ...runtimeContext,
+            userContextMessages: [
+                ...runtimeContext.userContextMessages,
+                cloneMessage(message),
+            ],
+        });
+    }
+    if (runtimeContext.currentUserContextMessage) {
+        return withRuntimePromptContext(prompt, {
+            ...runtimeContext,
+            currentUserContextMessage: {
+                ...runtimeContext.currentUserContextMessage,
+                content: appendContent(runtimeContext.currentUserContextMessage.content, message.content),
+            },
+            currentUserMessageIndex,
+        });
+    }
     return withRuntimePromptContext(prompt, {
         ...runtimeContext,
-        userContextMessages: [
-            ...runtimeContext.userContextMessages,
-            cloneMessage(message),
-        ],
+        currentUserContextMessage: cloneMessage(message),
+        currentUserMessageIndex,
     });
 }
 function appendTranscriptMessage(prompt, message) {
     const runtimeContext = ensureRuntimePromptContext(prompt);
+    const transcriptMessages = [...runtimeContext.transcriptMessages, cloneMessage(message)];
     return withRuntimePromptContext(prompt, {
         ...runtimeContext,
-        transcriptMessages: [...runtimeContext.transcriptMessages, cloneMessage(message)],
+        transcriptMessages,
+        currentUserMessageIndex: message.role === 'user'
+            ? transcriptMessages.length - 1
+            : runtimeContext.currentUserMessageIndex,
+        currentUserContextMessage: message.role === 'user'
+            ? undefined
+            : runtimeContext.currentUserContextMessage,
     });
 }
 function appendTranscriptMessages(prompt, messages) {
@@ -186,12 +328,17 @@ function appendTranscriptMessages(prompt, messages) {
 }
 function prependTranscriptMessages(prompt, messages) {
     const runtimeContext = ensureRuntimePromptContext(prompt);
+    const prependedMessages = cloneMessages(messages);
     return withRuntimePromptContext(prompt, {
         ...runtimeContext,
         transcriptMessages: [
-            ...cloneMessages(messages),
+            ...prependedMessages,
             ...runtimeContext.transcriptMessages,
         ],
+        currentUserContextMessage: runtimeContext.currentUserContextMessage,
+        currentUserMessageIndex: runtimeContext.currentUserMessageIndex === undefined
+            ? undefined
+            : runtimeContext.currentUserMessageIndex + prependedMessages.length,
     });
 }
 function replaceTranscriptMessages(prompt, messages) {
@@ -199,15 +346,46 @@ function replaceTranscriptMessages(prompt, messages) {
     return withRuntimePromptContext(prompt, {
         ...runtimeContext,
         transcriptMessages: cloneMessages(messages),
+        currentUserContextMessage: undefined,
+        currentUserMessageIndex: undefined,
     });
 }
 function getTranscriptMessages(prompt) {
     return cloneMessages(ensureRuntimePromptContext(prompt).transcriptMessages);
 }
+function getCurrentUserMessage(prompt) {
+    const runtimeContext = ensureRuntimePromptContext(prompt);
+    const currentUserMessageIndex = resolveCurrentUserMessageIndex(runtimeContext);
+    if (currentUserMessageIndex === -1) {
+        return undefined;
+    }
+    const message = runtimeContext.transcriptMessages[currentUserMessageIndex];
+    return message ? cloneMessage(message) : undefined;
+}
+function updateCurrentUserMessage(prompt, update) {
+    const runtimeContext = ensureRuntimePromptContext(prompt);
+    const currentUserMessageIndex = resolveCurrentUserMessageIndex(runtimeContext);
+    if (currentUserMessageIndex === -1) {
+        return prompt;
+    }
+    const currentUserMessage = runtimeContext.transcriptMessages[currentUserMessageIndex];
+    if (!currentUserMessage) {
+        return prompt;
+    }
+    const transcriptMessages = cloneMessages(runtimeContext.transcriptMessages);
+    transcriptMessages[currentUserMessageIndex] = update(cloneMessage(currentUserMessage));
+    return withRuntimePromptContext(prompt, {
+        ...runtimeContext,
+        transcriptMessages,
+        currentUserContextMessage: runtimeContext.currentUserContextMessage,
+        currentUserMessageIndex,
+    });
+}
 function buildInferenceParams(prompt) {
     const runtimeContext = ensureRuntimePromptContext(prompt);
     return {
         ...prompt.message,
-        messages: flattenRuntimeMessages(runtimeContext),
+        formatVersion: 'codebolt.llm.v2',
+        input: flattenRuntimeMessages(runtimeContext),
     };
 }

package/dist/unified/base/responseExecutor.d.ts

@@ -1,4 +1,4 @@
-import { AgentResponseExecutor, PostToolCallProcessor, PreToolCallProcessor, ResponseInput, ResponseOutput } from '@codebolt/types/agent';
+import { AgentResponseExecutor, AgentLocalTool, PostToolCallProcessor, PreToolCallProcessor, ResponseInput, ResponseOutput } from '@codebolt/types/agent';
 import { LoopDetectionService } from '../services/LoopDetectionService';
 export declare class ResponseExecutor implements AgentResponseExecutor {
     private preToolCallProcessors;
@@ -6,10 +6,12 @@ export declare class ResponseExecutor implements AgentResponseExecutor {
     private completed;
     private finalMessage;
     private loopDetectionService;
+    private localToolsByExecutionName;
     constructor(options: {
         preToolCallProcessors: PreToolCallProcessor[];
         postToolCallProcessors: PostToolCallProcessor[];
         loopDetectionService?: LoopDetectionService;
+        localToolsByExecutionName?: Map<string, AgentLocalTool>;
     });
     executeResponse(input: ResponseInput): Promise<ResponseOutput>;
     private parseToolCall;
@@ -17,8 +19,14 @@ export declare class ResponseExecutor implements AgentResponseExecutor {
     private getToolCalls;
     private hasExecutableToolCalls;
     private runRequiredCompaction;
+    private buildToolResultMessages;
+    private refreshNextMessageFromCompactedContext;
+    private extractCompactedContextMessages;
+    private isMessageObject;
+    private getMessagesAddedAfterInference;
     private executeTools;
     private sendFinalMessageToChat;
+    private getPendingAsyncTasks;
     private extractCompletionMessage;
     private formatCompletionValue;
     private executeSingleToolCall;