CodeCompanion.nvim 中代码块格式化问题的分析与解决

问题背景

在使用CodeCompanion.nvim插件时，部分用户反馈在获取AI助手返回的代码块时，偶尔会出现代码块未正确使用Markdown的三重反引号(```)格式进行包装的情况。这会导致Treesitter语法高亮无法正常工作，影响代码的可读性和编辑体验。

问题分析

经过深入调查，我们发现这一现象与插件的系统提示词设计有关。当前版本的系统提示中包含了"避免将整个响应包装在三重反引号中"的指令，这可能导致AI模型在某些情况下不会为代码块添加标准的Markdown格式标记。

技术细节

1. Treesitter依赖：Treesitter作为Neovim的语法分析引擎，依赖规范的Markdown代码块标记来实现语法高亮。当代码块缺少三重反引号时，Treesitter无法识别其为代码区域。

2. AI模型行为：不同AI模型(如Copilot、GPT-4.1等)对提示词的响应存在差异。部分模型可能过度遵循"避免包装整个响应"的提示，导致代码块格式化不完整。

3. 系统提示设计：原始系统提示旨在防止AI将整个响应(包括解释文本)都放入代码块中，但这一设计可能在某些情况下产生副作用。

解决方案

经过测试验证，调整系统提示词可以显著改善这一问题。建议修改为：

"Output the final code in a single code block in a markdown format wrapped in triple backticks, ensuring that only relevant code is included."

这一修改明确要求AI：

• 使用标准Markdown格式
• 仅包装相关代码部分
• 保持三重反引号的完整性

实施建议

1. 临时解决方案：用户可以自行修改本地配置中的系统提示词，添加明确的格式要求。

2. 长期方案：等待插件作者在后续版本中对系统提示进行优化调整。据悉，作者计划在即将发布的PR中对提示系统进行重大改进，包括允许工具组覆盖默认提示。

3. 模型选择：不同AI模型对提示的响应可能存在差异。用户可尝试切换模型(如Claude Sonnet)以获得更稳定的格式化结果。

最佳实践

1. 在自定义提示词时，明确指定所需的输出格式
2. 针对不同任务类型，可考虑设计专门的格式化规则
3. 定期更新插件版本以获取最新的提示词优化

通过以上调整，用户可以确保从AI助手获取的代码块始终具有规范的Markdown格式，从而保证Treesitter语法高亮的正常工作，提升开发体验。