fix: 调整 OpenAI 兼容请求 User-Agent

Author: XavierWangHX

src/services/api/openai/__tests__/thinking.test.ts

@@ -1,5 +1,41 @@
 import { describe, expect, test, beforeEach, afterEach } from 'bun:test'
 import { isOpenAIThinkingEnabled, buildOpenAIRequestBody } from '../requestBody.js'
+import { getOpenAICompatibleDefaultHeaders } from '../client.js'
+
+describe('getOpenAICompatibleDefaultHeaders', () => {
+  const originalEnv = {
+    OPENAI_USER_AGENT: process.env.OPENAI_USER_AGENT,
+  }
+
+  beforeEach(() => {
+    delete process.env.OPENAI_USER_AGENT
+  })
+
+  afterEach(() => {
+    if (originalEnv.OPENAI_USER_AGENT === undefined) {
+      delete process.env.OPENAI_USER_AGENT
+    } else {
+      process.env.OPENAI_USER_AGENT = originalEnv.OPENAI_USER_AGENT
+    }
+  })
+
+  test('does not override the official OpenAI default path', () => {
+    expect(getOpenAICompatibleDefaultHeaders()).toBeUndefined()
+  })
+
+  test('uses a generic user agent for OpenAI-compatible base URLs', () => {
+    expect(getOpenAICompatibleDefaultHeaders('https://sub2api.example/v1')).toEqual({
+      'User-Agent': 'opencode/1.14.22 (win32 10.0.26200; x64) ai-sdk/provider-utils/4.0.23 runtime/bun/1.3.13',
+    })
+  })
+
+  test('allows explicit user agent override', () => {
+    process.env.OPENAI_USER_AGENT = 'Mozilla/5.0 test-client'
+    expect(getOpenAICompatibleDefaultHeaders('https://sub2api.example/v1')).toEqual({
+      'User-Agent': 'Mozilla/5.0 test-client',
+    })
+  })
+})
 
 describe('isOpenAIThinkingEnabled', () => {
   const originalEnv = {
@@ -236,4 +272,4 @@ describe('buildOpenAIRequestBody — thinking params', () => {
     expect(body.tools).toBeUndefined()
     expect(body.tool_choice).toBeUndefined()
   })
-})
+})

src/services/api/openai/client.ts

@@ -2,7 +2,6 @@ import OpenAI from 'openai'
 import { openaiAdapter } from 'src/services/providerUsage/adapters/openai.js'
 import { updateProviderBuckets } from 'src/services/providerUsage/store.js'
 import { getProxyFetchOptions } from 'src/utils/proxy.js'
-import { isEnvTruthy } from 'src/utils/envUtils.js'
 
 /**
  * Environment variables:
@@ -15,6 +14,22 @@ import { isEnvTruthy } from 'src/utils/envUtils.js'
 
 let cachedClient: OpenAI | null = null
 
+const DEFAULT_COMPATIBLE_USER_AGENT = 'opencode/1.14.22 (win32 10.0.26200; x64) ai-sdk/provider-utils/4.0.23 runtime/bun/1.3.13'
+
+/**
+ * Some OpenAI-compatible gateways sit behind Cloudflare/sub2api rules that
+ * block SDK-looking user agents such as `OpenAI/JS ...`. Keep the official
+ * OpenAI path unchanged, but make custom base URLs look like a generic client.
+ */
+export function getOpenAICompatibleDefaultHeaders(baseURL?: string): Record<string, string> | undefined {
+  const userAgent = process.env.OPENAI_USER_AGENT || (baseURL ? DEFAULT_COMPATIBLE_USER_AGENT : undefined)
+  if (!userAgent) return undefined
+
+  return {
+    'User-Agent': userAgent,
+  }
+}
+
 /**
  * Wrap a fetch so that every response's rate-limit headers are fed into the
  * provider usage store. Errors in parsing must never break the request.
@@ -49,10 +64,12 @@ export function getOpenAIClient(options?: {
 
   const baseFetch = options?.fetchOverride ?? (globalThis.fetch as typeof fetch)
   const wrappedFetch = wrapFetchForUsage(baseFetch)
+  const defaultHeaders = getOpenAICompatibleDefaultHeaders(baseURL)
 
   const client = new OpenAI({
     apiKey,
     ...(baseURL && { baseURL }),
+    ...(defaultHeaders && { defaultHeaders }),
     maxRetries: options?.maxRetries ?? 0,
     timeout: parseInt(process.env.API_TIMEOUT_MS || String(600 * 1000), 10),
     dangerouslyAllowBrowser: true,