opencode-thinking-fix 2.0.0 → 3.1.4

package/proxy/core.js

@@ -0,0 +1,255 @@
+'use strict'
+
+import { createParser } from 'eventsource-parser'
+import crypto from 'node:crypto'
+import { homedir } from 'node:os'
+import { join, dirname } from 'node:path'
+import { appendFileSync, mkdirSync } from 'node:fs'
+
+export function deriveSessionId(modelName, parsedBody, authHeader) {
+  const messages = parsedBody?.messages
+  if (!Array.isArray(messages)) return ''
+  const firstUser = messages.find((m) => m && m.role === 'user')
+  if (!firstUser) return ''
+  const text = typeof firstUser.content === 'string'
+    ? firstUser.content
+    : Array.isArray(firstUser.content)
+      ? firstUser.content.map((p) => (p && typeof p.text === 'string' ? p.text : '')).join('')
+      : ''
+  if (!text) return ''
+  return 'derived:' + crypto.createHash('sha256')
+    .update(`${authHeader || ''}||${modelName}||${text}`).digest('hex').slice(0, 32)
+}
+
+const LOG_FILE = process.env.LOG_FILE === ''
+  ? ''
+  : (process.env.LOG_FILE || join(homedir(), '.local', 'share', 'opencode', 'thinking-fix.log'))
+
+export function writeLog(entry) {
+  if (!LOG_FILE) return
+  try {
+    mkdirSync(dirname(LOG_FILE), { recursive: true })
+    appendFileSync(LOG_FILE, JSON.stringify({ ts: new Date().toISOString(), ...entry }) + '\n', 'utf8')
+  } catch (error) {
+    console.error('[Proxy] log write error:', error instanceof Error ? error.message : error)
+  }
+}
+
+const DEBUG = process.env.DEBUG === '1'
+
+export const ROUTES = {
+  // R1 aliases: reasoning must not be echoed.
+  'deepseek-r1': { base: 'https://api.deepseek.com', reasoningKey: null },
+  'deepseek-v4-pro': { base: 'https://api.deepseek.com', reasoningKey: 'reasoning_content' },
+  'deepseek-v4-flash': { base: 'https://api.deepseek.com', reasoningKey: 'reasoning_content' },
+  'deepseek-chat': { base: 'https://api.deepseek.com', reasoningKey: 'reasoning_content' },
+  'deepseek-reasoner': { base: 'https://api.deepseek.com', reasoningKey: null },
+  deepseek: { base: 'https://api.deepseek.com', reasoningKey: 'reasoning_content' },
+  kimi: { base: 'https://api.moonshot.ai/v1', reasoningKey: 'reasoning_content' },
+  moonshot: { base: 'https://api.moonshot.ai/v1', reasoningKey: 'reasoning_content' },
+  glm: { base: 'https://open.bigmodel.cn/api/paas/v4', reasoningKey: 'reasoning_content' },
+  zhipu: { base: 'https://open.bigmodel.cn/api/paas/v4', reasoningKey: 'reasoning_content' },
+  minimax: { base: 'https://api.minimax.io/v1', reasoningKey: 'reasoning_details' },
+  mimo: { base: 'https://api.xiaomimimo.com/v1', reasoningKey: 'reasoning_content' },
+  qwen: { base: 'https://dashscope-intl.aliyuncs.com/compatible-mode/v1', reasoningKey: null },
+  gpt: { base: 'https://api.openai.com', reasoningKey: null },
+  o1: { base: 'https://api.openai.com', reasoningKey: null },
+  claude: { base: 'https://api.anthropic.com', reasoningKey: null },
+  anthropic: { base: 'https://api.anthropic.com', reasoningKey: null },
+  gemini: { base: 'https://generativelanguage.googleapis.com/v1beta/openai', reasoningKey: null },
+  llama: { base: 'https://api.together.xyz', reasoningKey: null },
+  mistral: { base: 'https://api.mistral.ai', reasoningKey: null },
+}
+
+export const DEFAULT_ROUTE = { base: 'https://api.deepseek.com', reasoningKey: null }
+
+export function route(modelName) {
+  if (!modelName) return DEFAULT_ROUTE
+  const lower = modelName.toLowerCase()
+  for (const [prefix, entry] of Object.entries(ROUTES)) {
+    if (lower.startsWith(prefix)) return entry
+  }
+  return DEFAULT_ROUTE
+}
+
+// Models served by opencode-go over the Anthropic /v1/messages wire format.
+const ANTHROPIC_WIRE_PREFIXES = ['minimax', 'qwen']
+
+export function fixedUpstreamRoute(modelName, requestUrl = '') {
+  const stripped = (modelName || '').toLowerCase().replace(/^opencode-go\//, '')
+  if (String(requestUrl).includes('/messages') && ANTHROPIC_WIRE_PREFIXES.some((p) => stripped.startsWith(p))) {
+    return { base: '', reasoningKey: 'anthropic' }
+  }
+  // Unconditional 'reasoning' for go mode: matches current working behavior
+  // for chat/completions models; /responses bodies have no `messages` array
+  // and pass through untouched.
+  return { base: '', reasoningKey: 'reasoning' }
+}
+
+export class LRUCache {
+  constructor(capacity) { this.cache = new Map(); this.capacity = capacity }
+  get(key) {
+    if (!this.cache.has(key)) return undefined
+    const value = this.cache.get(key)
+    this.cache.delete(key); this.cache.set(key, value)
+    return value
+  }
+  set(key, value) {
+    this.cache.delete(key)
+    this.cache.set(key, value)
+    while (this.cache.size > this.capacity) {
+      const oldest = this.cache.keys().next().value
+      this.cache.delete(oldest)
+      if (DEBUG) console.log(`[Cache] evicted session ${String(oldest).slice(-8)}`)
+    }
+  }
+  put(key, value) { this.set(key, value) }
+}
+
+const parsedCapacity = parseInt(process.env.CACHE_CAPACITY || '500', 10)
+const CACHE_CAPACITY = Number.isFinite(parsedCapacity) && parsedCapacity >= 0 ? parsedCapacity : 500
+const parsedMaxTurns = parseInt(process.env.MAX_TURNS_PER_SESSION || '200', 10)
+const MAX_TURNS_PER_SESSION = Number.isFinite(parsedMaxTurns) && parsedMaxTurns > 0 ? parsedMaxTurns : 200
+const parsedMaxReasoning = parseInt(process.env.MAX_REASONING_CHARS || '200000', 10)
+const MAX_REASONING_CHARS = Number.isFinite(parsedMaxReasoning) && parsedMaxReasoning > 0 ? parsedMaxReasoning : 200000
+
+class BoundedSessionMap extends Map {
+  set(key, value) {
+    super.set(key, typeof value === 'string' ? value.slice(0, MAX_REASONING_CHARS) : value)
+    while (this.size > MAX_TURNS_PER_SESSION) this.delete(this.keys().next().value)
+    return this
+  }
+}
+
+export const cache = new LRUCache(CACHE_CAPACITY)
+export const stats = { sessions: 0, hits: 0, misses: 0, stored: 0, requests: 0, passthrough: 0 }
+
+export function getSessionCache(sessionId) {
+  let sessionMap = cache.get(sessionId)
+  if (!sessionMap) { sessionMap = new BoundedSessionMap(); cache.set(sessionId, sessionMap); stats.sessions++ }
+  return sessionMap
+}
+
+export function patchRequestBody(body, sessionId, reasoningKey) {
+  const emptyReport = { assistantTurns: 0, missingText: 0, missingReasoning: 0, hits: 0, misses: 0, turns: [] }
+  if (!reasoningKey || typeof sessionId !== 'string' || !sessionId.trim() || sessionId.trim() === 'unknown') return { body, assistantCount: 0, modified: false, report: emptyReport }
+  let parsed
+  try { parsed = JSON.parse(body) } catch { return { body, assistantCount: 0, modified: false, report: emptyReport } }
+  if (!Array.isArray(parsed.messages)) return { body, assistantCount: 0, modified: false, report: emptyReport }
+
+  const sessionCache = getSessionCache(sessionId)
+  const report = { ...emptyReport, turns: [] }
+  let assistantIndex = 0
+  let modified = false
+  for (const msg of parsed.messages) {
+    if (msg.role !== 'assistant') continue
+    report.assistantTurns++
+    if (msg.content == null || msg.content === '') report.missingText++
+    const cached = sessionCache.get(assistantIndex)
+    let missingReasoning = false
+    if (reasoningKey === 'anthropic') {
+      // Anthropic wire: reasoning lives inside content[] as a thinking block.
+      const hasThinking = Array.isArray(msg.content) && msg.content.some((b) => b && (b.type === 'thinking' || b.type === 'redacted_thinking'))
+      missingReasoning = !hasThinking
+      if (missingReasoning) report.missingReasoning++
+      if (cached) {
+        if (Array.isArray(msg.content)) {
+          if (!hasThinking) { msg.content.unshift({ type: 'thinking', thinking: cached }); stats.hits++; report.hits++; report.turns.push({ index: assistantIndex, fields: ['content'], source: 'hit' }); modified = true }
+        } else if (typeof msg.content === 'string') {
+          msg.content = [{ type: 'thinking', thinking: cached }, { type: 'text', text: msg.content }]
+          stats.hits++; report.hits++; report.turns.push({ index: assistantIndex, fields: ['content'], source: 'hit' }); modified = true
+        }
+      } else {
+        // No hard-400 contract on MiniMax/Qwen anthropic endpoints; never fabricate.
+        if (missingReasoning) { stats.misses++; report.misses++ }
+      }
+    } else if (reasoningKey === 'reasoning_details') {
+      const hasDetails = Array.isArray(msg.reasoning_details) ? msg.reasoning_details.length > 0 : !!msg.reasoning_details
+      missingReasoning = !hasDetails && !msg.reasoning_content
+      if (missingReasoning) report.missingReasoning++
+      if (!hasDetails && !msg.reasoning_content && cached) {
+        msg.reasoning_details = [{ text: cached, type: 'thinking' }]; stats.hits++; report.hits++; report.turns.push({ index: assistantIndex, fields: ['reasoning_details'], source: 'hit' }); modified = true
+      } else if (missingReasoning) {
+        stats.misses++; report.misses++
+      }
+    } else if (!msg[reasoningKey] || msg[reasoningKey] === '') {
+      missingReasoning = true
+      report.missingReasoning++
+      if (cached) { msg[reasoningKey] = cached; stats.hits++ }
+      else { msg[reasoningKey] = ''; stats.misses++ }
+      if (cached) { report.hits++; report.turns.push({ index: assistantIndex, fields: [reasoningKey], source: 'hit' }) }
+      else { report.misses++; report.turns.push({ index: assistantIndex, fields: [reasoningKey], source: 'miss' }) }
+      modified = true
+    }
+    assistantIndex++
+  }
+  return { body: modified ? JSON.stringify(parsed) : body, assistantCount: assistantIndex, modified, report }
+}
+
+export function extractReasoningFromJson(text) {
+  try {
+    const parsed = JSON.parse(text)
+    // Anthropic buffered format: content[] with thinking blocks.
+    if (Array.isArray(parsed?.content)) {
+      const thinking = parsed.content
+        .filter((b) => b && b.type === 'thinking' && typeof b.thinking === 'string')
+        .map((b) => b.thinking)
+        .join('\n')
+      if (thinking) return thinking
+    }
+    // OpenAI buffered format.
+    const msg = parsed?.choices?.[0]?.message
+    if (msg) {
+      if (typeof msg.reasoning_content === 'string' && msg.reasoning_content) return msg.reasoning_content
+      if (typeof msg.reasoning === 'string' && msg.reasoning) return msg.reasoning
+      if (Array.isArray(msg.reasoning_details)) {
+        const reasoning = msg.reasoning_details.map((d) => d.text || '').join('')
+        if (reasoning) return reasoning
+      }
+    }
+  } catch { /* not JSON */ }
+  return null
+}
+
+export function createStreamParser(sessionId, baseIndex, onComplete) {
+  let reasoningBuffer = ''
+  const parser = createParser({
+    maxBufferSize: 1024 * 1024,
+    onEvent(event) {
+      if (!event.data || event.data === '[DONE]') {
+        if (reasoningBuffer) { onComplete(baseIndex, reasoningBuffer); reasoningBuffer = '' }
+        return
+      }
+      try {
+        const parsed = JSON.parse(event.data)
+        // Anthropic SSE: terminate on message_stop.
+        if (event.event === 'message_stop' || parsed.type === 'message_stop') {
+          if (reasoningBuffer) { onComplete(baseIndex, reasoningBuffer); reasoningBuffer = '' }
+          return
+        }
+        // Anthropic SSE: thinking deltas carry delta.thinking (type thinking_delta).
+        if (event.event === 'content_block_delta' || parsed.type === 'content_block_delta') {
+          const delta = parsed.delta
+          if (delta && typeof delta.thinking === 'string') reasoningBuffer += delta.thinking
+          return
+        }
+        // OpenAI SSE.
+        const delta = parsed.choices?.[0]?.delta
+        if (!delta) return
+        if (typeof delta.reasoning_content === 'string') reasoningBuffer += delta.reasoning_content
+        if (typeof delta.reasoning === 'string') reasoningBuffer += delta.reasoning
+        if (Array.isArray(delta.reasoning_details)) {
+          for (const detail of delta.reasoning_details) if (typeof detail.text === 'string') reasoningBuffer += detail.text
+        }
+        if (parsed.choices?.[0]?.finish_reason && reasoningBuffer) {
+          onComplete(baseIndex, reasoningBuffer); reasoningBuffer = ''
+        }
+      } catch { /* malformed SSE chunk, ignore */ }
+    },
+    onError(error) { if (DEBUG) console.error(`[Proxy] SSE parse error (${sessionId.slice(-8)}):`, error.message) },
+  })
+  return {
+    feed(chunk) { parser.feed(chunk.toString('utf8')) },
+    flush() { if (reasoningBuffer) { onComplete(baseIndex, reasoningBuffer); reasoningBuffer = '' } },
+  }
+}