@magpie-community/opencode-trae-auth 0.1.0 → 0.1.2

package/README.md

@@ -32,9 +32,10 @@ agent's `POST https://trae-api-cn.mchost.guru/api/agent/v3/llm_utils_chat`:
 - **Request:**
   - The messages go as `{role, content: [{type: "text", text}]}`.
   - The model goes as `config_name` and `model`.
-  - The function is `chat_v3`, the classic IDE's. If Trae answers 4001, 4023
-    or 1005, the plugin tries `solo_work_lite` (SOLO's) once and remembers
-    which one worked.
+  - The function is the one whose model list has the model (`chat_v3`, the
+    classic IDE's, first; else `solo_work_lite`, SOLO's Work mode; else
+    `solo_agent`, the TRAE agent's). If Trae answers 4001, 4023 or 1005,
+    the plugin tries the others once and remembers which one worked.
 - **Headers:** the IDE's: `Authorization: Cloud-IDE-JWT …`,
   `X-Cloudide-Token`, `x-app-id`, `x-ide-version`, the device headers and
   `x-uid`.
@@ -54,8 +55,10 @@ agent's `POST https://trae-api-cn.mchost.guru/api/agent/v3/llm_utils_chat`:
   - Earlier calls and their results go back in as text.
   - Images aren't sent.
 
-The model list is the account's own, from `/api/ide/v1/get_detail_param`.
-When that can't be read, the plugin uses the models Trae CN's `chat_v3` is
+The model list is the account's own, from `/api/ide/v1/get_detail_param`,
+asked for each of those functions and put together: SOLO and the TRAE
+agent list models `chat_v3` doesn't (deepseek-v4.1-flash is the TRAE
+agent's, `solo_agent`). When neither can be read, the plugin uses the models Trae CN's `chat_v3` is
 known to serve: GLM-5.2, GLM-5, Kimi K2.6, Qwen 3.7 Plus, DeepSeek V4 Pro
 and DeepSeek V4 Flash.
 

package/index.mjs

@@ -26,8 +26,10 @@ const SIGN_IN_TIMEOUT = 10 * 60 * 1000
 const EARLY_MS = 2 * 60 * 1000 // a token this close to its end is renewed before a request
 const LEAD_MS = 10 * 60 * 1000 // and this close, magpie renews it ahead of time (auth.refresh)
 const DAY = 24 * 3600 * 1000
-// the IDE's chat functions: the classic IDE's agent, then SOLO's
-const FUNCTIONS = ["chat_v3", "solo_work_lite"]
+// the IDE's chat functions: the classic IDE's agent, then SOLO's Work
+// mode, then the TRAE agent (solo_agent): Trae CN 3.3.104's agent-type map
+// files its models under "trae", and only that one has deepseek-v4.1-flash
+const FUNCTIONS = ["chat_v3", "solo_work_lite", "solo_agent"]
 
 const MODEL = { attachment: false, tool_call: true, reasoning: true, temperature: true, limit: { context: 128_000, output: 32_000 }, modalities: { input: ["text"], output: ["text"] } }
 // what Trae CN's chat_v3 is known to serve; the live list replaces it
@@ -689,7 +691,7 @@ async function openai(req, it) {
   return new Response(stream, { status: 200, headers: { "content-type": "text/event-stream", "cache-control": "no-cache" } })
 }
 
-// a model or function this account's chat_v3 doesn't take: SOLO's may
+// a model or function this account's chat_v3 doesn't take: another may
 const wrongFunction = (code) => ["4001", "4023", "1005"].includes(String(code))
 
 // ---- usage ------------------------------------------------------------------------
@@ -727,6 +729,8 @@ export const TraeCNAuthPlugin = async ({ client }) => {
   const renewed = new Map()
   // the chat function that served each account last
   const fnOf = new Map()
+  // the first chat function whose model list has each model
+  const listedBy = new Map()
 
   const save = async (auth) => {
     try {
@@ -814,10 +818,10 @@ export const TraeCNAuthPlugin = async ({ client }) => {
     return out
   }
 
-  // liveModels is the account's model list, as the IDE asks for it
-  const liveModels = async (a) => {
+  // listOf is one function's model list, as the IDE asks for it
+  const listOf = async (a, fn) => {
     const r = await post(a, apiOf(a) + "/api/ide/v1/get_detail_param", {
-      function: FUNCTIONS[0], config_names: null, need_prompt: false, current_config_info: null, poly_prompt: true, mode_type: null, agent_type: null,
+      function: fn, config_names: null, need_prompt: false, current_config_info: null, poly_prompt: true, mode_type: null, agent_type: null,
     })
     if (r.status === 401 || lapsedCode(errorOf(r.v).code)) throw new Expired("Trae CN's sign-in has expired; sign in again")
     if (r.status !== 200) throw new Error(`Trae CN models: ${statusLine(r.status)}`)
@@ -825,6 +829,28 @@ export const TraeCNAuthPlugin = async ({ client }) => {
     return list.filter((m) => m?.config_name)
   }
 
+  // liveModels is the account's model list: every chat function's, since
+  // SOLO lists models the classic IDE's chat_v3 doesn't (deepseek-v4.1-flash
+  // came to the TRAE agent's first, yetone/magpie#681); a model listed by one function
+  // only is asked through that one. One function's list failing leaves the
+  // others'; all failing is the error.
+  const liveModels = async (a) => {
+    const got = await Promise.allSettled(FUNCTIONS.map((fn) => listOf(a, fn)))
+    const expired = got.find((g) => g.status === "rejected" && g.reason instanceof Expired)
+    if (expired) throw expired.reason
+    if (got.every((g) => g.status === "rejected")) throw got[0].reason
+    const out = new Map()
+    got.forEach((g, i) => {
+      if (g.status !== "fulfilled") return
+      for (const m of g.value) {
+        const id = String(m.config_name)
+        if (!out.has(id)) out.set(id, { m, fn: FUNCTIONS[i] })
+      }
+    })
+    for (const [id, { fn }] of out) listedBy.set(id, fn)
+    return [...out.values()].map((x) => x.m)
+  }
+
   const modelOf = (provider, m) => {
     const id = String(m.config_name)
     const was = provider.models?.[id] ?? {}
@@ -889,7 +915,9 @@ export const TraeCNAuthPlugin = async ({ client }) => {
             }
             const signIn = a.renewed ? "renewed" : "kept"
             const who = a.uid || a.name
-            const fns = fnOf.has(who) ? [fnOf.get(who), ...FUNCTIONS.filter((f) => f !== fnOf.get(who))] : FUNCTIONS
+            // the function that lists the model first, then the one that
+            // served this account last, then the rest
+            const fns = [...new Set([listedBy.get(String(req.model)), fnOf.get(who), ...FUNCTIONS].filter(Boolean))]
             let last
             for (const fn of fns) {
               const res = await fetch(apiOf(a) + "/api/agent/v3/llm_utils_chat", {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@magpie-community/opencode-trae-auth",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "OpenCode provider plugin for Trae CN (trae.cn): your Trae CN account's models, free tier included, with Trae's own browser sign-in",
   "type": "module",
   "main": "./index.mjs",