初步添加claude接口

src/server/index.js

@@ -4,7 +4,7 @@ import path from 'path';
 import fs from 'fs';
 import { fileURLToPath } from 'url';
 import { generateAssistantResponse, generateAssistantResponseNoStream, getAvailableModels, generateImageForSD, closeRequester } from '../api/client.js';
-import { generateRequestBody, generateGeminiRequestBody, prepareImageRequest } from '../utils/utils.js';
+import { generateRequestBody, generateGeminiRequestBody, generateClaudeRequestBody, prepareImageRequest } from '../utils/utils.js';
 import logger from '../utils/logger.js';
 import config from '../config/config.js';
 import tokenManager from '../auth/token_manager.js';
@@ -301,10 +301,10 @@ app.use((req, res, next) => {
   if (req.path.startsWith('/v1/')) {
     const apiKey = config.security?.apiKey;
     if (apiKey) {
-      const authHeader = req.headers.authorization;
+      const authHeader = req.headers.authorization || req.headers['x-api-key'];
       const providedKey = authHeader?.startsWith('Bearer ') ? authHeader.slice(7) : authHeader;
       if (providedKey !== apiKey) {
-        logger.warn(`API Key 验证失败: ${req.method} ${req.path}`);
+        logger.warn(`API Key 验证失败: ${req.method} ${req.path} (提供的Key: ${providedKey ? providedKey.substring(0, 10) + '...' : '无'})`);
         return res.status(401).json({ error: 'Invalid API Key' });
       }
     }
@@ -611,6 +611,303 @@ app.post('/v1beta/models/:model\\:generateContent', (req, res) => {
   handleGeminiRequest(req, res, modelName, isStream);
 });
 
+// ==================== Claude API ====================
+
+// Claude 错误响应构造
+const buildClaudeErrorPayload = (error, statusCode) => {
+  let message = error.message || 'Internal server error';
+  if (error.isUpstreamApiError && error.rawBody) {
+    try {
+      const raw = typeof error.rawBody === 'string' ? JSON.parse(error.rawBody) : error.rawBody;
+      message = raw.error?.message || raw.message || message;
+    } catch {}
+  }
+
+  return {
+    type: "error",
+    error: {
+      type: statusCode === 401 ? "authentication_error" :
+            statusCode === 429 ? "rate_limit_error" :
+            statusCode === 400 ? "invalid_request_error" :
+            "api_error",
+      message: message
+    }
+  };
+};
+
+// Claude 流式响应工具
+const createClaudeStreamEvent = (eventType, data) => {
+  return `event: ${eventType}\ndata: ${JSON.stringify(data)}\n\n`;
+};
+
+// Claude 非流式响应构建
+const createClaudeResponse = (id, model, content, reasoning, toolCalls, stopReason, usage) => {
+  const contentBlocks = [];
+  
+  // 思维链内容（如果有）- Claude 格式用 thinking 类型
+  if (reasoning) {
+    contentBlocks.push({
+      type: "thinking",
+      thinking: reasoning
+    });
+  }
+  
+  // 文本内容
+  if (content) {
+    contentBlocks.push({
+      type: "text",
+      text: content
+    });
+  }
+  
+  // 工具调用
+  if (toolCalls && toolCalls.length > 0) {
+    for (const tc of toolCalls) {
+      try {
+        contentBlocks.push({
+          type: "tool_use",
+          id: tc.id,
+          name: tc.function.name,
+          input: JSON.parse(tc.function.arguments)
+        });
+      } catch (e) {
+        // 解析失败时传入空对象
+        contentBlocks.push({
+          type: "tool_use",
+          id: tc.id,
+          name: tc.function.name,
+          input: {}
+        });
+      }
+    }
+  }
+
+  return {
+    id: id,
+    type: "message",
+    role: "assistant",
+    content: contentBlocks,
+    model: model,
+    stop_reason: stopReason,
+    stop_sequence: null,
+    usage: usage ? {
+      input_tokens: usage.prompt_tokens || 0,
+      output_tokens: usage.completion_tokens || 0
+    } : { input_tokens: 0, output_tokens: 0 }
+  };
+};
+
+// Claude API 处理函数
+const handleClaudeRequest = async (req, res, isStream) => {
+  const { messages, model, system, tools, max_tokens, temperature, top_p, top_k, ...otherParams } = req.body;
+  
+  try {
+    if (!messages) {
+      return res.status(400).json(buildClaudeErrorPayload({ message: 'messages is required' }, 400));
+    }
+    
+    const token = await tokenManager.getToken();
+    if (!token) {
+      throw new Error('没有可用的token，请运行 npm run login 获取token');
+    }
+    
+    // 构建参数
+    const parameters = {
+      max_tokens: max_tokens || config.defaults.max_tokens,
+      temperature: temperature ?? config.defaults.temperature,
+      top_p: top_p ?? config.defaults.top_p,
+      top_k: top_k ?? config.defaults.top_k,
+      ...otherParams
+    };
+    
+    const requestBody = generateClaudeRequestBody(messages, model, parameters, tools, system, token);
+    
+    const msgId = `msg_${Date.now()}`;
+    const maxRetries = Number(config.retryTimes || 0);
+    const safeRetries = maxRetries > 0 ? Math.floor(maxRetries) : 0;
+    
+    if (isStream) {
+      setStreamHeaders(res);
+      const heartbeatTimer = createHeartbeat(res);
+      
+      try {
+        let contentIndex = 0;
+        let usageData = null;
+        let hasToolCall = false;
+        let currentBlockType = null;
+        let reasoningSent = false;
+        
+        // 发送 message_start
+        res.write(createClaudeStreamEvent('message_start', {
+          type: "message_start",
+          message: {
+            id: msgId,
+            type: "message",
+            role: "assistant",
+            content: [],
+            model: model,
+            stop_reason: null,
+            stop_sequence: null,
+            usage: { input_tokens: 0, output_tokens: 0 }
+          }
+        }));
+        
+        await with429Retry(
+          () => generateAssistantResponse(requestBody, token, (data) => {
+            if (data.type === 'usage') {
+              usageData = data.usage;
+            } else if (data.type === 'reasoning') {
+              // 思维链内容 - 使用 thinking 类型
+              if (!reasoningSent) {
+                // 开始思维块
+                res.write(createClaudeStreamEvent('content_block_start', {
+                  type: "content_block_start",
+                  index: contentIndex,
+                  content_block: { type: "thinking", thinking: "" }
+                }));
+                currentBlockType = 'thinking';
+                reasoningSent = true;
+              }
+              // 发送思维增量
+              res.write(createClaudeStreamEvent('content_block_delta', {
+                type: "content_block_delta",
+                index: contentIndex,
+                delta: { type: "thinking_delta", thinking: data.reasoning_content || '' }
+              }));
+            } else if (data.type === 'tool_calls') {
+              hasToolCall = true;
+              // 结束之前的块（如果有）
+              if (currentBlockType) {
+                res.write(createClaudeStreamEvent('content_block_stop', {
+                  type: "content_block_stop",
+                  index: contentIndex
+                }));
+                contentIndex++;
+              }
+              // 工具调用
+              for (const tc of data.tool_calls) {
+                try {
+                  const inputObj = JSON.parse(tc.function.arguments);
+                  res.write(createClaudeStreamEvent('content_block_start', {
+                    type: "content_block_start",
+                    index: contentIndex,
+                    content_block: { type: "tool_use", id: tc.id, name: tc.function.name, input: {} }
+                  }));
+                  // 发送 input 增量
+                  res.write(createClaudeStreamEvent('content_block_delta', {
+                    type: "content_block_delta",
+                    index: contentIndex,
+                    delta: { type: "input_json_delta", partial_json: JSON.stringify(inputObj) }
+                  }));
+                  res.write(createClaudeStreamEvent('content_block_stop', {
+                    type: "content_block_stop",
+                    index: contentIndex
+                  }));
+                  contentIndex++;
+                } catch (e) {
+                  // 解析失败，跳过
+                }
+              }
+              currentBlockType = null;
+            } else {
+              // 普通文本内容
+              if (currentBlockType === 'thinking') {
+                // 结束思维块
+                res.write(createClaudeStreamEvent('content_block_stop', {
+                  type: "content_block_stop",
+                  index: contentIndex
+                }));
+                contentIndex++;
+                currentBlockType = null;
+              }
+              if (currentBlockType !== 'text') {
+                // 开始文本块
+                res.write(createClaudeStreamEvent('content_block_start', {
+                  type: "content_block_start",
+                  index: contentIndex,
+                  content_block: { type: "text", text: "" }
+                }));
+                currentBlockType = 'text';
+              }
+              // 发送文本增量
+              res.write(createClaudeStreamEvent('content_block_delta', {
+                type: "content_block_delta",
+                index: contentIndex,
+                delta: { type: "text_delta", text: data.content || '' }
+              }));
+            }
+          }),
+          safeRetries,
+          'claude.stream '
+        );
+        
+        // 结束最后一个内容块
+        if (currentBlockType) {
+          res.write(createClaudeStreamEvent('content_block_stop', {
+            type: "content_block_stop",
+            index: contentIndex
+          }));
+        }
+        
+        // 发送 message_delta
+        const stopReason = hasToolCall ? 'tool_use' : 'end_turn';
+        res.write(createClaudeStreamEvent('message_delta', {
+          type: "message_delta",
+          delta: { stop_reason: stopReason, stop_sequence: null },
+          usage: usageData ? { output_tokens: usageData.completion_tokens || 0 } : { output_tokens: 0 }
+        }));
+        
+        // 发送 message_stop
+        res.write(createClaudeStreamEvent('message_stop', {
+          type: "message_stop"
+        }));
+        
+        clearInterval(heartbeatTimer);
+        res.end();
+      } catch (error) {
+        clearInterval(heartbeatTimer);
+        throw error;
+      }
+    } else {
+      // 非流式请求
+      req.setTimeout(0);
+      res.setTimeout(0);
+      
+      const { content, reasoningContent, toolCalls, usage } = await with429Retry(
+        () => generateAssistantResponseNoStream(requestBody, token),
+        safeRetries,
+        'claude.no_stream '
+      );
+      
+      const stopReason = toolCalls.length > 0 ? 'tool_use' : 'end_turn';
+      const response = createClaudeResponse(
+        msgId,
+        model,
+        content,
+        reasoningContent,
+        toolCalls,
+        stopReason,
+        usage
+      );
+      
+      res.json(response);
+    }
+  } catch (error) {
+    logger.error('Claude 请求失败:', error.message);
+    if (res.headersSent) return;
+    
+    const statusCode = Number(error.status) || 500;
+    const errorPayload = buildClaudeErrorPayload(error, statusCode);
+    res.status(statusCode).json(errorPayload);
+  }
+};
+
+// Claude Messages API 端点
+app.post('/v1/messages', (req, res) => {
+  const isStream = req.body.stream === true;
+  handleClaudeRequest(req, res, isStream);
+});
+
 const server = app.listen(config.server.port, config.server.host, () => {
   logger.info(`服务器已启动: ${config.server.host}:${config.server.port}`);
 });

src/utils/utils.js

@@ -5,7 +5,7 @@ import os from 'os';
 import { getReasoningSignature, getToolSignature } from './thoughtSignatureCache.js';
 import { setToolNameMapping } from './toolNameCache.js';
 
-// 思维链签名常量
+// 思维链签名常量 - 暂时123占位
 // Claude 模型签名
 const CLAUDE_THOUGHT_SIGNATURE = 'RXFRRENrZ0lDaEFDR0FJcVFKV1Bvcy9GV20wSmtMV2FmWkFEbGF1ZTZzQTdRcFlTc1NvbklmemtSNFo4c1dqeitIRHBOYW9hS2NYTE1TeTF3bjh2T1RHdE1KVjVuYUNQclZ5cm9DMFNETHk4M0hOSWsrTG1aRUhNZ3hvTTl0ZEpXUDl6UUMzOExxc2ZJakI0UkkxWE1mdWJ1VDQrZnY0Znp0VEoyTlhtMjZKL2daYi9HL1gwcmR4b2x0VE54empLemtLcEp0ZXRia2plb3NBcWlRSWlXUHloMGhVVTk1dHNha1dyNDVWNUo3MTJjZDNxdHQ5Z0dkbjdFaFk4dUllUC9CcThVY2VZZC9YbFpYbDc2bHpEbmdzL2lDZXlNY3NuZXdQMjZBTDRaQzJReXdibVQzbXlSZmpld3ZSaUxxOWR1TVNidHIxYXRtYTJ0U1JIRjI0Z0JwUnpadE1RTmoyMjR4bTZVNUdRNXlOSWVzUXNFNmJzRGNSV0RTMGFVOEZERExybmhVQWZQT2JYMG5lTGR1QnU1VGZOWW9NZGlRbTgyUHVqVE1xaTlmN0t2QmJEUUdCeXdyVXR2eUNnTEFHNHNqeWluZDRCOEg3N2ZJamt5blI3Q3ZpQzlIOTVxSENVTCt3K3JzMmsvV0sxNlVsbGlTK0pET3UxWXpPMWRPOUp3V3hEMHd5ZVU0a0Y5MjIxaUE5Z2lUd2djZXhSU2c4TWJVMm1NSjJlaGdlY3g0YjJ3QloxR0FFPQ==';
 // Gemini 思维链签名
@@ -564,10 +564,305 @@ function generateGeminiRequestBody(geminiBody, modelName, token){
   return requestBody;
 }
 
+// ==================== Claude API 转换函数 ====================
+
+/**
+ * 从 Claude 消息内容中提取图片
+ * Claude 格式: { type: "image", source: { type: "base64", media_type: "image/png", data: "..." } }
+ */
+function extractImagesFromClaudeContent(content) {
+  const result = { text: '', images: [] };
+
+  if (typeof content === 'string') {
+    result.text = content;
+    return result;
+  }
+
+  if (Array.isArray(content)) {
+    for (const item of content) {
+      if (item.type === 'text') {
+        result.text += item.text || '';
+      } else if (item.type === 'image') {
+        // Claude 格式的图片
+        const source = item.source;
+        if (source && source.type === 'base64' && source.data) {
+          result.images.push({
+            inlineData: {
+              mimeType: source.media_type || 'image/png',
+              data: source.data
+            }
+          });
+        }
+      }
+    }
+  }
+
+  return result;
+}
+
+/**
+ * 处理 Claude 用户消息
+ */
+function handleClaudeUserMessage(extracted, antigravityMessages) {
+  antigravityMessages.push({
+    role: "user",
+    parts: [
+      { text: extracted.text },
+      ...extracted.images
+    ]
+  });
+}
+
+/**
+ * 处理 Claude 助手消息（包含 tool_use）
+ */
+function handleClaudeAssistantMessage(message, antigravityMessages, enableThinking, actualModelName, sessionId) {
+  const lastMessage = antigravityMessages[antigravityMessages.length - 1];
+  const content = message.content;
+  
+  // 解析 content 数组
+  let textContent = '';
+  const toolCalls = [];
+  
+  if (typeof content === 'string') {
+    textContent = content;
+  } else if (Array.isArray(content)) {
+    for (const item of content) {
+      if (item.type === 'text') {
+        textContent += item.text || '';
+      } else if (item.type === 'tool_use') {
+        // Claude 的 tool_use 格式
+        const originalName = item.name;
+        const safeName = sanitizeToolName(originalName);
+        
+        const part = {
+          functionCall: {
+            id: item.id,
+            name: safeName,
+            args: {
+              query: JSON.stringify(item.input || {})
+            }
+          }
+        };
+        
+        // 记录工具名映射
+        if (sessionId && actualModelName && safeName !== originalName) {
+          setToolNameMapping(sessionId, actualModelName, safeName, originalName);
+        }
+        
+        toolCalls.push(part);
+      }
+    }
+  }
+  
+  const hasToolCalls = toolCalls.length > 0;
+  const hasContent = textContent && textContent.trim() !== '';
+  
+  if (lastMessage?.role === "model" && hasToolCalls && !hasContent) {
+    lastMessage.parts.push(...toolCalls);
+  } else {
+    const parts = [];
+    
+    // 思维链处理（与 OpenAI 相同）
+    if (enableThinking) {
+      const cachedSig = getReasoningSignature(sessionId, actualModelName);
+      const thoughtSignature = cachedSig || getThoughtSignatureForModel(actualModelName);
+      parts.push({ text: ' ', thought: true });
+      parts.push({ text: ' ', thoughtSignature });
+    }
+    
+    if (hasContent) parts.push({ text: textContent.trimEnd() });
+    parts.push(...toolCalls);
+    
+    antigravityMessages.push({
+      role: "model",
+      parts
+    });
+  }
+}
+
+/**
+ * 处理 Claude tool_result 消息
+ */
+function handleClaudeToolResult(message, antigravityMessages) {
+  const content = message.content;
+  
+  if (!Array.isArray(content)) return;
+  
+  for (const item of content) {
+    if (item.type !== 'tool_result') continue;
+    
+    const toolUseId = item.tool_use_id;
+    
+    // 从之前的 model 消息中找到对应的 functionCall name
+    let functionName = '';
+    for (let i = antigravityMessages.length - 1; i >= 0; i--) {
+      if (antigravityMessages[i].role === 'model') {
+        const parts = antigravityMessages[i].parts;
+        for (const part of parts) {
+          if (part.functionCall && part.functionCall.id === toolUseId) {
+            functionName = part.functionCall.name;
+            break;
+          }
+        }
+        if (functionName) break;
+      }
+    }
+    
+    const lastMessage = antigravityMessages[antigravityMessages.length - 1];
+    
+    // 提取工具结果内容
+    let resultContent = '';
+    if (typeof item.content === 'string') {
+      resultContent = item.content;
+    } else if (Array.isArray(item.content)) {
+      resultContent = item.content
+        .filter(c => c.type === 'text')
+        .map(c => c.text)
+        .join('');
+    }
+    
+    const functionResponse = {
+      functionResponse: {
+        id: toolUseId,
+        name: functionName,
+        response: {
+          output: resultContent
+        }
+      }
+    };
+    
+    // 如果上一条消息是 user 且包含 functionResponse，则合并
+    if (lastMessage?.role === "user" && lastMessage.parts.some(p => p.functionResponse)) {
+      lastMessage.parts.push(functionResponse);
+    } else {
+      antigravityMessages.push({
+        role: "user",
+        parts: [functionResponse]
+      });
+    }
+  }
+}
+
+/**
+ * 将 Claude 消息转换为 Antigravity 格式
+ */
+function claudeMessageToAntigravity(claudeMessages, enableThinking, actualModelName, sessionId) {
+  const antigravityMessages = [];
+  
+  for (const message of claudeMessages) {
+    if (message.role === "user") {
+      // 检查是否包含 tool_result
+      const content = message.content;
+      if (Array.isArray(content) && content.some(item => item.type === 'tool_result')) {
+        handleClaudeToolResult(message, antigravityMessages);
+      } else {
+        const extracted = extractImagesFromClaudeContent(content);
+        handleClaudeUserMessage(extracted, antigravityMessages);
+      }
+    } else if (message.role === "assistant") {
+      handleClaudeAssistantMessage(message, antigravityMessages, enableThinking, actualModelName, sessionId);
+    }
+  }
+  
+  return antigravityMessages;
+}
+
+/**
+ * 将 Claude 工具格式转换为 Antigravity 格式
+ * Claude: { name, description, input_schema: {...} }
+ * Antigravity/Gemini: { functionDeclarations: [{ name, description, parameters: {...} }] }
+ */
+function convertClaudeToolsToAntigravity(claudeTools, sessionId, actualModelName) {
+  if (!claudeTools || claudeTools.length === 0) return [];
+  
+  return claudeTools.map((tool) => {
+    // 清洗参数
+    const rawParams = tool.input_schema || {};
+    const cleanedParams = cleanParameters(rawParams) || {};
+    
+    // 确保顶层是合法的 JSON Schema 对象
+    if (cleanedParams.type === undefined) {
+      cleanedParams.type = 'object';
+    }
+    if (cleanedParams.type === 'object' && cleanedParams.properties === undefined) {
+      cleanedParams.properties = {};
+    }
+    
+    const originalName = tool.name;
+    const safeName = sanitizeToolName(originalName);
+    
+    // 缓存映射
+    if (sessionId && actualModelName && safeName !== originalName) {
+      setToolNameMapping(sessionId, actualModelName, safeName, originalName);
+    }
+    
+    return {
+      functionDeclarations: [
+        {
+          name: safeName,
+          description: tool.description || '',
+          parameters: cleanedParams
+        }
+      ]
+    };
+  });
+}
+
+/**
+ * 生成 Claude 请求体并转换为 Antigravity 格式
+ */
+function generateClaudeRequestBody(claudeMessages, modelName, parameters, claudeTools, systemPrompt, token) {
+  const enableThinking = isEnableThinking(modelName);
+  const actualModelName = modelMapping(modelName);
+  
+  // 合并 system 指令
+  const baseSystem = config.systemInstruction || '';
+  let mergedSystem = '';
+  
+  if (config.useContextSystemPrompt && systemPrompt) {
+    const parts = [];
+    if (baseSystem.trim()) parts.push(baseSystem.trim());
+    if (systemPrompt.trim()) parts.push(systemPrompt.trim());
+    mergedSystem = parts.join('\n\n');
+  } else {
+    mergedSystem = baseSystem;
+  }
+  
+  const requestBody = {
+    project: token.projectId,
+    requestId: generateRequestId(),
+    request: {
+      contents: claudeMessageToAntigravity(claudeMessages, enableThinking, actualModelName, token.sessionId),
+      tools: convertClaudeToolsToAntigravity(claudeTools, token.sessionId, actualModelName),
+      toolConfig: {
+        functionCallingConfig: {
+          mode: "VALIDATED"
+        }
+      },
+      generationConfig: generateGenerationConfig(parameters, enableThinking, actualModelName),
+      sessionId: token.sessionId
+    },
+    model: actualModelName,
+    userAgent: "antigravity"
+  };
+  
+  // 只有当有 system 指令时才添加
+  if (mergedSystem) {
+    requestBody.request.systemInstruction = {
+      role: "user",
+      parts: [{ text: mergedSystem }]
+    };
+  }
+  
+  return requestBody;
+}
+
 export{
   generateRequestId,
   generateRequestBody,
   generateGeminiRequestBody,
+  generateClaudeRequestBody,
   prepareImageRequest,
   getDefaultIp
 }