VSCode ChatGPT插件开发指南：从零搭建你的AI编程助手

9次阅读

共计 2301 个字符，预计需要花费 6 分钟才能阅读完成。

VSCode 作为最受欢迎的代码编辑器之一，其插件生态日益丰富。然而，现有 AI 编程助手插件普遍存在以下痛点：

响应延迟 ：多数插件直接调用远程 API，网络传输导致交互卡顿
上下文管理差 ：对话历史处理简单，未考虑 Token 长度限制
功能单一 ：仅实现基础问答，缺乏代码建议、错误诊断等深度集成

flowchart TD
    A[VSCode UI] -->| 用户输入 | B[插件入口]
    B --> C[上下文管理器]
    C --> D[API 调用封装层]
    D --> E[ChatGPT 服务]
    E --> D
    D --> F[响应处理器]
    F --> G[Markdown 渲染器]
    G --> A

/**
 * 带指数退避重试机制的 API 调用
 * @param prompt 用户输入
 * @param maxRetries 最大重试次数（默认 3 次）*/
async function callChatGPT(
  prompt: string,
  maxRetries = 3
): Promise<string> {
  let retryCount = 0;
  const baseDelay = 1000;

  while (retryCount < maxRetries) {
    try {
      const response = await axios.post(
        'https://api.openai.com/v1/chat/completions',
        {
          model: 'gpt-4',
          messages: [{role: 'user', content: prompt}]
        },
        {
          headers: {'Authorization': `Bearer ${apiKey}`,
            'Content-Type': 'application/json'
          }
        }
      );
      return response.data.choices[0].message.content;
    } catch (error) {if (retryCount === maxRetries - 1) throw error;

      const delay = baseDelay * Math.pow(2, retryCount);
      await new Promise(resolve => setTimeout(resolve, delay));
      retryCount++;
    }
  }
  throw new Error('Max retries exceeded');
}

// 基于 Token 计数的上下文压缩
function compressContext(history: ChatMessage[],
  maxTokens: number
): ChatMessage[] {let totalTokens = calculateTokens(history);
  const compressed = [...history];

  while (totalTokens > maxTokens && compressed.length > 1) {
    // 移除最早的非系统消息
    const index = compressed.findIndex(msg => msg.role !== 'system');
    if (index === -1) break;

    totalTokens -= calculateTokens([compressed[index]]);
    compressed.splice(index, 1);
  }

  return compressed;
}

vscode.languages.registerHoverProvider('*', {async provideHover(document, position) {const code = document.getText();
    const suggestion = await getCodeSuggestion(code);

    return new vscode.Hover(
      new vscode.MarkdownString(
        `**AI 建议 **:\n\n` +
        '```' + document.languageId + '\n' +
        suggestion.code + '\n```' +
        `\n\n${suggestion.explanation}`
      )
    );
  }
});

实现请求队列管理，避免突发流量
针对 429 响应自动降低请求频率
考虑使用多个 API 密钥轮询

// 代码片段净化函数
function sanitizeInput(code: string): string {
  const patterns = [/\b(?:password|api[_-]?key|secret)\s*=\s*['"][^'"]+['"]/gi,
    /\b(?:\d{1,3}\.){3}\d{1,3}\b/g
  ];

  return patterns.reduce((result, pattern) => result.replace(pattern, '[REDACTED]'),
    code
  );
}