mirror of
https://gitcode.com/JianFeeeee/ModelRouter.git
synced 2026-11-07 17:57:23 +00:00
Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 58073a6cbe | |||
| a5923b3797 |
@ -55,3 +55,31 @@ func TestBuddyCatalogFilterDropsNonChatModels(t *testing.T) {
|
||||
}
|
||||
t.Log("✓ buddy 目录过滤四类规则齐备，且未误伤 default-model 类别名")
|
||||
}
|
||||
|
||||
// ★ 已实测**不可调用**的模型必须也从目录里排除 —— 包括 agentRef 例外路径。
|
||||
//
|
||||
// 真实缺陷：`hy4-preview-x` 被服务端 agents 引用，于是 agentRef 例外把它放回
|
||||
// 目录；而它对上游**恒回 11102**（`model service info not found`，加/不加
|
||||
// reasoning_effort 都一样 ⇒ 与 effort 无关，是后端未开放）。
|
||||
// 用户选中就 400，错误文案指向模型 ⇒ 极易误判"模型坏了"。
|
||||
//
|
||||
// 参考用**白名单重建**（product.ts 的兜底表里没有它）达到同样效果。
|
||||
func TestBuddyCatalogDropsKnownUncallable(t *testing.T) {
|
||||
src, err := readBundledChannels(t)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
s := string(src)
|
||||
|
||||
if !strings.Contains(s, "local NOT_CALLABLE = {") {
|
||||
t.Fatal("必须有已实测不可调用的排除表（等价于参考的白名单）")
|
||||
}
|
||||
if !strings.Contains(s, `["hy4-preview-x"] = true`) {
|
||||
t.Error("hy4-preview-x 实测恒 11102，必须排除")
|
||||
}
|
||||
// 闸门必须生效在**所有**入口 —— 尤其是 agentRef 例外之后
|
||||
if !strings.Contains(s, "if NOT_CALLABLE[id] then return false end") {
|
||||
t.Error("NOT_CALLABLE 闸门必须在 buddyIsChatModel 内（覆盖 models 与 agentRef 两条路径）")
|
||||
}
|
||||
t.Log("✓ 不可调用模型已排除，且 agentRef 例外也过闸")
|
||||
}
|
||||
|
||||
73
internal/core/channels_lua_locals_test.go
Normal file
73
internal/core/channels_lua_locals_test.go
Normal file
@ -0,0 +1,73 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"os/exec"
|
||||
"regexp"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// ★ channels.lua 必须留出顶层 local 余量 —— 否则**下一次顶层改动就编译失败**。
|
||||
//
|
||||
// 实测事实（2026-10-16）：该文件的主 chunk 曾**顶到 Lua 上限 200**，
|
||||
// 往顶部加**哪怕 1 个**顶层 local 就报
|
||||
//
|
||||
// main function has more than 200 local variables
|
||||
//
|
||||
// 而这炸的不是某个功能 —— 是**整个渠道插件**（插件加载失败 ⇒ 全渠道不可用）。
|
||||
//
|
||||
// ⚠️ 判据为什么用 luajit 而不是 luac：
|
||||
//
|
||||
// 内核链接的是 libluajit-5.1.so.2；而 `luac`（Lua 5.4）对同一文件给出
|
||||
// **不同的**结论。用错工具会得到假绿。
|
||||
//
|
||||
// ⚠️ 判据为什么是「注入 N 个 local 仍能编译」而不是「数一遍 local」：
|
||||
//
|
||||
// 主 chunk 的 local 数受**作用域嵌套**影响，静态计数不可靠（同一个文件
|
||||
// 用不同脚本数出 228/232/248 三个数）。而"能不能再加"是**决定性**的，
|
||||
// 也正是我们真正需要的性质。
|
||||
func TestChannelsLuaHasLocalHeadroom(t *testing.T) {
|
||||
src, err := readBundledChannels(t)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := exec.LookPath("luajit"); err != nil {
|
||||
t.Skip("luajit 不在 PATH（判据需要与内核相同的 Lua 实现）")
|
||||
}
|
||||
|
||||
const headroom = 10
|
||||
for _, n := range []int{1, headroom} {
|
||||
probe := strings.Repeat("local __headroom = 1\n", n) + src
|
||||
cmd := exec.Command("luajit", "-e", "local f,e=loadfile(arg[1]); if not f then io.stderr:write(tostring(e)) os.exit(1) end", "/dev/stdin")
|
||||
cmd.Stdin = strings.NewReader(probe)
|
||||
if out, err := cmd.CombinedOutput(); err != nil {
|
||||
t.Fatalf("★ 顶层 local 余量不足：加 %d 个 local 就无法编译 ⇒ "+
|
||||
"下一次顶层改动会**整插件加载失败**（全渠道不可用）\n %s\n"+
|
||||
"修法：把纯常量收进 plugin.K 表（挂表字段不占 local 槽位）", n, strings.TrimSpace(string(out)))
|
||||
}
|
||||
}
|
||||
t.Logf("✓ channels.lua 仍有 ≥%d 个顶层 local 余量", headroom)
|
||||
}
|
||||
|
||||
// 配对判据：常量必须确实在 plugin.K 里（防止有人把余量"省"出来的同时又
|
||||
// 把常量改回裸 local —— 那会让下一个改动的余量再次归零）。
|
||||
func TestChannelsConstantsLiveInPluginK(t *testing.T) {
|
||||
src, err := readBundledChannels(t)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
s := string(src)
|
||||
if !strings.Contains(s, "plugin.K = {") {
|
||||
t.Fatal("必须有 plugin.K 常量表（它是 local 余量的来源）")
|
||||
}
|
||||
// 抽样：这些曾占用 local 槽位的纯常量必须已挂表
|
||||
for _, n := range []string{"STORE_NAME", "SESSION_TTL", "B64_ALPHABET", "MODELS_DEV_URL"} {
|
||||
if !regexp.MustCompile(`(?m)^\s+` + n + ` = `).MatchString(s) {
|
||||
t.Errorf("%s 应定义在 plugin.K 表内", n)
|
||||
}
|
||||
if regexp.MustCompile(`(?m)^local ` + n + ` = `).MatchString(s) {
|
||||
t.Errorf("%s 又变回裸 local 了（会吃掉 local 余量）", n)
|
||||
}
|
||||
}
|
||||
t.Log("✓ 纯常量已集中在 plugin.K")
|
||||
}
|
||||
116
internal/lua/adapter_buddy_test.go
Normal file
116
internal/lua/adapter_buddy_test.go
Normal file
@ -0,0 +1,116 @@
|
||||
package lua
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// loadBuddyAdapter 装载内置 buddy 适配器。
|
||||
func loadBuddyAdapter(t *testing.T) *VM {
|
||||
t.Helper()
|
||||
src, err := bundledAdapters.ReadFile("adapters/buddy.lua")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
vm := NewVM(t.TempDir())
|
||||
t.Cleanup(vm.Stop)
|
||||
if err := vm.LoadAdapterSource("buddy", string(src)); err != nil {
|
||||
t.Fatalf("装载失败: %v", err)
|
||||
}
|
||||
return vm
|
||||
}
|
||||
|
||||
// ★ buddy 适配器：两个厂商要求，各有实测依据。
|
||||
//
|
||||
// ① glm-5.1 **必须**带非空 reasoning_effort（否则上游 400 + 11150）
|
||||
// ② 流式请求必须带 Accept: text/event-stream（否则只回空帧、completion_tokens=0）
|
||||
func TestBuddyAdapterEffortAndHeaders(t *testing.T) {
|
||||
vm := loadBuddyAdapter(t)
|
||||
|
||||
// ① glm-5.1 → 补 medium
|
||||
out, err := vm.Transform("buddy", "transform_request",
|
||||
`{"model":"glm-5.1","messages":[{"role":"user","content":"hi"}],"max_tokens":100,"stream":true}`)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
var b map[string]interface{}
|
||||
if err := json.Unmarshal([]byte(out), &b); err != nil {
|
||||
t.Fatalf("%v\n%s", err, out)
|
||||
}
|
||||
if b["reasoning_effort"] != "medium" {
|
||||
t.Errorf("glm-5.1 必须补 reasoning_effort=medium（否则上游 11150），实得 %v", b["reasoning_effort"])
|
||||
}
|
||||
|
||||
// ② 其它模型**不补**（避免把单模型要求扩成全平台行为）
|
||||
out2, err := vm.Transform("buddy", "transform_request",
|
||||
`{"model":"hy3","messages":[{"role":"user","content":"hi"}],"max_tokens":100,"stream":true}`)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
var b2 map[string]interface{}
|
||||
json.Unmarshal([]byte(out2), &b2)
|
||||
if b2["reasoning_effort"] != nil {
|
||||
t.Errorf("hy3 不该被补 reasoning_effort（实测不需要），实得 %v", b2["reasoning_effort"])
|
||||
}
|
||||
|
||||
// ③ 调用方已给档位时**尊重它**
|
||||
out3, _ := vm.Transform("buddy", "transform_request",
|
||||
`{"model":"glm-5.1","messages":[{"role":"user","content":"hi"}],"reasoning_effort":"high","stream":true}`)
|
||||
var b3 map[string]interface{}
|
||||
json.Unmarshal([]byte(out3), &b3)
|
||||
if b3["reasoning_effort"] != "high" {
|
||||
t.Errorf("调用方给的档位必须保留，实得 %v", b3["reasoning_effort"])
|
||||
}
|
||||
|
||||
// ④ 头：Accept 必须是 SSE
|
||||
h, err := vm.BuildHeaders("buddy", map[string]interface{}{"api_key": "TOK"})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if h["Accept"] != "text/event-stream" {
|
||||
t.Errorf("Accept 必须是 text/event-stream（否则上游只回空帧），实得 %q", h["Accept"])
|
||||
}
|
||||
for _, k := range []string{"X-Domain", "X-Product", "X-Product-Code"} {
|
||||
if h[k] == "" {
|
||||
t.Errorf("缺少 buddy 身份头 %s", k)
|
||||
}
|
||||
}
|
||||
t.Logf("✓ glm-5.1 补 effort；hy3 不补；Accept=%s", h["Accept"])
|
||||
}
|
||||
|
||||
// 配对判据：协议转换必须与 openai 一致（buddy 只是补两个字段，不该改转换）。
|
||||
func TestBuddyAdapterKeepsOpenAIWire(t *testing.T) {
|
||||
vm := loadBuddyAdapter(t)
|
||||
chunk := `{"id":"x","object":"chat.completion.chunk","created":1,"model":"m","choices":[{"index":0,"delta":{"content":"hi","reasoning_content":"think"},"finish_reason":null}]}`
|
||||
out, err := vm.Transform("buddy", "transform_stream_chunk", chunk)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !strings.Contains(out, `"content":"hi"`) || !strings.Contains(out, `"reasoning_content":"think"`) {
|
||||
t.Errorf("流式转换必须保留 content 与 reasoning_content，实得 %s", out)
|
||||
}
|
||||
}
|
||||
|
||||
// ★ 回归护栏：适配器**必须**定义全部四个转换函数。
|
||||
//
|
||||
// 真实事故（2026-10-16）：我第一版把基座继承写成"在适配器里读兄弟适配器源码"，
|
||||
// 而适配器运行在**另一个沙箱 VM**（无 __llmsproxy_plugin_dir、无文件系统），
|
||||
// 于是基座加载**静默失败** ⇒ 适配器只有 transform_request，
|
||||
// **缺 transform_stream_chunk** ⇒ 内核无法解析 SSE ⇒ 上游明明回了 39 帧正文，
|
||||
// 网关却记 completion_tokens=0、客户端看到**空回复**。
|
||||
// 更糟的是：hy3/glm-5.3/kimi-k3-1 全被它弄坏，而 glm-5.2 因多源兜底
|
||||
// "看起来正常" ⇒ 症状指向错误方向。
|
||||
//
|
||||
// ⇒ 判据必须咬住「四个转换函数都存在」，而不是"某个模型能回话"
|
||||
//
|
||||
// （后者会被别的源兜底掩盖 —— 本会话已栽过两次）。
|
||||
func TestBuddyAdapterDefinesAllTransforms(t *testing.T) {
|
||||
vm := loadBuddyAdapter(t)
|
||||
for _, fn := range []string{"transform_request", "transform_response", "transform_stream_chunk", "transform_error"} {
|
||||
if !vm.HasFn("buddy", fn) {
|
||||
t.Errorf("适配器缺 %s ⇒ 内核该路径会静默降级（流式会变成空回复）", fn)
|
||||
}
|
||||
}
|
||||
t.Log("✓ 四个转换函数齐备")
|
||||
}
|
||||
201
internal/lua/adapters/buddy.lua
Normal file
201
internal/lua/adapters/buddy.lua
Normal file
@ -0,0 +1,201 @@
|
||||
local adapter = {}
|
||||
|
||||
adapter.name = "buddy"
|
||||
adapter.version = "1.0.0"
|
||||
adapter.endpoint = "/chat/completions"
|
||||
adapter.headers = {}
|
||||
|
||||
-- Claude-behind-OpenAI upstreams (tabitoken, 扇贝, …) convert /chat/completions
|
||||
-- to the Anthropic Messages API internally, so they inherit Anthropic's
|
||||
-- tool id rule: ^[a-zA-Z0-9_-]{1,64}$, enforced by rejecting the WHOLE request
|
||||
-- with "Invalid tool use format" / "tool_use.id: String should match pattern".
|
||||
-- Plain OpenAI has no such rule, so an OpenAI-compatible model can mint an id
|
||||
-- like "bash:0" (observed from moonshotai/kimi-k3). In a fan-out router that id
|
||||
-- is replayed to every other source, so one such id kills every Claude slot at
|
||||
-- once and an AUTO request falls through all tiers.
|
||||
--
|
||||
-- safe_tool_id is pure and deterministic, so a tool_calls entry and its
|
||||
-- matching tool_call_id are rewritten identically within one request. A
|
||||
-- rewritten id keeps an 8-hex digest of the ORIGINAL id, without which two
|
||||
-- distinct ids could collapse into one ("a:b" and "a_b") and become an
|
||||
-- unpaired/duplicate tool call. Already-legal ids are returned untouched, so
|
||||
-- well-behaved traffic is byte-identical to before.
|
||||
-- (anthropic.lua carries the same helper; Lua adapters have no shared prelude.)
|
||||
local TOOL_ID_MAX = 64
|
||||
|
||||
local function safe_tool_id(id)
|
||||
if type(id) ~= "string" or id == "" then return id end
|
||||
local clean = string.gsub(id, "[^A-Za-z0-9_-]", "_")
|
||||
if clean == id and #clean <= TOOL_ID_MAX then
|
||||
return clean
|
||||
end
|
||||
local digest = string.sub(sha256_hex(id), 1, 8)
|
||||
local keep = TOOL_ID_MAX - #digest - 1
|
||||
if #clean > keep then clean = string.sub(clean, 1, keep) end
|
||||
return clean .. "_" .. digest
|
||||
end
|
||||
|
||||
-- 需要补 reasoning_effort 的模型 → 补哪个档位。
|
||||
--
|
||||
-- 唯一出处：参考 product.ts:237 声明 glm-5.1 的 reasoningEfforts 只有 ['medium']。
|
||||
-- 但实测**任何非空档位都被接受**，故这里取 medium（与参考声明一致）。
|
||||
local NEED_EFFORT = {
|
||||
["glm-5.1"] = "medium",
|
||||
}
|
||||
|
||||
-- OpenAI /chat/completions format (pass-through, strip provider-specific fields)
|
||||
function adapter.transform_request(raw_body)
|
||||
local ok, req = pcall(json.decode, raw_body)
|
||||
if not ok then return raw_body end
|
||||
-- The gateway's private control fields are GONE from the body (v2): the
|
||||
-- neutral `thinking` field carries that intent and each adapter maps it in
|
||||
-- its own protocol terms. These upstreams have no thinking switch at all,
|
||||
-- so the field is simply dropped here — which is what the nil-ing below
|
||||
-- used to do, except the field being dropped is no longer the kernel's
|
||||
-- private one.
|
||||
req.thinking = nil
|
||||
req.extra_body = nil
|
||||
-- ★★ buddy 的 glm-5.1 **要求请求体里有非空的 reasoning_effort**，
|
||||
-- 否则上游回 400：
|
||||
-- {"code":11150,"msg":"the reasoning effort value is not supported
|
||||
-- by the current model"}
|
||||
-- 实测（2026-10-16，逐字对照实验）：
|
||||
-- glm-5.1 裸（无该字段） → 400 11150 ✗
|
||||
-- glm-5.1 + reasoning_effort:medium → 200 ✓
|
||||
-- glm-5.1 空串 → 400 11150 ✗
|
||||
-- 任何**非空**档位均可（low/medium/high/xhigh/max/minimal 逐个 200）
|
||||
-- 而其余 15 个 buddy 模型裸跑全 200（多带该字段也无害）—— 即这是
|
||||
-- glm-5.1 **独有**的要求。
|
||||
-- ⚠️ 只对**已实测需要**的模型补（而不是一律补）：一律补会把"某个模型的
|
||||
-- 特殊要求"扩成"全平台行为"，将来上游若对某些模型收紧就会一次性弄坏
|
||||
-- 所有模型。
|
||||
if NEED_EFFORT[req.model or ""] and
|
||||
not (type(req.reasoning_effort) == "string" and req.reasoning_effort ~= "") and
|
||||
req.reasoning_effort == nil then
|
||||
req.reasoning_effort = NEED_EFFORT[req.model]
|
||||
end
|
||||
if req.messages then
|
||||
for _, msg in ipairs(req.messages) do
|
||||
msg.reasoning_content = nil
|
||||
if msg.tool_call_id ~= nil then
|
||||
msg.tool_call_id = safe_tool_id(msg.tool_call_id)
|
||||
end
|
||||
if type(msg.tool_calls) == "table" then
|
||||
for _, tc in ipairs(msg.tool_calls) do
|
||||
if type(tc) == "table" and tc.id ~= nil then
|
||||
tc.id = safe_tool_id(tc.id)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
return json.encode(req)
|
||||
end
|
||||
|
||||
function adapter.transform_response(raw_body)
|
||||
-- 非流式 OpenAI 响应 → 统一信封。此前在 openai/groq/kimicode/mistral/github
|
||||
-- 五个适配器里**逐字节相同**（各 2206B，diff 为空），已抽到共享前导。
|
||||
return __prelude.openai_response(raw_body)
|
||||
end
|
||||
|
||||
function adapter.transform_stream_chunk(raw_chunk)
|
||||
local ok, chunk = pcall(json.decode, raw_chunk)
|
||||
if not ok then return "" end
|
||||
|
||||
-- OpenAI-style streams may attach usage to a chunk with empty choices
|
||||
-- (the final usage chunk). Keys must match Go's TokenUsage json tags
|
||||
-- (prompt/completion/total); the gateway re-emits standard *_tokens.
|
||||
local uses = nil
|
||||
if type(chunk.usage) == "table" then
|
||||
uses = {
|
||||
prompt = chunk.usage.prompt_tokens or chunk.usage.prompt or 0,
|
||||
completion = chunk.usage.completion_tokens or chunk.usage.completion or 0,
|
||||
total = chunk.usage.total_tokens or chunk.usage.total or 0,
|
||||
}
|
||||
if type(chunk.usage.prompt_tokens_details) == "table" and chunk.usage.prompt_tokens_details.cached_tokens ~= nil then
|
||||
uses.prompt_tokens_details = { cached_tokens = chunk.usage.prompt_tokens_details.cached_tokens }
|
||||
elseif (chunk.usage.prompt_cache_hit_tokens or 0) > 0 then
|
||||
uses.prompt_cache_hit_tokens = chunk.usage.prompt_cache_hit_tokens
|
||||
uses.prompt_cache_miss_tokens = chunk.usage.prompt_cache_miss_tokens or 0
|
||||
uses.prompt_tokens_details = { cached_tokens = chunk.usage.prompt_cache_hit_tokens }
|
||||
end
|
||||
end
|
||||
|
||||
if not chunk.choices or #chunk.choices == 0 then
|
||||
if uses ~= nil then
|
||||
-- usage-only chunk is not a content/finish signal; the gateway
|
||||
-- emits its own terminal stop chunk and merges this usage.
|
||||
return json.encode({ usage = uses, done = false })
|
||||
end
|
||||
return ""
|
||||
end
|
||||
local delta = chunk.choices[1].delta or {}
|
||||
local fr = chunk.choices[1].finish_reason
|
||||
|
||||
local finish = (type(fr) == "string" and fr ~= "") and fr or nil
|
||||
|
||||
local unified = {
|
||||
content = delta.content or "",
|
||||
done = (finish ~= nil)
|
||||
}
|
||||
if finish then
|
||||
unified.finish_reason = finish
|
||||
end
|
||||
if uses ~= nil then
|
||||
unified.usage = uses
|
||||
end
|
||||
if delta.reasoning_content then
|
||||
unified.reasoning_content = delta.reasoning_content
|
||||
end
|
||||
if delta.tool_calls then
|
||||
-- Sanitize on the way OUT too: an id this upstream happily minted (it
|
||||
-- does not validate them) becomes a landmine once the client replays
|
||||
-- it to a Claude upstream. Only the first fragment of a streamed call
|
||||
-- carries an id; later argument fragments have none and are untouched.
|
||||
for _, tc in ipairs(delta.tool_calls) do
|
||||
if type(tc) == "table" and tc.id ~= nil then
|
||||
tc.id = safe_tool_id(tc.id)
|
||||
end
|
||||
end
|
||||
unified.tool_calls = delta.tool_calls
|
||||
end
|
||||
return json.encode(unified)
|
||||
end
|
||||
|
||||
-- 错误收敛：标准 OpenAI 信封 {error:{message,...}}
|
||||
function adapter.transform_error(status, body)
|
||||
-- OpenAI 风格错误体抽消息。此前在 deepseek/github/groq/kimicode/
|
||||
-- openai/sensenova 六个适配器里逐字节相同（各 293B），已抽到共享前导。
|
||||
return __prelude.openai_error(status, body)
|
||||
end
|
||||
|
||||
|
||||
-- ★ build_headers：buddy 要求客户端身份头 + 流式的 Accept。
|
||||
--
|
||||
-- 实测（2026-10-16）：不发 `Accept: text/event-stream` 时上游接受请求（200）
|
||||
-- 但只回一帧空内容（completion_tokens=0），既不报错也不产出正文 —— 表现是
|
||||
-- "模型不回答"，极易误判成模型坏了或额度不足。
|
||||
-- 参考 buddy-adapter.ts 的 send() 固定发该头。
|
||||
--
|
||||
-- 身份头照参考 credentialRequestHeaders + CODEBUDDY 产品配置：
|
||||
-- X-Domain 用产品域（不用凭据里的 domain 快照，那是登录时的区域，
|
||||
-- 跨产品迁移后会指错区）；X-Product 是**部署类型**（SaaS）；X-Product-Code
|
||||
-- 是产品码（codebuddy）。
|
||||
function adapter.build_headers(meta)
|
||||
local h = {
|
||||
["Accept"] = "text/event-stream",
|
||||
["Content-Type"] = "application/json",
|
||||
["X-Domain"] = "copilot.tencent.com",
|
||||
["X-Product"] = "SaaS",
|
||||
["X-Product-Code"] = "codebuddy",
|
||||
["User-Agent"] = "CodeBuddyIDE/1.106.1",
|
||||
}
|
||||
-- Authorization 由内核在 buildHeaders 之后补（值为源的 api_key）——
|
||||
-- 这里显式给出同一份，保证与其它 buddy 请求形态一致。
|
||||
if type(meta) == "table" and type(meta.api_key) == "string" and meta.api_key ~= "" then
|
||||
h["Authorization"] = "Bearer " .. meta.api_key
|
||||
end
|
||||
return h
|
||||
end
|
||||
|
||||
return adapter
|
||||
@ -47,6 +47,36 @@ local plugin = {
|
||||
author = "llmsproxy",
|
||||
}
|
||||
|
||||
-- ★★ 顶层常量集中到 plugin.K：本文件主 chunk 的 local 数已**顶到 Lua 上限 200**，
|
||||
-- 再加一个顶层 local 就编译失败（实测 +1 即 `main function has more than 200
|
||||
-- local variables`），而且炸的是**整个渠道插件**（插件加载失败 = 全渠道不可用）。
|
||||
-- 挂表字段**不占** local 槽位，故把纯常量收进这里腾出余量。
|
||||
--
|
||||
-- ⚠️ 只搬**纯常量**（字符串/数字/自闭合字面量）：它们没有前向引用问题，
|
||||
-- 搬迁是纯机械的。函数保持 local（它们之间的调用顺序是真实依赖）。
|
||||
plugin.K = {
|
||||
CODEARTS_SNAP = "https://snap-access.cn-north-4.myhuaweicloud.com",
|
||||
CODEARTS_LOGIN_BASE = "https://devcloud.cn-north-4.huaweicloud.com/doer/redirect",
|
||||
HUAWEI_AUTH_BASE = "https://auth.huaweicloud.com/authui/login.html",
|
||||
CODEARTS_TICKET_PLUGIN_NAME = "snap_jetbrains",
|
||||
CODEARTS_TICKET_PLUGIN_VERSION = "26.3.3",
|
||||
QODER_UTC8_OFFSET_MS = 8 * 3600 * 1000,
|
||||
QODER_REFRESH_HOUR_UTC8 = 10,
|
||||
QODER_MACHINE_ENVIRONMENT = "3",
|
||||
STORE_NAME = ".channels-store.json",
|
||||
B64_ALPHABET = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/",
|
||||
B64_LOOKUP = {},
|
||||
SESSION_TTL = 900, -- 15 分钟，过期会话在读取时惰性清理
|
||||
MODELS_DEV_URL = "https://models.dev/api.json",
|
||||
MODELS_DEV_TTL = 6 * 3600, -- 6 小时（照参考的 TTL 量级）
|
||||
EXPIRING_WINDOW_DAYS_DEFAULT = 15,
|
||||
BALANCE_CACHE_TTL = 60,
|
||||
DEAD_MODEL_TTL_DAYS = 30,
|
||||
REMOTE_CATALOG_COOLDOWN = 30, -- 秒，照参考 REMOTE_CATALOG_COOLDOWN_MS
|
||||
INJECT_PLATFORM_LIMIT = 32,
|
||||
ZCODE_ORIGIN = "https://zcode.z.ai",
|
||||
}
|
||||
|
||||
-- 申请的内核权限。设置页会把"申请中但未授权"标红。
|
||||
-- 声明本插件要用的内核动作（运维据此授予；declared 与 granted 成对展示）。
|
||||
--
|
||||
@ -183,7 +213,7 @@ local PLATFORMS = {
|
||||
adapter = "channels-trae",
|
||||
-- 照参考 TRAE_FALLBACK_MODELS 全量（32 项）。其中 4 项带 isHidden
|
||||
-- （内部子 agent）—— 不进用户可选列表，单列在 models_hidden。
|
||||
models = {
|
||||
models = {
|
||||
"DeepSeek-V4-Flash-Official", "Doubao-Seed-2.1-Pro", "seed-code-pro-0430",
|
||||
"Doubao-Seed-2.1-Turbo", "Doubao-Seed-2.0-Code", "glm-5.2", "glm-5-turbo",
|
||||
"glm-5", "DeepSeek-V4-Pro", "DeepSeek-V4-Flash", "kimi-k3", "kimi-k2.7-code",
|
||||
@ -435,7 +465,11 @@ local PLATFORMS = {
|
||||
headers_fn = true,
|
||||
parse = "buddy_config",
|
||||
},
|
||||
adapter = "openai",
|
||||
-- ★ 用内置 buddy 适配器（而非通用 openai）：buddy 的 glm-5.1 要求请求体里
|
||||
-- 必须有非空 reasoning_effort（否则 400 + 11150），且流式请求必须带
|
||||
-- Accept: text/event-stream（否则上游只回空帧、completion_tokens=0）。
|
||||
-- 该适配器复用 openai 的全部转换，只补这两处厂商要求。
|
||||
adapter = "buddy",
|
||||
-- 计费/状态请求头：逐行移植参考 checkinHeaders（credits.ts）。
|
||||
-- 实测只发 Authorization 会被上游 401：buddy 要求身份标识头齐全。
|
||||
-- X-Domain 以产品配置为准（不用凭据里的 domain 快照，那是登录时的区域，
|
||||
@ -461,7 +495,10 @@ local PLATFORMS = {
|
||||
return h
|
||||
end,
|
||||
models = {
|
||||
"deepseek-v4-flash", "deepseek-v4-pro", "hy4-preview", "hy4-preview-x",
|
||||
-- 兜底目录（远端不可用时才用）。照参考 CODEBUDDY_FALLBACK_MODELS 的取舍：
|
||||
-- **只列实测可调用的**。hy4-preview-x 已实测恒 11102（后端未开放）故剔除
|
||||
-- —— 参考的兜底表里同样没有它（它靠白名单达到同样效果）。
|
||||
"deepseek-v4-flash", "deepseek-v4-pro", "hy4-preview",
|
||||
"glm-5.3", "glm-5.3-flash", "glm-5.2",
|
||||
},
|
||||
-- 签到状态（对齐参考 credits.ts：必须用 checkin-activity-status，不能用
|
||||
@ -1132,7 +1169,6 @@ local PLATFORMS = {
|
||||
}
|
||||
|
||||
-- 华为云 snap-access 引擎域名（连接与签名都用它）。
|
||||
local CODEARTS_SNAP = "https://snap-access.cn-north-4.myhuaweicloud.com"
|
||||
|
||||
-- CodeArts（华为云 snap-access）的请求签名与每日领取。
|
||||
--
|
||||
@ -1211,7 +1247,7 @@ end
|
||||
|
||||
-- codeartsAccountInfo(cred)：账号信息（statistics/plugin），判断是否积分账户。
|
||||
local function codeartsAccountInfo(cred)
|
||||
local req, err = codeartsSignedRequest("GET", CODEARTS_SNAP .. "/snap-manager/v1/statistics/plugin", cred, nil)
|
||||
local req, err = codeartsSignedRequest("GET", plugin.K.CODEARTS_SNAP .. "/snap-manager/v1/statistics/plugin", cred, nil)
|
||||
if not req then return nil, err end
|
||||
local resp, herr = __llmsproxy_http(req)
|
||||
if herr or not resp or resp.status ~= 200 then
|
||||
@ -1257,7 +1293,7 @@ local function codeartsClaim(plat, tok, args)
|
||||
|
||||
-- ② 活动列表
|
||||
local areq = codeartsSignedRequest("GET",
|
||||
CODEARTS_SNAP .. "/v1/ops/delivery?channel=IDE", cred, nil)
|
||||
plugin.K.CODEARTS_SNAP .. "/v1/ops/delivery?channel=IDE", cred, nil)
|
||||
if not areq then return nil, "活动列表请求构造失败" end
|
||||
local aresp, aerr = __llmsproxy_http(areq)
|
||||
if aerr or not aresp or aresp.status ~= 200 then
|
||||
@ -1292,7 +1328,7 @@ local function codeartsClaim(plat, tok, args)
|
||||
|
||||
-- ④ claim
|
||||
local body = json.encode({ campaignId = campaignId, channel = "IDE" })
|
||||
local creq = codeartsSignedRequest("POST", CODEARTS_SNAP .. "/v1/ops/claim", cred, body)
|
||||
local creq = codeartsSignedRequest("POST", plugin.K.CODEARTS_SNAP .. "/v1/ops/claim", cred, body)
|
||||
if not creq then return nil, "claim 请求构造失败" end
|
||||
local cresp, cerr = __llmsproxy_http(creq)
|
||||
if cerr or not cresp or cresp.status ~= 200 then
|
||||
@ -1306,7 +1342,7 @@ local function codeartsClaim(plat, tok, args)
|
||||
local benefit = cok and type(crec) == "table" and (((crec.data or {}).benefit) or nil) or nil
|
||||
if type(benefit) == "table" and benefit.id then
|
||||
local fbody = json.encode({ id = benefit.id, channel = "IDE" })
|
||||
local freq = codeartsSignedRequest("POST", CODEARTS_SNAP .. "/v1/ops/confirm", cred, fbody)
|
||||
local freq = codeartsSignedRequest("POST", plugin.K.CODEARTS_SNAP .. "/v1/ops/confirm", cred, fbody)
|
||||
if freq then
|
||||
local fresp, ferr = __llmsproxy_http(freq)
|
||||
-- 只在真正失败时降级提示，不改变整体成败（积分已在待确认态）
|
||||
@ -1353,7 +1389,7 @@ end
|
||||
-- OAUTH_THEME = 2（Dark） OAUTH_LOCALE = zh-cn
|
||||
-- code_challenge_method = **SHA-256**（不是 S256！portal 以此识别 OAuth，
|
||||
-- 错值会回退到旧 ticket 流程）
|
||||
-- 旧流程回退（ticket）：CODEARTS_LOGIN_BASE + /snap-manager/v1/login/ticket
|
||||
-- 旧流程回退（ticket）：plugin.K.CODEARTS_LOGIN_BASE + /snap-manager/v1/login/ticket
|
||||
-- ⚠️ 常量收进**表**（`plugin.consts`）而不是顶层 `local`。
|
||||
-- 原因很实际：本文件顶层 local 已**顶到 Lua 单函数 200 个上限**，多一个就让
|
||||
-- 整个插件**加载失败**（`too many local variables`）；报错在编译期，
|
||||
@ -1584,8 +1620,8 @@ local function codeartsTicketPoll(ticketId, secret, pluginName, pluginVersion)
|
||||
local resp = __llmsproxy_http({
|
||||
url = url, method = "GET",
|
||||
headers = { ["Content-Type"] = "application/json;charset=UTF-8",
|
||||
["plugin-name"] = pluginName or CODEARTS_TICKET_PLUGIN_NAME,
|
||||
["plugin-version"] = pluginVersion or CODEARTS_TICKET_PLUGIN_VERSION },
|
||||
["plugin-name"] = pluginName or plugin.K.CODEARTS_TICKET_PLUGIN_NAME,
|
||||
["plugin-version"] = pluginVersion or plugin.K.CODEARTS_TICKET_PLUGIN_VERSION },
|
||||
})
|
||||
if resp and resp.status == 200 then
|
||||
local ok, rec = pcall(json.decode, resp.body or "")
|
||||
@ -1614,12 +1650,8 @@ local function codeartsTicketPoll(ticketId, secret, pluginName, pluginVersion)
|
||||
end
|
||||
|
||||
-- 旧 ticket 流程（portal 判定新式不可用时回退）
|
||||
local CODEARTS_LOGIN_BASE = "https://devcloud.cn-north-4.huaweicloud.com/doer/redirect"
|
||||
local HUAWEI_AUTH_BASE = "https://auth.huaweicloud.com/authui/login.html"
|
||||
local CODEARTS_CREDENTIAL_ENDPOINT =
|
||||
"https://snap-access.cn-north-4.myhuaweicloud.com/snap-manager/v1/login/ticket"
|
||||
local CODEARTS_TICKET_PLUGIN_NAME = "snap_jetbrains"
|
||||
local CODEARTS_TICKET_PLUGIN_VERSION = "26.3.3"
|
||||
|
||||
-- 生成 PKCE 配对（照参考 generatePkcePair：verifier = 48 随机字节的 base64url，
|
||||
-- challenge = sha256(verifier) 的 base64url）。
|
||||
@ -1658,8 +1690,6 @@ end
|
||||
QODER_SASH_BASE = "https://openapi.qoder.sh"
|
||||
|
||||
-- UTC+8 偏移（毫秒）。活动按 UTC+8 结算，口径与参考一致。
|
||||
local QODER_UTC8_OFFSET_MS = 8 * 3600 * 1000
|
||||
local QODER_REFRESH_HOUR_UTC8 = 10
|
||||
|
||||
-- hasQoderCampaignRefreshedToday()：今天的活动列表是否已刷新。
|
||||
local function qoderCampaignRefreshed(nowMs)
|
||||
@ -1669,8 +1699,8 @@ local function qoderCampaignRefreshed(nowMs)
|
||||
-- **必须用算术平移**判 UTC+8，不能用 os.date("*t")（本机时区）：
|
||||
-- 活动按 UTC+8 结算，取本机时区在出差/改时区时会判错方向
|
||||
-- （偏东提前把当天记为已处理、真漏领；偏西一天判两次）。
|
||||
local utc8 = os.date("!*t", math.floor((nowMs + QODER_UTC8_OFFSET_MS) / 1000))
|
||||
return (utc8.hour or 0) >= QODER_REFRESH_HOUR_UTC8
|
||||
local utc8 = os.date("!*t", math.floor((nowMs + plugin.K.QODER_UTC8_OFFSET_MS) / 1000))
|
||||
return (utc8.hour or 0) >= plugin.K.QODER_REFRESH_HOUR_UTC8
|
||||
end
|
||||
|
||||
-- 读数字/字符串（容忍字符串形式与缺失），照参考 readNumber/readString
|
||||
@ -1888,7 +1918,6 @@ end
|
||||
-- 4. 两者都拿不到时**照常发请求**（不带这两个头）—— 保守降级，
|
||||
-- 纯插件登录、未装桌面端的用户没有该文件，不能让整个积分功能报错。
|
||||
-- 5. 缓存解析结果：拿不到时也记 null，不反复读盘/反复 spawn。
|
||||
local QODER_MACHINE_ENVIRONMENT = "3"
|
||||
|
||||
local function qoderMachineCache()
|
||||
plugin.state.qoderMachine = plugin.state.qoderMachine or { done = false }
|
||||
@ -1965,7 +1994,7 @@ local function qoderReadFromRuntimeInfo()
|
||||
f:write('{"account":""}')
|
||||
f:close()
|
||||
local cmd = ("timeout 20 %s %s %s < %s 2>/dev/null"):format(
|
||||
exe, QODER_MACHINE_ENVIRONMENT, "--account-stdin", tmp)
|
||||
exe, plugin.K.QODER_MACHINE_ENVIRONMENT, "--account-stdin", tmp)
|
||||
local h = io.popen(cmd)
|
||||
local out = h and h:read("*a") or ""
|
||||
if h then h:close() end
|
||||
@ -2559,7 +2588,6 @@ end
|
||||
-- 即便这个文件泄露，也不含任何 token。
|
||||
--
|
||||
-- 写入：tmp + rename 原子替换。读：缺失/损坏 → 用内置默认值（不阻塞启动）。
|
||||
local STORE_NAME = ".channels-store.json"
|
||||
|
||||
-- storeDir 求插件目录：优先 config.yaml 的 plugin_dir，否则用本文件所在目录。
|
||||
-- 插件与内核同机、有完整 OS 权限，读自己的配置是能力范围内的事。
|
||||
@ -2618,7 +2646,7 @@ local PERSIST_KEYS = {
|
||||
|
||||
local function persistState()
|
||||
local dir = storeDir()
|
||||
local path = dir .. "/" .. STORE_NAME
|
||||
local path = dir .. "/" .. plugin.K.STORE_NAME
|
||||
local snap = {}
|
||||
for _, k in ipairs(PERSIST_KEYS) do
|
||||
if plugin.state[k] ~= nil then snap[k] = plugin.state[k] end
|
||||
@ -2673,7 +2701,7 @@ end
|
||||
-- 迁移后删除遗留文件：留着它只会让「哪个文件权威」变成永久的疑团。
|
||||
local function loadOwnSnapshot()
|
||||
local dir = storeDir()
|
||||
local f = io.open(dir .. "/" .. STORE_NAME, "r")
|
||||
local f = io.open(dir .. "/" .. plugin.K.STORE_NAME, "r")
|
||||
if f then
|
||||
local txt = f:read("*a")
|
||||
f:close()
|
||||
@ -2709,7 +2737,7 @@ local function migrateLegacyKernelState()
|
||||
local written, werr = persistState()
|
||||
if written then
|
||||
os.remove(legacy)
|
||||
io.stderr:write("[channels] 已迁移遗留状态到 " .. STORE_NAME .. "\n")
|
||||
io.stderr:write("[channels] 已迁移遗留状态到 " .. plugin.K.STORE_NAME .. "\n")
|
||||
else
|
||||
io.stderr:write("[channels] 遗留状态迁移写入失败（保留原文件）: "
|
||||
.. tostring(werr) .. "\n")
|
||||
@ -2939,16 +2967,14 @@ end
|
||||
-- 为读一个 JWT 声明而给内核加一个通用解码函数，是把插件的事推给内核 ——
|
||||
-- 插件能做就该自己做。纯 Lua 实现约 20 行，无依赖、无内核改动。
|
||||
-- 接受标准/URL-safe 两种字母表，容忍缺失填充。
|
||||
local B64_ALPHABET = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
||||
local B64_LOOKUP = {}
|
||||
for i = 1, #B64_ALPHABET do B64_LOOKUP[B64_ALPHABET:sub(i, i)] = i - 1 end
|
||||
for i = 1, #plugin.K.B64_ALPHABET do plugin.K.B64_LOOKUP[plugin.K.B64_ALPHABET:sub(i, i)] = i - 1 end
|
||||
local function b64url_decode(input)
|
||||
if type(input) ~= "string" or input == "" then return nil end
|
||||
-- 归一化：URL-safe → 标准，去空白，去掉填充（下面按 4 位一组补零处理）
|
||||
local s = input:gsub("%s", ""):gsub("-", "+"):gsub("_", "/"):gsub("=", "")
|
||||
local out, acc, bits = {}, 0, 0
|
||||
for i = 1, #s do
|
||||
local v = B64_LOOKUP[s:sub(i, i)]
|
||||
local v = plugin.K.B64_LOOKUP[s:sub(i, i)]
|
||||
if v == nil then return nil end -- 非法字符：如实失败，不猜
|
||||
acc = acc * 64 + v
|
||||
bits = bits + 6
|
||||
@ -3317,7 +3343,23 @@ local function parseRemoteModels(kind, body, provider)
|
||||
end
|
||||
end
|
||||
end
|
||||
-- ★ 已实测**不可调用**的模型（上游恒回 11102 "model service info not found"）。
|
||||
--
|
||||
-- 为什么需要它：这类模型既**不是**前缀可判（不像 nes-/codewise-），也**不在**
|
||||
-- agent 引用清单之外 —— 实测 `hy4-preview-x` 被服务端 agents 引用，
|
||||
-- 于是会被下面的 agentRef 例外重新放回目录，用户选中就 400。
|
||||
-- 参考的做法是**白名单重建**（只保留编译期实测过的 id），本仓没有那份表，
|
||||
-- 故用等价的**黑名单**精确排除：只放实测确认调不通的，不做任何猜测。
|
||||
--
|
||||
-- ⚠️ 判据来源必须是**真实请求**，不是元数据：
|
||||
-- 实测 hy4-preview-x 加/不加 reasoning_effort 都是 11102 ⇒ 与 effort 无关，
|
||||
-- 是后端未开放（与 glm-5.1 的 11150 是**不同**原因，后者已被 adapter 修好）。
|
||||
-- 参考 product.ts 的兜底白名单里同样**没有** hy4-preview-x。
|
||||
local NOT_CALLABLE = {
|
||||
["hy4-preview-x"] = true,
|
||||
}
|
||||
local function buddyIsChatModel(id)
|
||||
if NOT_CALLABLE[id] then return false end
|
||||
if id == "auto" or id == "default" then return false end
|
||||
if id:sub(1, 4) == "nes-" or id:sub(1, 11) == "completion-"
|
||||
or id:sub(1, 9) == "codewise-" then return false end
|
||||
@ -3349,6 +3391,8 @@ local function parseRemoteModels(kind, body, provider)
|
||||
if type(id) == "string" then pushId(id) end
|
||||
end
|
||||
end
|
||||
-- agentRef 的例外也必须过 NOT_CALLABLE 闸：否则"被 agent 引用"会把
|
||||
-- 实测不可调的模型重新放回目录（hy4-preview-x 正是这样漏进来的）。
|
||||
for id in pairs(agentRef) do pushId(id) end
|
||||
|
||||
|
||||
@ -3923,7 +3967,6 @@ end
|
||||
-- 登录会话放 plugin.state.sessions：内核没有"登录"概念，这正是插件该自己干的
|
||||
-- 事。表结构 { id, platform, flow, url, user_code, nonce, state, account,
|
||||
-- created_at, expires_at }。nonce 只在 begin 响应里出现一次，存侧不外泄。
|
||||
local SESSION_TTL = 900 -- 15 分钟，过期会话在读取时惰性清理
|
||||
|
||||
newToken = function()
|
||||
-- 用 hmac 自己跟自己造一个不可猜的 id（内核给过 sha256_hex）
|
||||
@ -4942,7 +4985,7 @@ local function initLoginSession(plat, args, now)
|
||||
nonce = newToken():sub(1, 32),
|
||||
state = "pending",
|
||||
created_at = now,
|
||||
expires_at = now + SESSION_TTL,
|
||||
expires_at = now + plugin.K.SESSION_TTL,
|
||||
}
|
||||
if plat.id == "lobsterai" then
|
||||
local seed = newToken()
|
||||
@ -6052,8 +6095,6 @@ plugin.schedule = {
|
||||
-- ⚠️ 我们额外的一条：本插件是**服务端**进程、没有"渲染路径"，
|
||||
-- 但 5MB 的下载仍不能放在请求路径上 —— 故挂在 refresh_models 定时任务里，
|
||||
-- 与模型目录刷新同一节拍。请求路径只读缓存。
|
||||
local MODELS_DEV_URL = "https://models.dev/api.json"
|
||||
local MODELS_DEV_TTL = 6 * 3600 -- 6 小时（照参考的 TTL 量级）
|
||||
--
|
||||
-- ⚠️ **已知取不到，属设计约束而非故障**：该端点实测 **5.64 MB**，而内核的
|
||||
-- `__llmsproxy_http` 有 **4 MiB 响应上限**（`httpMaxResponse`，刻意的堆保护 ——
|
||||
@ -6116,11 +6157,11 @@ end
|
||||
local function refreshCapabilities()
|
||||
local cached = plugin.state.capabilitiesMeta
|
||||
local now = os.time and os.time() or 0
|
||||
if cached and cached.at and cached.at <= now and (now - cached.at) < MODELS_DEV_TTL then
|
||||
if cached and cached.at and cached.at <= now and (now - cached.at) < plugin.K.MODELS_DEV_TTL then
|
||||
return false, "未到刷新时间"
|
||||
end
|
||||
local resp, err = __llmsproxy_http({
|
||||
url = MODELS_DEV_URL, method = "GET", timeout_ms = 30000,
|
||||
url = plugin.K.MODELS_DEV_URL, method = "GET", timeout_ms = 30000,
|
||||
headers = { ["Accept"] = "application/json" },
|
||||
})
|
||||
if err or not resp or resp.status ~= 200 then
|
||||
@ -6271,7 +6312,6 @@ end
|
||||
-- → 17 > 15 ⇒ 全算永久 ⇒ 锁定后立刻「无可用账号」
|
||||
-- 中国版按 30 天发包，故"距到期"天然落在 17~30 天 —— 判据本身没错，这类池
|
||||
-- 想用锁定就得把窗口放宽（如设 31）。窗口**只能由用户显式改**，默认 15。
|
||||
local EXPIRING_WINDOW_DAYS_DEFAULT = 15
|
||||
|
||||
expiringWindowDays = function()
|
||||
-- 优先级：settings（面板可改）→ env（部署可改）→ 默认 15。
|
||||
@ -6291,7 +6331,7 @@ expiringWindowDays = function()
|
||||
local n = tonumber(raw)
|
||||
if n and n >= 0 then return n end
|
||||
end
|
||||
return EXPIRING_WINDOW_DAYS_DEFAULT
|
||||
return plugin.K.EXPIRING_WINDOW_DAYS_DEFAULT
|
||||
end
|
||||
|
||||
-- balanceTier(expiring, permanent, allow_permanent)：
|
||||
@ -6319,7 +6359,6 @@ end
|
||||
-- 开着、时间只向前流：一笔距到期 15.02 天的余额，在用户什么都不做的两分钟里就
|
||||
-- 越过了线。若把分类也缓存，选号会继续按"永久"处理一笔马上要作废的积分
|
||||
-- （锁定时更糟：本该可用的号被判成不可用）。故命中缓存也**每次重新分桶**。
|
||||
local BALANCE_CACHE_TTL = 60
|
||||
|
||||
-- （余额缓存改用通用 TTL 缓存后，这个助手不再需要）
|
||||
|
||||
@ -6335,7 +6374,7 @@ local function accountBalanceRaw(provider, accountId)
|
||||
if not tok or tok == "" then return nil end
|
||||
local plate = plat
|
||||
local act = acct
|
||||
return ttlCacheGet("balance", provider .. ":" .. accountId, BALANCE_CACHE_TTL, function()
|
||||
return ttlCacheGet("balance", provider .. ":" .. accountId, plugin.K.BALANCE_CACHE_TTL, function()
|
||||
-- 复用既有的余额抓取（credits_balances 用的同一个助手，避免两处口径漂移）
|
||||
local rec, err = fetchAccountJSON(plate.balance, tok,
|
||||
credentialFull(act.credentialRef), plate)
|
||||
@ -6475,7 +6514,6 @@ local function isModelGone(message)
|
||||
end
|
||||
|
||||
-- 失效记录 TTL（秒）。照参考默认 30 天；过期自动复活，避免永久隐藏。
|
||||
local DEAD_MODEL_TTL_DAYS = 30
|
||||
|
||||
local function deadModelsOf(provider)
|
||||
plugin.state.deadModels = plugin.state.deadModels or {}
|
||||
@ -6498,7 +6536,7 @@ end
|
||||
deadModelIds = function(provider)
|
||||
local p = deadModelsOf(provider)
|
||||
local now = os.time and os.time() or 0
|
||||
local ttl = DEAD_MODEL_TTL_DAYS * 86400
|
||||
local ttl = plugin.K.DEAD_MODEL_TTL_DAYS * 86400
|
||||
local out = {}
|
||||
for id, meta in pairs(p) do
|
||||
local at = type(meta) == "table" and tonumber(meta.at) or 0
|
||||
@ -6542,7 +6580,6 @@ end
|
||||
--
|
||||
-- ⚠️ 本 Lua 无协程（单 VM 单锁），做不了"共享同一个 Promise"。用**时间戳
|
||||
-- 闸门**达到等价效果：记录 in_flight_at（防同轮重复）+ retry_at（失败冷却）。
|
||||
local REMOTE_CATALOG_COOLDOWN = 30 -- 秒，照参考 REMOTE_CATALOG_COOLDOWN_MS
|
||||
|
||||
local function catalogGateOf(provider)
|
||||
plugin.state.catalogGate = plugin.state.catalogGate or {}
|
||||
@ -6560,7 +6597,7 @@ local function catalogGateCooling(provider, now)
|
||||
local g = catalogGateOf(provider)
|
||||
if not g.retry_at then return false end
|
||||
if now >= g.retry_at then return false end
|
||||
if g.retry_at - now > REMOTE_CATALOG_COOLDOWN then return false end
|
||||
if g.retry_at - now > plugin.K.REMOTE_CATALOG_COOLDOWN then return false end
|
||||
return true
|
||||
end
|
||||
|
||||
@ -6578,7 +6615,7 @@ end
|
||||
local function catalogGateEnd(provider, got, now)
|
||||
local g = catalogGateOf(provider)
|
||||
g.in_flight_at = nil
|
||||
g.retry_at = got and nil or (now + REMOTE_CATALOG_COOLDOWN)
|
||||
g.retry_at = got and nil or (now + plugin.K.REMOTE_CATALOG_COOLDOWN)
|
||||
end
|
||||
|
||||
-- refresh_models()：定时把远端模型目录拉回来并重同步所有源。
|
||||
@ -6700,9 +6737,9 @@ end
|
||||
-- 出差/改时区会把日界判错方向。
|
||||
local function todayKey()
|
||||
local nowMs = (os.time and os.time() or 0) * 1000
|
||||
local utc8 = os.date("!*d", math.floor((nowMs + QODER_UTC8_OFFSET_MS) / 1000))
|
||||
local utc8m = os.date("!*m", math.floor((nowMs + QODER_UTC8_OFFSET_MS) / 1000))
|
||||
local utc8y = os.date("!*Y", math.floor((nowMs + QODER_UTC8_OFFSET_MS) / 1000))
|
||||
local utc8 = os.date("!*d", math.floor((nowMs + plugin.K.QODER_UTC8_OFFSET_MS) / 1000))
|
||||
local utc8m = os.date("!*m", math.floor((nowMs + plugin.K.QODER_UTC8_OFFSET_MS) / 1000))
|
||||
local utc8y = os.date("!*Y", math.floor((nowMs + plugin.K.QODER_UTC8_OFFSET_MS) / 1000))
|
||||
return tostring(utc8y) .. "-" .. tostring(utc8m) .. "-" .. tostring(utc8)
|
||||
end
|
||||
|
||||
@ -8235,7 +8272,6 @@ end
|
||||
-- 语义：整表替换（与 set_endpoints 的按字段覆盖不同）—— 注入是声明式的
|
||||
-- “这就是本轮我要用的目录”，避免测试之间残留状态。传空数组即清空。
|
||||
-- 容量上限：注入属于运维/测试能力，不承担生产目录职责，限 32 条够用。
|
||||
local INJECT_PLATFORM_LIMIT = 32
|
||||
-- ============================================================================
|
||||
-- 服务端浏览器栈的运维面（依赖预检 / 取凭据 / 停栈）
|
||||
-- ============================================================================
|
||||
@ -8395,8 +8431,8 @@ function plugin.inject_platforms(req)
|
||||
if type(list) ~= "table" then
|
||||
return { status = 400, json = { error = "platforms 必须是数组" } }
|
||||
end
|
||||
if #list > INJECT_PLATFORM_LIMIT then
|
||||
return { status = 400, json = { error = "最多注入 " .. INJECT_PLATFORM_LIMIT .. " 个平台" } }
|
||||
if #list > plugin.K.INJECT_PLATFORM_LIMIT then
|
||||
return { status = 400, json = { error = "最多注入 " .. plugin.K.INJECT_PLATFORM_LIMIT .. " 个平台" } }
|
||||
end
|
||||
-- 校验：每条必须有 id，且不得与内建目录重名（注入不能劫持真实厂商）。
|
||||
local builtin = {}
|
||||
@ -9231,7 +9267,6 @@ end
|
||||
-- isAutoCheckinExcluded）：代价发生在调用期间（拉 headful Chromium +
|
||||
-- 阿里云按设备限流 150 次/小时），"等它报错再判定"来不及。
|
||||
-- * 用户**手动**点领取时带上 captcha param 即可走通（面板交互）。
|
||||
local ZCODE_ORIGIN = "https://zcode.z.ai"
|
||||
|
||||
-- zcodeHeaders：照参考 buildZcodeHeaders。X-Device-Mid 是**硬需求**
|
||||
-- （缺它 billing 全家桶回 400）。
|
||||
@ -9255,7 +9290,7 @@ end
|
||||
-- zcodeClaimablePlans(cred)：查可领 plan（preview）。
|
||||
local function zcodeClaimablePlans(cred)
|
||||
local av = tostring(cred.app_version or "1.0.0")
|
||||
local url = ZCODE_ORIGIN .. "/api/v1/zcode-plan/billing/preview"
|
||||
local url = plugin.K.ZCODE_ORIGIN .. "/api/v1/zcode-plan/billing/preview"
|
||||
.. "?app_version=" .. rfc3986Escape(av) .. "&platform=win32"
|
||||
local resp, err = __llmsproxy_http({
|
||||
url = url, method = "GET",
|
||||
@ -9285,7 +9320,7 @@ function claimZcode(plat, tok, args)
|
||||
|
||||
-- ① 补激活上报（照参考：不需要 Authorization）
|
||||
__llmsproxy_http({
|
||||
url = ZCODE_ORIGIN .. "/api/v1/event/report", method = "POST",
|
||||
url = plugin.K.ZCODE_ORIGIN .. "/api/v1/event/report", method = "POST",
|
||||
headers = { ["Content-Type"] = "application/json" },
|
||||
body = json.encode({ event = "activation", app_version = tostring(cred.app_version or "1.0.0") }),
|
||||
})
|
||||
@ -9314,7 +9349,7 @@ function claimZcode(plat, tok, args)
|
||||
local granted, failedMsg = 0, nil
|
||||
for _, planId in ipairs(plans) do
|
||||
local resp, err = __llmsproxy_http({
|
||||
url = ZCODE_ORIGIN .. "/api/v1/zcode-plan/billing/claim", method = "POST",
|
||||
url = plugin.K.ZCODE_ORIGIN .. "/api/v1/zcode-plan/billing/claim", method = "POST",
|
||||
headers = zcodeHeaders(cred, {
|
||||
authorization = "Bearer " .. tostring(jwt),
|
||||
captcha = { param = captcha.param, region = captcha.region or "cn" },
|
||||
|
||||
Reference in New Issue
Block a user