@codebolt/agent 6.1.20 → 6.1.21

package/dist/processor-pieces/messageModifiers/memoryImportModifier.js

@@ -37,6 +37,7 @@ exports.MemoryImportModifier = void 0;
 const base_1 = require("../base");
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
+const promptContext_1 = require("../../unified/base/promptContext");
 class MemoryImportModifier extends base_1.BaseMessageModifier {
     constructor(options = {}) {
         super();
@@ -54,8 +55,7 @@ class MemoryImportModifier extends base_1.BaseMessageModifier {
             if (!this.options.enableMemoryImport) {
                 return createdMessage;
             }
-            // Get the user message content to process
-            const userMessage = createdMessage.message.messages.find(msg => msg.role === 'user');
+            const userMessage = (0, promptContext_1.getCurrentUserMessage)(createdMessage);
             if (!userMessage || typeof userMessage.content !== 'string') {
                 return createdMessage;
             }
@@ -105,23 +105,17 @@ class MemoryImportModifier extends base_1.BaseMessageModifier {
             if (processedImports.length === 0) {
                 return createdMessage;
             }
-            // Update the user message with processed content
-            const messages = createdMessage.message.messages.map(msg => {
-                if (msg.role === 'user' && msg === userMessage) {
-                    return {
-                        ...msg,
-                        content
-                    };
-                }
-                return msg;
-            });
+            const updatedMessage = (0, promptContext_1.updateCurrentUserMessage)(createdMessage, (message) => ({
+                ...message,
+                content,
+            }));
             return Promise.resolve({
+                ...updatedMessage,
                 message: {
-                    ...createdMessage.message,
-                    messages
+                    ...updatedMessage.message,
                 },
                 metadata: {
-                    ...createdMessage.metadata,
+                    ...updatedMessage.metadata,
                     memoryImportsProcessed: true,
                     importedFiles: processedImports,
                     totalImports: processedImports.length

package/dist/processor-pieces/messageModifiers/toolInjectionModifier.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ToolInjectionModifier = void 0;
 const base_1 = require("../base");
 const agentToolLoader_1 = require("../../unified/utils/agentToolLoader");
+const promptContext_1 = require("../../unified/base/promptContext");
 class ToolInjectionModifier extends base_1.BaseMessageModifier {
     constructor(options = {}) {
         super();
@@ -19,7 +20,7 @@ class ToolInjectionModifier extends base_1.BaseMessageModifier {
     }
     async modify(originalRequest, createdMessage) {
         try {
-            let mentionedMCPs = originalRequest.mentionedMCPs || [];
+            const mentionedMCPs = originalRequest.mentionedMCPs || [];
             let tools = await (0, agentToolLoader_1.listAgentAvailableTools)(Array.isArray(mentionedMCPs) ? mentionedMCPs : []);
             // Filter tools if allowedTools is specified
             if (this.options.allowedTools && this.options.allowedTools.length > 0) {
@@ -30,17 +31,12 @@ class ToolInjectionModifier extends base_1.BaseMessageModifier {
             }
             switch (this.options.toolsLocation) {
                 case 'InsidePrompt':
-                    //@ts-ignore
                     return this.addToolsInsidePrompt(createdMessage, tools);
-                //@ts-ignore
                 case 'SystemMessage':
-                    //@ts-ignore
                     return this.addToolsAsSystemMessage(createdMessage, tools);
                 case 'Tool':
-                    //@ts-ignore
                     return this.addToolsAsToolCalls(createdMessage, tools);
                 default:
-                    //@ts-ignore
                     return this.addToolsAsToolCalls(createdMessage, tools);
             }
         }
@@ -51,23 +47,24 @@ class ToolInjectionModifier extends base_1.BaseMessageModifier {
     }
     addToolsInsidePrompt(createdMessage, tools) {
         const toolsInfo = this.generateToolsInfo(tools);
-        // Modify the last user message to include tools information
-        const modifiedMessages = createdMessage.message.messages.map((message, index) => {
-            if (message.role === 'user' && index === createdMessage.message.messages.length - 1) {
-                return {
-                    ...message,
-                    content: `${message.content}\n\n--- Available Tools ---\n${toolsInfo}`
-                };
-            }
-            return message;
-        });
+        const updatedMessage = (0, promptContext_1.updateCurrentUserMessage)(createdMessage, (message) => ({
+            ...message,
+            content: typeof message.content === 'string'
+                ? `${message.content}\n\n--- Available Tools ---\n${toolsInfo}`
+                : [
+                    ...(Array.isArray(message.content)
+                        ? message.content
+                        : [{ type: 'text', text: String(message.content) }]),
+                    { type: 'text', text: `--- Available Tools ---\n${toolsInfo}` },
+                ],
+        }));
         return {
+            ...updatedMessage,
             message: {
-                ...createdMessage.message,
-                messages: modifiedMessages
+                ...updatedMessage.message,
             },
             metadata: {
-                ...createdMessage.metadata,
+                ...updatedMessage.metadata,
                 toolsInjected: true,
                 toolsLocation: 'InsidePrompt',
                 toolsCount: tools.length
@@ -83,7 +80,7 @@ class ToolInjectionModifier extends base_1.BaseMessageModifier {
         return {
             message: {
                 ...createdMessage.message,
-                messages: [toolsMessage, ...createdMessage.message.messages]
+                input: [toolsMessage, ...createdMessage.message.input]
             },
             metadata: {
                 ...createdMessage.metadata,

package/dist/processor-pieces/postInferenceProcessors/loopDetectionModifier.js

@@ -21,32 +21,21 @@ class LoopDetectionModifier extends base_1.BasePostInferenceProcessor {
             }
             const currentTime = Date.now();
             // Add LLM response to history
-            if (llmResponseMessage.content) {
+            const responseContent = llmResponseMessage.output_text || llmResponseMessage.content;
+            if (responseContent) {
                 this.messageHistory.push({
-                    content: llmResponseMessage.content,
+                    content: typeof responseContent === 'string' ? responseContent : JSON.stringify(responseContent),
                     timestamp: currentTime,
-                    role: llmResponseMessage.role
+                    role: llmResponseMessage.role || 'assistant'
                 });
             }
-            // Add any choice messages from LLM response to history
-            if (llmResponseMessage.choices) {
-                for (const choice of llmResponseMessage.choices) {
-                    if (choice.message && choice.message.content) {
-                        this.messageHistory.push({
-                            content: choice.message.content,
-                            timestamp: currentTime,
-                            role: choice.message.role
-                        });
-                    }
-                }
-            }
             // Add the original sent message to history if it exists
-            for (const message of llmMessageSent.message.messages) {
+            for (const message of llmMessageSent.message.input) {
                 if (typeof message.content === 'string') {
                     this.messageHistory.push({
                         content: message.content,
                         timestamp: currentTime,
-                        role: message.role
+                        role: message.role || 'user'
                     });
                 }
             }
@@ -61,11 +50,11 @@ class LoopDetectionModifier extends base_1.BasePostInferenceProcessor {
                     role: 'system',
                     content: `[Loop Detection Warning] Similar messages detected in recent conversation. This may indicate a conversational loop. Consider rephrasing your request or providing more specific information.`
                 };
-                const updatedMessages = [...nextPrompt.message.messages, warningMessage];
+                const updatedMessages = [...nextPrompt.message.input, warningMessage];
                 return Promise.resolve({
                     message: {
                         ...nextPrompt.message,
-                        messages: updatedMessages
+                        input: updatedMessages
                     },
                     metadata: {
                         ...nextPrompt.metadata,

package/dist/processor-pieces/postToolCallProcessors/conversationCompactorModifier.d.ts

@@ -238,7 +238,7 @@ export declare class ConversationCompactorModifier extends BasePostToolCallProce
      * Forces compression regardless of threshold (for testing or manual trigger)
      */
     forceCompress(messages: MessageObject[]): Promise<{
-        messages: MessageObject[];
+        input: MessageObject[];
         metadata: Partial<CompressionMetadata>;
     }>;
 }

package/dist/processor-pieces/postToolCallProcessors/conversationCompactorModifier.js

@@ -304,7 +304,7 @@ class ConversationCompactorModifier extends basePostToolCallProcessor_1.BasePost
         }
         // If under budget, no truncation needed
         if (toolResponseTokens <= this.options.toolResponseTokenBudget) {
-            return { messages: result, truncated: false, tokensSaved: 0 };
+            return { input: result, truncated: false, tokensSaved: 0 };
         }
         // Iterate from oldest to newest tool messages for truncation
         // (We truncate older tool outputs first to preserve recent context)
@@ -344,7 +344,7 @@ class ConversationCompactorModifier extends basePostToolCallProcessor_1.BasePost
                 }
             }
         }
-        return { messages: result, truncated, tokensSaved };
+        return { input: result, truncated, tokensSaved };
     }
     // ========================================================================
     // History Split Point Calculation (Phase 2)
@@ -518,7 +518,7 @@ CRITICAL RULES:
         try {
             const prompt = this.getCompressionPrompt(historyToCompress);
             const response = await codeboltjs_1.default.llm.inference({
-                messages: [
+                input: [
                     { role: 'system', content: 'You are a precise conversation compression assistant.' },
                     { role: 'user', content: prompt }
                 ],
@@ -554,7 +554,7 @@ If the snapshot is accurate and complete, respond with just the original snapsho
 If there are inaccuracies or missing critical information, provide a corrected version.
 Keep the same XML format.`;
             const response = await codeboltjs_1.default.llm.inference({
-                messages: [
+                input: [
                     { role: 'system', content: 'You are verifying a conversation compression for accuracy.' },
                     { role: 'user', content: verificationPrompt }
                 ]
@@ -680,7 +680,7 @@ Keep the same XML format.`;
         if (this.options.enableLogging && deduplicatedCount > 0) {
             console.log(`[ConversationCompactor] Deduplicated ${deduplicatedCount} file reads`);
         }
-        return { messages: result, deduplicatedCount };
+        return { input: result, deduplicatedCount };
     }
     // ========================================================================
     // Compression Strategies
@@ -732,7 +732,7 @@ Keep the same XML format.`;
      */
     async compressSmart(messages) {
         // Phase 1: Truncate large tool outputs
-        const { messages: truncatedMessages, truncated } = this.truncateLargeToolOutputs(messages);
+        const { input: truncatedMessages, truncated } = this.truncateLargeToolOutputs(messages);
         // Check if truncation alone was sufficient
         const tokensAfterTruncation = this.countMessageTokens(truncatedMessages);
         const threshold = this.options.modelTokenLimit * this.options.compressionTokenThreshold;
@@ -789,7 +789,7 @@ Keep the same XML format.`;
      */
     async compressSummarize(messages) {
         // Phase 1: Truncate large tool outputs
-        const { messages: truncatedMessages, truncated } = this.truncateLargeToolOutputs(messages);
+        const { input: truncatedMessages, truncated } = this.truncateLargeToolOutputs(messages);
         // Phase 2: Find split point
         const splitIndex = this.findCompressionSplitIndex(truncatedMessages);
         if (splitIndex === 0) {
@@ -869,7 +869,7 @@ Please continue from where the previous conversation left off. Do not ask the us
         const { nextPrompt, rawLLMResponseMessage, tokenLimit } = input;
         try {
             // Safety check: ensure messages array exists
-            if (!((_a = nextPrompt === null || nextPrompt === void 0 ? void 0 : nextPrompt.message) === null || _a === void 0 ? void 0 : _a.messages) || !Array.isArray(nextPrompt.message.messages)) {
+            if (!((_a = nextPrompt === null || nextPrompt === void 0 ? void 0 : nextPrompt.message) === null || _a === void 0 ? void 0 : _a.input) || !Array.isArray(nextPrompt.message.input)) {
                 if (this.options.enableLogging) {
                     console.warn('[ConversationCompactor] No messages array found, skipping compression');
                 }
@@ -878,7 +878,7 @@ Please continue from where the previous conversation left off. Do not ask the us
                     shouldExit: false
                 };
             }
-            const messages = nextPrompt.message.messages;
+            const messages = nextPrompt.message.input;
             // Get model token limit: prefer tokenLimit from LLM response, then lookup by model name, then fallback to config
             const modelName = rawLLMResponseMessage === null || rawLLMResponseMessage === void 0 ? void 0 : rawLLMResponseMessage.model;
             const modelTokenLimit = tokenLimit !== null && tokenLimit !== void 0 ? tokenLimit : (modelName ? getModelTokenLimit(modelName) : this.options.modelTokenLimit);
@@ -981,7 +981,7 @@ Please continue from where the previous conversation left off. Do not ask the us
                 };
             }
             // Pre-compression: deduplicate file reads to reduce token waste
-            const { messages: dedupedMessages } = this.deduplicateFileReads(messages);
+            const { input: dedupedMessages } = this.deduplicateFileReads(messages);
             // Apply compression based on strategy
             let result;
             switch (this.options.compactStrategy) {
@@ -1056,7 +1056,7 @@ Please continue from where the previous conversation left off. Do not ask the us
                 nextPrompt: {
                     message: {
                         ...nextPrompt.message,
-                        messages: result.compressedMessages
+                        input: result.compressedMessages
                     },
                     metadata: {
                         ...nextPrompt.metadata,
@@ -1079,10 +1079,10 @@ Please continue from where the previous conversation left off. Do not ask the us
                         ...nextPrompt.metadata,
                         compression: {
                             status: CompressionStatus.FAILED_SUMMARIZATION_ERROR,
-                            originalTokenCount: this.countMessageTokens(nextPrompt.message.messages),
-                            newTokenCount: this.countMessageTokens(nextPrompt.message.messages),
+                            originalTokenCount: this.countMessageTokens(nextPrompt.message.input),
+                            newTokenCount: this.countMessageTokens(nextPrompt.message.input),
                             messagesCompressed: 0,
-                            messagesPreserved: nextPrompt.message.messages.length,
+                            messagesPreserved: nextPrompt.message.input.length,
                             toolOutputsTruncated: false,
                             failedAttempts: this.failedCompressionAttempts,
                             timestamp: new Date().toISOString(),
@@ -1125,7 +1125,7 @@ Please continue from where the previous conversation left off. Do not ask the us
             ? this.compressSummarize(messages)
             : this.compressSmart(messages));
         return {
-            messages: result.compressedMessages,
+            input: result.compressedMessages,
             metadata: result.metadata
         };
     }

package/dist/processor-pieces/postToolCallProcessors/shellProcessorModifier.js

@@ -42,7 +42,7 @@ class ShellProcessorModifier extends base_1.BasePostToolCallProcessor {
             // Also process any shell injections in the next prompt messages
             const updatedMessages = [];
             let contentModified = false;
-            for (const message of processedNextPrompt.message.messages) {
+            for (const message of processedNextPrompt.message.input) {
                 if (typeof message.content === 'string') {
                     let processedContent = message.content;
                     // Replace {{args}} placeholders if metadata has args
@@ -69,7 +69,7 @@ class ShellProcessorModifier extends base_1.BasePostToolCallProcessor {
                 processedNextPrompt = {
                     message: {
                         ...processedNextPrompt.message,
-                        messages: updatedMessages
+                        input: updatedMessages
                     },
                     metadata: {
                         ...processedNextPrompt.metadata,
@@ -109,7 +109,7 @@ class ShellProcessorModifier extends base_1.BasePostToolCallProcessor {
                     processedPrompt = {
                         message: {
                             ...processedPrompt.message,
-                            messages: [...processedPrompt.message.messages, systemMessage]
+                            input: [...processedPrompt.message.input, systemMessage]
                         },
                         metadata: {
                             ...processedPrompt.metadata,

package/dist/processor-pieces/preInferenceProcessors/chatCompressionModifier.js

@@ -104,12 +104,12 @@ class ChatCompressionModifier extends base_1.BasePreInferenceProcessor {
                     };
                 }
             }
-            const compressionResult = await this.tryCompressChat(createdMessage.message.messages, this.options.force || false);
+            const compressionResult = await this.tryCompressChat(createdMessage.message.input, this.options.force || false);
             if (compressionResult.compressionStatus === CompressionStatus.COMPRESSED) {
                 return {
                     message: {
                         ...createdMessage.message,
-                        messages: compressionResult.compressedMessages || createdMessage.message.messages
+                        input: compressionResult.compressedMessages || createdMessage.message.input
                     },
                     metadata: {
                         ...createdMessage.metadata,

package/dist/processor-pieces/utils/messageModifierHelper.js

@@ -7,7 +7,7 @@ const mergeMessages = (existing, additional) => {
         message: {
             ...existing.message,
             ...additional.message,
-            messages: [...existing.message.messages, ...additional.message.messages],
+            input: [...existing.message.input, ...additional.message.input],
         },
         metadata: {
             ...existing.metadata,
@@ -31,7 +31,7 @@ const addSystemMessage = (message, systemContent) => {
     return {
         message: {
             ...message.message,
-            messages: [systemMessage, ...message.message.messages]
+            input: [systemMessage, ...message.message.input]
         },
         metadata: {
             ...message.metadata,
@@ -50,7 +50,7 @@ const addUserContext = (message, contextKey, contextValue) => {
     return {
         message: {
             ...message.message,
-            messages: [...message.message.messages, contextMessage]
+            input: [...message.message.input, contextMessage]
         },
         metadata: {
             ...message.metadata,

package/dist/types/libFunctionTypes.d.ts

@@ -232,6 +232,11 @@ export interface CodeboltAPI {
         listMcpFromServers: (servers: string[]) => Promise<{
             data: OpenAITool[];
         }>;
+        getRegisteredTools: () => Promise<{
+            data?: {
+                tools: OpenAITool[];
+            };
+        }>;
         getTools: (mcps: any[]) => Promise<{
             data: OpenAITool[];
         }>;

package/dist/unified/agent/agent.d.ts

@@ -57,8 +57,11 @@ export declare class Agent implements AgentInterface {
     private readonly maxTurns;
     private readonly localToolSchemas;
     private readonly localToolsByExecutionName;
+    private readonly runtimeToolSetId;
     constructor(config: AgentOptions);
     run(message: string | FlatUserMessage, options?: ProcessedMessage | AgentRunOptions): Promise<AgentRunResult>;
+    private registerRuntimeToolsForSearch;
+    private unregisterRuntimeToolsForSearch;
     processMessage(message: string | FlatUserMessage, options?: ProcessedMessage | AgentRunOptions): Promise<AgentRunResult>;
     execute(reqMessage: FlatUserMessage): Promise<{
         success: boolean;
@@ -72,8 +75,15 @@ export declare class Agent implements AgentInterface {
     getPreToolCallProcessors(): PreToolCallProcessor[];
     getPostToolCallProcessors(): PostToolCallProcessor[];
     private resolveRunContext;
+    private getResumeMessageModifiers;
     private isProcessedMessage;
     private applyCompaction;
+    private hydratePromptFromServerCompaction;
+    private isServerCompactionCurrent;
+    private extractCurrentRunMessages;
+    private isCurrentRunUserMessage;
+    private extractServerCompactedMessages;
+    private isServerCompactedMessage;
     private tryRecoverPrompt;
     private refreshAvailableTools;
     private getAllowedToolNames;