genaicode 0.14.1 → 0.15.0

package/dist/main/ui/backend/endpoints/edit-message-endpoint.js

@@ -2,17 +2,20 @@ import { registerEndpoint } from '../api-handlers.js';
 registerEndpoint((router, service) => {
     router.post('/edit-message', async (req, res) => {
         try {
-            const { messageId, newContent } = req.body;
+            const { messageId, newContent, newData } = req.body;
             if (!messageId || typeof messageId !== 'string') {
                 return res.status(400).json({ error: 'Invalid message ID' });
             }
-            if (!newContent || typeof newContent !== 'string') {
-                return res.status(400).json({ error: 'Invalid message content' });
+            if (newContent === undefined && newData === undefined) {
+                return res.status(400).json({ error: 'Invalid message content or data' });
             }
-            if (newContent.trim().length === 0) {
-                return res.status(400).json({ error: 'Message content cannot be empty' });
+            // Allow empty content if newData is present (e.g. updating planning data)
+            if (newContent !== undefined && (typeof newContent !== 'string' || newContent.trim().length === 0)) {
+                if (!newData) {
+                    return res.status(400).json({ error: 'Message content cannot be empty' });
+                }
             }
-            const success = await service.editMessage(messageId, newContent);
+            const success = await service.editMessage(messageId, newContent, newData);
             if (success) {
                 res.json({ message: 'Message edited successfully' });
             }

package/dist/main/ui/backend/endpoints/edit-message-endpoint.js.map

@@ -1 +1 @@
-{"version":3,"file":"edit-message-endpoint.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/edit-message-endpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QAC9C,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC,IAGrC,CAAC;YAEF,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAChD,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAC/D,CAAC;YAED,IAAI,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;gBAClD,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,yBAAyB,EAAE,CAAC,CAAC;YACpE,CAAC;YAED,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnC,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC,CAAC;YAC5E,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YAEjE,IAAI,OAAO,EAAE,CAAC;gBACZ,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAC/C,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACjD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,6CAA6C,EAAE,CAAC,CAAC;YACjF,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"edit-message-endpoint.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/edit-message-endpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QAC9C,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,GAAG,CAAC,IAI9C,CAAC;YAEF,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAChD,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;YAC/D,CAAC;YAED,IAAI,UAAU,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBACtD,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC,CAAC;YAC5E,CAAC;YAED,0EAA0E;YAC1E,IAAI,UAAU,KAAK,SAAS,IAAI,CAAC,OAAO,UAAU,KAAK,QAAQ,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;gBACnG,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,OAAO,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,iCAAiC,EAAE,CAAC,CAAC;gBAC5E,CAAC;YACH,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YAE1E,IAAI,OAAO,EAAE,CAAC;gBACZ,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,6BAA6B,EAAE,CAAC,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAC/C,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACjD,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,6CAA6C,EAAE,CAAC,CAAC;YACjF,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/main/ui/backend/endpoints/index.d.ts

@@ -10,3 +10,4 @@ import './usage-endpoint.js';
 import './config-endpoint.js';
 import './iteration-endpoint.js';
 import './context-endpoint.js';
+import './compress-context-endpoint.js';

package/dist/main/ui/backend/endpoints/index.js

@@ -10,4 +10,5 @@ import './usage-endpoint.js';
 import './config-endpoint.js';
 import './iteration-endpoint.js';
 import './context-endpoint.js';
+import './compress-context-endpoint.js';
 //# sourceMappingURL=index.js.map

package/dist/main/ui/backend/endpoints/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/index.ts"],"names":[],"mappings":"AAAA,OAAO,uBAAuB,CAAC;AAC/B,OAAO,4BAA4B,CAAC;AACpC,OAAO,+BAA+B,CAAC;AACvC,OAAO,gCAAgC,CAAC;AACxC,OAAO,gCAAgC,CAAC;AACxC,OAAO,mCAAmC,CAAC;AAC3C,OAAO,qCAAqC,CAAC;AAC7C,OAAO,wBAAwB,CAAC;AAChC,OAAO,qBAAqB,CAAC;AAC7B,OAAO,sBAAsB,CAAC;AAC9B,OAAO,yBAAyB,CAAC;AACjC,OAAO,uBAAuB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/index.ts"],"names":[],"mappings":"AAAA,OAAO,uBAAuB,CAAC;AAC/B,OAAO,4BAA4B,CAAC;AACpC,OAAO,+BAA+B,CAAC;AACvC,OAAO,gCAAgC,CAAC;AACxC,OAAO,gCAAgC,CAAC;AACxC,OAAO,mCAAmC,CAAC;AAC3C,OAAO,qCAAqC,CAAC;AAC7C,OAAO,wBAAwB,CAAC;AAChC,OAAO,qBAAqB,CAAC;AAC7B,OAAO,sBAAsB,CAAC;AAC9B,OAAO,yBAAyB,CAAC;AACjC,OAAO,uBAAuB,CAAC;AAC/B,OAAO,gCAAgC,CAAC"}

package/dist/main/ui/backend/endpoints/interrupt-execution-endpoint.js

@@ -1,5 +1,5 @@
 import { registerEndpoint } from '../api-handlers.js';
-import { requestInterrupt } from '../../../../prompt/steps/step-ask-question/handlers/container-task/commands/interrupt-controller.js';
+import { requestInterrupt } from '../../../../prompt/steps/step-iterate/handlers/container-task/commands/interrupt-controller.js';
 registerEndpoint((router, service) => {
     router.post('/interrupt-execution', async (req, res) => {
         try {

package/dist/main/ui/backend/endpoints/interrupt-execution-endpoint.js.map

@@ -1 +1 @@
-{"version":3,"file":"interrupt-execution-endpoint.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/interrupt-execution-endpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qGAAqG,CAAC;AAEvI,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QACrD,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YAE3B,IAAI,KAAK,KAAK,mBAAmB,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;gBAClC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACpB,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,wDAAwD,EAAE,CAAC,CAAC;gBAClF,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,qDAAqD,EAAE,CAAC,CAAC;gBACzF,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,+CAA+C;gBAC/C,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;gBACnC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,oCAAoC,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"interrupt-execution-endpoint.js","sourceRoot":"","sources":["../../../../../src/main/ui/backend/endpoints/interrupt-execution-endpoint.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gGAAgG,CAAC;AAElI,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,sBAAsB,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QACrD,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YAE3B,IAAI,KAAK,KAAK,mBAAmB,EAAE,CAAC;gBAClC,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;gBAClC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACpB,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,wDAAwD,EAAE,CAAC,CAAC;gBAClF,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,qDAAqD,EAAE,CAAC,CAAC;gBACzF,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,+CAA+C;gBAC/C,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;gBACnC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,oCAAoC,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,gDAAgD,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/main/ui/backend/service.d.ts

@@ -1,14 +1,14 @@
 import { CodegenOptions } from '../../codegen-types.js';
 import { AiServiceType } from '../../../ai-service/service-configurations-types.js';
 import { RcConfig } from '../../config-types.js';
-import { ServiceConfigUpdate, SanitizedServiceConfigurations } from '../common/api-types.js';
+import { ServiceConfigUpdate, SanitizedServiceConfigurations, ContextFile } from '../common/api-types.js';
 import { ContentProps } from '../../common/content-bus-types.js';
 import { UsageMetrics } from '../../common/cost-collector.js';
 import { CodegenResult, ConfirmationProps, Question } from '../common/api-types.js';
-import { FunctionCall, PromptImageMediaType, PromptItemImage } from '../../../ai-service/common-types.js';
+import { FunctionCall, PromptImageMediaType, PromptItem, PromptItemImage } from '../../../ai-service/common-types.js';
 import { ModelType } from '../../../ai-service/common-types.js';
 import { AppContextProvider } from '../../common/app-context-bus.js';
-import { ActionType, StructuredQuestionForm, StructuredQuestionResponse } from '../../../prompt/steps/step-ask-question/step-ask-question-types.js';
+import { ActionType, StructuredQuestionForm, StructuredQuestionResponse } from '../../../prompt/steps/step-iterate/step-iterate-types.js';
 export interface ImageData {
     buffer: Buffer;
     mimetype: PromptImageMediaType;
@@ -17,23 +17,28 @@ export interface ImageData {
 export declare class Service implements AppContextProvider {
     private executionStatus;
     private currentQuestion;
-    private askQuestionConversation;
+    private iterateConversation;
     private codegenOptions;
     private content;
     private pausePromiseResolve;
     private securityToken;
+    private getContext;
     private contextStorage;
-    constructor(codegenOptions: CodegenOptions);
+    constructor(codegenOptions: CodegenOptions, getContext: () => PromptItem[]);
+    /**
+     * Expose the current prompt context items (PromptItem[])
+     */
+    getPromptContext(): PromptItem[];
     /**
      * Edit a message in the conversation
      * @param messageId ID of the message to edit
      * @param newContent New content for the message
+     * @param newData New data for the message (optional)
      * @returns true if the message was successfully edited, false otherwise
      * @throws Error if the message ID is invalid or if editing is not allowed
      */
-    editMessage(messageId: string, newContent: string): Promise<boolean>;
+    editMessage(messageId: string, newContent: string, newData?: Record<string, unknown>): Promise<boolean>;
     /**
-     * Get service configurations in a safe format that excludes sensitive data.
      * This method is used by external API endpoints.
      */
     getServiceConfigurations(): SanitizedServiceConfigurations;
@@ -52,6 +57,40 @@ export declare class Service implements AppContextProvider {
     interruptExecution(): Promise<void>;
     pauseExecution(): Promise<void>;
     resumeExecution(): Promise<void>;
+    /**
+     * Optimize the current context
+     */
+    optimizeContext(): Promise<{
+        success: boolean;
+        message?: string;
+        stats?: {
+            tokensBefore: number;
+            tokensAfter: number;
+            savings: number;
+            percentage: number;
+        };
+    }>;
+    /**
+     * Compress the current context
+     */
+    compressContext(): Promise<{
+        success: boolean;
+        message?: string;
+        stats?: {
+            tokensBefore: number;
+            tokensAfter: number;
+            savings: number;
+            percentage: number;
+        };
+    }>;
+    getContextPreview(): Promise<{
+        preview: {
+            type: string;
+            summary: string;
+            tokenCount: number;
+        }[];
+        totalTokens: number;
+    }>;
     getExecutionStatus(): Promise<string>;
     getCurrentQuestion(): Promise<Question | null>;
     askQuestion(question: string, confirmation: ConfirmationProps): Promise<{
@@ -94,4 +133,26 @@ export declare class Service implements AppContextProvider {
      * Clear all context values
      */
     clearAllContext(): Promise<void>;
+    /**
+     * Get list of files currently in the conversation context
+     * @returns Array of unique file paths
+     */
+    getContextFiles(): Promise<ContextFile[]>;
+    /**
+     * Remove files from the conversation context by setting their content to null
+     * @param filePaths Array of file paths to remove
+     * @returns Number of files removed
+     */
+    removeFilesFromContext(filePaths: string[]): Promise<number>;
+    /**
+     * Get all project files (not just those in context)
+     * @returns Array of file paths
+     */
+    getAllProjectFiles(): Promise<string[]>;
+    /**
+     * Add files to the conversation context by fetching their content
+     * @param filePaths Array of file paths to add
+     * @returns Number of files added
+     */
+    addFilesToContext(filePaths: string[]): Promise<number>;
 }

package/dist/main/ui/backend/service.js

@@ -5,54 +5,90 @@ import { abortController } from '../../common/abort-controller.js';
 import { getUsageMetrics } from '../../common/cost-collector.js';
 import { editMessage, putSystemMessage } from '../../common/content-bus.js';
 import { getGenerateContentFunctions } from '../../codegen.js';
-import { ModelType } from '../../../ai-service/common-types.js';
 import { getSanitizedServiceConfigurations, updateServiceConfig } from '../../../ai-service/service-configurations.js';
+import { estimateTokenCount } from '../../../prompt/token-estimator.js';
+import { executeStepContextOptimization } from '../../../prompt/steps/step-context-optimization.js';
+import { executeStepContextCompression } from '../../../prompt/steps/step-context-compression.js';
+import { getFilesContextSizeFromPrompt } from '../../../prompt/context-utils.js';
 export class Service {
     executionStatus = 'idle';
     currentQuestion = null;
-    askQuestionConversation = [];
+    iterateConversation = [];
     codegenOptions;
     content = [];
     pausePromiseResolve = null;
     securityToken;
+    getContext = () => [];
     // Context storage for managing application context
     contextStorage = new Map();
-    constructor(codegenOptions) {
+    constructor(codegenOptions, getContext) {
         this.codegenOptions = codegenOptions;
         this.securityToken = this.generateToken();
+        this.getContext = getContext;
+    }
+    /**
+     * Expose the current prompt context items (PromptItem[])
+     */
+    getPromptContext() {
+        return this.getContext();
     }
     /**
      * Edit a message in the conversation
      * @param messageId ID of the message to edit
      * @param newContent New content for the message
+     * @param newData New data for the message (optional)
      * @returns true if the message was successfully edited, false otherwise
      * @throws Error if the message ID is invalid or if editing is not allowed
      */
-    async editMessage(messageId, newContent) {
+    async editMessage(messageId, newContent, newData) {
         // Basic input validation
-        if (!messageId || !newContent.trim()) {
-            throw new Error('Invalid message ID or content');
+        if (!messageId || (newContent === undefined && newData === undefined)) {
+            throw new Error('Invalid message ID, content or data');
         }
         // Find the message in the content array
         const contentItem = this.content.find((item) => item.message?.id === messageId);
         if (!contentItem || !contentItem.message) {
             throw new Error('Message not found');
         }
-        // Check if editing is allowed for this message type
-        if (contentItem.message.type === 'system') {
-            throw new Error('System messages cannot be edited');
-        }
         // Check if the message is from an ongoing iteration
         const currentIterationId = this.getCurrentIterationId();
         if (contentItem.message.iterationId !== currentIterationId || this.executionStatus === 'idle') {
             throw new Error('Can only edit messages from the currently ongoing iteration');
         }
+        // Prevent editing after submit (when no question is active)
+        if (!this.currentQuestion) {
+            throw new Error('Cannot edit message after it has been submitted or when no question is active');
+        }
+        // Apply newData if provided
+        if (newData !== undefined && contentItem.message) {
+            // Validate codegenPlanning data structure
+            if (newData.name === 'codegenPlanning') {
+                const args = newData.args;
+                if (!args ||
+                    typeof args.problemAnalysis !== 'string' ||
+                    typeof args.codeChanges !== 'string' ||
+                    !Array.isArray(args.affectedFiles)) {
+                    throw new Error('Invalid codegenPlanning data structure');
+                }
+            }
+            // Validate codegenSummary data structure
+            else if (newData.name === 'codegenSummary') {
+                const args = newData.args;
+                if (!args ||
+                    typeof args.explanation !== 'string' ||
+                    !Array.isArray(args.fileUpdates) ||
+                    !Array.isArray(args.contextPaths)) {
+                    throw new Error('Invalid codegenSummary data structure');
+                }
+            }
+            contentItem.message.data = newData;
+            contentItem.data = newData;
+        }
         // Try to edit the message using content bus
-        const success = editMessage(contentItem, newContent);
+        const success = editMessage(contentItem, newContent !== undefined ? newContent : contentItem.message.content, newData);
         return success;
     }
     /**
-     * Get service configurations in a safe format that excludes sensitive data.
      * This method is used by external API endpoints.
      */
     getServiceConfigurations() {
@@ -134,20 +170,18 @@ export class Service {
                     text: prompt,
                 },
             ], {
-                functionDefs: modelType === ModelType.REASONING
-                    ? []
-                    : [
-                        {
-                            name: 'printMessage',
-                            description: 'Print a message',
-                            parameters: {
-                                type: 'object',
-                                properties: { message: { type: 'string' } },
-                                required: ['message'],
-                            },
+                functionDefs: [
+                    {
+                        name: 'printMessage',
+                        description: 'Print a message',
+                        parameters: {
+                            type: 'object',
+                            properties: { message: { type: 'string' } },
+                            required: ['message'],
                         },
-                    ],
-                requiredFunctionName: modelType === ModelType.REASONING ? 'reasoningInferenceResponse' : 'printMessage',
+                    },
+                ],
+                requiredFunctionName: 'printMessage',
                 temperature,
                 modelType,
                 expectedResponseType: {
@@ -187,6 +221,127 @@ export class Service {
             }
         }
     }
+    /**
+     * Optimize the current context
+     */
+    async optimizeContext() {
+        try {
+            const prompt = this.getContext();
+            const generateContentFns = getGenerateContentFunctions();
+            const generateContentFn = generateContentFns[this.codegenOptions.aiService];
+            if (!generateContentFn) {
+                throw new Error(`AI service ${this.codegenOptions.aiService} is not available`);
+            }
+            // Estimate tokens before
+            const tokensBefore = estimateTokenCount(JSON.stringify(prompt));
+            // Execute optimization
+            await executeStepContextOptimization(generateContentFn, prompt, this.codegenOptions);
+            // Estimate tokens after
+            const tokensAfter = estimateTokenCount(JSON.stringify(prompt));
+            const filesContextSize = getFilesContextSizeFromPrompt(prompt);
+            putSystemMessage('Context optimization applied', { filesContextSize, contextSize: tokensAfter });
+            return {
+                success: true,
+                stats: {
+                    tokensBefore,
+                    tokensAfter,
+                    savings: tokensBefore - tokensAfter,
+                    percentage: tokensBefore > 0 ? Math.round(((tokensBefore - tokensAfter) / tokensBefore) * 100) : 0,
+                },
+            };
+        }
+        catch (error) {
+            console.error('Error optimizing context:', error);
+            return {
+                success: false,
+                message: error instanceof Error ? error.message : 'Unknown error during context optimization',
+            };
+        }
+    }
+    /**
+     * Compress the current context
+     */
+    async compressContext() {
+        try {
+            const prompt = this.getContext();
+            const generateContentFns = getGenerateContentFunctions();
+            const generateContentFn = generateContentFns[this.codegenOptions.aiService];
+            if (!generateContentFn) {
+                throw new Error(`AI service ${this.codegenOptions.aiService} is not available`);
+            }
+            // Estimate tokens before
+            const tokensBefore = estimateTokenCount(JSON.stringify(prompt));
+            // Execute compression
+            await executeStepContextCompression(generateContentFn, prompt, this.codegenOptions);
+            // Estimate tokens after
+            const tokensAfter = estimateTokenCount(JSON.stringify(prompt));
+            const filesContextSize = getFilesContextSizeFromPrompt(prompt);
+            putSystemMessage('Context compression applied', {
+                tokensBefore,
+                tokensAfter,
+                filesContextSize,
+                contextSize: tokensAfter,
+            });
+            return {
+                success: true,
+                stats: {
+                    tokensBefore,
+                    tokensAfter,
+                    savings: tokensBefore - tokensAfter,
+                    percentage: tokensBefore > 0 ? Math.round(((tokensBefore - tokensAfter) / tokensBefore) * 100) : 0,
+                },
+            };
+        }
+        catch (error) {
+            console.error('Error compressing context:', error);
+            return {
+                success: false,
+                message: error instanceof Error ? error.message : 'Unknown error during context compression',
+            };
+        }
+    }
+    async getContextPreview() {
+        const prompt = this.getContext();
+        const preview = prompt.map((item) => {
+            let summary = '';
+            if (item.type === 'user') {
+                const text = item.text ?? '';
+                summary = text ? text.substring(0, 100) + (text.length > 100 ? '...' : '') : 'User Input';
+                if (Array.isArray(item.images) && item.images.length > 0) {
+                    summary += ` [${item.images.length} images]`;
+                }
+                if (item.functionResponses?.length) {
+                    const name = item.functionResponses.map((fr) => fr.name).join(', ');
+                    summary += ` Function Response${name ? `: ${name}` : ''}`;
+                }
+            }
+            else if (item.type === 'assistant') {
+                // assistant may contain text and/or functionCalls
+                const text = item.text;
+                if (text) {
+                    summary = text.substring(0, 100) + (text.length > 100 ? '...' : '');
+                }
+                else if (Array.isArray(item.functionCalls)) {
+                    const names = item.functionCalls.map((fc) => fc?.name).filter(Boolean);
+                    summary = names.length ? `Assistant: ${names.join(', ')}` : 'Assistant';
+                }
+                else {
+                    summary = 'Assistant';
+                }
+            }
+            else if (item.type === 'systemPrompt') {
+                const text = item.text;
+                summary = text ? text.substring(0, 100) + (text.length > 100 ? '...' : '') : 'System Prompt';
+            }
+            return {
+                type: item.type,
+                summary,
+                tokenCount: estimateTokenCount(JSON.stringify(item)),
+            };
+        });
+        const totalTokens = estimateTokenCount(JSON.stringify(prompt));
+        return { preview, totalTokens };
+    }
     async getExecutionStatus() {
         return this.executionStatus;
     }
@@ -202,7 +357,7 @@ export class Service {
         };
         await this.waitForQuestionAnswer();
         console.log('Question answer wait finished.');
-        const conversationItem = this.askQuestionConversation.find((q) => q.id === questionId);
+        const conversationItem = this.iterateConversation.find((q) => q.id === questionId);
         if (!conversationItem) {
             // Should ideally not happen if waitForQuestionAnswer resolved correctly
             throw new Error(`Conversation item not found for question ID: ${questionId}`);
@@ -220,7 +375,7 @@ export class Service {
         };
         await this.waitForQuestionAnswer();
         console.log('Structured question answer wait finished.');
-        const conversationItem = this.askQuestionConversation.find((q) => q.id === questionId);
+        const conversationItem = this.iterateConversation.find((q) => q.id === questionId);
         if (!conversationItem || !conversationItem.structuredResponse) {
             // Should ideally not happen if waitForQuestionAnswer resolved correctly
             throw new Error(`Conversation item or structured response not found for question ID: ${questionId}`);
@@ -240,7 +395,7 @@ export class Service {
                 mediaType: image.mimetype,
             }));
             // Store the answer and processed images in the conversation history
-            this.askQuestionConversation.push({
+            this.iterateConversation.push({
                 id: this.currentQuestion.id,
                 question: this.currentQuestion.text,
                 answer: answer,
@@ -345,5 +500,123 @@ export class Service {
     async clearAllContext() {
         this.contextStorage.clear();
     }
+    /**
+     * Get list of files currently in the conversation context
+     * @returns Array of unique file paths
+     */
+    async getContextFiles() {
+        const contextFilesMap = new Map();
+        for (const item of this.getContext()) {
+            for (const response of item.functionResponses ?? []) {
+                if (response.name === 'getSourceCode' && response.content) {
+                    try {
+                        const sourceCode = JSON.parse(response.content);
+                        for (const [filePath, fileData] of Object.entries(sourceCode)) {
+                            if (fileData && typeof fileData === 'object' && 'content' in fileData && fileData.content !== null) {
+                                const content = String(fileData.content);
+                                const tokenCount = estimateTokenCount(content);
+                                contextFilesMap.set(filePath, tokenCount);
+                            }
+                        }
+                    }
+                    catch (error) {
+                        console.warn('Failed to parse getSourceCode response while fetching context files:', error);
+                    }
+                }
+            }
+        }
+        return Array.from(contextFilesMap.entries())
+            .map(([path, tokenCount]) => ({ path, tokenCount }))
+            .sort((a, b) => a.path.localeCompare(b.path));
+    }
+    /**
+     * Remove files from the conversation context by setting their content to null
+     * @param filePaths Array of file paths to remove
+     * @returns Number of files removed
+     */
+    async removeFilesFromContext(filePaths) {
+        if (!Array.isArray(filePaths) || filePaths.length === 0) {
+            return 0;
+        }
+        const filesToRemove = new Set(filePaths);
+        const removedFiles = new Set();
+        for (const item of this.getContext()) {
+            for (const response of item.functionResponses ?? []) {
+                if (response.name === 'getSourceCode' && response.content) {
+                    try {
+                        const sourceCode = JSON.parse(response.content);
+                        let modified = false;
+                        for (const filePath of filesToRemove) {
+                            if (sourceCode[filePath] &&
+                                typeof sourceCode[filePath] === 'object' &&
+                                'content' in sourceCode[filePath] &&
+                                sourceCode[filePath].content !== null) {
+                                sourceCode[filePath].content = null;
+                                removedFiles.add(filePath);
+                                modified = true;
+                            }
+                        }
+                        if (modified) {
+                            response.content = JSON.stringify(sourceCode);
+                        }
+                    }
+                    catch (error) {
+                        console.warn('Failed to parse getSourceCode response while removing context files:', error);
+                    }
+                }
+            }
+        }
+        if (removedFiles.size > 0) {
+            const filesContextSize = getFilesContextSizeFromPrompt(this.getContext());
+            putSystemMessage('Context reduction applied', { removed: Array.from(removedFiles), filesContextSize });
+        }
+        return removedFiles.size;
+    }
+    /**
+     * Get all project files (not just those in context)
+     * @returns Array of file paths
+     */
+    async getAllProjectFiles() {
+        const { getSourceFiles } = await import('../../../files/find-files.js');
+        return getSourceFiles();
+    }
+    /**
+     * Add files to the conversation context by fetching their content
+     * @param filePaths Array of file paths to add
+     * @returns Number of files added
+     */
+    async addFilesToContext(filePaths) {
+        if (!Array.isArray(filePaths) || filePaths.length === 0) {
+            return 0;
+        }
+        const { getSourceCode } = await import('../../../files/read-files.js');
+        const { putSystemMessage } = await import('../../common/content-bus.js');
+        try {
+            // Fetch the content of the requested files
+            const sourceCode = getSourceCode({ filterPaths: filePaths, forceAll: true }, this.codegenOptions);
+            // Create a getSourceCode function response and inject it into the context
+            const functionResponse = {
+                name: 'getSourceCode',
+                content: JSON.stringify(sourceCode),
+            };
+            // Find the most recent prompt item in the context to attach the response
+            const contextItems = this.getContext();
+            if (contextItems.length > 0) {
+                const lastItem = contextItems[contextItems.length - 1];
+                if (!lastItem.functionResponses) {
+                    lastItem.functionResponses = [];
+                }
+                lastItem.functionResponses.push(functionResponse);
+            }
+            const addedCount = filePaths.length;
+            const filesContextSize = getFilesContextSizeFromPrompt(this.getContext());
+            putSystemMessage('Files added to context', { added: filePaths, filesContextSize });
+            return addedCount;
+        }
+        catch (error) {
+            console.error('Failed to add files to context:', error);
+            throw error;
+        }
+    }
 }
 //# sourceMappingURL=service.js.map