genaicode 0.6.1 → 0.8.0

package/dist/eval/prompt2.js

@@ -0,0 +1,125 @@
+import { getSystemPrompt } from '../prompt/systemprompt';
+export const DEBUG_PROMPT2 = [
+    {
+        type: 'systemPrompt',
+        systemPrompt: getSystemPrompt({ verbose: false, askQuestion: true, interactive: false, ui: true }),
+    },
+    {
+        type: 'user',
+        text: 'Hello, GenAIcode!',
+    },
+    {
+        type: 'assistant',
+        text: 'Hello there! I guess you have a task for me today. Before we start, could you please provide me with the current source code of your application, and the image assets?',
+        functionCalls: [
+            {
+                name: 'getSourceCode',
+            },
+            {
+                name: 'getImageAssets',
+            },
+        ],
+    },
+    {
+        type: 'user',
+        functionResponses: [
+            {
+                name: 'getSourceCode',
+                content: JSON.stringify({
+                    '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers': {
+                        'cancel-code-generation.ts': { summary: 'Handles the cancellation of code generation.' },
+                        'confirm-code-generation.ts': { summary: 'Handles confirmation of code generation' },
+                        'context-optimization.ts': { summary: 'Handles context optimization' },
+                        'default-action.ts': { summary: 'Default action handler' },
+                        'handle-request-answer-with-image.ts': {
+                            summary: 'Handles request to generate image and display it to user',
+                        },
+                        'handle-request-answer.ts': { summary: 'Handles user answer requests' },
+                        'handler-utils.ts': { summary: 'Utility functions for step-ask-question handlers' },
+                        'remove-files-from-context.ts': { summary: 'Handles removing files from context' },
+                        'request-files-content.ts': { summary: 'Handles requesting content of missing files' },
+                        'request-permissions.ts': { summary: 'Handles requesting permissions from user' },
+                        'start-code-generation.ts': { summary: 'Handles the start of code generation.' },
+                    },
+                }),
+            },
+            {
+                name: 'getImageAssets',
+                content: '{"/Users/gtanczyk/src/codegen/media/logo-dark.png":{"mimeType":"image/png","width":1200,"height":686},"/Users/gtanczyk/src/codegen/media/logo.png":{"mimeType":"image/png","width":1200,"height":686},"/Users/gtanczyk/src/codegen/src/vite-genaicode/media/demo-for-readme.gif":{"mimeType":"image/gif","width":769,"height":465}}',
+            },
+        ],
+        cache: true,
+        text: 'Sure, here is the application source code and image assets.',
+    },
+    {
+        type: 'assistant',
+        text: "Thank you, I'm ready to assist you with your request.",
+    },
+    {
+        type: 'user',
+        text: 'Hello',
+    },
+    {
+        type: 'assistant',
+        text: 'Hello, How can I assist you today?',
+    },
+    {
+        type: 'user',
+        text: 'we will be working on files from src/prompt/steps/step-ask-question/handlers, I want you to analyze them and provide me with a summary of the content of each file.',
+    },
+    //   {
+    //     type: 'assistant',
+    //     text: 'Please provide the content of the following files from the src/prompt/steps/step-ask-question/handlers directory: \n- cancel-code-generation.ts\n- confirm-code-generation.ts \n- context-optimization.ts\n- default-action.ts\n- handle-request-answer-with-image.ts\n- handle-request-answer.ts\n- handler-utils.ts\n- remove-files-from-context.ts\n- request-files-content.ts\n- request-permissions.ts\n- start-code-generation.ts',
+    //     functionCalls: [
+    //       {
+    //         name: 'getSourceCode',
+    //         id: 'toolu_019TwZHC29M4DDNGqt72yKek_source',
+    //         args: {
+    //           filePaths: [
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/cancel-code-generation.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/confirm-code-generation.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/context-optimization.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/default-action.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/handle-request-answer-with-image.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/handle-request-answer.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/handler-utils.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/remove-files-from-context.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/request-files-content.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/request-permissions.ts',
+    //             '/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/start-code-generation.ts',
+    //           ],
+    //         },
+    //       },
+    //     ],
+    //   },
+    //   {
+    //     type: 'user',
+    //     text: 'All requested file contents have been provided.',
+    //     functionResponses: [
+    //       {
+    //         name: 'getSourceCode',
+    //         call_id: 'toolu_019TwZHC29M4DDNGqt72yKek_source',
+    //         content:
+    //           "{\"/Users/gtanczyk/src/codegen/src/main/codegen-types.ts\":{\"content\":\"import { FunctionDef, GenerateContentFunction } from '../ai-service/common';\\nimport { ActionHandler } from '../prompt/steps/step-ask-question/step-ask-question-types';\\n\\n/** Example: {@link ../../examples/genaicode_plugins/grok_ai_service.ts} */\\nexport type PluginAiServiceType = `plugin:${string}`;\\n\\n/** Example: {@link ../../examples/genaicode_plugins/nonsense_action_handlers.ts} */\\nexport type PluginActionType = `plugin:${string}`;\\n\\nexport type AiServiceType =\\n  | 'vertex-ai'\\n  | 'ai-studio'\\n  | 'vertex-ai-claude'\\n  | 'chat-gpt'\\n  | 'anthropic'\\n  | PluginAiServiceType;\\n\\nexport type ImagenType = 'vertex-ai' | 'dall-e';\\n\\nexport interface UploadedImage {\\n  base64url: string;\\n  mediaType: 'image/jpeg' | 'image/png' | 'image/gif' | 'image/webp';\\n  originalName: string;\\n}\\n\\nexport interface CodegenOptions {\\n  explicitPrompt?: string;\\n  taskFile?: string;\\n  considerAllFiles?: boolean;\\n  allowFileCreate?: boolean;\\n  allowFileDelete?: boolean;\\n  allowDirectoryCreate?: boolean;\\n  allowFileMove?: boolean;\\n  vision?: boolean;\\n  imagen?: ImagenType;\\n  aiService: AiServiceType;\\n\\n  disableContextOptimization?: boolean;\\n  temperature?: number;\\n  cheap?: boolean;\\n  dryRun?: boolean;\\n  verbose?: boolean;\\n  requireExplanations?: boolean;\\n  geminiBlockNone?: boolean;\\n  disableInitialLint?: boolean;\\n  contentMask?: string;\\n  ignorePatterns?: string[];\\n  askQuestion?: boolean;\\n  interactive?: boolean;\\n  ui?: boolean;\\n  uiPort?: number;\\n  uiFrameAncestors?: string[];\\n  disableCache?: boolean;\\n  dependencyTree?: boolean;\\n  historyEnabled?: boolean;\\n\\n  disableAiServiceFallback?: boolean;\\n  conversationSummaryEnabled?: boolean;\\n  images?: UploadedImage[];\\n  isDev?: boolean;\\n}\\n\\ninterface ExecutorArgs {\\n  [key: string]: unknown;\\n}\\n\\nexport type OperationExecutor = (args: ExecutorArgs, options: CodegenOptions) => Promise<void>;\\n\\nexport type Operation = {\\n  executor: OperationExecutor;\\n  def: FunctionDef;\\n};\\n\\nexport interface Plugin {\\n  name: string;\\n  aiServices?: Record<string, GenerateContentFunction>;\\n  operations?: Record<string, Operation>;\\n  actionHandlers?: Record<\\n    string,\\n    {\\n      /** The action handler implementation */\\n      handler: ActionHandler;\\n      /**\\n       * Description of what this action handler does.\\n       * This description will be included in the askQuestion function definition\\n       * to help the AI understand when to use this action.\\n       */\\n      description: string;\\n    }\\n  >;\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/main/codegen.ts\":{\"content\":\"import { exec } from 'child_process';\\nimport util from 'util';\\n\\nimport * as cliParams from '../cli/cli-params.js';\\nimport { validateCliParams } from '../cli/validate-cli-params.js';\\nimport { generateContent as generateContentVertexAi } from '../ai-service/vertex-ai.js';\\nimport { generateContent as generateContentGPT } from '../ai-service/chat-gpt.js';\\nimport { generateContent as generateContentAnthropic } from '../ai-service/anthropic.js';\\nimport { generateContent as generateContentVertexAiClaude } from '../ai-service/vertex-ai-claude.js';\\nimport { generateContent as generateContentAiStudio } from '../ai-service/ai-studio.js';\\nimport { generateImage as generateImageDallE } from '../ai-service/dall-e.js';\\nimport { generateImage as generateImageVertexAi } from '../ai-service/vertex-ai-imagen.js';\\n\\nimport { promptService } from '../prompt/prompt-service.js';\\nimport { updateFiles } from '../files/update-files.js';\\nimport { rcConfig } from '../main/config.js';\\nimport { AiServiceType, CodegenOptions, ImagenType } from './codegen-types.js';\\nimport { getLintFixPrompt } from '../prompt/prompt-codegen.js';\\nimport { printHelpMessage } from '../cli/cli-options.js';\\nimport { FunctionCall, GenerateContentFunction, GenerateImageFunction } from '../ai-service/common.js';\\nimport { getCodeGenPrompt } from '../prompt/prompt-codegen.js';\\n\\nimport { runInteractiveMode } from './interactive/codegen-interactive.js';\\nimport { runCodegenUI } from './ui/codegen-ui.js';\\nimport {\\n  putSystemMessage,\\n  putUserMessage,\\n  setCurrentIterationId,\\n  unsetCurrentIterationId,\\n} from './common/content-bus.js';\\nimport { refreshFiles } from '../files/find-files.js';\\nimport { getRegisteredAiServices } from './plugin-loader.js';\\nimport { stringToAiServiceType } from './codegen-utils.js';\\n\\n/** Executes codegen */\\nexport async function runCodegen(isDev = false): Promise<void> {\\n  // Print to console the received parameters\\n  console.log(`Received parameters: ${process.argv.slice(2).join(' ')}`);\\n\\n  validateCliParams();\\n\\n  // Handle --help option\\n  if (cliParams.helpRequested) {\\n    printHelpMessage();\\n    return;\\n  }\\n\\n  const options: CodegenOptions = {\\n    explicitPrompt: cliParams.explicitPrompt,\\n    taskFile: cliParams.taskFile,\\n    considerAllFiles: cliParams.considerAllFiles,\\n    dependencyTree: cliParams.dependencyTree,\\n\\n    allowFileCreate: cliParams.allowFileCreate,\\n    allowFileDelete: cliParams.allowFileDelete,\\n    allowDirectoryCreate: cliParams.allowDirectoryCreate,\\n    allowFileMove: cliParams.allowFileMove,\\n\\n    aiService: stringToAiServiceType(cliParams.aiService),\\n    vision: cliParams.vision,\\n    imagen: cliParams.imagen,\\n\\n    disableContextOptimization: cliParams.disableContextOptimization,\\n    temperature: cliParams.temperature,\\n    cheap: cliParams.cheap,\\n    dryRun: cliParams.dryRun,\\n    verbose: cliParams.verbosePrompt,\\n    requireExplanations: !cliParams.disableExplanations,\\n    geminiBlockNone: cliParams.geminiBlockNone,\\n    disableInitialLint: cliParams.disableInitialLint,\\n    contentMask: cliParams.contentMask,\\n    ignorePatterns: cliParams.ignorePatterns,\\n    askQuestion: cliParams.askQuestion,\\n    disableCache: cliParams.disableCache,\\n    interactive: cliParams.interactive,\\n    ui: cliParams.ui,\\n    uiPort: cliParams.uiPort,\\n    disableAiServiceFallback: cliParams.disableAiServiceFallback,\\n    historyEnabled: !cliParams.disableHistory,\\n    conversationSummaryEnabled: !cliParams.disableConversationSummary,\\n\\n    isDev,\\n  };\\n\\n  if (cliParams.ui) {\\n    await runCodegenUI(options);\\n  } else if (cliParams.interactive) {\\n    // Handle interactive mode\\n    await runInteractiveMode(options);\\n  } else {\\n    console.log('Executing codegen in non-interactive mode');\\n    await runCodegenIteration(options);\\n  }\\n}\\n\\nexport async function runCodegenIteration(\\n  options: CodegenOptions,\\n  abortSignal?: AbortSignal,\\n  waitIfPaused: () => Promise<void> = () => Promise.resolve(),\\n) {\\n  refreshFiles();\\n\\n  setCurrentIterationId();\\n\\n  putUserMessage(\\n    options.explicitPrompt ?? options.taskFile ?? 'Run codegen iteration without explicit prompt.',\\n    undefined,\\n    undefined,\\n    options.images,\\n  );\\n\\n  if (rcConfig.lintCommand && !options.disableInitialLint) {\\n    try {\\n      putSystemMessage(`Executing lint command: ${rcConfig.lintCommand}`);\\n      await execPromise(rcConfig.lintCommand, { cwd: rcConfig.rootDir });\\n      putSystemMessage('Lint command executed successfully');\\n    } catch (error) {\\n      const { stderr, stdout } = error as { stdout: string; stderr: string };\\n      putSystemMessage(\\n        'Lint command failed. Aborting codegen, please fix lint issues before running codegen, or use --disable-initial-lint',\\n      );\\n      console.log('Lint errors:', stdout, stderr);\\n      unsetCurrentIterationId();\\n      return;\\n    }\\n  } else if (rcConfig.lintCommand && options.disableInitialLint) {\\n    console.log('Initial lint was skipped.');\\n  }\\n\\n  if (abortSignal?.aborted) {\\n    unsetCurrentIterationId();\\n    throw new Error('Codegen iteration aborted');\\n  }\\n\\n  await waitIfPaused();\\n\\n  putSystemMessage('Generating response');\\n  try {\\n    const functionCalls = await promptService(\\n      getGenerateContentFunctions(),\\n      GENERATE_IMAGE_FNS,\\n      getCodeGenPrompt(options),\\n      waitIfPaused,\\n    );\\n    console.log('Received function calls:', functionCalls);\\n\\n    await waitIfPaused();\\n\\n    if (functionCalls.length === 0) {\\n      putSystemMessage('No updates to apply');\\n    } else if (options.dryRun) {\\n      putSystemMessage('Dry run mode, not updating files');\\n    } else {\\n      putSystemMessage('Update files');\\n      await updateFiles(\\n        functionCalls.filter((call) => call.name !== 'explanation' && call.name !== 'getSourceCode'),\\n        options,\\n      );\\n      putSystemMessage('Initial updates applied');\\n\\n      if (abortSignal?.aborted) {\\n        throw new Error('Codegen iteration aborted after initial updates');\\n      }\\n\\n      await waitIfPaused();\\n\\n      // Check if lintCommand is specified in .genaicoderc\\n      if (rcConfig.lintCommand && functionCalls.length > 0) {\\n        try {\\n          putSystemMessage(`Executing lint command: ${rcConfig.lintCommand}`);\\n          await execPromise(rcConfig.lintCommand, { cwd: rcConfig.rootDir });\\n          putSystemMessage('Lint command executed successfully');\\n        } catch (error) {\\n          putSystemMessage('Lint command failed. Attempting to fix issues...');\\n\\n          // Prepare the lint error output for the second pass\\n          const firstLintError = error as { stdout: string; stderr: string };\\n          const lintErrorPrompt = getLintFixPrompt(\\n            rcConfig.lintCommand,\\n            options,\\n            firstLintError.stdout,\\n            firstLintError.stderr,\\n          );\\n\\n          putSystemMessage('Generating response for lint fixes');\\n          const lintFixFunctionCalls = (await promptService(\\n            getGenerateContentFunctions(),\\n            GENERATE_IMAGE_FNS,\\n            {\\n              prompt: lintErrorPrompt,\\n              options: { ...options, considerAllFiles: true },\\n            },\\n            waitIfPaused,\\n          )) as FunctionCall[];\\n\\n          console.log('Received function calls for lint fixes:', lintFixFunctionCalls);\\n\\n          await waitIfPaused();\\n\\n          putSystemMessage('Applying lint fixes');\\n          updateFiles(\\n            lintFixFunctionCalls.filter((call) => call.name !== 'explanation' && call.name !== 'getSourceCode'),\\n            options,\\n          );\\n\\n          if (abortSignal?.aborted) {\\n            throw new Error('Codegen iteration aborted after lint fixes');\\n          }\\n\\n          // Run lint command again to verify fixes\\n          try {\\n            putSystemMessage(`Re-running lint command: ${rcConfig.lintCommand}`);\\n            await execPromise(rcConfig.lintCommand);\\n            putSystemMessage('Lint command executed successfully after fixes');\\n          } catch (secondLintError) {\\n            const error = secondLintError as { stdout: string; stderr: string };\\n            putSystemMessage('Lint command still failing after fixes. Manual intervention may be required.');\\n            console.log('Lint errors:', error.stdout, error.stderr);\\n          }\\n        }\\n      }\\n\\n      console.log('Done!');\\n    }\\n  } catch (error) {\\n    if (error instanceof Error) {\\n      if (error.name === 'AbortError' || error.message.includes('interrupted')) {\\n        putSystemMessage('Codegen iteration was interrupted');\\n      } else if (error.message.includes('Rate limit exceeded')) {\\n        putSystemMessage(\\n          'Rate limit exceeded. Consider switching to a different AI service or waiting before retrying.',\\n        );\\n      } else {\\n        putSystemMessage(`An error occurred during codegen: ${error.message}`);\\n      }\\n    } else {\\n      putSystemMessage('An unknown error occurred during codegen');\\n    }\\n    console.error('Error details:', error);\\n  } finally {\\n    unsetCurrentIterationId();\\n  }\\n}\\n\\n// helper functions and consts\\n\\nconst execPromise = util.promisify(exec);\\n\\nfunction getGenerateContentFunctions(): Record<AiServiceType, GenerateContentFunction> {\\n  return {\\n    'vertex-ai-claude': generateContentVertexAiClaude,\\n    'vertex-ai': generateContentVertexAi,\\n    'ai-studio': generateContentAiStudio,\\n    anthropic: generateContentAnthropic,\\n    'chat-gpt': generateContentGPT,\\n    ...Object.fromEntries(getRegisteredAiServices().entries()),\\n  };\\n}\\n\\nconst GENERATE_IMAGE_FNS: Record<ImagenType, GenerateImageFunction> = {\\n  'dall-e': generateImageDallE,\\n  'vertex-ai': generateImageVertexAi,\\n} as const;\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/prompt-service.ts\":{\"content\":\"import assert from 'node:assert';\\nimport fs from 'fs';\\nimport mime from 'mime-types';\\n\\nimport { getSystemPrompt } from './systemprompt.js';\\nimport { getFunctionDefs } from './function-calling.js';\\nimport { getSourceCode, getImageAssets } from '../files/read-files.js';\\nimport {\\n  PromptItem,\\n  FunctionDef,\\n  FunctionCall,\\n  GenerateContentFunction,\\n  GenerateImageFunction,\\n} from '../ai-service/common.js';\\nimport { importantContext } from '../main/config.js';\\nimport { AiServiceType, CodegenOptions, ImagenType } from '../main/codegen-types.js';\\nimport { executeStepAskQuestion } from './steps/step-ask-question/step-ask-question.js';\\nimport { validateAndRecoverSingleResult } from './steps/step-validate-recover.js';\\nimport { executeStepVerifyPatch } from './steps/step-verify-patch.js';\\nimport { executeStepGenerateImage } from './steps/step-generate-image.js';\\nimport { executeStepContextOptimization } from './steps/step-context-optimization.js';\\nimport { StepResult } from './steps/steps-types.js';\\nimport { CodegenPrompt } from './prompt-codegen.js';\\nimport { putSystemMessage } from '../main/common/content-bus.js';\\nimport { handleAiServiceFallback } from './ai-service-fallback.js';\\nimport { summarizeSourceCode } from './steps/step-summarization.js';\\nimport { executeStepHistoryUpdate, getCurrentHistory } from './steps/step-history-update.js';\\nimport { executeStepGenerateSummary } from './steps/step-generate-summary.js';\\nimport { getSourceCodeTree } from '../files/source-code-tree.js';\\n\\n/** A function that communicates with model using */\\nexport async function promptService(\\n  generateContentFns: Record<AiServiceType, GenerateContentFunction>,\\n  generateImageFns: Record<ImagenType, GenerateImageFunction>,\\n  codegenPrompt: CodegenPrompt,\\n  waitIfPaused: () => Promise<void> = () => Promise.resolve(),\\n): Promise<FunctionCall[]> {\\n  const generateContentFn: GenerateContentFunction = async (...args) => {\\n    return await handleAiServiceFallback(\\n      generateContentFns,\\n      codegenPrompt.options.aiService,\\n      codegenPrompt.options,\\n      ...args,\\n    );\\n  };\\n\\n  const generateImageFn: GenerateImageFunction = (...args) => {\\n    assert(codegenPrompt.options.imagen, 'imagen value must be provided');\\n    return generateImageFns[codegenPrompt.options.imagen](...args);\\n  };\\n\\n  const { result, prompt } = await executePromptService(\\n    generateContentFn,\\n    generateImageFn,\\n    codegenPrompt,\\n    waitIfPaused,\\n  );\\n\\n  if (codegenPrompt.options.historyEnabled) {\\n    await executeStepHistoryUpdate(generateContentFn, prompt, codegenPrompt.options);\\n  }\\n\\n  return result;\\n}\\n\\nasync function executePromptService(\\n  generateContentFn: GenerateContentFunction,\\n  generateImageFn: GenerateImageFunction,\\n  codegenPrompt: CodegenPrompt,\\n  waitIfPaused: () => Promise<void> = () => Promise.resolve(),\\n): Promise<{ result: FunctionCall[]; prompt: PromptItem[] }> {\\n  const messages = prepareMessages(codegenPrompt);\\n\\n  // First stage: summarize the source code\\n  if (!codegenPrompt.options.disableContextOptimization) {\\n    await summarizeSourceCode(\\n      generateContentFn,\\n      getSourceCode({ forceAll: true }, codegenPrompt.options),\\n      codegenPrompt.options,\\n    );\\n  }\\n\\n  // Second stage: generate code generation summary, which should not take a lot of output tokens\\n  const getSourceCodeRequest: FunctionCall = { name: 'getSourceCode' };\\n\\n  const prompt: PromptItem[] = [\\n    { type: 'systemPrompt', systemPrompt: getSystemPrompt(codegenPrompt.options) },\\n    { type: 'user', text: messages.suggestSourceCode },\\n    {\\n      type: 'assistant',\\n      text: messages.requestSourceCode,\\n      functionCalls: [getSourceCodeRequest, ...(codegenPrompt.options.historyEnabled ? [{ name: 'readHistory' }] : [])],\\n    },\\n  ];\\n\\n  const getSourceCodeResponse: PromptItem = {\\n    type: 'user',\\n    functionResponses: [\\n      { name: 'getSourceCode', content: messages.sourceCode },\\n      ...(codegenPrompt.options.historyEnabled ? [{ name: 'readHistory', content: getCurrentHistory() }] : []),\\n    ],\\n    cache: true,\\n  };\\n  prompt.push(getSourceCodeResponse);\\n\\n  if (codegenPrompt.options.vision) {\\n    prompt.slice(-1)[0].text = messages.suggestImageAssets;\\n    prompt.push(\\n      { type: 'assistant', text: messages.requestImageAssets, functionCalls: [{ name: 'getImageAssets' }] },\\n      { type: 'user', functionResponses: [{ name: 'getImageAssets', content: messages.imageAssets }] },\\n    );\\n  }\\n\\n  prompt.slice(-1)[0].text = messages.prompt;\\n\\n  // Add uploaded images to the prompt if available\\n  if (codegenPrompt.options.images && codegenPrompt.options.images.length > 0 && codegenPrompt.options.vision) {\\n    prompt.slice(-1)[0].images = codegenPrompt.options.images.map((img) => ({\\n      base64url: img.base64url,\\n      mediaType: img.mediaType,\\n    }));\\n  }\\n\\n  // Initial summary based on first user input\\n  await executeStepGenerateSummary(generateContentFn, prompt, codegenPrompt.options);\\n\\n  // Execute the context optimization step\\n  if (!codegenPrompt.options.disableContextOptimization) {\\n    const optimizationResult = await executeStepContextOptimization(generateContentFn, prompt, codegenPrompt.options);\\n\\n    if (optimizationResult === StepResult.BREAK) {\\n      return { result: [], prompt };\\n    }\\n  }\\n\\n  // Execute the ask question step\\n  if (codegenPrompt.options.askQuestion !== false && (codegenPrompt.options.interactive || codegenPrompt.options.ui)) {\\n    const askQuestionResult = await executeStepAskQuestion(\\n      generateContentFn,\\n      generateImageFn,\\n      prompt,\\n      getFunctionDefs(),\\n      codegenPrompt.options.temperature ?? 0.7,\\n      messages,\\n      codegenPrompt.options,\\n    );\\n\\n    // Summary based on the ask-question conversation history (may be different from the initial summary)\\n    await executeStepGenerateSummary(generateContentFn, prompt, codegenPrompt.options);\\n\\n    if (askQuestionResult === StepResult.BREAK) {\\n      return { result: [], prompt };\\n    }\\n  } else if (codegenPrompt.options.askQuestion === false) {\\n    console.log('Ask question is not enabled.');\\n    // Also there is no need to generate conversation summary\\n  }\\n\\n  const baseRequest: [PromptItem[], FunctionDef[], string, number, boolean, CodegenOptions] = [\\n    prompt,\\n    getFunctionDefs(),\\n    'codegenSummary',\\n    codegenPrompt.options.temperature ?? 0.7,\\n    codegenPrompt.options.cheap ?? false,\\n    codegenPrompt.options,\\n  ];\\n  let baseResult = await generateContentFn(...baseRequest);\\n\\n  let codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');\\n\\n  if (codegenSummaryRequest) {\\n    // Second stage: for each file request the actual code updates\\n    putSystemMessage('Received codegen summary, will collect partial updates', codegenSummaryRequest.args);\\n\\n    baseResult = await validateAndRecoverSingleResult(baseRequest, baseResult, generateContentFn);\\n    codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');\\n\\n    // Sometimes the result happens to be a string\\n    assert(Array.isArray(codegenSummaryRequest?.args?.fileUpdates), 'fileUpdates is not an array');\\n    assert(Array.isArray(codegenSummaryRequest?.args.contextPaths), 'contextPaths is not an array');\\n\\n    if (!codegenPrompt.options.disableContextOptimization) {\\n      console.log('Optimize with context paths.');\\n      // Monkey patch the initial getSourceCode, do not send parts of source code that are consider irrelevant\\n      getSourceCodeRequest.args = {\\n        filePaths: [\\n          ...codegenSummaryRequest.args.fileUpdates.map((file: { filePath: string }) => file.filePath),\\n          ...codegenSummaryRequest.args.contextPaths,\\n          ...(importantContext.files ?? []),\\n        ],\\n      };\\n      getSourceCodeResponse.functionResponses!.find((item) => item.name === 'getSourceCode')!.content =\\n        messages.contextSourceCode(getSourceCodeRequest.args?.filePaths as string[]);\\n    }\\n\\n    // Store the first stage response entirely in conversation history\\n    prompt.push({ type: 'assistant', functionCalls: baseResult });\\n    prompt.push({\\n      type: 'user',\\n      functionResponses: baseResult.map((call) => ({ name: call.name, call_id: call.id })),\\n      cache: true,\\n    });\\n\\n    const result: FunctionCall[] = [];\\n\\n    for (const file of codegenSummaryRequest!.args.fileUpdates) {\\n      putSystemMessage('Collecting partial update for: ' + file.filePath + ' using tool: ' + file.updateToolName, file);\\n      console.log('- Prompt:', file.prompt);\\n      console.log('- Temperature', file.temperature);\\n      console.log('- Cheap', file.cheap);\\n      if (codegenPrompt.options.vision) {\\n        console.log('- Context image assets', file.contextImageAssets);\\n      }\\n\\n      // Check if execution is paused before proceeding\\n      await waitIfPaused();\\n\\n      // this is needed, otherwise we will get an error\\n      if (prompt.slice(-1)[0].type === 'user') {\\n        prompt.slice(-1)[0].text = file.prompt ?? messages.partialPromptTemplate(file.filePath);\\n      } else {\\n        prompt.push({ type: 'user', text: file.prompt ?? messages.partialPromptTemplate(file.filePath) });\\n      }\\n\\n      if (codegenPrompt.options.vision && file.contextImageAssets) {\\n        prompt.slice(-1)[0].images = file.contextImageAssets.map((path: string) => ({\\n          path,\\n          base64url: fs.readFileSync(path, 'base64'),\\n          mediaType: mime.lookup(path) || '',\\n        }));\\n      }\\n\\n      const partialRequest: [PromptItem[], FunctionDef[], string, number, boolean, CodegenOptions] = [\\n        prompt,\\n        getFunctionDefs(),\\n        file.updateToolName,\\n        file.temperature ?? codegenPrompt.options.temperature,\\n        file.cheap === true,\\n        codegenPrompt.options,\\n      ];\\n      let partialResult = await generateContentFn(...partialRequest);\\n\\n      putSystemMessage('Received partial update', partialResult);\\n\\n      // Validate if function call is compliant with the schema\\n      partialResult = await validateAndRecoverSingleResult(partialRequest, partialResult, generateContentFn);\\n\\n      // Handle image generation requests\\n      const generateImageCall = partialResult.find((call) => call.name === 'generateImage');\\n      if (generateImageCall) {\\n        partialResult.push(await executeStepGenerateImage(generateImageFn, generateImageCall));\\n      }\\n\\n      // Verify if patchFile is one of the functions called, and test if patch is valid and can be applied successfully\\n      const patchFileCall = partialResult.find((call) => call.name === 'patchFile');\\n      if (patchFileCall) {\\n        partialResult = await executeStepVerifyPatch(\\n          patchFileCall.args as { filePath: string; patch: string },\\n          generateContentFn,\\n          prompt,\\n          getFunctionDefs(),\\n          file.temperature ?? codegenPrompt.options.temperature,\\n          file.cheap === true,\\n          codegenPrompt.options,\\n        );\\n      }\\n\\n      // add the code gen result to the context, as the subsequent code gen may depend on the result\\n      prompt.push(\\n        { type: 'assistant', functionCalls: partialResult },\\n        {\\n          type: 'user',\\n          text: 'Update applied.',\\n          functionResponses: partialResult.map((call) => ({ name: call.name, call_id: call.id })),\\n        },\\n      );\\n\\n      result.push(...partialResult);\\n    }\\n\\n    return { result, prompt };\\n  } else {\\n    // This is unexpected, if happens probably means no code updates.\\n    putSystemMessage('Did not receive codegen summary, returning result.');\\n    return { result: baseResult, prompt };\\n  }\\n}\\n\\n/**\\n * Function to prepare messages for AI services\\n */\\nfunction prepareMessages(codegen: CodegenPrompt) {\\n  return {\\n    suggestSourceCode: 'I should provide you with application source code.',\\n    requestSourceCode: 'Please provide application source code.',\\n    suggestImageAssets: 'I should also provide you with a summary of application image assets',\\n    requestImageAssets: 'Please provide summary of application image assets.',\\n    prompt:\\n      codegen.prompt +\\n      '\\\\n Start from generating codegen summary, this summary will be used as a context to generate updates, so make sure that it contains useful information.',\\n    sourceCode: JSON.stringify(\\n      getSourceCodeTree(getSourceCode({ taskFile: codegen.options.taskFile }, codegen.options)),\\n    ),\\n    contextSourceCode: (paths: string[], pathsOnly: boolean = false) =>\\n      JSON.stringify(\\n        getSourceCodeTree(\\n          Object.fromEntries(\\n            Object.entries(\\n              getSourceCode(\\n                { filterPaths: paths, taskFile: codegen.options.taskFile, forceAll: true },\\n                codegen.options,\\n              ),\\n            ).filter(([path]) => !pathsOnly || paths.includes(path)),\\n          ),\\n        ),\\n      ),\\n    imageAssets: JSON.stringify(getImageAssets()),\\n    partialPromptTemplate(path: string) {\\n      return `Thank you for providing the summary, now suggest changes for the \\\\`${path}\\\\` file using appropriate tools.`;\\n    },\\n  };\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/cancel-code-generation.ts\":{\"content\":\"import { putSystemMessage } from '../../../../main/common/content-bus.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';\\n\\nexport async function handleCancelCodeGeneration({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult> {\\n  putSystemMessage('Assistant requested to stop code generation. Exiting...');\\n  return {\\n    breakLoop: true,\\n    stepResult: StepResult.BREAK,\\n    items: [\\n      {\\n        assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },\\n        user: { type: 'user', text: 'Code generation cancelled.' },\\n      },\\n    ],\\n  };\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/confirm-code-generation.ts\":{\"content\":\"import { putSystemMessage } from '../../../../main/common/content-bus.js';\\nimport { askUserForConfirmationWithAnswer } from '../../../../main/common/user-actions.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';\\n\\nexport async function handleConfirmCodeGeneration({\\n  askQuestionCall,\\n  options,\\n}: ActionHandlerProps): Promise<ActionResult> {\\n  const userConfirmation = await askUserForConfirmationWithAnswer(\\n    'The assistant is ready to start code generation. Do you want to proceed?',\\n    'Start code generation',\\n    'Continue conversation',\\n    true,\\n  );\\n  if (userConfirmation.options?.aiService) {\\n    options.aiService = userConfirmation.options.aiService;\\n  }\\n  if (userConfirmation.confirmed) {\\n    putSystemMessage('Proceeding with code generation.');\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.CONTINUE,\\n      items: [\\n        {\\n          assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },\\n          user: { type: 'user', text: userConfirmation.answer || 'Confirmed. Proceed with code generation.' },\\n        },\\n      ],\\n    };\\n  } else {\\n    putSystemMessage('Declined. Continuing the conversation.');\\n    return {\\n      breakLoop: false,\\n      stepResult: StepResult.CONTINUE,\\n      items: [\\n        {\\n          assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },\\n          user: {\\n            type: 'user',\\n            text: userConfirmation.answer || 'Declined. Please continue the conversation.',\\n            functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n          },\\n        },\\n      ],\\n    };\\n  }\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/context-optimization.ts\":{\"content\":\"import { refreshFiles } from '../../../../files/find-files.js';\\nimport { putSystemMessage } from '../../../../main/common/content-bus.js';\\nimport { askUserForConfirmation } from '../../../../main/common/user-actions.js';\\nimport { executeStepContextOptimization } from '../../step-context-optimization.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult, UserItem, AssistantItem } from '../step-ask-question-types.js';\\n\\nexport async function handleContextOptimization({\\n  askQuestionCall,\\n  prompt,\\n  options,\\n  generateContentFn,\\n}: ActionHandlerProps): Promise<ActionResult> {\\n  const userConfirmation = await askUserForConfirmation(\\n    'The assistant suggests optimizing the context to reduce token usage, cost, and latency. Do you want to proceed?',\\n    false,\\n  );\\n\\n  const user: UserItem = {\\n    type: 'user',\\n    text: userConfirmation ? 'Context optimization applied.' : 'Context optimization not applied.',\\n    functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n  };\\n\\n  const assistant: AssistantItem = {\\n    type: 'assistant',\\n    text: askQuestionCall.args?.content ?? '',\\n    functionCalls: [askQuestionCall],\\n  };\\n\\n  if (userConfirmation) {\\n    // the request may be caused be an appearance of a new file, so lets refresh\\n    refreshFiles();\\n\\n    // Execute context optimization step\\n    putSystemMessage('Executing context optimization step.');\\n    await executeStepContextOptimization(generateContentFn, [...prompt, assistant, user], options);\\n  }\\n\\n  return {\\n    breakLoop: false,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant,\\n        user,\\n      },\\n    ],\\n  };\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/default-action.ts\":{\"content\":\"import { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';\\n\\nexport async function handleDefaultAction({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult> {\\n  return {\\n    breakLoop: false,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },\\n        user: {\\n          type: 'user',\\n          text: \\\"I don't want to start the code generation yet, let's talk a bit more.\\\",\\n          functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n        },\\n      },\\n    ],\\n  };\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/handle-request-answer-with-image.ts\":{\"content\":\"import { askUserForInput } from '../../../../main/common/user-actions.js';\\nimport { putAssistantMessage, putSystemMessage } from '../../../../main/common/content-bus.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult, RequestAnswerWithImageArgs } from '../step-ask-question-types.js';\\nimport { executeStepGenerateImage } from '../../step-generate-image.js';\\nimport { getFunctionDefs } from '../../../function-calling.js';\\nimport { FunctionCall } from '../../../../ai-service/common.js';\\n\\nexport async function handleRequestAnswerWithImage({\\n  askQuestionCall,\\n  options,\\n  generateContentFn,\\n  prompt,\\n  generateImageFn,\\n}: ActionHandlerProps): Promise<ActionResult> {\\n  if (!options.imagen) {\\n    putSystemMessage('Image generation is not enabled. Please enable it using --imagen parameter.');\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  const [requestAnswerWithImageCall] = (await generateContentFn(\\n    [\\n      ...prompt,\\n      {\\n        type: 'assistant',\\n        text: askQuestionCall.args?.content ?? '',\\n        functionCalls: [askQuestionCall],\\n      },\\n      {\\n        type: 'user',\\n        text: 'Yes, you can request the files contents.',\\n        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n      },\\n    ],\\n    getFunctionDefs(),\\n    'requestAnswerWithImage',\\n    0.7,\\n    true,\\n    options,\\n  )) as [FunctionCall<RequestAnswerWithImageArgs> | undefined];\\n\\n  if (!requestAnswerWithImageCall) {\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  // Get the image generation request from the args\\n  const imageGenerationRequest = requestAnswerWithImageCall.args;\\n  if (!imageGenerationRequest?.prompt) {\\n    putSystemMessage('Image generation request with prompt is required for requestAnswerWithImage action.');\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  // Create a temporary file path for the generated image\\n  const tempImagePath = `/tmp/question-image-${Date.now()}.png`;\\n\\n  // Generate the image using generateImage function call\\n  const generateImageCall = {\\n    name: 'generateImage',\\n    args: {\\n      prompt: imageGenerationRequest.prompt,\\n      filePath: tempImagePath,\\n      width: 1024,\\n      height: 1024,\\n      explanation: 'Generating image to support the question',\\n      ...(imageGenerationRequest.contextImage && { contextImagePath: imageGenerationRequest.contextImage }),\\n    },\\n  };\\n\\n  // Execute image generation\\n  const downloadFileCall = await executeStepGenerateImage(generateImageFn, generateImageCall);\\n\\n  // If image generation failed, the downloadFileCall will be an explanation\\n  if (!downloadFileCall.args?.downloadUrl) {\\n    putSystemMessage('Failed to generate image for the question.');\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  const response = await fetch(downloadFileCall.args?.downloadUrl);\\n  const arrayBuffer = await response.arrayBuffer();\\n  const buffer = Buffer.from(arrayBuffer);\\n\\n  putAssistantMessage(\\n    'Here is the image to support the question:',\\n    generateImageCall.args,\\n    [],\\n    [\\n      {\\n        base64url: buffer.toString('base64'),\\n        mediaType: 'image/png',\\n      },\\n    ],\\n  );\\n\\n  // Get user's response with the image displayed\\n  const inputResponse = await askUserForInput('Your answer', askQuestionCall.args?.content ?? '');\\n  if (inputResponse.options?.aiService) {\\n    options.aiService = inputResponse.options.aiService;\\n  }\\n\\n  // Return the conversation items with the generated image\\n  return {\\n    breakLoop: false,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant: {\\n          type: 'assistant',\\n          text: askQuestionCall.args?.content ?? '',\\n          functionCalls: [askQuestionCall, generateImageCall, downloadFileCall],\\n        },\\n        user: {\\n          type: 'user',\\n          text: inputResponse.answer,\\n          functionResponses: [\\n            {\\n              name: 'askQuestion',\\n              call_id: askQuestionCall.id,\\n              content: undefined,\\n            },\\n          ],\\n        },\\n      },\\n    ],\\n  };\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/handle-request-answer.ts\":{\"content\":\"import { askUserForInput } from '../../../../main/common/user-actions.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';\\n\\nexport async function handleRequestAnswer({ askQuestionCall, options }: ActionHandlerProps): Promise<ActionResult> {\\n  const response = await askUserForInput('Your answer', askQuestionCall.args?.content ?? '');\\n  if (response.options?.aiService) {\\n    options.aiService = response.options.aiService;\\n  }\\n  return {\\n    breakLoop: false,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [askQuestionCall] },\\n        user: {\\n          type: 'user',\\n          text: response.answer,\\n          functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n        },\\n      },\\n    ],\\n  };\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/handler-utils.ts\":{\"content\":\"\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/remove-files-from-context.ts\":{\"content\":\"import { FunctionCall, PromptItem } from '../../../../ai-service/common.js';\\nimport { getSourceCodeTree, parseSourceCodeTree } from '../../../../files/source-code-tree.js';\\nimport { putSystemMessage } from '../../../../main/common/content-bus.js';\\nimport { getFunctionDefs } from '../../../function-calling.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { getSourceCodeResponse } from '../../steps-utils.js';\\nimport { ActionHandlerProps, ActionResult, RemoveFilesFromContextArgs } from '../step-ask-question-types.js';\\n\\nexport async function handleRemoveFilesFromContext({\\n  askQuestionCall,\\n  generateContentFn,\\n  prompt,\\n  options,\\n}: ActionHandlerProps): Promise<ActionResult> {\\n  const [removeFilesFromContextCall] = (await generateContentFn(\\n    [\\n      ...prompt,\\n      {\\n        type: 'assistant',\\n        text: askQuestionCall.args?.content ?? '',\\n        functionCalls: [askQuestionCall],\\n      },\\n      {\\n        type: 'user',\\n        text: 'Yes, you can request the removal of files from context.',\\n        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n      },\\n    ],\\n    getFunctionDefs(),\\n    'removeFilesFromContext',\\n    0.7,\\n    true,\\n    options,\\n  )) as [FunctionCall<RemoveFilesFromContextArgs> | undefined];\\n\\n  if (!removeFilesFromContextCall) {\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  const filesToRemove = removeFilesFromContextCall.args?.filePaths ?? [];\\n  let userText = '';\\n\\n  if (filesToRemove.length > 0) {\\n    removeFileContentsFromPrompt(prompt, filesToRemove);\\n    putSystemMessage('Context reduction applied', filesToRemove);\\n    userText = 'Context reduction applied';\\n  } else {\\n    userText = 'No specific files were provided for context reduction. The context remains unchanged.';\\n  }\\n\\n  return {\\n    breakLoop: false,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant: {\\n          type: 'assistant',\\n          text: askQuestionCall.args?.content ?? '',\\n          functionCalls: [askQuestionCall, removeFilesFromContextCall],\\n        },\\n        user: {\\n          type: 'user',\\n          text: userText,\\n          functionResponses: [\\n            { name: 'askQuestion', call_id: askQuestionCall.id, content: undefined },\\n            { name: 'removeFilesFromContext', call_id: removeFilesFromContextCall.id, content: undefined },\\n          ],\\n        },\\n      },\\n    ],\\n  };\\n}\\n\\nfunction removeFileContentsFromPrompt(prompt: PromptItem[], filesToRemove: string[]) {\\n  const response = getSourceCodeResponse(prompt);\\n  if (!response || !response.content) {\\n    throw new Error('Could not find source code response');\\n  }\\n  const contentObj = parseSourceCodeTree(JSON.parse(response.content));\\n  filesToRemove.forEach((file) => {\\n    if (contentObj[file] && 'content' in contentObj[file]) {\\n      contentObj[file].content = null;\\n    }\\n  });\\n  response.content = JSON.stringify(getSourceCodeTree(contentObj));\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/request-files-content.ts\":{\"content\":\"import { FunctionCall } from '../../../../ai-service/common.js';\\nimport { getSourceFiles, refreshFiles } from '../../../../files/find-files.js';\\nimport { getSourceCode } from '../../../../files/read-files.js';\\nimport { getFunctionDefs } from '../../../function-calling.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport {\\n  ActionHandlerProps,\\n  ActionResult,\\n  AssistantItem,\\n  RequestFilesContentArgs,\\n  UserItem,\\n} from '../step-ask-question-types.js';\\n\\nexport async function handleRequestFilesContent({\\n  askQuestionCall,\\n  options,\\n  prompt,\\n  generateContentFn,\\n}: ActionHandlerProps): Promise<ActionResult> {\\n  const [requestFilesContentCall] = (await generateContentFn(\\n    [\\n      ...prompt,\\n      {\\n        type: 'assistant',\\n        text: askQuestionCall.args?.content ?? '',\\n        functionCalls: [askQuestionCall],\\n      },\\n      {\\n        type: 'user',\\n        text: 'Yes, you can request the files contents.',\\n        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n      },\\n    ],\\n    getFunctionDefs(),\\n    'requestFilesContent',\\n    0.7,\\n    true,\\n    options,\\n  )) as [FunctionCall<RequestFilesContentArgs> | undefined];\\n\\n  if (!requestFilesContentCall) {\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  const requestedFiles = requestFilesContentCall.args?.filePaths ?? [];\\n\\n  // the request may be caused be an appearance of a new file, so lets refresh\\n  refreshFiles();\\n\\n  const { legitimateFiles, illegitimateFiles } = categorizeLegitimateFiles(requestedFiles);\\n\\n  const sourceCallId = (askQuestionCall.id ?? askQuestionCall.name) + '_source';\\n  const assistant: AssistantItem = {\\n    type: 'assistant',\\n    text: askQuestionCall.args?.content ?? '',\\n    functionCalls: [\\n      askQuestionCall,\\n      requestFilesContentCall,\\n      { name: 'getSourceCode', id: sourceCallId, args: { filePaths: legitimateFiles } },\\n    ],\\n  };\\n\\n  const sourceCode = getSourceCode({ filterPaths: legitimateFiles, forceAll: true }, options);\\n  const user: UserItem = {\\n    type: 'user',\\n    text:\\n      illegitimateFiles.length > 0\\n        ? `Some files are not legitimate and their content cannot be provided:\\\\n\\\\n${illegitimateFiles.join('\\\\n')}`\\n        : 'All requested file contents have been provided.',\\n    functionResponses: [\\n      { name: 'askQuestion', call_id: askQuestionCall.id, content: undefined },\\n      {\\n        name: 'requestFilesContent',\\n        call_id: requestFilesContentCall.id,\\n        content: JSON.stringify({ filePaths: requestedFiles }),\\n      },\\n      {\\n        name: 'getSourceCode',\\n        call_id: sourceCallId,\\n        content: JSON.stringify(sourceCode),\\n      },\\n    ],\\n    cache: true,\\n  };\\n\\n  return { breakLoop: false, stepResult: StepResult.CONTINUE, items: [{ assistant, user }] };\\n}\\n\\nfunction categorizeLegitimateFiles(requestedFiles: string[]): {\\n  legitimateFiles: string[];\\n  illegitimateFiles: string[];\\n} {\\n  const legitimateFiles: string[] = [];\\n  const illegitimateFiles: string[] = [];\\n\\n  requestedFiles.forEach((filePath) => {\\n    if (isFilePathLegitimate(filePath)) {\\n      legitimateFiles.push(filePath);\\n    } else {\\n      illegitimateFiles.push(filePath);\\n    }\\n  });\\n\\n  return { legitimateFiles, illegitimateFiles };\\n}\\n\\nfunction isFilePathLegitimate(filePath: string): boolean {\\n  return getSourceFiles().includes(filePath);\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/request-permissions.ts\":{\"content\":\"import { FunctionCall } from '../../../../ai-service/common.js';\\nimport { CodegenOptions } from '../../../../main/codegen-types.js';\\nimport { askUserForConfirmation } from '../../../../main/common/user-actions.js';\\nimport { getFunctionDefs } from '../../../function-calling.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult, RequestPermissionsArgs, UserItem } from '../step-ask-question-types.js';\\n\\nexport async function handleRequestPermissions({\\n  askQuestionCall,\\n  options,\\n  prompt,\\n  generateContentFn,\\n}: ActionHandlerProps): Promise<ActionResult> {\\n  const [requestPermissionsCall] = (await generateContentFn(\\n    [\\n      ...prompt,\\n      {\\n        type: 'assistant',\\n        text: askQuestionCall.args?.content ?? '',\\n        functionCalls: [askQuestionCall],\\n      },\\n      {\\n        type: 'user',\\n        text: 'Yes, you can request the permissions.',\\n        functionResponses: [{ name: 'askQuestion', call_id: askQuestionCall.id, content: undefined }],\\n      },\\n    ],\\n    getFunctionDefs(),\\n    'requestPermissions',\\n    0.7,\\n    true,\\n    options,\\n  )) as [FunctionCall<RequestPermissionsArgs> | undefined];\\n\\n  if (!requestPermissionsCall) {\\n    return {\\n      breakLoop: true,\\n      stepResult: StepResult.BREAK,\\n      items: [],\\n    };\\n  }\\n\\n  const userConfirmation = await askUserForConfirmation(\\n    'The assistant is requesting additional permissions. Do you want to grant them?',\\n    false,\\n  );\\n\\n  const user: UserItem = {\\n    type: 'user',\\n    text: userConfirmation ? 'Permissions granted.' : 'Permission request denied.',\\n    functionResponses: [\\n      { name: 'askQuestion', call_id: askQuestionCall.id, content: undefined },\\n      {\\n        name: 'requestPermissions',\\n        call_id: requestPermissionsCall.id,\\n        content: JSON.stringify({ confirmed: userConfirmation.confirmed }),\\n      },\\n    ],\\n  };\\n\\n  if (userConfirmation.confirmed) {\\n    updatePermissions(requestPermissionsCall, options);\\n  }\\n\\n  return {\\n    breakLoop: false,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant: {\\n          type: 'assistant',\\n          text: askQuestionCall.args?.content ?? '',\\n          functionCalls: [requestPermissionsCall, askQuestionCall],\\n        },\\n        user,\\n      },\\n    ],\\n  };\\n}\\n\\nfunction updatePermissions(requestPermissionsCall: FunctionCall<RequestPermissionsArgs>, options: CodegenOptions) {\\n  const permissions = requestPermissionsCall.args;\\n  if (permissions) {\\n    if ('enableImagen' in permissions && permissions.enableImagen) {\\n      options.imagen = options.aiService === 'chat-gpt' ? 'dall-e' : 'vertex-ai';\\n    }\\n    if ('enableVision' in permissions && permissions.enableVision) options.vision = true;\\n    if ('allowDirectoryCreate' in permissions && permissions.allowDirectoryCreate) options.allowDirectoryCreate = true;\\n    if ('allowFileCreate' in permissions && permissions.allowFileCreate) options.allowFileCreate = true;\\n    if ('allowFileDelete' in permissions && permissions.allowFileDelete) options.allowFileDelete = true;\\n    if ('allowFileMove' in permissions && permissions.allowFileMove) options.allowFileMove = true;\\n  }\\n}\\n\"},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers/start-code-generation.ts\":{\"content\":\"import { putSystemMessage } from '../../../../main/common/content-bus.js';\\nimport { StepResult } from '../../steps-types.js';\\nimport { ActionHandlerProps, ActionResult } from '../step-ask-question-types.js';\\n\\nexport async function handleStartCodeGeneration({ askQuestionCall }: ActionHandlerProps): Promise<ActionResult> {\\n  putSystemMessage('Proceeding with code generation.');\\n  return {\\n    breakLoop: true,\\n    stepResult: StepResult.CONTINUE,\\n    items: [\\n      {\\n        assistant: { type: 'assistant', text: askQuestionCall.args?.content ?? '', functionCalls: [] },\\n        user: { type: 'user', text: 'Proceeding with code generation.' },\\n      },\\n    ],\\n  };\\n}\\n\"}}",
+    //       },
+    //     ],
+    //     cache: true,
+    //   },
+    //   {
+    //     type: 'assistant',
+    //     text: 'The assistant is ready to start code generation. Do you want to proceed?',
+    //   },
+    //   {
+    //     type: 'user',
+    //     text: 'Why do you want to start code generation?',
+    //   },
+    //   {
+    //     type: 'assistant',
+    //     text: 'The assistant is ready to start code generation. Do you want to proceed?',
+    //   },
+    //   {
+    //     type: 'user',
+    //     text: 'no, please analyze those files first, and say YOLO.',
+    //   },
+];
+//# sourceMappingURL=prompt2.js.map

package/dist/eval/prompt2.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt2.js","sourceRoot":"","sources":["../../src/eval/prompt2.js"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAEzD,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B;QACE,IAAI,EAAE,cAAc;QACpB,YAAY,EAAE,eAAe,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;KACnG;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,mBAAmB;KAC1B;IACD;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,yKAAyK;QAC/K,aAAa,EAAE;YACb;gBACE,IAAI,EAAE,eAAe;aACtB;YACD;gBACE,IAAI,EAAE,gBAAgB;aACvB;SACF;KACF;IACD;QACE,IAAI,EAAE,MAAM;QACZ,iBAAiB,EAAE;YACjB;gBACE,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;oBACtB,yEAAyE,EAAE;wBACzE,2BAA2B,EAAE,EAAE,OAAO,EAAE,8CAA8C,EAAE;wBACxF,4BAA4B,EAAE,EAAE,OAAO,EAAE,yCAAyC,EAAE;wBACpF,yBAAyB,EAAE,EAAE,OAAO,EAAE,8BAA8B,EAAE;wBACtE,mBAAmB,EAAE,EAAE,OAAO,EAAE,wBAAwB,EAAE;wBAC1D,qCAAqC,EAAE;4BACrC,OAAO,EAAE,0DAA0D;yBACpE;wBACD,0BAA0B,EAAE,EAAE,OAAO,EAAE,8BAA8B,EAAE;wBACvE,kBAAkB,EAAE,EAAE,OAAO,EAAE,kDAAkD,EAAE;wBACnF,8BAA8B,EAAE,EAAE,OAAO,EAAE,qCAAqC,EAAE;wBAClF,0BAA0B,EAAE,EAAE,OAAO,EAAE,6CAA6C,EAAE;wBACtF,wBAAwB,EAAE,EAAE,OAAO,EAAE,0CAA0C,EAAE;wBACjF,0BAA0B,EAAE,EAAE,OAAO,EAAE,uCAAuC,EAAE;qBACjF;iBACF,CAAC;aACH;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,OAAO,EACL,qUAAqU;aACxU;SACF;QACD,KAAK,EAAE,IAAI;QACX,IAAI,EAAE,6DAA6D;KACpE;IACD;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,uDAAuD;KAC9D;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,OAAO;KACd;IACD;QACE,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,oCAAoC;KAC3C;IACD;QACE,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,qKAAqK;KAC5K;IACD,MAAM;IACN,yBAAyB;IACzB,ubAAub;IACvb,uBAAuB;IACvB,UAAU;IACV,iCAAiC;IACjC,uDAAuD;IACvD,kBAAkB;IAClB,yBAAyB;IACzB,mHAAmH;IACnH,oHAAoH;IACpH,iHAAiH;IACjH,2GAA2G;IAC3G,6HAA6H;IAC7H,kHAAkH;IAClH,0GAA0G;IAC1G,sHAAsH;IACtH,kHAAkH;IAClH,gHAAgH;IAChH,kHAAkH;IAClH,eAAe;IACf,aAAa;IACb,WAAW;IACX,SAAS;IACT,OAAO;IACP,MAAM;IACN,oBAAoB;IACpB,+DAA+D;IAC/D,2BAA2B;IAC3B,UAAU;IACV,iCAAiC;IACjC,4DAA4D;IAC5D,mBAAmB;IACnB,uygDAAuygD;IACvygD,WAAW;IACX,SAAS;IACT,mBAAmB;IACnB,OAAO;IACP,MAAM;IACN,yBAAyB;IACzB,wFAAwF;IACxF,OAAO;IACP,MAAM;IACN,oBAAoB;IACpB,yDAAyD;IACzD,OAAO;IACP,MAAM;IACN,yBAAyB;IACzB,wFAAwF;IACxF,OAAO;IACP,MAAM;IACN,oBAAoB;IACpB,mEAAmE;IACnE,OAAO;CACR,CAAC"}

package/dist/files/read-files.d.ts

@@ -1,20 +1,43 @@
 import { CodegenOptions } from '../main/codegen-types.js';
-export type SourceCodeMap = Record<string, {
+/**
+ * Represents a dependency in the source code
+ */
+export interface DependencyInfo {
+    /** Path of the dependeny */
+    path: string;
+    /** Type of dependency */
+    type: 'local' | 'external';
+}
+/**
+ * Represents file content with optional dependencies
+ */
+export interface FileContent {
+    /** The actual content of the file */
     content: string | null;
-} | {
-    summary: string;
-    tokenCount: number;
-}>;
+    /** Optional list of dependencies */
+    dependencies?: DependencyInfo[];
+}
+/**
+ * Represents a file summary with optional dependencies
+ */
+export interface FileSummary {
+    /** Summary of the file content */
+    summary?: string;
+    /** Optional list of dependencies */
+    dependencies?: DependencyInfo[];
+}
+export type SourceCodeMap = Record<string, FileContent | FileSummary>;
 type ImageAssetsMap = Record<string, {
     mimeType: string | false;
     width: number;
     height: number;
 }>;
 /** Print source code of all source files */
-export declare function getSourceCode({ filterPaths, taskFile, forceAll, }: {
+export declare function getSourceCode({ filterPaths, taskFile, forceAll, ignoreImportantFiles, }: {
     filterPaths?: string[];
     taskFile?: string | undefined;
     forceAll?: boolean;
+    ignoreImportantFiles?: boolean;
 }, options: CodegenOptions): SourceCodeMap;
 /** Get image asset files summary */
 export declare function getImageAssets(): ImageAssetsMap;

package/dist/files/read-files.js

@@ -10,39 +10,60 @@ import { getSummary } from '../prompt/steps/step-summarization.js';
 /**
  * Read contents of source files and create a map with file path as key and file content as value
  */
-function readSourceFiles({ contentMask, ignorePatterns }, filterPaths, forceAll = false) {
+function readSourceFiles({ contentMask, ignorePatterns }, filterPaths, forceAll = false, ignoreImportantFiles = false) {
     const sourceCode = {};
-    const importantFiles = new Set(importantContext.files || []);
+    const importantFiles = ignoreImportantFiles ? new Set() : new Set(importantContext.files || []);
     for (const file of getSourceFiles()) {
+        if (!fs.existsSync(file)) {
+            continue;
+        }
         if (!filterPaths || filterPaths.includes(file) || importantFiles.has(file)) {
+            const summary = getSummary(file);
             // Always include important files
             if (importantFiles.has(file)) {
-                sourceCode[file] = { content: fs.readFileSync(file, 'utf-8') };
+                const content = fs.readFileSync(file, 'utf-8');
+                sourceCode[file] = {
+                    content,
+                    dependencies: summary?.dependencies,
+                };
                 continue;
             }
-            const summary = getSummary(file);
             // Apply content mask filter if it's set
             if (!filterPaths && contentMask && !forceAll) {
                 const relativePath = path.relative(rcConfig.rootDir, file);
                 if (!relativePath.startsWith(contentMask)) {
-                    sourceCode[file] = summary ? { ...summary } : { content: null };
+                    sourceCode[file] = summary
+                        ? {
+                            summary: summary.summary,
+                            ...(summary.dependencies?.length ? { dependencies: summary.dependencies } : {}),
+                        }
+                        : { content: null };
                     continue;
                 }
             }
-            if (!forceAll && ignorePatterns?.some((pattern) => globRegex(pattern).test(file))) {
-                sourceCode[file] = summary ? { ...summary } : { content: null };
+            if (!forceAll) {
+                sourceCode[file] =
+                    !ignorePatterns?.some((pattern) => globRegex(pattern).test(file)) && summary
+                        ? {
+                            summary: summary.summary,
+                            ...(summary.dependencies?.length ? { dependencies: summary.dependencies } : {}),
+                        }
+                        : { content: null };
             }
             else {
                 const content = fs.readFileSync(file, 'utf-8');
-                sourceCode[file] = { content, ...summary };
+                sourceCode[file] = {
+                    content,
+                    dependencies: summary?.dependencies,
+                };
             }
         }
     }
     return sourceCode;
 }
 /** Print source code of all source files */
-export function getSourceCode({ filterPaths, taskFile, forceAll, }, options) {
-    const sourceCode = readSourceFiles(options, filterPaths, forceAll);
+export function getSourceCode({ filterPaths, taskFile, forceAll, ignoreImportantFiles, }, options) {
+    const sourceCode = readSourceFiles(options, filterPaths, forceAll, ignoreImportantFiles);
     if (taskFile && !sourceCode[taskFile]) {
         sourceCode[taskFile] = {
             content: fs.readFileSync(taskFile, 'utf-8'),

package/dist/files/read-files.js.map

@@ -1 +1 @@
-{"version":3,"file":"read-files.js","sourceRoot":"","sources":["../../src/files/read-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,MAAM,MAAM,YAAY,CAAC;AAChC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,SAAS,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AAanE;;GAEG;AACH,SAAS,eAAe,CACtB,EAAE,WAAW,EAAE,cAAc,EAAkB,EAC/C,WAAsB,EACtB,QAAQ,GAAG,KAAK;IAEhB,MAAM,UAAU,GAAkB,EAAE,CAAC;IACrC,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAE7D,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3E,iCAAiC;YACjC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7B,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC/D,SAAS;YACX,CAAC;YAED,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAEjC,wCAAwC;YACxC,IAAI,CAAC,WAAW,IAAI,WAAW,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBAC3D,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC1C,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;oBAChE,SAAS;gBACX,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,IAAI,cAAc,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;gBAClF,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAClE,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC/C,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,aAAa,CAC3B,EACE,WAAW,EACX,QAAQ,EACR,QAAQ,GAKT,EACD,OAAuB;IAEvB,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;IAEnE,IAAI,QAAQ,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,UAAU,CAAC,QAAQ,CAAC,GAAG;YACrB,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;IAClD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,oCAAoC;AACpC,MAAM,UAAU,cAAc;IAC5B,MAAM,WAAW,GAAmB,EAAE,CAAC;IACvC,KAAK,MAAM,IAAI,IAAI,kBAAkB,EAAE,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,WAAW,CAAC,IAAI,CAAC,GAAG;YAClB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YAC3B,KAAK,EAAE,UAAU,CAAC,KAAM;YACxB,MAAM,EAAE,UAAU,CAAC,MAAO;SAC3B,CAAC;IACJ,CAAC;IACD,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC;IACnD,OAAO,WAAW,CAAC;AACrB,CAAC"}
+{"version":3,"file":"read-files.js","sourceRoot":"","sources":["../../src/files/read-files.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,MAAM,MAAM,YAAY,CAAC;AAChC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,SAAS,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AA2CnE;;GAEG;AACH,SAAS,eAAe,CACtB,EAAE,WAAW,EAAE,cAAc,EAAkB,EAC/C,WAAsB,EACtB,QAAQ,GAAG,KAAK,EAChB,oBAAoB,GAAG,KAAK;IAE5B,MAAM,UAAU,GAAkB,EAAE,CAAC;IACrC,MAAM,cAAc,GAAG,oBAAoB,CAAC,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAEhG,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,SAAS;QACX,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3E,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;YAEjC,iCAAiC;YACjC,IAAI,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC/C,UAAU,CAAC,IAAI,CAAC,GAAG;oBACjB,OAAO;oBACP,YAAY,EAAE,OAAO,EAAE,YAAY;iBACpC,CAAC;gBACF,SAAS;YACX,CAAC;YAED,wCAAwC;YACxC,IAAI,CAAC,WAAW,IAAI,WAAW,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBAC3D,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC1C,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO;wBACxB,CAAC,CAAC;4BACE,OAAO,EAAE,OAAO,CAAC,OAAO;4BACxB,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;yBAChF;wBACH,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;oBAEtB,SAAS;gBACX,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,UAAU,CAAC,IAAI,CAAC;oBACd,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,OAAO;wBAC1E,CAAC,CAAC;4BACE,OAAO,EAAE,OAAO,CAAC,OAAO;4BACxB,GAAG,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;yBAChF;wBACH,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC1B,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAC/C,UAAU,CAAC,IAAI,CAAC,GAAG;oBACjB,OAAO;oBACP,YAAY,EAAE,OAAO,EAAE,YAAY;iBACpC,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,aAAa,CAC3B,EACE,WAAW,EACX,QAAQ,EACR,QAAQ,EACR,oBAAoB,GAMrB,EACD,OAAuB;IAEvB,MAAM,UAAU,GAAG,eAAe,CAAC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,oBAAoB,CAAC,CAAC;IAEzF,IAAI,QAAQ,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,UAAU,CAAC,QAAQ,CAAC,GAAG;YACrB,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;IAClD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,oCAAoC;AACpC,MAAM,UAAU,cAAc;IAC5B,MAAM,WAAW,GAAmB,EAAE,CAAC;IACvC,KAAK,MAAM,IAAI,IAAI,kBAAkB,EAAE,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,WAAW,CAAC,IAAI,CAAC,GAAG;YAClB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YAC3B,KAAK,EAAE,UAAU,CAAC,KAAM;YACxB,MAAM,EAAE,UAAU,CAAC,MAAO;SAC3B,CAAC;IACJ,CAAC;IACD,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC;IACnD,OAAO,WAAW,CAAC;AACrB,CAAC"}

package/dist/files/read-files.test.js

@@ -7,9 +7,11 @@ import { getSourceFiles, getImageAssetFiles } from './find-files.js';
 import { verifySourceCodeLimit } from '../prompt/limits.js';
 import * as cliParams from '../cli/cli-params.js';
 import { rcConfig } from '../main/config.js';
+import { getSummary } from '../prompt/steps/step-summarization.js';
 vi.mock('fs');
 vi.mock('mime-types');
 vi.mock('image-size');
+vi.mock('../prompt/steps/step-summarization.js');
 vi.mock('./find-files.js', () => ({
     getImageAssetFiles: vi.fn(),
     getSourceFiles: vi.fn(),
@@ -29,44 +31,48 @@ describe('read-files', () => {
     beforeEach(() => {
         vi.resetAllMocks();
         vi.mocked(cliParams).contentMask = undefined;
+        vi.mocked(getSummary).mockImplementation((file) => ({ summary: `Summary of ${file}`, dependencies: [] }));
     });
     afterEach(() => {
         vi.clearAllMocks();
     });
     describe('getSourceCode', () => {
-        it('should return source code for all files', () => {
+        it('should return source code summaries for all files', () => {
             const mockFiles = ['/home/project/file1.js', '/home/project/file2.js'];
             vi.mocked(getSourceFiles).mockReturnValue(mockFiles);
+            vi.mocked(fs).existsSync.mockReturnValue(true);
             vi.mocked(fs).readFileSync.mockImplementation((file) => `Content of ${file}`);
             rcConfig.rootDir = '/home/project';
-            const result = getSourceCode({ taskFile: undefined }, { aiService: 'vertex-ai' });
+            const result = getSourceCode({ taskFile: undefined }, { aiService: 'vertex-ai', askQuestion: false });
             expect(result).toEqual({
-                '/home/project/file1.js': { content: 'Content of /home/project/file1.js' },
-                '/home/project/file2.js': { content: 'Content of /home/project/file2.js' },
+                '/home/project/file1.js': { summary: 'Summary of /home/project/file1.js' },
+                '/home/project/file2.js': { summary: 'Summary of /home/project/file2.js' },
             });
             expect(verifySourceCodeLimit).toHaveBeenCalled();
         });
         it('should apply content mask when specified', () => {
             const mockFiles = ['/home/project/file1.js', '/home/project/subfolder/file2.js'];
             vi.mocked(getSourceFiles).mockReturnValue(mockFiles);
+            vi.mocked(fs).existsSync.mockReturnValue(true);
             vi.mocked(fs).readFileSync.mockImplementation((file) => `Content of ${file}`);
             rcConfig.rootDir = '/home/project';
-            const result = getSourceCode({ taskFile: undefined }, { aiService: 'vertex-ai', contentMask: 'subfolder' });
+            const result = getSourceCode({ taskFile: undefined }, { aiService: 'vertex-ai', contentMask: 'subfolder', askQuestion: false });
             expect(result).toEqual({
-                '/home/project/file1.js': { content: null },
+                '/home/project/file1.js': { summary: 'Summary of /home/project/file1.js' },
                 '/home/project/subfolder/file2.js': {
-                    content: 'Content of /home/project/subfolder/file2.js',
+                    summary: 'Summary of /home/project/subfolder/file2.js',
                 },
             });
         });
         it('should include task file when specified', () => {
             const mockFiles = ['/home/project/file1.js'];
             vi.mocked(getSourceFiles).mockReturnValue(mockFiles);
+            vi.mocked(fs).existsSync.mockReturnValue(true);
             vi.mocked(fs).readFileSync.mockImplementation((file) => `Content of ${file}`);
             rcConfig.rootDir = '/home/project';
-            const result = getSourceCode({ taskFile: '/home/project/task.md' }, { aiService: 'vertex-ai' });
+            const result = getSourceCode({ taskFile: '/home/project/task.md' }, { aiService: 'vertex-ai', askQuestion: false });
             expect(result).toEqual({
-                '/home/project/file1.js': { content: 'Content of /home/project/file1.js' },
+                '/home/project/file1.js': { summary: 'Summary of /home/project/file1.js' },
                 '/home/project/task.md': { content: 'Content of /home/project/task.md' },
             });
         });

package/dist/files/read-files.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"read-files.test.js","sourceRoot":"","sources":["../../src/files/read-files.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,MAAM,MAAM,YAAY,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC3B,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,QAAQ,EAAE,IAAI;IACd,WAAW,EAAE,IAAI;IACjB,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;IACtC,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,SAAS,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,CAAC,wBAAwB,EAAE,wBAAwB,CAAC,CAAC;YACvE,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;YAElF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;aAC3E,CAAC,CAAC;YACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,SAAS,GAAG,CAAC,wBAAwB,EAAE,kCAAkC,CAAC,CAAC;YACjF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,CAAC,CAAC;YAE5G,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;gBAC3C,kCAAkC,EAAE;oBAClC,OAAO,EAAE,6CAA6C;iBACvD;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,CAAC,wBAAwB,CAAC,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,uBAAuB,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;YAEhG,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,uBAAuB,EAAE,EAAE,OAAO,EAAE,kCAAkC,EAAE;aACzE,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,cAAc,GAAG,CAAC,0BAA0B,EAAE,0BAA0B,CAAC,CAAC;YAChF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAC9D,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YAC1G,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;YAEhC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,0BAA0B,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;gBAC9E,0BAA0B,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;aAChF,CAAC,CAAC;YACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"read-files.test.js","sourceRoot":"","sources":["../../src/files/read-files.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAC9B,OAAO,MAAM,MAAM,YAAY,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrE,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,KAAK,SAAS,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AAEnE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AACtB,EAAE,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;AACjD,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;IAC3B,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAC/B,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,QAAQ,EAAE,IAAI;IACd,WAAW,EAAE,IAAI;IACjB,mBAAmB,EAAE,IAAI;IACzB,cAAc,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,QAAQ,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;IACtC,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7C,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,cAAc,IAAI,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC5G,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,SAAS,GAAG,CAAC,wBAAwB,EAAE,wBAAwB,CAAC,CAAC;YACvE,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;YAEtG,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;aAC3E,CAAC,CAAC;YACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,SAAS,GAAG,CAAC,wBAAwB,EAAE,kCAAkC,CAAC,CAAC;YACjF,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAC1B,EAAE,QAAQ,EAAE,SAAS,EAAE,EACvB,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CACzE,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,kCAAkC,EAAE;oBAClC,OAAO,EAAE,6CAA6C;iBACvD;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,CAAC,wBAAwB,CAAC,CAAC;YAC7C,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YACrD,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;YAC9E,QAAQ,CAAC,OAAO,GAAG,eAAe,CAAC;YAEnC,MAAM,MAAM,GAAG,aAAa,CAC1B,EAAE,QAAQ,EAAE,uBAAuB,EAAE,EACrC,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,wBAAwB,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE;gBAC1E,uBAAuB,EAAE,EAAE,OAAO,EAAE,kCAAkC,EAAE;aACzE,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,cAAc,GAAG,CAAC,0BAA0B,EAAE,0BAA0B,CAAC,CAAC;YAChF,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;YAC9D,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YAC1G,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;YAEhC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,0BAA0B,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;gBAC9E,0BAA0B,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;aAChF,CAAC,CAAC;YACH,MAAM,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/files/source-code-tree.d.ts

@@ -0,0 +1,17 @@
+import { SourceCodeMap, FileContent, FileSummary } from '../files/read-files.js';
+/**
+ * Directory structure representing source code files with their content or summaries and dependencies
+ */
+export type SourceCodeTree = {
+    [directoryPath: string]: {
+        [filePath: string]: FileContent | FileSummary;
+    };
+};
+/**
+ * Converts flat SourceCodeMap into a directory tree structure
+ */
+export declare function getSourceCodeTree(sourceCode: SourceCodeMap): SourceCodeTree;
+/**
+ * Converts directory tree structure back into flat SourceCodeMap
+ */
+export declare function parseSourceCodeTree(sourceCodeTree: SourceCodeTree): SourceCodeMap;

package/dist/files/source-code-tree.js

@@ -0,0 +1,70 @@
+import path from 'path';
+import { rcConfig } from '../main/config.js';
+/**
+ * Type guard to check if an object is a FileContent with dependencies
+ */
+function hasContent(obj) {
+    return 'content' in obj;
+}
+/**
+ * Type guard to check if an object is a FileSummary with dependencies
+ */
+function hasSummary(obj) {
+    return 'summary' in obj;
+}
+/**
+ * Converts flat SourceCodeMap into a directory tree structure
+ */
+export function getSourceCodeTree(sourceCode) {
+    const result = {};
+    for (const [filePath, fileData] of Object.entries(sourceCode)) {
+        if (!filePath.startsWith(rcConfig.rootDir)) {
+            continue;
+        }
+        const dirPath = path.dirname(filePath);
+        const fileName = path.basename(filePath);
+        if (!result[dirPath]) {
+            result[dirPath] = {};
+        }
+        // Handle both content and summary cases while preserving dependencies
+        if (hasContent(fileData)) {
+            result[dirPath][fileName] = {
+                content: fileData.content,
+                ...(fileData.dependencies && { dependencies: fileData.dependencies }),
+            };
+        }
+        else if (hasSummary(fileData)) {
+            result[dirPath][fileName] = {
+                summary: fileData.summary,
+                ...(fileData.dependencies && { dependencies: fileData.dependencies }),
+            };
+        }
+    }
+    return result;
+}
+/**
+ * Converts directory tree structure back into flat SourceCodeMap
+ */
+export function parseSourceCodeTree(sourceCodeTree) {
+    const result = {};
+    for (const [dirPath, files] of Object.entries(sourceCodeTree)) {
+        for (const [filePath, fileData] of Object.entries(files)) {
+            const fullPath = `${dirPath}/${filePath}`;
+            // Handle both content and summary cases while preserving dependencies
+            if (hasContent(fileData)) {
+                result[fullPath] = {
+                    content: fileData.content,
+                    ...(fileData.dependencies && { dependencies: fileData.dependencies }),
+                };
+            }
+            else if (hasSummary(fileData)) {
+                result[fullPath] = {
+                    summary: fileData.summary,
+                    ...(fileData.dependencies && { dependencies: fileData.dependencies }),
+                };
+            }
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=source-code-tree.js.map

package/dist/files/source-code-tree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"source-code-tree.js","sourceRoot":"","sources":["../../src/files/source-code-tree.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C;;GAEG;AACH,SAAS,UAAU,CAAC,GAA8B;IAChD,OAAO,SAAS,IAAI,GAAG,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,GAA8B;IAChD,OAAO,SAAS,IAAI,GAAG,CAAC;AAC1B,CAAC;AAWD;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,UAAyB;IACzD,MAAM,MAAM,GAAmB,EAAE,CAAC;IAElC,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC9D,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3C,SAAS;QACX,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YACrB,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;QACvB,CAAC;QAED,sEAAsE;QACtE,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzB,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,GAAG;gBAC1B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;aACtE,CAAC;QACJ,CAAC;aAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,GAAG;gBAC1B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;aACtE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,cAA8B;IAChE,MAAM,MAAM,GAAkB,EAAE,CAAC;IAEjC,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC9D,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzD,MAAM,QAAQ,GAAG,GAAG,OAAO,IAAI,QAAQ,EAAE,CAAC;YAE1C,sEAAsE;YACtE,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzB,MAAM,CAAC,QAAQ,CAAC,GAAG;oBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;iBACtE,CAAC;YACJ,CAAC;iBAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAChC,MAAM,CAAC,QAAQ,CAAC,GAAG;oBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,YAAY,EAAE,CAAC;iBACtE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}