From 2d7dafedb774e1a1d162a7f8ffc1c8cab28e83ea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=B0=8F=E6=B5=B7?= <7836246@qq.com> Date: Wed, 4 Mar 2026 17:46:04 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=20OpenAI=20Chat=20Co?= =?UTF-8?q?mpletions=20API=20=E5=85=BC=E5=AE=B9=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 POST /v1/chat/completions 端点，支持流式和非流式 - 完整支持 OpenAI 格式的工具调用 (function calling) - 支持 system/user/assistant/tool 四种角色消息 - 自动将 OpenAI 请求转换为 Anthropic 格式，复用现有 Cursor 管道 - 流式响应遵循 OpenAI SSE 规范 (data: [DONE] 结束标志) - 新增 openai-types.ts 和 openai-handler.ts - 更新启动信息展示两种 API 端点 - 更新 README 文档说明 OpenAI 兼容用法 --- README.md | 38 ++-- src/index.ts | 21 ++- src/openai-handler.ts | 400 ++++++++++++++++++++++++++++++++++++++++++ src/openai-types.ts | 106 +++++++++++ 4 files changed, 549 insertions(+), 16 deletions(-) create mode 100644 src/openai-handler.ts create mode 100644 src/openai-types.ts diff --git a/README.md b/README.md index a27052d..b2f9cb0 100644 --- a/README.md +++ b/README.md @@ -1,14 +1,18 @@ # Cursor2API v2 -将 Cursor 文档页免费 AI 对话接口代理转换为 **Anthropic Messages API**，可直接对接 **Claude Code**。 +将 Cursor 文档页免费 AI 对话接口代理转换为 **Anthropic Messages API** 和 **OpenAI Chat Completions API**，可直接对接 **Claude Code**、**ChatBox**、**LobeChat** 等各类客户端。 ## 原理 ``` ┌─────────────┐ ┌──────────────┐ ┌──────────────┐ -│ Claude Code │────▶│ cursor2api │────▶│ Cursor API │ -│ (Anthropic │ │ (代理+转换) │ │ /api/chat │ -│ Messages) │◀────│ │◀────│ │ +│ Claude Code │────▶│ │────▶│ │ +│ (Anthropic) │ │ │ │ │ +│ │◀────│ │◀────│ │ +├─────────────┤ │ cursor2api │ │ Cursor API │ +│ ChatBox 等 │────▶│ (代理+转换) │ │ /api/chat │ +│ (OpenAI) │ │ │ │ │ +│ │◀────│ │◀────│ │ └─────────────┘ └──────────────┘ └──────────────┘ ``` @@ -22,6 +26,7 @@ ## 核心特性 - **Anthropic Messages API 完整兼容** - `/v1/messages` 流式/非流式 +- **OpenAI Chat Completions API 兼容** - `/v1/chat/completions` 流式/非流式 + 工具调用 - **提示词注入工具能力** - 让 Claude Code 的 Bash、Read、Write 等工具全部可用 - **Node.js/TypeScript** - 无需外部进程生成 x-is-human token - **Chrome TLS 指纹** - 模拟真实浏览器请求头 @@ -62,19 +67,28 @@ export ANTHROPIC_BASE_URL=http://localhost:3010 claude ``` +### 6. 配合 OpenAI 兼容客户端（ChatBox、LobeChat 等） + +在客户端设置中填入： +- **API Base URL**: `http://localhost:3010/v1` +- **API Key**: 任意值（如 `sk-xxx`，不做校验） +- **Model**: 任意值（实际使用 config.yaml 中配置的模型） + ## 项目结构 ``` cursor2api/ ├── src/ -│ ├── index.ts # 入口 + Express 服务 -│ ├── config.ts # 配置管理 -│ ├── types.ts # 类型定义 -│ ├── cursor-client.ts # Cursor API 客户端 + Token 生成 -│ ├── converter.ts # 协议转换 + 工具提示词注入 -│ └── handler.ts # Anthropic API 处理器 -├── jscode/ # x-is-human token 生成脚本 -├── config.yaml # 配置文件 +│ ├── index.ts # 入口 + Express 服务 +│ ├── config.ts # 配置管理 +│ ├── types.ts # Anthropic/Cursor 类型定义 +│ ├── openai-types.ts # OpenAI 类型定义 +│ ├── cursor-client.ts # Cursor API 客户端 + Token 生成 +│ ├── converter.ts # 协议转换 + 工具提示词注入 +│ ├── handler.ts # Anthropic API 处理器 +│ └── openai-handler.ts # OpenAI API 处理器 +├── jscode/ # x-is-human token 生成脚本 +├── config.yaml # 配置文件 ├── package.json └── tsconfig.json ``` diff --git a/src/index.ts b/src/index.ts index 73df436..cc99a24 100644 --- a/src/index.ts +++ b/src/index.ts @@ -10,6 +10,7 @@ import express from 'express'; import { getConfig } from './config.js'; import { loadScripts } from './cursor-client.js'; import { handleMessages, listModels, countTokens } from './handler.js'; +import { handleOpenAIChatCompletions } from './openai-handler.js'; const app = express(); const config = getConfig(); @@ -35,6 +36,10 @@ app.use((_req, res, next) => { app.post('/v1/messages', handleMessages); app.post('/messages', handleMessages); +// OpenAI Chat Completions API（兼容） +app.post('/v1/chat/completions', handleOpenAIChatCompletions); +app.post('/chat/completions', handleOpenAIChatCompletions); + // Token 计数 app.post('/v1/messages/count_tokens', countTokens); app.post('/messages/count_tokens', countTokens); @@ -52,14 +57,16 @@ app.get('/', (_req, res) => { res.json({ name: 'cursor2api', version: '2.0.0', - description: 'Cursor Docs AI → Anthropic Messages API Proxy', + description: 'Cursor Docs AI → Anthropic & OpenAI API Proxy', endpoints: { - messages: 'POST /v1/messages', + anthropic_messages: 'POST /v1/messages', + openai_chat: 'POST /v1/chat/completions', models: 'GET /v1/models', health: 'GET /health', }, usage: { claude_code: 'export ANTHROPIC_BASE_URL=http://localhost:' + config.port, + openai_compatible: 'OPENAI_BASE_URL=http://localhost:' + config.port + '/v1', }, }); }); @@ -77,10 +84,16 @@ app.listen(config.port, () => { console.log(` ║ Server: http://localhost:${config.port} ║`); console.log(' ║ Model: ' + config.cursorModel.padEnd(26) + '║'); console.log(' ╠══════════════════════════════════════╣'); - console.log(' ║ Claude Code 使用方式: ║'); + console.log(' ║ API Endpoints: ║'); + console.log(' ║ • Anthropic: /v1/messages ║'); + console.log(' ║ • OpenAI: /v1/chat/completions ║'); + console.log(' ╠══════════════════════════════════════╣'); + console.log(' ║ Claude Code: ║'); console.log(` ║ export ANTHROPIC_BASE_URL= ║`); console.log(` ║ http://localhost:${config.port} ║`); - console.log(' ║ claude ║'); + console.log(' ║ OpenAI 兼容: ║'); + console.log(` ║ OPENAI_BASE_URL= ║`); + console.log(` ║ http://localhost:${config.port}/v1 ║`); console.log(' ╚══════════════════════════════════════╝'); console.log(''); }); diff --git a/src/openai-handler.ts b/src/openai-handler.ts new file mode 100644 index 0000000..aa50593 --- /dev/null +++ b/src/openai-handler.ts @@ -0,0 +1,400 @@ +/** + * openai-handler.ts - OpenAI Chat Completions API 兼容处理器 + * + * 将 OpenAI 格式请求转换为内部 Anthropic 格式，复用现有 Cursor 交互管道 + * 支持流式和非流式响应、工具调用 + */ + +import type { Request, Response } from 'express'; +import { v4 as uuidv4 } from 'uuid'; +import type { + OpenAIChatRequest, + OpenAIMessage, + OpenAIChatCompletion, + OpenAIChatCompletionChunk, + OpenAIToolCall, +} from './openai-types.js'; +import type { + AnthropicRequest, + AnthropicMessage, + AnthropicContentBlock, + AnthropicTool, + CursorSSEEvent, +} from './types.js'; +import { convertToCursorRequest, parseToolCalls, hasToolCalls } from './converter.js'; +import { sendCursorRequest, sendCursorRequestFull } from './cursor-client.js'; +import { getConfig } from './config.js'; + +function chatId(): string { + return 'chatcmpl-' + uuidv4().replace(/-/g, '').substring(0, 24); +} + +function toolCallId(): string { + return 'call_' + uuidv4().replace(/-/g, '').substring(0, 24); +} + +// ==================== 请求转换：OpenAI → Anthropic ==================== + +/** + * 将 OpenAI Chat Completions 请求转换为内部 Anthropic 格式 + * 这样可以完全复用现有的 convertToCursorRequest 管道 + */ +function convertToAnthropicRequest(body: OpenAIChatRequest): AnthropicRequest { + const messages: AnthropicMessage[] = []; + let systemPrompt: string | undefined; + + for (const msg of body.messages) { + switch (msg.role) { + case 'system': + // OpenAI system → Anthropic system + systemPrompt = (systemPrompt ? systemPrompt + '\n\n' : '') + extractOpenAIContent(msg); + break; + + case 'user': + messages.push({ + role: 'user', + content: extractOpenAIContent(msg), + }); + break; + + case 'assistant': { + // 助手消息可能包含 tool_calls + const blocks: AnthropicContentBlock[] = []; + const textContent = extractOpenAIContent(msg); + if (textContent) { + blocks.push({ type: 'text', text: textContent }); + } + + if (msg.tool_calls && msg.tool_calls.length > 0) { + for (const tc of msg.tool_calls) { + let args: Record = {}; + try { + args = JSON.parse(tc.function.arguments); + } catch { + args = { input: tc.function.arguments }; + } + blocks.push({ + type: 'tool_use', + id: tc.id, + name: tc.function.name, + input: args, + }); + } + } + + messages.push({ + role: 'assistant', + content: blocks.length > 0 ? blocks : (textContent || ''), + }); + break; + } + + case 'tool': { + // OpenAI tool result → Anthropic tool_result + messages.push({ + role: 'user', + content: [{ + type: 'tool_result', + tool_use_id: msg.tool_call_id, + content: extractOpenAIContent(msg), + }] as AnthropicContentBlock[], + }); + break; + } + } + } + + // 转换工具定义：OpenAI function → Anthropic tool + const tools: AnthropicTool[] | undefined = body.tools?.map(t => ({ + name: t.function.name, + description: t.function.description, + input_schema: t.function.parameters || { type: 'object', properties: {} }, + })); + + return { + model: body.model, + messages, + max_tokens: body.max_tokens || body.max_completion_tokens || 8192, + stream: body.stream, + system: systemPrompt, + tools, + temperature: body.temperature, + top_p: body.top_p, + stop_sequences: body.stop + ? (Array.isArray(body.stop) ? body.stop : [body.stop]) + : undefined, + }; +} + +/** + * 从 OpenAI 消息中提取文本内容 + */ +function extractOpenAIContent(msg: OpenAIMessage): string { + if (msg.content === null || msg.content === undefined) return ''; + if (typeof msg.content === 'string') return msg.content; + if (Array.isArray(msg.content)) { + return msg.content + .filter(p => p.type === 'text' && p.text) + .map(p => p.text!) + .join('\n'); + } + return String(msg.content); +} + +// ==================== 主处理入口 ==================== + +export async function handleOpenAIChatCompletions(req: Request, res: Response): Promise { + const body = req.body as OpenAIChatRequest; + + console.log(`[OpenAI] 收到请求: model=${body.model}, messages=${body.messages?.length}, stream=${body.stream}, tools=${body.tools?.length ?? 0}`); + + try { + // Step 1: OpenAI → Anthropic 格式 + const anthropicReq = convertToAnthropicRequest(body); + + // Step 2: Anthropic → Cursor 格式（复用现有管道） + const cursorReq = convertToCursorRequest(anthropicReq); + + if (body.stream) { + await handleOpenAIStream(res, cursorReq, body); + } else { + await handleOpenAINonStream(res, cursorReq, body); + } + } catch (err: unknown) { + const message = err instanceof Error ? err.message : String(err); + console.error(`[OpenAI] 请求处理失败:`, message); + res.status(500).json({ + error: { + message, + type: 'server_error', + code: 'internal_error', + }, + }); + } +} + +// ==================== 流式处理（OpenAI SSE 格式） ==================== + +async function handleOpenAIStream( + res: Response, + cursorReq: ReturnType, + body: OpenAIChatRequest, +): Promise { + res.writeHead(200, { + 'Content-Type': 'text/event-stream', + 'Cache-Control': 'no-cache', + 'Connection': 'keep-alive', + 'X-Accel-Buffering': 'no', + }); + + const id = chatId(); + const created = Math.floor(Date.now() / 1000); + const model = body.model; + const hasTools = (body.tools?.length ?? 0) > 0; + + // 发送 role delta（OpenAI 流式第一个 chunk 通常包含 role） + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: { role: 'assistant', content: '' }, + finish_reason: null, + }], + }); + + let fullResponse = ''; + let sentText = ''; + + try { + await sendCursorRequest(cursorReq, (event: CursorSSEEvent) => { + if (event.type !== 'text-delta' || !event.delta) return; + + fullResponse += event.delta; + + // 工具模式：缓冲直到完成 + if (hasTools && hasToolCalls(fullResponse)) { + return; + } + + // 实时流式推送文本 + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: { content: event.delta }, + finish_reason: null, + }], + }); + sentText += event.delta; + }); + + // 流完成后处理 + let finishReason: 'stop' | 'tool_calls' = 'stop'; + + if (hasTools && hasToolCalls(fullResponse)) { + const { toolCalls, cleanText } = parseToolCalls(fullResponse); + + if (toolCalls.length > 0) { + finishReason = 'tool_calls'; + + // 发送工具调用前的剩余文本 + const matchLen = findMatchLength(cleanText, sentText); + const unsentCleanText = cleanText.substring(matchLen).trim(); + + if (unsentCleanText) { + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: { content: unsentCleanText }, + finish_reason: null, + }], + }); + } + + // 发送每个工具调用 + for (let i = 0; i < toolCalls.length; i++) { + const tc = toolCalls[i]; + // 工具调用开始（包含 id、name） + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: { + tool_calls: [{ + index: i, + id: toolCallId(), + type: 'function', + function: { + name: tc.name, + arguments: JSON.stringify(tc.arguments), + }, + }], + }, + finish_reason: null, + }], + }); + } + } else { + // 误报：发送剩余文本 + const unsentText = fullResponse.substring(sentText.length); + if (unsentText) { + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: { content: unsentText }, + finish_reason: null, + }], + }); + } + } + } + + // 发送完成 chunk + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: {}, + finish_reason: finishReason, + }], + }); + + // OpenAI 流式结束标志 + res.write('data: [DONE]\n\n'); + + } catch (err: unknown) { + const message = err instanceof Error ? err.message : String(err); + // 在流中发送错误（非标准，但部分客户端可以处理） + writeOpenAISSE(res, { + id, object: 'chat.completion.chunk', created, model, + choices: [{ + index: 0, + delta: { content: `\n\n[Error: ${message}]` }, + finish_reason: 'stop', + }], + }); + res.write('data: [DONE]\n\n'); + } + + res.end(); +} + +// ==================== 非流式处理 ==================== + +async function handleOpenAINonStream( + res: Response, + cursorReq: ReturnType, + body: OpenAIChatRequest, +): Promise { + const fullText = await sendCursorRequestFull(cursorReq); + const hasTools = (body.tools?.length ?? 0) > 0; + + console.log(`[OpenAI] 原始响应 (${fullText.length} chars): ${fullText.substring(0, 300)}...`); + + let content: string | null = fullText; + let toolCalls: OpenAIToolCall[] | undefined; + let finishReason: 'stop' | 'tool_calls' = 'stop'; + + if (hasTools) { + const parsed = parseToolCalls(fullText); + + if (parsed.toolCalls.length > 0) { + finishReason = 'tool_calls'; + content = parsed.cleanText || null; + + toolCalls = parsed.toolCalls.map(tc => ({ + id: toolCallId(), + type: 'function' as const, + function: { + name: tc.name, + arguments: JSON.stringify(tc.arguments), + }, + })); + } + } + + const response: OpenAIChatCompletion = { + id: chatId(), + object: 'chat.completion', + created: Math.floor(Date.now() / 1000), + model: body.model, + choices: [{ + index: 0, + message: { + role: 'assistant', + content, + ...(toolCalls ? { tool_calls: toolCalls } : {}), + }, + finish_reason: finishReason, + }], + usage: { + prompt_tokens: 100, + completion_tokens: Math.ceil(fullText.length / 4), + total_tokens: 100 + Math.ceil(fullText.length / 4), + }, + }; + + res.json(response); +} + +// ==================== 工具函数 ==================== + +function writeOpenAISSE(res: Response, data: OpenAIChatCompletionChunk): void { + res.write(`data: ${JSON.stringify(data)}\n\n`); + // @ts-expect-error flush exists on ServerResponse when compression is used + if (typeof res.flush === 'function') res.flush(); +} + +/** + * 找到 cleanText 中已经发送过的文本长度 + */ +function findMatchLength(cleanText: string, sentText: string): number { + for (let i = Math.min(cleanText.length, sentText.length); i >= 0; i--) { + if (cleanText.startsWith(sentText.substring(0, i))) { + return i; + } + } + return 0; +} diff --git a/src/openai-types.ts b/src/openai-types.ts new file mode 100644 index 0000000..237e53d --- /dev/null +++ b/src/openai-types.ts @@ -0,0 +1,106 @@ +// ==================== OpenAI API Types ==================== + +export interface OpenAIChatRequest { + model: string; + messages: OpenAIMessage[]; + stream?: boolean; + temperature?: number; + top_p?: number; + max_tokens?: number; + max_completion_tokens?: number; + tools?: OpenAITool[]; + tool_choice?: string | { type: string; function?: { name: string } }; + stop?: string | string[]; + n?: number; + frequency_penalty?: number; + presence_penalty?: number; +} + +export interface OpenAIMessage { + role: 'system' | 'user' | 'assistant' | 'tool'; + content: string | OpenAIContentPart[] | null; + name?: string; + // assistant tool_calls + tool_calls?: OpenAIToolCall[]; + // tool result + tool_call_id?: string; +} + +export interface OpenAIContentPart { + type: 'text' | 'image_url'; + text?: string; + image_url?: { url: string; detail?: string }; +} + +export interface OpenAITool { + type: 'function'; + function: { + name: string; + description?: string; + parameters?: Record; + }; +} + +export interface OpenAIToolCall { + id: string; + type: 'function'; + function: { + name: string; + arguments: string; + }; +} + +// ==================== OpenAI Response Types ==================== + +export interface OpenAIChatCompletion { + id: string; + object: 'chat.completion'; + created: number; + model: string; + choices: OpenAIChatChoice[]; + usage: { + prompt_tokens: number; + completion_tokens: number; + total_tokens: number; + }; +} + +export interface OpenAIChatChoice { + index: number; + message: { + role: 'assistant'; + content: string | null; + tool_calls?: OpenAIToolCall[]; + }; + finish_reason: 'stop' | 'tool_calls' | 'length' | null; +} + +// ==================== OpenAI Stream Types ==================== + +export interface OpenAIChatCompletionChunk { + id: string; + object: 'chat.completion.chunk'; + created: number; + model: string; + choices: OpenAIStreamChoice[]; +} + +export interface OpenAIStreamChoice { + index: number; + delta: { + role?: 'assistant'; + content?: string | null; + tool_calls?: OpenAIStreamToolCall[]; + }; + finish_reason: 'stop' | 'tool_calls' | 'length' | null; +} + +export interface OpenAIStreamToolCall { + index: number; + id?: string; + type?: 'function'; + function: { + name?: string; + arguments: string; + }; +}