@magpie-community/opencode-trae-auth 0.2.5 → 0.2.7

package/README.md

@@ -69,6 +69,18 @@ agent's `POST https://trae-api-cn.mchost.guru/api/agent/v3/llm_utils_chat`:
     `solo_work_lite`, SOLO's Work mode; `solo_agent`, the TRAE agent's;
     `solo_agent_lite`). If Trae answers 4001, 4023 or 1005, the plugin tries
     the others once and remembers which one worked.
+  - `session_id` is the conversation's, as the IDE keeps one per conversation:
+    the session magpie (or OpenCode) names the request by, else the chat's
+    first user message, which every later turn repeats, together with the
+    account the request is made as. Trae counts a conversation by it, so a
+    fresh one per request made one user send that takes several steps (a tool
+    result sent back, a retry, another function answering) count as a
+    conversation of its own — several rows in the usage detail, each under no
+    title. One conversation answered by two accounts (magpie failing over, or
+    rotating, inside it) is named twice then, as two conversations, so Trae
+    never reads one account's conversation as another's; that costs the two
+    halves of such a conversation their single row. `request_id` is the
+    request's own, fresh on every one.
 - **Headers:** TRAE SOLO CN 0.1.69's (`x-ide-version`/`x-app-version`
   0.1.69, version code 20260917; Trae offers a model only to clients new
   enough for it): `Authorization: Cloud-IDE-JWT …`, `X-Cloudide-Token`,
@@ -126,7 +138,15 @@ a bigger `context_window_tokens.max` is listed again as its Max,
 can name the `__max` model, a SOLO list the windows). It asks the
 `__max` model through the function that names it, with `max_tokens` set
 and `user_message_context.model_info.prompt_max_tokens` the window less
-it, as Max mode does.
+it, as Max mode does. The lists needn't agree on a model's *name*: chat_v3
+calls deepseek-v4.1-flash "DeepSeek-V4-Flash 正式版", as it once called
+DeepSeek-V4-Flash-Official, where the SOLO lists call it
+"DeepSeek-V4.1-Flash" (yetone/magpie#681). Trae's usage page books a
+request under the name of the list it was served through, so the plugin
+takes the name most of a model's lists give — the one Trae's own client is
+served under — and asks that model, and its Max, through a list carrying
+it. Trae renamed its official entries on 2026-10-09, leaving the name its
+lists mostly agree on the only thing that tells a misnamed entry apart.
 
 Reasoning: the models reason, but Trae's request takes no effort or
 thinking level, so the plugin lists no variants and a level an agent

package/index.mjs

@@ -9,7 +9,7 @@
 // The account's fetch also sends Trae CN's own pages on api.trae.cn
 // (/trae/api/…) as the account, for magpie's daily check-in (每日签到:
 // /trae/api/v2/ug/checkin_credits/status, then /claim).
-import { generateKeyPairSync, randomBytes, randomUUID, randomInt, sign } from "node:crypto"
+import { createHash, generateKeyPairSync, randomBytes, randomUUID, randomInt, sign } from "node:crypto"
 import { createServer, STATUS_CODES } from "node:http"
 
 const SITES = {
@@ -767,14 +767,72 @@ const nativeTools = (tools) =>
     return { type: "function", function: { name: f.name, description: f.description ?? "", parameters: typeof f.parameters === "string" ? f.parameters : JSON.stringify(f.parameters ?? {}) } }
   })
 
+// SESSION carries the conversation magpie (or OpenCode) names a request by,
+// from the chat.headers hook to the loader's fetch. It goes no further.
+const SESSION = "x-magpie-trae-session"
+
+// SALT is what the id is derived with, so it says nothing of the words it was
+// made from. New in each process, as Trae's own session ids are one session's
+// own within a run.
+const SALT = randomBytes(16).toString("hex")
+
+// conversationID is the conversation a chat belongs to, as the UUID the
+// request's session_id wants: the session magpie or OpenCode names (its
+// chat.headers hook), else the chat's first user message, which every later
+// turn of the conversation repeats. "" when neither names one — a body that
+// isn't a chat, or one with no user message in it — and the caller then makes
+// a fresh id per request, as before: a made-up conversation of the request's
+// own is the very splitting this is here to end.
+//
+// who is the account the request is made as (its uid, else its name), and it
+// is part of what the id is made of, so one conversation answered by two
+// accounts (magpie failing over, or rotating, inside it) is two conversations
+// as Trae is concerned — an id a fresh random one never linked, and one that
+// would let Trae read one conversation as another's. The cost is that such a
+// conversation's two halves are counted apart again, which is what it did
+// before this named one at all.
+function conversationID(chat, session, who = "") {
+  const account = String(who ?? "").trim()
+  if (session) return uuidOf("magpie-trae-session|" + account + "|" + session)
+  const first = (Array.isArray(chat?.messages) ? chat.messages : []).find((m) => m?.role === "user")
+  const text = partText(first?.content ?? "")
+  return text.trim() ? uuidOf("magpie-trae-first|" + account + "|" + text) : ""
+}
+
+// uuidOf is a version 4 UUID made of a hash's first 16 bytes: the shape
+// randomUUID() returns, since that is what session_id carried before.
+function uuidOf(s) {
+  const b = Buffer.from(createHash("sha256").update(SALT + "|" + s).digest().subarray(0, 16))
+  b[6] = (b[6] & 0x0f) | 0x40
+  b[8] = (b[8] & 0x3f) | 0x80
+  const h = b.toString("hex")
+  return `${h.slice(0, 8)}-${h.slice(8, 12)}-${h.slice(12, 16)}-${h.slice(16, 20)}-${h.slice(20)}`
+}
+
+// sessionOf takes the session a chat.headers hook named off the headers, with
+// the header: it goes no further than here. Its length is capped as magpie's
+// own session ids are, the id being derived from it anyway.
+function sessionOf(headers) {
+  const s = String(headers.get(SESSION) ?? "").trim()
+  headers.delete(SESSION)
+  return s.slice(0, 128)
+}
+
 // chatBody is the request for llm_utils_chat; modelName is the model the
 // function's list names for the config (its __dev one), when it named one,
 // and most the max_tokens it gives that model (0: none given), which a
 // request asking more is held to. A Max model (max: {base, window}) asks
 // the config base by its __max model, max_tokens always set, and gives
 // the prompt the room the window has beside it, as the IDE's Max mode does.
-function chatBody(req, fn, modelName, most = 0, max = null) {
-  const session = randomUUID()
+//
+// session is the conversation's id, made by conversationID, so every request
+// of one conversation carries one: Trae counts a conversation by the session
+// the requests name, and a request naming a new session each time is counted
+// as a conversation of its own — one user send that takes several steps (a
+// tool result sent back, a retry, another function answering) shows up in the
+// usage detail as several, each under no title at all. request_id stays a
+// fresh id per request, being that request's own.
+function chatBody(req, fn, modelName, most = 0, max = null, session = "") {
   const config = max?.base ?? req.model
   const body = {
     messages: traeMessages(req),
@@ -783,8 +841,8 @@ function chatBody(req, fn, modelName, most = 0, max = null) {
     model: config,
     ...(modelName ? { model_name: modelName } : {}),
     stream: true, // Trae answers in SSE either way
-    request_id: session,
-    session_id: session,
+    request_id: randomUUID(),
+    session_id: session || randomUUID(),
   }
   const asked = req.max_completion_tokens ?? req.max_tokens
   if (Number.isFinite(asked) && asked > 0) body.max_tokens = most > 0 ? Math.min(Math.floor(asked), most) : Math.floor(asked)
@@ -1984,24 +2042,40 @@ const makePlugin = (site) => async ({ client }) => {
     // deepseek-v4.1-flash "DeepSeek-V4-Flash 正式版", as it does
     // DeepSeek-V4-Flash-Official, where SOLO's lists call it
     // DeepSeek-V4.1-Flash (yetone/magpie#681) — Trae's usage page then books
-    // it under that name too. Such an entry is misnamed: its name is another
-    // model's in the same list, and other lists name it otherwise.
+    // a request under the name of the list it was served through. Such an
+    // entry is misnamed: its name is another model's in the same list, and
+    // other lists name it otherwise.
     const nameOf = (m) => String(m.display_config?.display_name || m.display_name || m.display_model_name || m.config_name)
-    const names = new Map() // id → the names its lists give it
+    const names = new Map() // id → the names its lists give it, and how many give each
     for (const fn of site.functions) {
       for (const m of lists[fn] ?? []) {
         if (!chatModel(m, site)) continue
-        const id = String(m.config_name)
-        names.set(id, (names.get(id) ?? new Set()).add(nameOf(m)))
+        const id = String(m.config_name), name = nameOf(m)
+        const by = names.get(id) ?? new Map()
+        by.set(name, (by.get(name) ?? 0) + 1)
+        names.set(id, by)
       }
     }
+    // the name most of a model's lists give it: the one Trae's own client is
+    // served under, and so the one its usage page books it as. Trae renames
+    // entries when it likes, and its 2026-10-09 rename of
+    // DeepSeek-V4-Flash-Official left chat_v3 the only list still calling
+    // deepseek-v4.1-flash by another's name — which neither signal above
+    // catches. A tie leaves the name to its entry.
+    const agreed = new Map() // id → that name, "" when the lists tie
+    for (const [id, by] of names) {
+      const top = Math.max(...by.values())
+      const most = [...by].filter(([, n]) => n === top)
+      agreed.set(id, most.length === 1 ? most[0][0] : "")
+    }
     const misnamed = (fn, m) => {
       const id = String(m.config_name), name = nameOf(m)
       return names.get(id).size > 1 && (lists[fn] ?? []).some((o) => chatModel(o, site) && String(o.config_name) !== id && nameOf(o) === name)
     }
     // a function's entry is worth more when it names a __dev model, then
-    // when it isn't misnamed; the first of the best is taken
-    const worth = (fn, m) => (devModel(m) ? 2 : 0) + (misnamed(fn, m) ? 0 : 1)
+    // when it isn't misnamed, then when it carries the name its lists agree
+    // on; the first of the best is taken
+    const worth = (fn, m) => (devModel(m) ? 2 : 0) + (misnamed(fn, m) ? 0 : 1) + (nameOf(m) === agreed.get(String(m.config_name)) ? 1 : 0)
     const out = new Map()
     for (const fn of site.functions) {
       for (const m of lists[fn] ?? []) {
@@ -2017,8 +2091,11 @@ const makePlugin = (site) => async ({ client }) => {
       if (dev) modelNames.set(id, { fn, name: dev.model_name, most: Number(dev.max_tokens) || 0 })
       else modelNames.delete(id)
       if (out.has(id + MAX)) continue
-      // its Max is asked through the function whose entry names the __max model
-      const entries = [{ m, fn }, ...site.functions.flatMap((f) => (lists[f] ?? []).filter((o) => o !== m && chatModel(o, site) && String(o.config_name) === id).map((o) => ({ m: o, fn: f })))]
+      // its Max is asked through the function whose entry names the __max
+      // model, the entries carrying the agreed name first: Trae books the
+      // Max under that name too
+      const plain = (e) => (nameOf(e.m) === agreed.get(id) ? 0 : 1)
+      const entries = [{ m, fn }, ...site.functions.flatMap((f) => (lists[f] ?? []).filter((o) => o !== m && chatModel(o, site) && String(o.config_name) === id).map((o) => ({ m: o, fn: f })))].sort((a, b) => plain(a) - plain(b))
       const max = maxModel(...entries.map((e) => e.m))
       maxes.set(id, max)
       if (max) {
@@ -2078,6 +2155,14 @@ const makePlugin = (site) => async ({ client }) => {
         models: { ...site.models, ...(was.models ?? {}) },
       }
     },
+    // the conversation's id, which the requests below carry as their
+    // session_id: magpie's (or OpenCode's) session for this request, when it
+    // names one
+    "chat.headers": async (input, output) => {
+      const id = input?.model?.providerID ?? input?.provider?.info?.id
+      if (id !== site.id || !input?.sessionID) return
+      output.headers[SESSION] = String(input.sessionID).slice(0, 128)
+    },
     provider: {
       id: site.id,
       // the account's own list, when Trae answers; else the list above
@@ -2156,11 +2241,16 @@ const makePlugin = (site) => async ({ client }) => {
               if (outer.aborted) ac.abort()
               else outer.addEventListener("abort", () => ac.abort(), { once: true })
             }
+            // the conversation this request belongs to, from the chat.headers
+            // hook (magpie's session) or the chat's own first user message, as
+            // the account answering it, so a failover inside one conversation
+            // is not one Trae reads as another's
+            const session = conversationID(req, sessionOf(new Headers(init.headers ?? r0?.headers ?? {})), who)
             for (const fn of [...new Set(fns)]) {
               const res = await fetch(apiOf(a, site) + "/api/agent/v3/llm_utils_chat", {
                 method: "POST",
                 headers: ideHeaders(a, site, { Accept: "text/event-stream", "X-Request-ID": randomUUID() }),
-                body: JSON.stringify(chatBody(req, fn, named?.fn === fn || max ? named.name : "", named?.fn === fn || max ? named.most : 0, max)),
+                body: JSON.stringify(chatBody(req, fn, named?.fn === fn || max ? named.name : "", named?.fn === fn || max ? named.most : 0, max, session)),
                 signal: ac.signal,
               })
               if (!res.ok) {
@@ -2211,4 +2301,4 @@ const makePlugin = (site) => async ({ client }) => {
 export const TraeCNAuthPlugin = makePlugin(SITES["trae-cn"])
 export const TraeGlobalAuthPlugin = makePlugin(SITES["trae-global"])
 
-export const _internal = { HOSTS: SITES["trae-cn"].hosts, MODELS: SITES["trae-cn"].models, TextTools, NativeCalls, looseJSON, glmCall, toolNamed, traeMessages, chatBody, credits, dollarUsageOf, whenOf, newDevice, proofOf, SITES }
+export const _internal = { HOSTS: SITES["trae-cn"].hosts, MODELS: SITES["trae-cn"].models, TextTools, NativeCalls, looseJSON, glmCall, toolNamed, traeMessages, chatBody, credits, dollarUsageOf, whenOf, newDevice, proofOf, SITES, SESSION, conversationID }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@magpie-community/opencode-trae-auth",
-  "version": "0.2.5",
+  "version": "0.2.7",
   "description": "OpenCode provider plugins for Trae: Trae CN (trae.cn) and Trae Global (trae.ai), free tier included, with Trae's own browser sign-in",
   "type": "module",
   "main": "./index.mjs",