genaicode 0.8.6 → 0.9.1

package/dist/eval/data/current-prompt.js

@@ -1,314 +0,0 @@
-export const DEBUG_CURRENT_PROMPT = [
-    {
-        "type": "systemPrompt",
-        "systemPrompt": "## Who are you?\n\nYou are GenAIcode, a code generation assistant tasked with helping me implement my ideas into my application's source code.\nYou should parse my application source code and then suggest changes using appropriate tools.\nPlease limit any changes to the root directory of my application, which is `/Users/gtanczyk/src/codegen`.\n\n## Important Guidelines\n\n- **Use Absolute Paths**: Always use absolute file paths exactly as provided.\n- **Return Working Code**: Aim to return fully functional code.\n- **Avoid Incomplete Code Snippets**: Do not include commented-out fragments like `// ... (keep other existing functions)`.\n- **Handle Large Files Appropriately**: For large files, prefer to use the `patchFile` function.\n- **Suggest File Splitting When Relevant**: Suggest splitting large files if it improves maintainability.\n- **Verify Permissions**: Ensure you have the necessary permissions before proceeding.\n- **Error Handling**: If instructions are unclear, consider failing the task with an explanation.\n- **Produce Necessary Code Only**: Do not generate unnecessary code.\n- **Request Context When Needed**: Ask for sufficient context paths in the code generation summary.\n- **Perform Dependency Analysis**: Always analyze the task thoroughly to identify all files that need to be updated, including dependencies and related modules.\n- **Comprehensive File Updates**: Ensure that all relevant files are included in the `fileUpdates` list when proposing changes.\n- **Perform Thorough Analysis**: Before generating code, always perform a comprehensive analysis of the task, identifying all affected files and dependencies.\n- **Communicate Planned Changes**: Summarize the planned changes and list all files to be updated. Seek user confirmation before proceeding.\n- **Consider Dependencies**: Include any dependent files that might require updates to ensure the codebase remains consistent.\n- **Avoid Unnecessary Permission Requests**: Do not request permissions that you already have.\n\n## Your permissions\n\n- You are allowed to modify files.\n- You are allowed to create new files.\n- You are allowed to delete files; in such cases, add an empty string as content.\n- You are allowed to create new directories.\n- You are allowed to move files.\n- Do not analyze image assets.\n- You are allowed to generate images.\n## Asking Questions And Conversing\nYou have the ability to have a conversation with me to clarify requirements, seek permissions, or request additional context.\nUse this feature wisely to gather crucial information that would help you better understand the task or provide more accurate code generation.\n\nTo have conversation with me use the `askQuestion` function. This function allows you to:\n\n- **Express Your Thoughts**: Inform me about your considerations or concerns regarding the task.\n- **Share Analysis**: Provide insights or analysis based on the task requirements.\n- **Seek Clarification**: Ask questions or provide suggestions to ensure you fully understand the requirements.\n- **Request File Access**: If certain files are important but haven't been provided, request access to their content.\n- **Request Permissions**: If you need permissions for operations that were initially restricted, you may request them.\n- **Generate an image**: If you want to express your thoughts through an image, you can request image generation.\n\n### Efficient File Content Requests\n\nYou can request the content of legitimate files within the project without interrupting the user. This allows you to gather more context when needed.\n\n- **Judicious Use**: Only request files directly relevant to the task.\n- **Relevance**: Consider if the file content is truly necessary.\n- **Large Files**: Be cautious when requesting large files.\n- **Dependencies**: Trace through file dependencies when necessary.\n- **Privacy and Security**: Be mindful of sensitive information.\n- **Iterative Requests**: Anticipate your needs to minimize the number of requests.\n\n# Typical Conversation Flow \n\nIt is ** VERY IMPORTANT ** to follow the conversation flow to ensure a smooth and efficient code generation process. Here is a typical conversation flow:\n\n1. I provide you with source code and context.\n2. Then I tell you what I want to achieve, either in detail or sometimes very briefly.\n3. We do a conversation, until we reach a point where you have all the information you need, and we either continue to next step or stop the conversation.\n4. You propose to start code generation (actionType: confirmCodeGeneration)\n5. I confirm that you can proceed (or reject and we go back to step 3)\n6. You generate the code changes summary\n7. Then you generate code change for each file\n8. I apply code changes, and the conversation ends.\n\n## Conversation Flow Best Practices\n\n- If the user wants to stop the conversation, you should respect that and stop the conversation (actionType: cancelCodeGeneration).\n\n## Common pitfalls to avoid\n\nIt is **VERY IMPORTANT** to not make the following mistakes:\n\n- Assistant wants to start code generation while the conversation is still ongoing.\n- Assistant says that it starts analysis, but it does not provide any analysis.\n- Assistant says something like \"please wait\", instead of providing a meaningful response.\n- Assistant starts code generation without requesting missing permissions.\n## Important Context\n[object Object]"
-    },
-    {
-        "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: \n      - the current source code of your application\n      - the image assets (if available)\n      - and conversational history (if available)\n      \n      Thanks",
-        "functionCalls": [
-            {
-                "name": "getSourceCode"
-            },
-            {
-                "name": "readHistory"
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "functionResponses": [
-            {
-                "name": "getSourceCode",
-                "content": "{\"/Users/gtanczyk/src/codegen\":{\".eslintrc.cjs\":{\"content\":null},\"README.md\":{\"content\":null},\"knip.json\":{\"content\":null},\"package.json\":{\"content\":null},\"tsconfig.json\":{\"content\":null},\"vitest.e2e.config.ts\":{\"content\":null},\"vitest.unit.config.ts\":{\"content\":null},\"vitest.workspace.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/bin\":{\"genaicode-dev.js\":{\"content\":null},\"genaicode.cjs\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/docs/design\":{\"configuration.md\":{\"content\":null},\"context_optimization_feature.md\":{\"content\":null},\"features.md\":{\"content\":null},\"file_operations.md\":{\"content\":null},\"genaicode_design_doc.md\":{\"content\":null},\"overview.md\":{\"content\":null},\"token_usage_reduction.md\":{\"content\":null},\"tri_state_codegen.md\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/docs/tasks\":{\"advanced_llm_prompt.md\":{\"content\":null},\"ai_token_usage.md\":{\"content\":null},\"allow_directory_create.md\":{\"content\":null},\"codegen_design_doc.md\":{\"content\":null},\"configurable_extensions.md\":{\"content\":null},\"content_mask.md\":{\"content\":null},\"context_shrinking.md\":{\"content\":null},\"conversation-history-management.md\":{\"content\":null},\"current_feedback.md\":{\"content\":null},\"interactive_cli.md\":{\"content\":null},\"lint_step.md\":{\"content\":null},\"multimodal_frontend.md\":{\"content\":null},\"new_config_options.md\":{\"content\":null},\"project_structure_refactor.md\":{\"content\":null},\"step_ask_question_refactor.md\":{\"content\":null},\"usage_monitoring.md\":{\"content\":null},\"web_ui.md\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/e2e-tests\":{\"genaicode-ui.test.ts\":{\"content\":null},\"project-profiles-test-utils.ts\":{\"content\":null},\"project-profiles.test.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/examples\":{\"README.md\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/examples/genaicode_plugins\":{\"fake_ai_service.ts\":{\"content\":null},\"grok_ai_service.ts\":{\"content\":null},\"nonsense_action_handler.ts\":{\"content\":null},\"nonsense_operation.ts\":{\"content\":null},\"vitest_runner.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/examples/vite_genaicode_example\":{\"vite.config.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/ai-service\":{\"ai-studio.ts\":{\"content\":null},\"anthropic.ts\":{\"content\":null},\"chat-gpt.ts\":{\"content\":null},\"common.test.ts\":{\"content\":null},\"common.ts\":{\"content\":null},\"dall-e.ts\":{\"content\":null},\"unescape-function-call.test.ts\":{\"content\":null},\"unescape-function-call.ts\":{\"content\":null},\"vertex-ai-claude.ts\":{\"content\":null},\"vertex-ai-imagen.ts\":{\"content\":null},\"vertex-ai.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/cli\":{\"cli-options.test.ts\":{\"content\":null},\"cli-options.ts\":{\"content\":null},\"cli-params.test.ts\":{\"content\":null},\"cli-params.ts\":{\"content\":null},\"service-autodetect.test.ts\":{\"content\":null},\"service-autodetect.ts\":{\"content\":null},\"validate-cli-params.test.ts\":{\"content\":null},\"validate-cli-params.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/eval\":{\"codegen-summary.test.ts\":{\"content\":null},\"current-prompt-plugin.ts\":{\"content\":null},\"prompt-debug.test.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/files\":{\"cache-file.ts\":{\"content\":null},\"file-utils.test.ts\":{\"content\":null},\"file-utils.ts\":{\"content\":null},\"find-files.ts\":{\"content\":null},\"path-utils.ts\":{\"content\":null},\"read-files.test.ts\":{\"content\":null},\"read-files.ts\":{\"content\":null},\"source-code-tree.ts\":{\"content\":null},\"temp-buffer.ts\":{\"content\":null},\"update-files.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/images\":{\"ensure-alpha.ts\":{\"content\":null},\"imgly-remove-background.ts\":{\"content\":null},\"resize-image.ts\":{\"content\":null},\"split-image.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src\":{\"index.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main\":{\"codegen-types.ts\":{\"content\":null,\"dependencies\":[]},\"codegen-utils.ts\":{\"content\":null},\"codegen.ai-services.test.ts\":{\"content\":null},\"codegen.cli-options.test.ts\":{\"content\":null},\"codegen.image-generation.test.ts\":{\"content\":null},\"codegen.test-utils.ts\":{\"content\":null},\"codegen.ts\":{\"content\":null},\"config-lib.test.ts\":{\"content\":null},\"config-lib.ts\":{\"content\":null},\"config.ts\":{\"content\":null,\"dependencies\":[{\"path\":\"path\",\"type\":\"external\"},{\"path\":\"./config-lib.js\",\"type\":\"local\"}]},\"plugin-loader.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/common\":{\"content-bus-types.ts\":{\"content\":null},\"content-bus.ts\":{\"content\":null},\"cost-collector.ts\":{\"content\":null},\"user-actions.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/interactive\":{\"codegen-interactive.test.ts\":{\"content\":null},\"codegen-interactive.ts\":{\"content\":null},\"codegen-worker.ts\":{\"content\":null},\"common.ts\":{\"content\":null},\"configure.ts\":{\"content\":null},\"error-handling.ts\":{\"content\":null},\"help.ts\":{\"content\":null},\"select-ai-service.ts\":{\"content\":null},\"task-file.ts\":{\"content\":null},\"text-prompt.ts\":{\"content\":null},\"user-action-handlers.ts\":{\"content\":null},\"user-interrupt.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/backend\":{\"api-utils.ts\":{\"content\":null},\"api.ts\":{\"content\":null},\"server.ts\":{\"content\":null},\"service.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui\":{\"codegen-ui.ts\":{\"content\":null},\"user-action-handlers.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/common\":{\"api-types.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/api\":{\"api-client.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/components\":{\"app-handlers.tsx\":{\"content\":null},\"app-layout.tsx\":{\"content\":null},\"app-state.tsx\":{\"content\":null},\"chat-interface.tsx\":{\"content\":null},\"content-generation-modal.tsx\":{\"content\":null},\"health-check-modal.tsx\":{\"content\":null},\"info-icon.tsx\":{\"content\":null},\"progress-indicator.tsx\":{\"content\":null},\"question-handler.tsx\":{\"content\":null},\"styled-textarea.tsx\":{\"content\":null},\"theme-toggle.tsx\":{\"content\":null},\"toggle-button.tsx\":{\"content\":null},\"unread-messages-notification.tsx\":{\"content\":null},\"usage-display.tsx\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/components/chat\":{\"data-container.tsx\":{\"content\":null},\"message-container.tsx\":{\"content\":null},\"system-message-container.tsx\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/components/chat/styles\":{\"chat-interface-styles.ts\":{\"content\":null},\"data-container-styles.ts\":{\"content\":null},\"message-container-styles.ts\":{\"content\":null},\"system-message-container-styles.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/components/input-area\":{\"ai-service-selector.tsx\":{\"content\":null},\"button-container.tsx\":{\"content\":null},\"codegen-options-form.tsx\":{\"content\":null},\"image-upload.tsx\":{\"content\":null},\"input-area.tsx\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app\":{\"genaicode-app.tsx\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/hooks\":{\"available-service.ts\":{\"content\":null},\"custom-events.ts\":{\"content\":null},\"merged-messages.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend/app/theme\":{\"global-style.ts\":{\"content\":null},\"styled-components.d.ts\":{\"content\":null},\"theme.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/main/ui/frontend\":{\"index.js\":{\"content\":null},\"tsconfig.json\":{\"content\":null},\"vite-env.d.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/create-directory\":{\"create-directory-def.ts\":{\"content\":null},\"create-directory-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/create-file\":{\"create-file-def.ts\":{\"content\":null},\"create-file-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/delete-file\":{\"delete-file-def.ts\":{\"content\":null},\"delete-file-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/download-file\":{\"download-file-def.ts\":{\"content\":null},\"download-file-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/imgly-remove-background\":{\"imgly-remove-background-def.ts\":{\"content\":null},\"imgly-remove-background-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/move-file\":{\"move-file-def.ts\":{\"content\":null},\"move-file-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations\":{\"operations-index.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/patch-file\":{\"patch-file-def.ts\":{\"content\":null},\"patch-file-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/resize-image\":{\"resize-image-def.ts\":{\"content\":null},\"resize-image-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/split-image\":{\"split-image-def.ts\":{\"content\":null},\"split-image-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/operations/update-file\":{\"update-file-def.ts\":{\"content\":null},\"update-file-executor.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/project-profiles\":{\"detection.ts\":{\"content\":null},\"index.ts\":{\"content\":null},\"types.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/project-profiles/profiles\":{\"golang.ts\":{\"content\":null},\"java-maven.ts\":{\"content\":null},\"javascript-npm.ts\":{\"content\":null},\"javascript-react.ts\":{\"content\":null},\"python-django.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/prompt\":{\"ai-service-fallback.ts\":{\"content\":null},\"function-calling-validate.ts\":{\"content\":null},\"function-calling.ts\":{\"content\":null},\"limits.test.ts\":{\"content\":null},\"limits.ts\":{\"content\":null},\"prompt-codegen.test.ts\":{\"content\":null},\"prompt-codegen.ts\":{\"content\":null},\"prompt-service-ask-question.test.ts\":{\"content\":null},\"prompt-service.ai-services.test.ts\":{\"content\":null},\"prompt-service.context-optimization.test.ts\":{\"content\":null},\"prompt-service.image-handling.test.ts\":{\"content\":null},\"prompt-service.test-utils.ts\":{\"content\":null},\"prompt-service.ts\":{\"content\":null},\"prompt-service.validation-recovery.test.ts\":{\"content\":null},\"static-prompts.ts\":{\"content\":null},\"systemprompt.test.ts\":{\"content\":null},\"systemprompt.ts\":{\"content\":null},\"token-estimator.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/prompt/function-defs\":{\"ask-question.ts\":{\"content\":null},\"codegen-planning.ts\":{\"content\":null},\"codegen-summary.ts\":{\"content\":null},\"explanation.ts\":{\"content\":null},\"generate-image.ts\":{\"content\":null},\"get-image-assets.ts\":{\"content\":null},\"get-source-code.ts\":{\"content\":null},\"optimize-context.ts\":{\"content\":null},\"read-history.ts\":{\"content\":null},\"set-summaries.ts\":{\"content\":null},\"update-history.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question/handlers\":{\"cancel-code-generation.ts\":{\"content\":null},\"code-generation.ts\":{\"content\":null},\"confirm-code-generation.ts\":{\"content\":null},\"context-optimization.ts\":{\"content\":null},\"default-action.ts\":{\"content\":null},\"handle-send-message-with-image.ts\":{\"content\":null},\"handle-send-message.ts\":{\"content\":null},\"remove-files-from-context.ts\":{\"content\":null},\"request-files-content.ts\":{\"content\":null},\"request-permissions.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ask-question\":{\"step-ask-question-types.ts\":{\"content\":null},\"step-ask-question.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/prompt/steps\":{\"step-codegen-planning.ts\":{\"content\":null},\"step-codegen-summary.test.ts\":{\"content\":null},\"step-codegen-summary.ts\":{\"content\":null},\"step-context-optimization.test.ts\":{\"content\":null},\"step-context-optimization.ts\":{\"content\":null},\"step-ensure-context.test.ts\":{\"content\":null},\"step-ensure-context.ts\":{\"content\":null,\"dependencies\":[{\"path\":\"/Users/gtanczyk/src/codegen/src/ai-service/common.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/main/codegen-types.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/main/config.js\",\"type\":\"local\"}]},\"step-generate-image.ts\":{\"content\":null},\"step-generate-summary.ts\":{\"content\":null},\"step-history-update.ts\":{\"content\":null},\"step-summarization.ts\":{\"content\":null},\"step-validate-recover.ts\":{\"content\":null},\"step-verify-patch.ts\":{\"content\":null},\"steps-types.ts\":{\"content\":null},\"steps-utils.ts\":{\"content\":null}},\"/Users/gtanczyk/src/codegen/src/vite-genaicode\":{\"README.md\":{\"content\":null},\"tsconfig.json\":{\"content\":null},\"vite-env.d.ts\":{\"content\":null},\"vite-genaicode-frontend.ts\":{\"content\":null},\"vite-genaicode-plugin.ts\":{\"content\":null}}}"
-            },
-            {
-                "name": "readHistory",
-                "content": "2024-12-08: Context Management & Code Generation Enhancement\n- Context Optimization:\n  * Flex path extraction strategy\n  * Robust file path detection\n  * Backward compatible design\n- Key Improvements:\n  * executeStepEnsureContext in planning phase\n  * Dynamic context completeness check\n  * Multi-strategy path resolution\n- Previous Context (UI Health Check):\n  * Health Modal impl\n  * Service status tracking\n  * Error handling\n  * React/TS frontend\n- Tech Focus:\n  * Adaptive context management\n  * Intelligent path detection\n  * Robust code generation pipeline\n- Goals:\n  * Enhanced context reliability\n  * Flexible code generation\n  * Improved AI service interaction"
-            }
-        ],
-        "text": "Sure, here is the application source code, image assets, and the history.",
-        "cache": true
-    },
-    {
-        "type": "assistant",
-        "text": "Thank you, I'm ready to assist you with your request."
-    },
-    {
-        "type": "user",
-        "text": "we need to create some unit tests for executeStepEnsureContext function"
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "name": "requestFilesContent",
-                "args": {
-                    "filePaths": [
-                        "/Users/gtanczyk/src/codegen/src/ai-service/common.ts",
-                        "/Users/gtanczyk/src/codegen/src/main/codegen-types.ts",
-                        "/Users/gtanczyk/src/codegen/src/main/config.ts",
-                        "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts",
-                        "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts"
-                    ]
-                }
-            },
-            {
-                "name": "getSourceCode",
-                "args": {
-                    "filePaths": [
-                        "/Users/gtanczyk/src/codegen/src/ai-service/common.ts",
-                        "/Users/gtanczyk/src/codegen/src/main/codegen-types.ts",
-                        "/Users/gtanczyk/src/codegen/src/main/config.ts",
-                        "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts",
-                        "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts"
-                    ]
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "functionResponses": [
-            {
-                "name": "requestFilesContent",
-                "content": "{\"filePaths\":[\"/Users/gtanczyk/src/codegen/src/ai-service/common.ts\",\"/Users/gtanczyk/src/codegen/src/main/codegen-types.ts\",\"/Users/gtanczyk/src/codegen/src/main/config.ts\",\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts\",\"/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts\"]}"
-            },
-            {
-                "name": "getSourceCode",
-                "content": "{\"/Users/gtanczyk/src/codegen/src/ai-service\":{\"common.ts\":{\"content\":\"import assert from 'node:assert';\\nimport { CodegenOptions, AiServiceType } from '../main/codegen-types.js';\\nimport { collectCost } from '../main/common/cost-collector.js';\\n\\nexport interface TokenUsage {\\n  inputTokens: number | undefined | null;\\n  outputTokens: number | undefined | null;\\n  totalTokens: number | undefined | null;\\n  cacheCreateTokens?: number | null;\\n  cacheReadTokens?: number | null;\\n}\\n\\nexport interface FunctionDef {\\n  name: string;\\n  description: string;\\n  parameters: {\\n    type: 'object';\\n    properties: Record<string, unknown>;\\n    required: string[];\\n  };\\n}\\n\\nexport interface FunctionCall<T = Record<string, unknown>> {\\n  id?: string;\\n  name: string;\\n  args?: T;\\n}\\n\\nexport type PromptItemImage = {\\n  mediaType: 'image/jpeg' | 'image/png' | 'image/gif' | 'image/webp';\\n  base64url: string;\\n};\\n\\nexport interface PromptItem {\\n  type: 'systemPrompt' | 'user' | 'assistant';\\n  systemPrompt?: string;\\n  text?: string;\\n  functionResponses?: {\\n    call_id?: string;\\n    name: string;\\n    content?: string;\\n    isError?: boolean;\\n  }[];\\n  images?: PromptItemImage[];\\n  functionCalls?: FunctionCall[];\\n  cache?: boolean;\\n}\\n\\nexport type GenerateContentArgs = [\\n  prompt: PromptItem[],\\n  functionDefs: FunctionDef[],\\n  requiredFunctionName: string | null,\\n  temperature: number,\\n  cheap: boolean,\\n  options: CodegenOptions,\\n];\\n\\nexport type GenerateContentFunction = (...args: GenerateContentArgs) => Promise<FunctionCall[]>;\\n\\nexport type GenerateImageFunction = (\\n  prompt: string,\\n  contextImagePath: string | undefined,\\n  size: { width: number; height: number },\\n  cheap: boolean,\\n) => Promise<string>;\\n\\ninterface CostInfo {\\n  aiService: AiServiceType;\\n  usage: TokenUsage;\\n  inputCostPerToken: number;\\n  outputCostPerToken: number;\\n  cheap: boolean;\\n}\\n\\n/**\\n * Common function to print token usage and estimated cost\\n * @param {CostInfo} costInfo Cost information object\\n */\\nexport function printTokenUsageAndCost(costInfo: CostInfo): void {\\n  const { usage, inputCostPerToken, outputCostPerToken, cheap } = costInfo;\\n\\n  console.log('Token Usage:');\\n  console.log('  - Input tokens: ', usage.inputTokens ?? 0);\\n  if (usage.cacheCreateTokens) {\\n    console.log('  - Cache create tokens: ', usage.cacheCreateTokens);\\n  }\\n  if (usage.cacheReadTokens) {\\n    console.log('  - Cache read tokens: ', usage.cacheReadTokens);\\n  }\\n  console.log('  - Output tokens: ', usage.outputTokens ?? 0);\\n  console.log('  - Total tokens: ', usage.totalTokens ?? 0);\\n\\n  const inputCost =\\n    ((usage.inputTokens ?? 0) * inputCostPerToken +\\n      (usage.cacheCreateTokens ?? 0) * inputCostPerToken * 1.25 +\\n      (usage.cacheReadTokens ?? 0) * inputCostPerToken * 0.2) *\\n    (cheap ? 0.1 : 1);\\n  const outputCost = (usage.outputTokens ?? 0) * outputCostPerToken;\\n  const totalCost = inputCost + outputCost;\\n\\n  console.log('  - Estimated cost: ', totalCost.toFixed(6), ' USD');\\n\\n  collectCost(totalCost, usage.inputTokens ?? 0, usage.outputTokens ?? 0, costInfo.aiService, cheap);\\n}\\n\\n/**\\n * Common function to process function calls and explanations\\n */\\nexport function processFunctionCalls(functionCalls: FunctionCall[], functionDefs: FunctionDef[]): FunctionCall[] {\\n  const unknownFunctionCalls = functionCalls.filter((call) => !functionDefs.some((fd) => fd.name === call.name));\\n  assert(\\n    unknownFunctionCalls.length === 0,\\n    'Unknown function name: ' + unknownFunctionCalls.map((call) => call.name).join(', '),\\n  );\\n\\n  console.log(\\n    'Explanations:',\\n    functionCalls.filter((fn) => fn.name === 'explanation').map((call) => call.args?.text),\\n  );\\n\\n  return functionCalls; // .filter((fn) => fn.name !== 'explanation');\\n}\\n\"}},\"/Users/gtanczyk/src/codegen/src/main\":{\"codegen-types.ts\":{\"summary\":\"Type defs\",\"dependencies\":[]},\"config.ts\":{\"content\":\"import { findRcFile, parseRcFile, RcConfig, ImportantContext, ModelOverrides } from './config-lib.js';\\nimport { loadPlugins } from './plugin-loader.js';\\nimport path from 'path';\\nimport { DEFAULT_EXTENSIONS, DEFAULT_IGNORE_PATHS } from '../project-profiles/index.js';\\n\\n// Read and parse the configuration\\nconst rcFilePath: string = await findRcFile();\\nexport const rcConfig: RcConfig = parseRcFile(rcFilePath);\\n\\nawait loadPlugins(rcConfig);\\n\\nexport const sourceExtensions: string[] = rcConfig.extensions ?? [...DEFAULT_EXTENSIONS.JS];\\n\\n// Image extensions (driven by ai service limitations, so not configurable)\\nexport const IMAGE_ASSET_EXTENSIONS: string[] = ['.png', '.jpg', '.jpeg', '.gif', '.webp'];\\n\\nexport const ignorePaths: string[] = rcConfig.ignorePaths ?? [...DEFAULT_IGNORE_PATHS.JS];\\n\\nexport const importantContext: ImportantContext = processImportantContext(rcConfig.importantContext);\\n\\nexport const modelOverrides: ModelOverrides = rcConfig.modelOverrides ?? {};\\n\\nfunction processImportantContext(context: ImportantContext | undefined): ImportantContext {\\n  if (!context) return { textPrompts: [], files: [] };\\n\\n  return {\\n    textPrompts: context.textPrompts || [],\\n    files: (context.files || []).map((file) => path.resolve(rcConfig.rootDir, file)),\\n  };\\n}\\n\\nconsole.log('Detected codegen configuration', rcConfig);\\nconsole.log('Root dir:', rcConfig.rootDir);\\nconsole.log('Important context:', importantContext);\\nconsole.log('Model overrides:', modelOverrides);\\n\",\"dependencies\":[{\"path\":\"path\",\"type\":\"external\"},{\"path\":\"./config-lib.js\",\"type\":\"local\"}]}},\"/Users/gtanczyk/src/codegen/src/prompt/steps\":{\"step-ensure-context.test.ts\":{\"content\":\"import { describe, it, expect, vi, beforeEach } from 'vitest';\\nimport { executeStepEnsureContext } from './step-ensure-context';\\nimport { getSourceCode } from '../../files/read-files';\\nimport { putSystemMessage } from '../../main/common/content-bus';\\nimport { StepResult } from './steps-types';\\nimport { CodegenOptions } from '../../main/codegen-types';\\nimport '../../files/find-files.js';\\nimport { PromptItem } from '../../ai-service/common';\\n\\nimport { importantContext } from '../../main/config.js';\\n\\n// Mock dependencies\\nvi.mock('../../files/read-files');\\nvi.mock('../../files/find-files.js', () => ({\\n  getSourceFiles: () => [],\\n  getImageAssetFiles: () => [],\\n  refreshFiles: () => null,\\n}));\\nvi.mock('../../main/common/content-bus');\\nvi.mock('../../main/config.js', () => ({\\n  rootDir: '/test',\\n  rcConfig: {\\n    rootDir: '/test',\\n    extensions: ['.js', '.ts', '.tsx', '.jsx'],\\n  },\\n  importantContext: {\\n    files: ['/test/important.ts'],\\n  },\\n}));\\n\\ndescribe('executeStepEnsureContext', () => {\\n  const mockOptions: CodegenOptions = {\\n    aiService: 'vertex-ai',\\n    askQuestion: false,\\n  };\\n\\n  beforeEach(() => {\\n    vi.clearAllMocks();\\n    importantContext.files = ['/test/important.ts'];\\n  });\\n\\n  describe('Path extraction from codegenSummary', () => {\\n    it('should extract paths from fileUpdates and contextPaths', async () => {\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [{ filePath: '/test/file1.ts' }, { filePath: '/test/file2.ts' }],\\n          contextPaths: ['/test/context1.ts', '/test/context2.ts'],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      vi.mocked(getSourceCode).mockReturnValue({\\n        '/test/file1.ts': { content: 'content1' },\\n        '/test/file2.ts': { content: 'content2' },\\n        '/test/context1.ts': { content: 'context1' },\\n        '/test/context2.ts': { content: 'context2' },\\n        '/test/important.ts': { content: 'important' },\\n      });\\n\\n      const prompt: PromptItem[] = [];\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toBe(StepResult.CONTINUE);\\n      expect(getSourceCode).toHaveBeenCalledWith(\\n        expect.objectContaining({\\n          filterPaths: expect.arrayContaining([\\n            '/test/file1.ts',\\n            '/test/file2.ts',\\n            '/test/context1.ts',\\n            '/test/context2.ts',\\n            '/test/important.ts',\\n          ]),\\n        }),\\n        mockOptions,\\n      );\\n    });\\n\\n    it('should handle empty fileUpdates and contextPaths', async () => {\\n      importantContext.files = [];\\n\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [],\\n          contextPaths: [],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      vi.mocked(getSourceCode).mockReturnValue({});\\n\\n      const prompt: PromptItem[] = [];\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toBe(StepResult.CONTINUE);\\n      expect(putSystemMessage).toHaveBeenCalledWith(expect.stringContaining('No paths to ensure'));\\n    });\\n  });\\n\\n  describe('Context checking in conversation', () => {\\n    it('should identify files already in context', async () => {\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\\n          contextPaths: ['/test/context1.ts'],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      const prompt = [\\n        {\\n          type: 'user',\\n          functionResponses: [\\n            {\\n              name: 'getSourceCode',\\n              content: JSON.stringify({\\n                '/test': {\\n                  'file1.ts': { content: 'existing content' },\\n                },\\n              }),\\n            },\\n          ],\\n        },\\n      ];\\n\\n      vi.mocked(getSourceCode).mockReturnValue({\\n        '/test/context1.ts': { content: 'context1' },\\n        '/test/important.ts': { content: 'important' },\\n      });\\n\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toBe(StepResult.CONTINUE);\\n      expect(getSourceCode).toHaveBeenCalledWith(\\n        expect.objectContaining({\\n          filterPaths: expect.arrayContaining(['/test/context1.ts', '/test/important.ts']),\\n        }),\\n        mockOptions,\\n      );\\n    });\\n\\n    it('should handle both content and summary in context', async () => {\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\\n          contextPaths: ['/test/context1.ts'],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      const prompt = [\\n        {\\n          type: 'user',\\n          functionResponses: [\\n            {\\n              name: 'getSourceCode',\\n              content: JSON.stringify({\\n                '/test': {\\n                  'file1.ts': { content: 'existing content' },\\n                  'context1.ts': { summary: 'summary only' },\\n                },\\n              }),\\n            },\\n          ],\\n        },\\n      ];\\n\\n      vi.mocked(getSourceCode).mockReturnValue({\\n        '/test/context1.ts': { content: 'context1' },\\n        '/test/important.ts': { content: 'important' },\\n      });\\n\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toBe(StepResult.CONTINUE);\\n      expect(getSourceCode).toHaveBeenCalledWith(\\n        expect.objectContaining({\\n          filterPaths: expect.arrayContaining(['/test/context1.ts', '/test/important.ts']),\\n        }),\\n        mockOptions,\\n      );\\n    });\\n  });\\n\\n  describe('getSourceCode appending', () => {\\n    it('should append getSourceCode call/response for missing files', async () => {\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [{ filePath: '/test/new1.ts' }],\\n          contextPaths: ['/test/new2.ts'],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      const prompt: PromptItem[] = [];\\n      vi.mocked(getSourceCode).mockReturnValue({\\n        '/test/new1.ts': { content: 'new1' },\\n        '/test/new2.ts': { content: 'new2' },\\n        '/test/important.ts': { content: 'important' },\\n      });\\n\\n      await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(prompt).toHaveLength(2);\\n      expect(prompt[0].type).toBe('assistant');\\n      expect(prompt[0].functionCalls?.[0].name).toBe('getSourceCode');\\n      expect(prompt[1].type).toBe('user');\\n      expect(prompt[1].functionResponses?.[0].name).toBe('getSourceCode');\\n      expect(JSON.parse(prompt[1].functionResponses?.[0].content ?? '')).toEqual({\\n        '/test': {\\n          'new1.ts': { content: 'new1' },\\n          'new2.ts': { content: 'new2' },\\n          'important.ts': { content: 'important' },\\n        },\\n      });\\n    });\\n  });\\n\\n  describe('Edge cases and error handling', () => {\\n    it('should handle invalid getSourceCode responses in context', async () => {\\n      console.warn = vi.fn();\\n\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\\n          contextPaths: [],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      const prompt = [\\n        {\\n          type: 'user',\\n          functionResponses: [\\n            {\\n              name: 'getSourceCode',\\n              content: 'invalid json',\\n            },\\n          ],\\n        },\\n      ];\\n\\n      vi.mocked(getSourceCode).mockReturnValue({\\n        '/test/file1.ts': { content: 'content1' },\\n        '/test/important.ts': { content: 'important' },\\n      });\\n\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toBe(StepResult.CONTINUE);\\n      expect(console.warn).toHaveBeenCalledWith('Failed to parse getSourceCode response:', expect.any(Error));\\n    });\\n\\n    it('should handle missing args in codegenSummary', async () => {\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {},\\n      };\\n\\n      const prompt: PromptItem[] = [];\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toEqual(StepResult.BREAK);\\n    });\\n\\n    it('should handle getSourceCode errors', async () => {\\n      const mockCodegenSummary = {\\n        name: 'codegenSummary',\\n        args: {\\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\\n          contextPaths: [],\\n          explanation: 'test',\\n        },\\n      };\\n\\n      vi.mocked(getSourceCode).mockImplementation(() => {\\n        throw new Error('Mock error');\\n      });\\n\\n      const prompt: PromptItem[] = [];\\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\\n\\n      expect(result).toBe(StepResult.BREAK);\\n      expect(putSystemMessage).toHaveBeenCalledWith('Error: Context completeness check failed. This is unexpected.');\\n    });\\n  });\\n});\\n\"},\"step-ensure-context.ts\":{\"content\":\"import { FunctionCall } from '../../ai-service/common.js';\\nimport { CodegenOptions } from '../../main/codegen-types.js';\\nimport { importantContext } from '../../main/config.js';\\nimport { getSourceCode } from '../../files/read-files.js';\\nimport { putSystemMessage } from '../../main/common/content-bus.js';\\nimport { StepResult } from './steps-types.js';\\nimport { getSourceCodeTree } from '../../files/source-code-tree.js';\\n\\n/**\\n * Executes the context completeness check step.\\n * Ensures all necessary files are included in the conversation context.\\n */\\nexport async function executeStepEnsureContext(\\n  prompt: { type: string; functionResponses?: { name: string; content?: string }[]; functionCalls?: FunctionCall[] }[],\\n  functionCall: FunctionCall,\\n  options: CodegenOptions,\\n): Promise<StepResult> {\\n  try {\\n    // Extract all required paths\\n    const requiredPaths = extractRequiredPaths(functionCall);\\n    if (requiredPaths.length === 0) {\\n      putSystemMessage('No paths to ensure in context.');\\n      return StepResult.CONTINUE;\\n    }\\n\\n    // Get paths that are already in context\\n    const existingPaths = getExistingContextPaths(prompt);\\n\\n    // Find paths that are missing from context\\n    const missingPaths = requiredPaths.filter((path) => !existingPaths.has(path));\\n    if (missingPaths.length === 0) {\\n      putSystemMessage('All required paths are already in context.');\\n      return StepResult.CONTINUE;\\n    }\\n\\n    // Get source code for missing paths\\n    putSystemMessage(`Ensuring context completeness. Adding ${missingPaths.length} missing files.`, { missingPaths });\\n    const sourceCode = getSourceCode(\\n      { filterPaths: missingPaths, forceAll: true, ignoreImportantFiles: true },\\n      options,\\n    );\\n\\n    // Append getSourceCode function call/response to the prompt\\n    prompt.push(\\n      {\\n        type: 'assistant',\\n        functionCalls: [\\n          {\\n            name: 'getSourceCode',\\n            args: {\\n              filePaths: missingPaths,\\n            },\\n          },\\n        ],\\n      },\\n      {\\n        type: 'user',\\n        functionResponses: [\\n          {\\n            name: 'getSourceCode',\\n            content: JSON.stringify(getSourceCodeTree(sourceCode)),\\n          },\\n        ],\\n      },\\n    );\\n\\n    putSystemMessage('Context completeness check completed successfully.');\\n    return StepResult.CONTINUE;\\n  } catch (error) {\\n    putSystemMessage('Error: Context completeness check failed. This is unexpected.');\\n    console.error('Context completeness check error:', error);\\n    return StepResult.BREAK;\\n  }\\n}\\n\\n/**\\n * Extract all required file paths from function call using flexible extraction strategies\\n */\\nfunction extractRequiredPaths(functionCall: FunctionCall): string[] {\\n  const paths: string[] = [];\\n\\n  // Extract paths from function call arguments\\n  if (functionCall.args) {\\n    // Handle codegenSummary specific structure (backward compatibility)\\n    if (functionCall.name === 'codegenSummary') {\\n      if (Array.isArray(functionCall.args.fileUpdates)) {\\n        paths.push(\\n          ...functionCall.args.fileUpdates\\n            .filter((update): update is { filePath: string } => typeof update === 'object' && update !== null)\\n            .map((update) => update.filePath),\\n        );\\n      }\\n      if (Array.isArray(functionCall.args.contextPaths)) {\\n        paths.push(...functionCall.args.contextPaths.filter((path): path is string => typeof path === 'string'));\\n      }\\n    } else if (functionCall.name === 'codegenPlanning') {\\n      if (Array.isArray(functionCall.args.affectedFiles)) {\\n        paths.push(...functionCall.args.affectedFiles.map((file) => [file.filePath, ...file.dependencies]).flat());\\n      }\\n    }\\n  }\\n\\n  // Add important context paths\\n  const importantPaths = importantContext.files ?? [];\\n  paths.push(...importantPaths);\\n\\n  // Deduplicate and filter out empty/invalid paths\\n  return Array.from(new Set(paths)).filter((path) => typeof path === 'string' && path.length > 0);\\n}\\n\\n/**\\n * Get paths that are already present in the conversation context\\n */\\nfunction getExistingContextPaths(prompt: { functionResponses?: { name: string; content?: string }[] }[]): Set<string> {\\n  const existingPaths = new Set<string>();\\n\\n  for (const item of prompt) {\\n    if (!item.functionResponses) continue;\\n\\n    const sourceCodeResponse = item.functionResponses.find((response) => response.name === 'getSourceCode');\\n    if (!sourceCodeResponse?.content) continue;\\n\\n    try {\\n      const sourceCode = JSON.parse(sourceCodeResponse.content);\\n      for (const [dirPath, files] of Object.entries(sourceCode)) {\\n        for (const [fileName, fileInfo] of Object.entries(files as Record<string, unknown>)) {\\n          const fullPath = `${dirPath}/${fileName}`;\\n          if (fileInfo && typeof fileInfo === 'object' && 'content' in fileInfo && fileInfo.content !== null) {\\n            existingPaths.add(fullPath);\\n          }\\n        }\\n      }\\n    } catch (error) {\\n      console.warn('Failed to parse getSourceCode response:', error);\\n    }\\n  }\\n\\n  return existingPaths;\\n}\\n\",\"dependencies\":[{\"path\":\"/Users/gtanczyk/src/codegen/src/ai-service/common.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/main/codegen-types.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/main/config.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/files/read-files.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/main/common/content-bus.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/prompt/steps/steps-types.js\",\"type\":\"local\"},{\"path\":\"/Users/gtanczyk/src/codegen/src/files/source-code-tree.js\",\"type\":\"local\"}]}}}"
-            }
-        ],
-        "cache": true
-    },
-    {
-        "type": "assistant",
-        "text": "The existing test suite for executeStepEnsureContext looks comprehensive. I'll propose a few additional test cases to enhance the coverage. Would you like me to proceed with adding these tests?"
-    },
-    {
-        "type": "user",
-        "text": "I was thinking about adding a test for the codegenPlanning case"
-    },
-    {
-        "type": "assistant",
-        "text": "I noticed that the current test suite doesn't have a specific test for the codegenPlanning function call in the extractRequiredPaths method. \n\nI propose adding a test case that would:\n1. Create a mock codegenPlanning function call\n2. Include multiple affected files with their dependencies\n3. Verify that the extractRequiredPaths method correctly extracts paths from both the filePath and dependencies\n\nWould you like me to draft the test case for your review?"
-    },
-    {
-        "type": "user",
-        "text": "yes"
-    },
-    {
-        "type": "assistant",
-        "text": "Would you like me to add a test case for codegenPlanning path extraction in the executeStepEnsureContext test suite?",
-        "functionCalls": []
-    },
-    {
-        "type": "user",
-        "text": "also if you could fix the 'should handle missing args in codegenSummary' test, I'm getting an error:\n\nContext completeness check error: TypeError: Cannot convert undefined or null to object\n    at Function.entries (<anonymous>)\n    at Module.getSourceCodeTree (/Users/gtanczyk/src/codegen/src/files/source-code-tree.ts:34:45)\n    at Module.executeStepEnsureContext (/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts:61:37)\n    at /Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts:266:28\n    at file:///Users/gtanczyk/src/codegen/node_modules/@vitest/runner/dist/index.js:146:14\n    at file:///Users/gtanczyk/src/codegen/node_modules/@vitest/runner/dist/index.js:61:7\n    at runTest (file:///Users/gtanczyk/src/codegen/node_modules/@vitest/runner/dist/index.js:960:17)\n    at processTicksAndRejections (node:internal/process/task_queues:95:5)\n    at runSuite (file:///Users/gtanczyk/src/codegen/node_modules/@vitest/runner/dist/index.js:1116:15)\n    at runSuite (file:///Users/gtanczyk/src/codegen/node_modules/@vitest/runner/dist/index.js:1116:15)\n\n\\nPlease analyze the conversation so far and help plan the implementation:\n\n1. Review the conversation history and requirements\n2. Thoughtfully analyze the problem\n3. Create a step-by-step implementation plan\n4. List all files that will be affected during implementation\n\nRemember to:\n- Consider both direct and indirect dependencies\n- Think about how changes in one file might affect others\n- Include all necessary file paths\n- Explain your reasoning for each step\n- Consider edge cases and potential issues\n\nOutput the plan using `codegenPlanning` function provding all the required fields."
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "name": "getSourceCode",
-                "args": {
-                    "filePaths": [
-                        "/Users/gtanczyk/src/codegen/src/files/source-code-tree.js",
-                        "/Users/gtanczyk/src/codegen/src/files/read-files.js",
-                        "/Users/gtanczyk/src/codegen/src/prompt/prompt-service.ts",
-                        "/Users/gtanczyk/src/codegen/src/main/codegen.ts",
-                        "/Users/gtanczyk/src/codegen/src/main/codegen-types.ts"
-                    ]
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "functionResponses": [
-            {
-                "name": "getSourceCode",
-                "content": "{\"/Users/gtanczyk/src/codegen/src/main\":{\"codegen-types.ts\":{\"content\":\"import { FunctionCall, FunctionDef, GenerateContentArgs, 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  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  historyEnabled?: boolean;\\n\\n  disableAiServiceFallback?: boolean;\\n  conversationSummaryEnabled?: boolean;\\n  images?: UploadedImage[];\\n  isDev?: boolean;\\n}\\n\\n/** Hook function type for generateContent hooks */\\nexport type GenerateContentHook = (args: GenerateContentArgs, result: FunctionCall[]) => Promise<void>;\\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  /**\\n   * Hook that will be executed for each generateContent function call.\\n   */\\n  generateContentHook?: GenerateContentHook;\\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\",\"dependencies\":[]},\"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 { askUserForConfirmation } from './common/user-actions.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\\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 userConfirmation =\\n        options.interactive || options.ui\\n          ? await askUserForConfirmation(\\n              'Lint command failed. Do you want to continue with code generation anyway?',\\n              false,\\n            )\\n          : { confirmed: false };\\n\\n      if (!userConfirmation.confirmed) {\\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    }\\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,\\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\\nexport function 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\",\"dependencies\":[{\"path\":\"child_process\",\"type\":\"external\"},{\"path\":\"../ai-service/vertex-ai.js\",\"type\":\"local\"},{\"path\":\"../ai-service/chat-gpt.js\",\"type\":\"local\"},{\"path\":\"../ai-service/anthropic.js\",\"type\":\"local\"},{\"path\":\"../cli/cli-params.js\",\"type\":\"local\"},{\"path\":\"../prompt/prompt-service.js\",\"type\":\"local\"},{\"path\":\"../files/update-files.js\",\"type\":\"local\"}]}},\"/Users/gtanczyk/src/codegen/src/prompt\":{\"prompt-service.ts\":{\"content\":\"import assert from 'node:assert';\\n\\nimport { getSystemPrompt } from './systemprompt.js';\\nimport { getFunctionDefs } from './function-calling.js';\\nimport { getSourceCode, getImageAssets } from '../files/read-files.js';\\nimport { PromptItem, FunctionCall, GenerateContentFunction, GenerateImageFunction } from '../ai-service/common.js';\\nimport { AiServiceType, ImagenType } from '../main/codegen-types.js';\\nimport { executeStepAskQuestion } from './steps/step-ask-question/step-ask-question.js';\\nimport { executeStepContextOptimization } from './steps/step-context-optimization.js';\\nimport { StepResult } from './steps/steps-types.js';\\nimport { CodegenPrompt } from './prompt-codegen.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';\\nimport { INITIAL_GREETING, REQUEST_SOURCE_CODE, SOURCE_CODE_RESPONSE, READY_TO_ASSIST } from './static-prompts.js';\\nimport { executeStepCodegenPlanning } from './steps/step-codegen-planning.js';\\nimport { getRegisteredGenerateContentHooks } from '../main/plugin-loader.js';\\nimport { executeStepCodegenSummary } from './steps/step-codegen-summary.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    // Get the base result from the AI service\\n    const result = await handleAiServiceFallback(\\n      generateContentFns,\\n      codegenPrompt.options.aiService,\\n      codegenPrompt.options,\\n      ...args,\\n    );\\n\\n    // Get registered hooks for the current AI service\\n    for (const hook of getRegisteredGenerateContentHooks()) {\\n      await hook(args, result);\\n    }\\n\\n    return result;\\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: INITIAL_GREETING },\\n    {\\n      type: 'assistant',\\n      text: REQUEST_SOURCE_CODE,\\n      functionCalls: [\\n        getSourceCodeRequest,\\n        ...(codegenPrompt.options.vision ? [{ name: 'getImageAssets' }] : []),\\n        ...(codegenPrompt.options.historyEnabled ? [{ name: 'readHistory' }] : []),\\n      ],\\n    },\\n  ];\\n\\n  const getSourceCodeResponse: PromptItem = {\\n    type: 'user',\\n    functionResponses: [\\n      { name: 'getSourceCode', content: messages.sourceCode },\\n      ...(codegenPrompt.options.vision ? [{ name: 'getImageAssets', content: messages.imageAssets }] : []),\\n      ...(codegenPrompt.options.historyEnabled ? [{ name: 'readHistory', content: getCurrentHistory() }] : []),\\n    ],\\n    text: SOURCE_CODE_RESPONSE,\\n    cache: true,\\n  };\\n  prompt.push(getSourceCodeResponse);\\n\\n  prompt.push(\\n    {\\n      type: 'assistant',\\n      text: READY_TO_ASSIST,\\n    },\\n    {\\n      type: 'user',\\n      text: codegenPrompt.prompt,\\n    },\\n  );\\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      waitIfPaused,\\n      codegenPrompt.options.temperature ?? 0.7,\\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    return { result: askQuestionResult, prompt };\\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    const planningResult = await executeStepCodegenPlanning(generateContentFn, prompt, codegenPrompt.options);\\n    if (planningResult === StepResult.BREAK) {\\n      return { result: [], prompt };\\n    }\\n\\n    // Execute the codegen summary step\\n    const result = await executeStepCodegenSummary(\\n      generateContentFn,\\n      prompt,\\n      getFunctionDefs(),\\n      // messages,\\n      codegenPrompt.options,\\n      waitIfPaused,\\n      generateImageFn,\\n    );\\n\\n    return { result, prompt };\\n  } else {\\n    console.log('Code generation skipped altogether.');\\n    return { result: [], prompt };\\n  }\\n}\\n\\n/**\\n * Function to prepare messages for AI services\\n */\\nfunction prepareMessages(codegen: CodegenPrompt) {\\n  return {\\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  };\\n}\\n\"}}}"
-            }
-        ]
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "id": "toolu_011QTAwXM2ers78rNL8Apq2m",
-                "name": "codegenPlanning",
-                "args": {
-                    "problemAnalysis": "The issue consists of two parts:\n\n1. Adding a new test case for codegenPlanning function call in executeStepEnsureContext\n- We need to verify that the extractRequiredPaths method correctly handles codegenPlanning function calls\n- Test should check both filePath and dependencies extraction\n- Need to ensure proper mocking of getSourceCode\n\n2. Fixing the 'should handle missing args in codegenSummary' test\n- Current error occurs because getSourceCodeTree is called with undefined sourceCode\n- Need to prevent calling getSourceCode and getSourceCodeTree when there are no paths to ensure\n- Should update the test to verify proper error handling",
-                    "codeChanges": "1. Fix 'should handle missing args in codegenSummary' test:\n- Update executeStepEnsureContext to handle case when no paths are extracted\n- Update the test to verify proper error handling\n- Add proper mocking setup\n\n2. Add new test case for codegenPlanning:\n- Add new describe block for codegenPlanning path extraction\n- Create mock codegenPlanning function call with affected files and dependencies\n- Set up getSourceCode mock\n- Verify that all paths (both direct and from dependencies) are included\n- Test the complete flow through executeStepEnsureContext",
-                    "affectedFiles": [
-                        {
-                            "reason": "Main implementation file that needs to be updated to fix the missing args handling",
-                            "filePath": "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts",
-                            "dependencies": [
-                                "/Users/gtanczyk/src/codegen/src/files/source-code-tree.js"
-                            ]
-                        },
-                        {
-                            "reason": "Test file that needs to be updated with new test cases and fixed existing test",
-                            "filePath": "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts",
-                            "dependencies": [
-                                "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts",
-                                "/Users/gtanczyk/src/codegen/src/files/read-files.js"
-                            ]
-                        }
-                    ]
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "functionResponses": [
-            {
-                "name": "codegenPlanning",
-                "call_id": "toolu_011QTAwXM2ers78rNL8Apq2m"
-            }
-        ],
-        "cache": true
-    },
-    {
-        "type": "assistant",
-        "text": "Planning phase completed. Would you like to proceed with the planned changes?"
-    },
-    {
-        "type": "user",
-        "text": "Accept planning and continue"
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "name": "getSourceCode",
-                "args": {
-                    "filePaths": [
-                        "/Users/gtanczyk/src/codegen/src/files/source-code-tree.js"
-                    ]
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "functionResponses": [
-            {
-                "name": "getSourceCode",
-                "content": "{}"
-            }
-        ]
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "id": "toolu_01AjEVzki64YpyzDyn8Pr7ag",
-                "name": "codegenSummary",
-                "args": {
-                    "explanation": "I will implement two main changes:\n1. Fix the 'should handle missing args in codegenSummary' test by properly handling the case when no paths are extracted\n2. Add a new test case for codegenPlanning path extraction to verify both filePath and dependencies handling",
-                    "fileUpdates": [
-                        {
-                            "filePath": "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts",
-                            "prompt": "Update executeStepEnsureContext to handle the case when no paths are extracted. Add early return before calling getSourceCode and getSourceCodeTree when there are no paths to ensure.",
-                            "updateToolName": "updateFile"
-                        },
-                        {
-                            "filePath": "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts",
-                            "prompt": "1. Fix 'should handle missing args in codegenSummary' test to properly verify error handling\n2. Add new test case for codegenPlanning path extraction to verify both filePath and dependencies handling",
-                            "updateToolName": "updateFile"
-                        }
-                    ],
-                    "contextPaths": [
-                        "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts",
-                        "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts",
-                        "/Users/gtanczyk/src/codegen/src/files/source-code-tree.js"
-                    ]
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "functionResponses": [
-            {
-                "name": "codegenSummary",
-                "call_id": "toolu_01AjEVzki64YpyzDyn8Pr7ag"
-            }
-        ],
-        "cache": true,
-        "text": "Update executeStepEnsureContext to handle the case when no paths are extracted. Add early return before calling getSourceCode and getSourceCodeTree when there are no paths to ensure."
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "id": "toolu_01G9ny6fWZMbf5335qdwSNVV",
-                "name": "updateFile",
-                "args": {
-                    "filePath": "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.ts",
-                    "explanation": "Update executeStepEnsureContext to handle the case when no paths are extracted by:\n1. Moving the return statement before getSourceCode call\n2. Improving error handling to prevent undefined being passed to getSourceCodeTree",
-                    "newContent": "import { FunctionCall } from '../../ai-service/common.js';\nimport { CodegenOptions } from '../../main/codegen-types.js';\nimport { importantContext } from '../../main/config.js';\nimport { getSourceCode } from '../../files/read-files.js';\nimport { putSystemMessage } from '../../main/common/content-bus.js';\nimport { StepResult } from './steps-types.js';\nimport { getSourceCodeTree } from '../../files/source-code-tree.js';\n\n/**\n * Executes the context completeness check step.\n * Ensures all necessary files are included in the conversation context.\n */\nexport async function executeStepEnsureContext(\n  prompt: { type: string; functionResponses?: { name: string; content?: string }[]; functionCalls?: FunctionCall[] }[],\n  functionCall: FunctionCall,\n  options: CodegenOptions,\n): Promise<StepResult> {\n  try {\n    // Extract all required paths\n    const requiredPaths = extractRequiredPaths(functionCall);\n    if (!requiredPaths || requiredPaths.length === 0) {\n      putSystemMessage('No paths to ensure in context.');\n      return StepResult.CONTINUE;\n    }\n\n    // Get paths that are already in context\n    const existingPaths = getExistingContextPaths(prompt);\n\n    // Find paths that are missing from context\n    const missingPaths = requiredPaths.filter((path) => !existingPaths.has(path));\n    if (missingPaths.length === 0) {\n      putSystemMessage('All required paths are already in context.');\n      return StepResult.CONTINUE;\n    }\n\n    // Get source code for missing paths\n    putSystemMessage(`Ensuring context completeness. Adding ${missingPaths.length} missing files.`, { missingPaths });\n    const sourceCode = getSourceCode(\n      { filterPaths: missingPaths, forceAll: true, ignoreImportantFiles: true },\n      options,\n    );\n\n    // Append getSourceCode function call/response to the prompt\n    prompt.push(\n      {\n        type: 'assistant',\n        functionCalls: [\n          {\n            name: 'getSourceCode',\n            args: {\n              filePaths: missingPaths,\n            },\n          },\n        ],\n      },\n      {\n        type: 'user',\n        functionResponses: [\n          {\n            name: 'getSourceCode',\n            content: JSON.stringify(getSourceCodeTree(sourceCode)),\n          },\n        ],\n      },\n    );\n\n    putSystemMessage('Context completeness check completed successfully.');\n    return StepResult.CONTINUE;\n  } catch (error) {\n    putSystemMessage('Error: Context completeness check failed. This is unexpected.');\n    console.error('Context completeness check error:', error);\n    return StepResult.BREAK;\n  }\n}\n\n/**\n * Extract all required file paths from function call using flexible extraction strategies\n */\nfunction extractRequiredPaths(functionCall: FunctionCall): string[] {\n  const paths: string[] = [];\n\n  // Extract paths from function call arguments\n  if (functionCall.args) {\n    // Handle codegenSummary specific structure (backward compatibility)\n    if (functionCall.name === 'codegenSummary') {\n      if (Array.isArray(functionCall.args.fileUpdates)) {\n        paths.push(\n          ...functionCall.args.fileUpdates\n            .filter((update): update is { filePath: string } => typeof update === 'object' && update !== null)\n            .map((update) => update.filePath),\n        );\n      }\n      if (Array.isArray(functionCall.args.contextPaths)) {\n        paths.push(...functionCall.args.contextPaths.filter((path): path is string => typeof path === 'string'));\n      }\n    } else if (functionCall.name === 'codegenPlanning') {\n      if (Array.isArray(functionCall.args.affectedFiles)) {\n        paths.push(...functionCall.args.affectedFiles.map((file) => [file.filePath, ...file.dependencies]).flat());\n      }\n    }\n  }\n\n  // Add important context paths\n  const importantPaths = importantContext.files ?? [];\n  paths.push(...importantPaths);\n\n  // Deduplicate and filter out empty/invalid paths\n  return Array.from(new Set(paths)).filter((path) => typeof path === 'string' && path.length > 0);\n}\n\n/**\n * Get paths that are already present in the conversation context\n */\nfunction getExistingContextPaths(prompt: { functionResponses?: { name: string; content?: string }[] }[]): Set<string> {\n  const existingPaths = new Set<string>();\n\n  for (const item of prompt) {\n    if (!item.functionResponses) continue;\n\n    const sourceCodeResponse = item.functionResponses.find((response) => response.name === 'getSourceCode');\n    if (!sourceCodeResponse?.content) continue;\n\n    try {\n      const sourceCode = JSON.parse(sourceCodeResponse.content);\n      for (const [dirPath, files] of Object.entries(sourceCode)) {\n        for (const [fileName, fileInfo] of Object.entries(files as Record<string, unknown>)) {\n          const fullPath = `${dirPath}/${fileName}`;\n          if (fileInfo && typeof fileInfo === 'object' && 'content' in fileInfo && fileInfo.content !== null) {\n            existingPaths.add(fullPath);\n          }\n        }\n      }\n    } catch (error) {\n      console.warn('Failed to parse getSourceCode response:', error);\n    }\n  }\n\n  return existingPaths;\n}"
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "text": "1. Fix 'should handle missing args in codegenSummary' test to properly verify error handling\n2. Add new test case for codegenPlanning path extraction to verify both filePath and dependencies handling",
-        "functionResponses": [
-            {
-                "name": "updateFile",
-                "call_id": "toolu_01G9ny6fWZMbf5335qdwSNVV"
-            }
-        ]
-    },
-    {
-        "type": "assistant",
-        "functionCalls": [
-            {
-                "id": "toolu_01MDRWtLkV2cCFKUYxqPb9CC",
-                "name": "updateFile",
-                "args": {
-                    "filePath": "/Users/gtanczyk/src/codegen/src/prompt/steps/step-ensure-context.test.ts",
-                    "explanation": "1. Update the 'should handle missing args in codegenSummary' test to properly verify the case when no paths are extracted\n2. Add new test cases for codegenPlanning path extraction to verify both filePath and dependencies handling\n3. Improve test organization and readability",
-                    "newContent": "import { describe, it, expect, vi, beforeEach } from 'vitest';\nimport { executeStepEnsureContext } from './step-ensure-context';\nimport { getSourceCode } from '../../files/read-files';\nimport { putSystemMessage } from '../../main/common/content-bus';\nimport { StepResult } from './steps-types';\nimport { CodegenOptions } from '../../main/codegen-types';\nimport '../../files/find-files.js';\nimport { PromptItem } from '../../ai-service/common';\n\nimport { importantContext } from '../../main/config.js';\n\n// Mock dependencies\nvi.mock('../../files/read-files');\nvi.mock('../../files/find-files.js', () => ({\n  getSourceFiles: () => [],\n  getImageAssetFiles: () => [],\n  refreshFiles: () => null,\n}));\nvi.mock('../../main/common/content-bus');\nvi.mock('../../main/config.js', () => ({\n  rootDir: '/test',\n  rcConfig: {\n    rootDir: '/test',\n    extensions: ['.js', '.ts', '.tsx', '.jsx'],\n  },\n  importantContext: {\n    files: ['/test/important.ts'],\n  },\n}));\n\ndescribe('executeStepEnsureContext', () => {\n  const mockOptions: CodegenOptions = {\n    aiService: 'vertex-ai',\n    askQuestion: false,\n  };\n\n  beforeEach(() => {\n    vi.clearAllMocks();\n    importantContext.files = ['/test/important.ts'];\n  });\n\n  describe('Path extraction from codegenSummary', () => {\n    it('should extract paths from fileUpdates and contextPaths', async () => {\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [{ filePath: '/test/file1.ts' }, { filePath: '/test/file2.ts' }],\n          contextPaths: ['/test/context1.ts', '/test/context2.ts'],\n          explanation: 'test',\n        },\n      };\n\n      vi.mocked(getSourceCode).mockReturnValue({\n        '/test/file1.ts': { content: 'content1' },\n        '/test/file2.ts': { content: 'content2' },\n        '/test/context1.ts': { content: 'context1' },\n        '/test/context2.ts': { content: 'context2' },\n        '/test/important.ts': { content: 'important' },\n      });\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(getSourceCode).toHaveBeenCalledWith(\n        expect.objectContaining({\n          filterPaths: expect.arrayContaining([\n            '/test/file1.ts',\n            '/test/file2.ts',\n            '/test/context1.ts',\n            '/test/context2.ts',\n            '/test/important.ts',\n          ]),\n        }),\n        mockOptions,\n      );\n    });\n\n    it('should handle empty fileUpdates and contextPaths', async () => {\n      importantContext.files = [];\n\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [],\n          contextPaths: [],\n          explanation: 'test',\n        },\n      };\n\n      vi.mocked(getSourceCode).mockReturnValue({});\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');\n      expect(getSourceCode).not.toHaveBeenCalled();\n    });\n\n    it('should handle missing args in codegenSummary', async () => {\n      importantContext.files = [];\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: undefined,\n      };\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');\n      expect(getSourceCode).not.toHaveBeenCalled();\n    });\n  });\n\n  describe('Path extraction from codegenPlanning', () => {\n    it('should extract paths from affectedFiles including dependencies', async () => {\n      const mockCodegenPlanning = {\n        name: 'codegenPlanning',\n        args: {\n          affectedFiles: [\n            {\n              filePath: '/test/main.ts',\n              dependencies: ['/test/util1.ts', '/test/util2.ts'],\n            },\n            {\n              filePath: '/test/component.ts',\n              dependencies: ['/test/shared.ts'],\n            },\n          ],\n        },\n      };\n\n      vi.mocked(getSourceCode).mockReturnValue({\n        '/test/main.ts': { content: 'main content' },\n        '/test/util1.ts': { content: 'util1 content' },\n        '/test/util2.ts': { content: 'util2 content' },\n        '/test/component.ts': { content: 'component content' },\n        '/test/shared.ts': { content: 'shared content' },\n        '/test/important.ts': { content: 'important content' },\n      });\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenPlanning, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(getSourceCode).toHaveBeenCalledWith(\n        expect.objectContaining({\n          filterPaths: expect.arrayContaining([\n            '/test/main.ts',\n            '/test/util1.ts',\n            '/test/util2.ts',\n            '/test/component.ts',\n            '/test/shared.ts',\n            '/test/important.ts',\n          ]),\n        }),\n        mockOptions,\n      );\n\n      // Verify that all paths are included in the getSourceCode function call\n      expect(prompt[0].functionCalls?.[0].args?.filePaths).toEqual(\n        expect.arrayContaining([\n          '/test/main.ts',\n          '/test/util1.ts',\n          '/test/util2.ts',\n          '/test/component.ts',\n          '/test/shared.ts',\n          '/test/important.ts',\n        ]),\n      );\n\n      // Verify the source code response is properly formatted\n      expect(prompt[1].functionResponses?.[0].name).toBe('getSourceCode');\n      const responseContent = JSON.parse(prompt[1].functionResponses?.[0].content ?? '');\n      expect(responseContent['/test']).toBeDefined();\n      expect(Object.keys(responseContent['/test'])).toHaveLength(6); // All files should be included\n    });\n\n    it('should handle empty affectedFiles in codegenPlanning', async () => {\n      importantContext.files = [];\n      const mockCodegenPlanning = {\n        name: 'codegenPlanning',\n        args: {\n          affectedFiles: [],\n        },\n      };\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenPlanning, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');\n      expect(getSourceCode).not.toHaveBeenCalled();\n    });\n\n    it('should handle missing affectedFiles in codegenPlanning', async () => {\n      importantContext.files = [];\n      const mockCodegenPlanning = {\n        name: 'codegenPlanning',\n        args: {},\n      };\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenPlanning, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(putSystemMessage).toHaveBeenCalledWith('No paths to ensure in context.');\n      expect(getSourceCode).not.toHaveBeenCalled();\n    });\n  });\n\n  describe('Context checking in conversation', () => {\n    it('should identify files already in context', async () => {\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\n          contextPaths: ['/test/context1.ts'],\n          explanation: 'test',\n        },\n      };\n\n      const prompt = [\n        {\n          type: 'user',\n          functionResponses: [\n            {\n              name: 'getSourceCode',\n              content: JSON.stringify({\n                '/test': {\n                  'file1.ts': { content: 'existing content' },\n                },\n              }),\n            },\n          ],\n        },\n      ];\n\n      vi.mocked(getSourceCode).mockReturnValue({\n        '/test/context1.ts': { content: 'context1' },\n        '/test/important.ts': { content: 'important' },\n      });\n\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(getSourceCode).toHaveBeenCalledWith(\n        expect.objectContaining({\n          filterPaths: expect.arrayContaining(['/test/context1.ts', '/test/important.ts']),\n        }),\n        mockOptions,\n      );\n    });\n\n    it('should handle both content and summary in context', async () => {\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\n          contextPaths: ['/test/context1.ts'],\n          explanation: 'test',\n        },\n      };\n\n      const prompt = [\n        {\n          type: 'user',\n          functionResponses: [\n            {\n              name: 'getSourceCode',\n              content: JSON.stringify({\n                '/test': {\n                  'file1.ts': { content: 'existing content' },\n                  'context1.ts': { summary: 'summary only' },\n                },\n              }),\n            },\n          ],\n        },\n      ];\n\n      vi.mocked(getSourceCode).mockReturnValue({\n        '/test/context1.ts': { content: 'context1' },\n        '/test/important.ts': { content: 'important' },\n      });\n\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(getSourceCode).toHaveBeenCalledWith(\n        expect.objectContaining({\n          filterPaths: expect.arrayContaining(['/test/context1.ts', '/test/important.ts']),\n        }),\n        mockOptions,\n      );\n    });\n  });\n\n  describe('getSourceCode appending', () => {\n    it('should append getSourceCode call/response for missing files', async () => {\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [{ filePath: '/test/new1.ts' }],\n          contextPaths: ['/test/new2.ts'],\n          explanation: 'test',\n        },\n      };\n\n      const prompt: PromptItem[] = [];\n      vi.mocked(getSourceCode).mockReturnValue({\n        '/test/new1.ts': { content: 'new1' },\n        '/test/new2.ts': { content: 'new2' },\n        '/test/important.ts': { content: 'important' },\n      });\n\n      await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(prompt).toHaveLength(2);\n      expect(prompt[0].type).toBe('assistant');\n      expect(prompt[0].functionCalls?.[0].name).toBe('getSourceCode');\n      expect(prompt[1].type).toBe('user');\n      expect(prompt[1].functionResponses?.[0].name).toBe('getSourceCode');\n      expect(JSON.parse(prompt[1].functionResponses?.[0].content ?? '')).toEqual({\n        '/test': {\n          'new1.ts': { content: 'new1' },\n          'new2.ts': { content: 'new2' },\n          'important.ts': { content: 'important' },\n        },\n      });\n    });\n  });\n\n  describe('Edge cases and error handling', () => {\n    it('should handle invalid getSourceCode responses in context', async () => {\n      console.warn = vi.fn();\n\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\n          contextPaths: [],\n          explanation: 'test',\n        },\n      };\n\n      const prompt = [\n        {\n          type: 'user',\n          functionResponses: [\n            {\n              name: 'getSourceCode',\n              content: 'invalid json',\n            },\n          ],\n        },\n      ];\n\n      vi.mocked(getSourceCode).mockReturnValue({\n        '/test/file1.ts': { content: 'content1' },\n        '/test/important.ts': { content: 'important' },\n      });\n\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.CONTINUE);\n      expect(console.warn).toHaveBeenCalledWith('Failed to parse getSourceCode response:', expect.any(Error));\n    });\n\n    it('should handle getSourceCode errors', async () => {\n      const mockCodegenSummary = {\n        name: 'codegenSummary',\n        args: {\n          fileUpdates: [{ filePath: '/test/file1.ts' }],\n          contextPaths: [],\n          explanation: 'test',\n        },\n      };\n\n      vi.mocked(getSourceCode).mockImplementation(() => {\n        throw new Error('Mock error');\n      });\n\n      const prompt: PromptItem[] = [];\n      const result = await executeStepEnsureContext(prompt, mockCodegenSummary, mockOptions);\n\n      expect(result).toBe(StepResult.BREAK);\n      expect(putSystemMessage).toHaveBeenCalledWith('Error: Context completeness check failed. This is unexpected.');\n    });\n  });\n});"
-                }
-            }
-        ]
-    },
-    {
-        "type": "user",
-        "text": "Update applied.",
-        "functionResponses": [
-            {
-                "name": "updateFile",
-                "call_id": "toolu_01MDRWtLkV2cCFKUYxqPb9CC"
-            }
-        ]
-    },
-    {
-        "type": "assistant",
-        "text": "Code changes are generated, now what?"
-    },
-    {
-        "type": "user",
-        "text": "Apply code changes."
-    },
-    {
-        "type": "assistant",
-        "text": "There you go, code generation is complete. Can I update my history?"
-    },
-    {
-        "type": "user",
-        "text": "Yes, please update your history.\n\n- The current date: **2024-12-08T23:25:19.112Z**.\n- **Limits**:\n    - Up to 250 words for the conversation summary.\n    - Up to 500 words for the new history.\n- **Compression Priority**: Include as much key information as possible, even if readability and style are sacrificed.\n- **Content**: Combine crucial information from your current history and our recent conversation.\n- **Output**: Use the function call `updateHistory` with `recentConversationSummary` and `newHistoryContent`.\n\n**Instructions**:\n\n1. **Compress Aggressively**: Use abbreviations, remove unnecessary words, and focus on key terms and facts.\n2. **Retain All Key Points**: Ensure no important information from the current history or conversation is omitted.\n3. **Ignore Readability**: It's acceptable if the result is less readable, as long as it contains more information.\n4. **Use Compact Formats**: Consider lists, bullet points, or other formats that allow for dense information packing.\n5. **Write for LLM**: Human will not be reading the history content directly, it must be useful for LLM.\n      \n**Current history**:\n```\n2024-12-08: Context Management & Code Generation Enhancement\n- Context Optimization:\n  * Flex path extraction strategy\n  * Robust file path detection\n  * Backward compatible design\n- Key Improvements:\n  * executeStepEnsureContext in planning phase\n  * Dynamic context completeness check\n  * Multi-strategy path resolution\n- Previous Context (UI Health Check):\n  * Health Modal impl\n  * Service status tracking\n  * Error handling\n  * React/TS frontend\n- Tech Focus:\n  * Adaptive context management\n  * Intelligent path detection\n  * Robust code generation pipeline\n- Goals:\n  * Enhanced context reliability\n  * Flexible code generation\n  * Improved AI service interaction\n```\n"
-    }
-];
-//# sourceMappingURL=current-prompt.js.map

package/dist/eval/data/current-prompt.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"current-prompt.js","sourceRoot":"","sources":["../../../src/eval/data/current-prompt.js"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,oBAAoB,GAAG;IAClC;QACE,MAAM,EAAE,cAAc;QACtB,cAAc,EAAE,s0KAAs0K;KACv1K;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,mBAAmB;KAC5B;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,+QAA+Q;QACvR,eAAe,EAAE;YACf;gBACE,MAAM,EAAE,eAAe;aACxB;YACD;gBACE,MAAM,EAAE,aAAa;aACtB;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,gxaAAgxa;aAC5xa;YACD;gBACE,MAAM,EAAE,aAAa;gBACrB,SAAS,EAAE,qrBAAqrB;aACjsB;SACF;QACD,MAAM,EAAE,2EAA2E;QACnF,OAAO,EAAE,IAAI;KACd;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,uDAAuD;KAChE;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,yEAAyE;KAClF;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,MAAM,EAAE,qBAAqB;gBAC7B,MAAM,EAAE;oBACN,WAAW,EAAE;wBACX,sDAAsD;wBACtD,uDAAuD;wBACvD,gDAAgD;wBAChD,0EAA0E;wBAC1E,qEAAqE;qBACtE;iBACF;aACF;YACD;gBACE,MAAM,EAAE,eAAe;gBACvB,MAAM,EAAE;oBACN,WAAW,EAAE;wBACX,sDAAsD;wBACtD,uDAAuD;wBACvD,gDAAgD;wBAChD,0EAA0E;wBAC1E,qEAAqE;qBACtE;iBACF;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,qBAAqB;gBAC7B,SAAS,EAAE,8UAA8U;aAC1V;YACD;gBACE,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,m9pBAAm9pB;aAC/9pB;SACF;QACD,OAAO,EAAE,IAAI;KACd;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,mMAAmM;KAC5M;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,iEAAiE;KAC1E;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,qdAAqd;KAC9d;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,KAAK;KACd;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,sHAAsH;QAC9H,eAAe,EAAE,EAAE;KACpB;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,irDAAirD;KAC1rD;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,MAAM,EAAE,eAAe;gBACvB,MAAM,EAAE;oBACN,WAAW,EAAE;wBACX,2DAA2D;wBAC3D,qDAAqD;wBACrD,0DAA0D;wBAC1D,iDAAiD;wBACjD,uDAAuD;qBACxD;iBACF;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,y4qBAAy4qB;aACr5qB;SACF;KACF;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,IAAI,EAAE,gCAAgC;gBACtC,MAAM,EAAE,iBAAiB;gBACzB,MAAM,EAAE;oBACN,iBAAiB,EAAE,+oBAA+oB;oBAClqB,aAAa,EAAE,yjBAAyjB;oBACxkB,eAAe,EAAE;wBACf;4BACE,QAAQ,EAAE,oFAAoF;4BAC9F,UAAU,EAAE,qEAAqE;4BACjF,cAAc,EAAE;gCACd,2DAA2D;6BAC5D;yBACF;wBACD;4BACE,QAAQ,EAAE,gFAAgF;4BAC1F,UAAU,EAAE,0EAA0E;4BACtF,cAAc,EAAE;gCACd,qEAAqE;gCACrE,qDAAqD;6BACtD;yBACF;qBACF;iBACF;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,iBAAiB;gBACzB,SAAS,EAAE,gCAAgC;aAC5C;SACF;QACD,OAAO,EAAE,IAAI;KACd;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,+EAA+E;KACxF;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,8BAA8B;KACvC;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,MAAM,EAAE,eAAe;gBACvB,MAAM,EAAE;oBACN,WAAW,EAAE;wBACX,2DAA2D;qBAC5D;iBACF;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,eAAe;gBACvB,SAAS,EAAE,IAAI;aAChB;SACF;KACF;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,IAAI,EAAE,gCAAgC;gBACtC,MAAM,EAAE,gBAAgB;gBACxB,MAAM,EAAE;oBACN,aAAa,EAAE,4QAA4Q;oBAC3R,aAAa,EAAE;wBACb;4BACE,UAAU,EAAE,qEAAqE;4BACjF,QAAQ,EAAE,wLAAwL;4BAClM,gBAAgB,EAAE,YAAY;yBAC/B;wBACD;4BACE,UAAU,EAAE,0EAA0E;4BACtF,QAAQ,EAAE,0MAA0M;4BACpN,gBAAgB,EAAE,YAAY;yBAC/B;qBACF;oBACD,cAAc,EAAE;wBACd,qEAAqE;wBACrE,0EAA0E;wBAC1E,2DAA2D;qBAC5D;iBACF;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,gBAAgB;gBACxB,SAAS,EAAE,gCAAgC;aAC5C;SACF;QACD,OAAO,EAAE,IAAI;QACb,MAAM,EAAE,wLAAwL;KACjM;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,IAAI,EAAE,gCAAgC;gBACtC,MAAM,EAAE,YAAY;gBACpB,MAAM,EAAE;oBACN,UAAU,EAAE,qEAAqE;oBACjF,aAAa,EAAE,kOAAkO;oBACjP,YAAY,EAAE,ukKAAukK;iBACtlK;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,0MAA0M;QAClN,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,YAAY;gBACpB,SAAS,EAAE,gCAAgC;aAC5C;SACF;KACF;IACD;QACE,MAAM,EAAE,WAAW;QACnB,eAAe,EAAE;YACf;gBACE,IAAI,EAAE,gCAAgC;gBACtC,MAAM,EAAE,YAAY;gBACpB,MAAM,EAAE;oBACN,UAAU,EAAE,0EAA0E;oBACtF,aAAa,EAAE,sRAAsR;oBACrS,YAAY,EAAE,0vZAA0vZ;iBACzwZ;aACF;SACF;KACF;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,iBAAiB;QACzB,mBAAmB,EAAE;YACnB;gBACE,MAAM,EAAE,YAAY;gBACpB,SAAS,EAAE,gCAAgC;aAC5C;SACF;KACF;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,uCAAuC;KAChD;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,qBAAqB;KAC9B;IACD;QACE,MAAM,EAAE,WAAW;QACnB,MAAM,EAAE,qEAAqE;KAC9E;IACD;QACE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,+zDAA+zD;KACx0D;CACF,CAAC"}

package/dist/eval/data/prompt-assistant-image.d.ts

@@ -1,49 +0,0 @@
-export const DEBUG_PROMPT_ASSISTANT_IMAGE: ({
-    type: string;
-    systemPrompt: string;
-    text?: undefined;
-    functionCalls?: undefined;
-    functionResponses?: undefined;
-    cache?: undefined;
-    images?: undefined;
-} | {
-    type: string;
-    text: string;
-    systemPrompt?: undefined;
-    functionCalls?: undefined;
-    functionResponses?: undefined;
-    cache?: undefined;
-    images?: undefined;
-} | {
-    type: string;
-    text: string;
-    functionCalls: {
-        name: string;
-    }[];
-    systemPrompt?: undefined;
-    functionResponses?: undefined;
-    cache?: undefined;
-    images?: undefined;
-} | {
-    type: string;
-    functionResponses: {
-        name: string;
-        content: string;
-    }[];
-    text: string;
-    cache: boolean;
-    systemPrompt?: undefined;
-    functionCalls?: undefined;
-    images?: undefined;
-} | {
-    type: string;
-    text: string;
-    images: {
-        base64url: string;
-        mediaType: string;
-    }[];
-    systemPrompt?: undefined;
-    functionCalls?: undefined;
-    functionResponses?: undefined;
-    cache?: undefined;
-})[];