VSCode中Claude插件开发入门：从零搭建你的第一个AI助手

6次阅读

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

原生支持多轮对话上下文保持，更适合复杂任务分解
响应结构包含完整的元数据，便于实现精细化交互控制
允许自定义 temperature 等参数，平衡创造力和确定性输出

深度集成 GitHub 代码库
基于单次提示的补全模式
上下文自动从当前文件推断

混合本地 + 云端模型架构
支持对话式代码重构
内置终端操作能力

显式对话历史管理
支持超长上下文窗口(100K tokens)
可定制的角色设定模板

mkdir claude-extension
cd claude-extension
npm init -y

npm install --save-dev \
  @types/vscode \
  @types/node \
  typescript \
  eslint \
  prettier \
  ts-node \
  @anthropic-ai/sdk

创建 src/auth.ts 实现凭证管理：

import * as vscode from 'vscode';

export class AuthManager {
  private static KEY = 'claude-api-key';

  static async getKey(context: vscode.ExtensionContext): Promise<string | undefined> {return await context.secrets.get(this.KEY);
  }

  static async setKey(context: vscode.ExtensionContext, key: string): Promise<void> {await context.secrets.store(this.KEY, key);
  }
}

interface ClaudeMessage {
  role: 'user' | 'assistant';
  content: string;
}

interface ClaudeResponse {
  completion: string;
  stop_reason: string;
  model: string;
}

import {Anthropic} from '@anthropic-ai/sdk';

export class ClaudeChat {private history: ClaudeMessage[] = [];
  private client: Anthropic;

  constructor(apiKey: string) {this.client = new Anthropic({ apiKey});
  }

  async sendMessage(prompt: string): Promise<string> {this.history.push({ role: 'user', content: prompt});

    const response = await this.client.completions.create({
      model: 'claude-2',
      prompt: this.formatHistory(),
      max_tokens_to_sample: 1000,
    });

    this.history.push({
      role: 'assistant',
      content: response.completion,
    });

    return response.completion;
  }

  private formatHistory(): string {
    return this.history
      .map(msg => `${msg.role}: ${msg.content}`)
      .join('\n\n');
  }
}

实现请求队列管理
添加 retry-after 头处理
重要操作添加本地缓存

function sanitizeInput(input: string): string {
  return input
    .replace(/\b\d{4}-\d{4}-\d{4}-\d{4}\b/g, '[CREDIT_CARD]')
    .replace(/\b\w+@\w+\.\w+\b/g, '[EMAIL]');
}

async function withRetry<T>(fn: () => Promise<T>,
  maxAttempts = 3
): Promise<T> {
  let attempt = 0;
  while (attempt < maxAttempts) {
    try {return await fn();
    } catch (error) {if (++attempt >= maxAttempts) throw error;
      await new Promise(res => setTimeout(res, 1000 * attempt));
    }
  }
  throw new Error('Max retries exceeded');
}