github/claude-relay-service
1
0
Fork 0
mirror of https://github.com/Wei-Shaw/claude-relay-service.git synced 2026-01-23 18:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复 OpenAI 兼容路由的 Claude API 认证和 Function Calling 支持

Browse Source 
- 添加必需的系统消息 "You are Claude Code, Anthropic's official CLI for Claude."
- 修改 anthropic-beta header 为 OAuth-only 模式 (oauth-2025-04-20)
- 不再传递客户端 headers，使用固定的 4 个必需 headers
- 增强流式响应的 Function Calling 支持，正确处理 tool_use 事件
- 支持自定义 beta header 参数，允许不同路由使用不同的认证模式

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
shaw
2025-07-22 14:38:30 +08:00
parent 6ceed5c3ee
commit b2ad2a4a61
3 changed files with 46 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/services/claudeRelayService.js
View File

@@ -18,7 +18,7 @@ class ClaudeRelayService {
}
// 🚀 转发请求到Claude API
async relayRequest(requestBody, apiKeyData, clientRequest, clientResponse, clientHeaders) {
async relayRequest(requestBody, apiKeyData, clientRequest, clientResponse, clientHeaders, options = {}) {
let upstreamRequest = null;
try {
@@ -89,7 +89,8 @@ class ClaudeRelayService {
accessToken,
proxyAgent,
clientHeaders,
(req) => { upstreamRequest = req; }
(req) => { upstreamRequest = req; },
options
);
// 移除监听器（请求成功完成）
@@ -325,7 +326,7 @@ class ClaudeRelayService {
}
// 🔗 发送请求到Claude API
async _makeClaudeRequest(body, accessToken, proxyAgent, clientHeaders, onRequest) {
async _makeClaudeRequest(body, accessToken, proxyAgent, clientHeaders, onRequest, requestOptions = {}) {
return new Promise((resolve, reject) => {
const url = new URL(this.claudeApiUrl);
@@ -352,8 +353,10 @@ class ClaudeRelayService {
options.headers['User-Agent'] = 'claude-cli/1.0.53 (external, cli)';
}
if (this.betaHeader) {
options.headers['anthropic-beta'] = this.betaHeader;
// 使用自定义的 betaHeader 或默认值
const betaHeader = requestOptions?.betaHeader !== undefined ? requestOptions.betaHeader : this.betaHeader;
if (betaHeader) {
options.headers['anthropic-beta'] = betaHeader;
}
const req = https.request(options, (res) => {
@@ -445,7 +448,7 @@ class ClaudeRelayService {
}
// 🌊 处理流式响应（带usage数据捕获）
async relayStreamRequestWithUsageCapture(requestBody, apiKeyData, responseStream, clientHeaders, usageCallback, streamTransformer = null) {
async relayStreamRequestWithUsageCapture(requestBody, apiKeyData, responseStream, clientHeaders, usageCallback, streamTransformer = null, options = {}) {
try {
// 调试日志：查看API Key数据（流式请求）
logger.info('🔍 [Stream] API Key data received:', {
@@ -495,7 +498,7 @@ class ClaudeRelayService {
const proxyAgent = await this._getProxyAgent(accountId);
// 发送流式请求并捕获usage数据
return await this._makeClaudeStreamRequestWithUsageCapture(processedBody, accessToken, proxyAgent, clientHeaders, responseStream, usageCallback, accountId, sessionHash, streamTransformer);
return await this._makeClaudeStreamRequestWithUsageCapture(processedBody, accessToken, proxyAgent, clientHeaders, responseStream, usageCallback, accountId, sessionHash, streamTransformer, options);
} catch (error) {
logger.error('❌ Claude stream relay with usage capture failed:', error);
throw error;
@@ -503,7 +506,7 @@ class ClaudeRelayService {
}
// 🌊 发送流式请求到Claude API（带usage数据捕获）
async _makeClaudeStreamRequestWithUsageCapture(body, accessToken, proxyAgent, clientHeaders, responseStream, usageCallback, accountId, sessionHash, streamTransformer = null) {
async _makeClaudeStreamRequestWithUsageCapture(body, accessToken, proxyAgent, clientHeaders, responseStream, usageCallback, accountId, sessionHash, streamTransformer = null, requestOptions = {}) {
return new Promise((resolve, reject) => {
const url = new URL(this.claudeApiUrl);
@@ -530,8 +533,10 @@ class ClaudeRelayService {
options.headers['User-Agent'] = 'claude-cli/1.0.53 (external, cli)';
}
if (this.betaHeader) {
options.headers['anthropic-beta'] = this.betaHeader;
// 使用自定义的 betaHeader 或默认值
const betaHeader = requestOptions?.betaHeader !== undefined ? requestOptions.betaHeader : this.betaHeader;
if (betaHeader) {
options.headers['anthropic-beta'] = betaHeader;
}
const req = https.request(options, (res) => {
@@ -736,7 +741,7 @@ class ClaudeRelayService {
}
// 🌊 发送流式请求到Claude API
async _makeClaudeStreamRequest(body, accessToken, proxyAgent, clientHeaders, responseStream) {
async _makeClaudeStreamRequest(body, accessToken, proxyAgent, clientHeaders, responseStream, requestOptions = {}) {
return new Promise((resolve, reject) => {
const url = new URL(this.claudeApiUrl);
@@ -763,8 +768,10 @@ class ClaudeRelayService {
options.headers['User-Agent'] = 'claude-cli/1.0.53 (external, cli)';
}
if (this.betaHeader) {
options.headers['anthropic-beta'] = this.betaHeader;
// 使用自定义的 betaHeader 或默认值
const betaHeader = requestOptions?.betaHeader !== undefined ? requestOptions.betaHeader : this.betaHeader;
if (betaHeader) {
options.headers['anthropic-beta'] = betaHeader;
}
const req = https.request(options, (res) => {