@codebolt/agent 6.1.20 → 6.1.21

package/dist/unified/base/initialPromptGenerator.js

@@ -18,14 +18,22 @@ const mergeFlagLists = (existing, mentioned) => {
 };
 const formatFlagContext = (flags, mentionedFlags) => {
     const normalizedMentionedFlags = normalizeFlagList(mentionedFlags);
-    if (!flags && normalizedMentionedFlags.length === 0)
+    const userFlags = (flags === null || flags === void 0 ? void 0 : flags.user) || [];
+    const projectFlags = (flags === null || flags === void 0 ? void 0 : flags.project) || [];
+    const threadFlags = mergeFlagLists(flags === null || flags === void 0 ? void 0 : flags.thread, normalizedMentionedFlags);
+    const effectiveFlags = mergeFlagLists(flags === null || flags === void 0 ? void 0 : flags.effective, normalizedMentionedFlags);
+    if (userFlags.length === 0 &&
+        projectFlags.length === 0 &&
+        threadFlags.length === 0 &&
+        effectiveFlags.length === 0) {
         return null;
+    }
     return [
         '<flags>',
-        `user: ${formatFlagList(flags === null || flags === void 0 ? void 0 : flags.user)}`,
-        `project: ${formatFlagList(flags === null || flags === void 0 ? void 0 : flags.project)}`,
-        `thread: ${formatFlagList(mergeFlagLists(flags === null || flags === void 0 ? void 0 : flags.thread, normalizedMentionedFlags))}`,
-        `effective: ${formatFlagList(mergeFlagLists(flags === null || flags === void 0 ? void 0 : flags.effective, normalizedMentionedFlags))}`,
+        `user: ${formatFlagList(userFlags)}`,
+        `project: ${formatFlagList(projectFlags)}`,
+        `thread: ${formatFlagList(threadFlags)}`,
+        `effective: ${formatFlagList(effectiveFlags)}`,
         '</flags>',
     ].join('\n');
 };
@@ -97,6 +105,9 @@ class InitialPromptGenerator {
         if (options.baseSystemPrompt !== undefined) {
             this.baseSystemPrompt = options.baseSystemPrompt;
         }
+        if (options.initialPrompt !== undefined) {
+            this.initialPrompt = options.initialPrompt;
+        }
     }
     /**
      * Process and modify input messages using the message modifier pattern
@@ -106,19 +117,32 @@ class InitialPromptGenerator {
             if (this.enableLogging) {
                 // console.log('[InitialPromptGenerator] Processing message:', input);
             }
-            let createdMessage = {
-                message: {
-                    messages: [],
-                    tools: []
-                },
-                metadata: {
-                    timestamp: new Date().toISOString(),
-                    messageId: input.messageId,
-                    threadId: input.threadId
+            const isResumedPrompt = this.initialPrompt !== undefined;
+            let createdMessage = isResumedPrompt
+                ? {
+                    ...this.initialPrompt,
+                    metadata: {
+                        ...this.initialPrompt.metadata,
+                        timestamp: new Date().toISOString(),
+                        messageId: input.messageId,
+                        threadId: input.threadId,
+                        resumedPrompt: true,
+                    },
                 }
-            };
+                : {
+                    message: {
+                        formatVersion: 'codebolt.llm.v2',
+                        input: [],
+                        tools: []
+                    },
+                    metadata: {
+                        timestamp: new Date().toISOString(),
+                        messageId: input.messageId,
+                        threadId: input.threadId
+                    }
+                };
             createdMessage = (0, promptContext_1.syncProcessedMessageWithRuntimeContext)(createdMessage);
-            if (this.baseSystemPrompt !== undefined) {
+            if (!isResumedPrompt && this.baseSystemPrompt !== undefined) {
                 createdMessage = (0, promptContext_1.setSystemPrompt)(createdMessage, this.baseSystemPrompt);
             }
             createdMessage = (0, promptContext_1.appendTranscriptMessage)(createdMessage, {
@@ -141,7 +165,7 @@ class InitialPromptGenerator {
                     console.error(`[InitialPromptGenerator] Error in message modifier:`, error);
                 }
             }
-            let { todos } = await codeboltjs_1.default.todo.getAllIncompleteTodos();
+            const { todos } = await codeboltjs_1.default.todo.getAllIncompleteTodos();
             if (todos && todos.length == 0) {
                 createdMessage = (0, promptContext_1.appendUserContextMessage)(createdMessage, {
                     role: 'user',
@@ -150,7 +174,7 @@ class InitialPromptGenerator {
             }
             if (this.enableLogging) {
                 // console.log('[InitialPromptGenerator] Processing completed:', {
-                //     messageCount: createdMessage.message.messages.length,
+                //     messageCount: createdMessage.message.input.length,
                 //     metadata: createdMessage.metadata
                 // });
             }
@@ -185,7 +209,6 @@ class InitialPromptGenerator {
     updateProcessors(processors) {
         this.processors = processors;
     }
-    ;
     getProcessors() {
         return this.processors;
     }

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

@@ -1,5 +1,6 @@
 import type { ProcessedMessage } from '@codebolt/types/agent';
 import type { LLMInferenceParams, MessageObject } from '@codebolt/types/sdk';
+export declare function isGeneratedUserContextMessage(message: MessageObject): boolean;
 export declare function syncProcessedMessageWithRuntimeContext(prompt: ProcessedMessage): ProcessedMessage;
 export declare function reconcileRuntimePromptContext(prompt: ProcessedMessage): ProcessedMessage;
 export declare function setSystemPrompt(prompt: ProcessedMessage, systemPrompt: string): ProcessedMessage;
@@ -10,4 +11,6 @@ export declare function appendTranscriptMessages(prompt: ProcessedMessage, messa
 export declare function prependTranscriptMessages(prompt: ProcessedMessage, messages: MessageObject[]): ProcessedMessage;
 export declare function replaceTranscriptMessages(prompt: ProcessedMessage, messages: MessageObject[]): ProcessedMessage;
 export declare function getTranscriptMessages(prompt: ProcessedMessage): MessageObject[];
+export declare function getCurrentUserMessage(prompt: ProcessedMessage): MessageObject | undefined;
+export declare function updateCurrentUserMessage(prompt: ProcessedMessage, update: (message: MessageObject) => MessageObject): ProcessedMessage;
 export declare function buildInferenceParams(prompt: ProcessedMessage): LLMInferenceParams;

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

@@ -19,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;