codebee 0.1.4 → 0.1.6

package/app/core/modelhub.py

@@ -114,6 +114,22 @@ def _normalize_ids(data):
 _LITE = ("mini", "flash", "lite", "nano", "small", "tiny", "8b", "7b", "4b")
 _HEAVY = ("opus", "pro", "max", "ultra", "plus", "heavy", "codex")
 
+# 图片输入命名惯例：vision/omni 词段、glm-4v 式「数字+v」段；vl 子串另判
+# （qwen-vl / internvl2 / cogvlm 等视觉家族的通用命名根，文本模型名含 vl 的极罕见）
+_IMAGE_IN_RE = re.compile(
+    r"(?:^|[-_.])(?:vision|visual|omni)(?:[-_.0-9]|$)|[-_.]\dv(?:[-_.]|$)")
+
+
+def _auto_image_in(name):
+    """按模型名启发式预填「支持图片输入」。claude/gemini 全系原生多模态直接标；
+    其余只认显式命名（vision 词段、vl 子串、数字+v），宁缺勿滥——漏标的明模型
+    用户手动打开即可，误标只是运行时网关显式报错、开关改回即恢复。
+    仅用于新模型的初始声明。"""
+    n = (name or "").lower()
+    if n.startswith(("claude", "gemini")):
+        return True
+    return bool(_IMAGE_IN_RE.search(n)) or "vl" in n
+
 
 def _auto_priority(name):
     """按模型名启发式估强弱：分越高越强（优先级越靠前）。仅用于新模型的初始排序。"""
@@ -251,11 +267,14 @@ def refresh_models(provider_id):
                 # 保留 hidden：用户删掉的模型刷新时不能被重新带回
                 item = {"name": n, "enabled": bool(o.get("enabled", True)),
                         "priority": o.get("priority", 0),
-                        "hidden": bool(o.get("hidden"))}
+                        "hidden": bool(o.get("hidden")),
+                        # 模态声明同理：用户/预填设过的 image_in 刷新时不能被抹掉
+                        "image_in": bool(o.get("image_in"))}
                 (hidden if item["hidden"] else existing).append(item)
             else:
                 fresh.append({"name": n, "enabled": True,
-                              "auto": _auto_priority(n)})
+                              "auto": _auto_priority(n),
+                              "image_in": _auto_image_in(n)})
         # 已删除但本次未返回的条目也保留，否则下次拉取会当作新模型“复活”
         hidden += [dict(o) for n, o in old.items()
                    if o.get("hidden") and n not in names]
@@ -435,6 +454,7 @@ def model_ops(provider_id, names, op):
         if changed:
             _promote(models, newly_enabled)  # 启用的置顶，停用的退到启用块之后
             _save(data)
+            sync_binding_alerts()  # 模型启停/恢复 → 绑定链死活立即刷新告警
         return changed, ""
 
 
@@ -454,6 +474,7 @@ def _restore_all(provider_id):
             m["enabled"] = True
         _promote(models, names)  # 恢复 = 重新启用：同样置顶
         _save(data)
+        sync_binding_alerts()
         return None
 
 
@@ -729,15 +750,48 @@ def note_key_ok(provider_id, key_id):
     with _LOCK:
         data = _load()
         prov = next((p for p in data.get("providers", []) if p.get("id") == provider_id), None)
-        if not prov or not isinstance(prov.get("keys"), list):
+        if not prov:
             return
-        target = next((k for k in prov["keys"] if str(k.get("id")) == str(key_id)), None)
+        target = next((k for k in (prov.get("keys") or []) if str(k.get("id")) == str(key_id)), None)
         if target is None or not target.get("last_error"):
             return
         target.pop("last_error", None)
         _save(data)
 
 
+def _is_codex_target(target):
+    return (target or "").strip().lower() in ("codex-cli", "codex", "codex-code")
+
+
+def note_codex_wire_dead(provider_id, minutes=30):
+    """codex 撞上 wire 不兼容的供应商 → 供应商级冷却（自动绕开的记账位）。
+
+    codex 0.154 起只支持 responses wire（chat 被官方移除），讯飞等只有
+    chat completions 的 MaaS 每次 404/启动即拒。runner 按 404+no Route
+    matched / wire_api no longer supported 特征自动调用本函数；链展开与
+    路由绑定加分据此自动绕开，无需人工改绑定（2026-09-17 mo-so 实测）。
+    """
+    import time as _t
+    if not provider_id:
+        return
+    with _LOCK:
+        data = _load()
+        prov = next((p for p in data.get("providers", []) if p.get("id") == provider_id), None)
+        if not prov:
+            return
+        prov["codex_wire_dead_until"] = _t.time() + max(1, int(minutes)) * 60
+        _save(data)
+
+
+def codex_wire_blocked(prov):
+    """该供应商是否处于 codex wire 不兼容冷却期。"""
+    import time as _t
+    try:
+        return float((prov or {}).get("codex_wire_dead_until") or 0) > _t.time()
+    except Exception:
+        return False
+
+
 def _is_private_host(url):
     """主机是私网/环回 IP 字面量或 localhost 时返回 True（自动放行内网网关）。"""
     import urllib.parse
@@ -873,6 +927,8 @@ def providers_op(ids, op):
                     not (p.get("id") in sel and p.get("enabled", True)),
                     not bool(p.get("enabled", True))))
         _save(data)
+        if changed or op == "delete":
+            sync_binding_alerts()  # 厂商启停/删除 → 绑定链死活立即刷新告警
         return changed, ("" if changed else "所选供应商已是目标状态")
 
 
@@ -1813,6 +1869,14 @@ def _model_bindable(prov, model):
     return True
 
 
+def _model_image_in(prov, model):
+    """模型是否声明支持图片输入：models[] 条目 image_in；缺省/查不到=False（纯文本）。"""
+    for m in ((prov or {}).get("models") or []):
+        if isinstance(m, dict) and m.get("name") == model:
+            return bool(m.get("image_in"))
+    return False
+
+
 def bind_agent(agent, difficulty="default"):
     """按绑定生成应用了供应商/模型覆盖的 agent 副本；无绑定时原样返回。"""
     r = resolve_binding(agent.get("id"), difficulty) or resolve_binding(agent.get("kind"), difficulty)
@@ -1907,6 +1971,56 @@ def _protocol_candidates(prov):
             if (caps.get(p) or {}).get("base")]
 
 
+def bindable_protocols(agent_kind_or_id):
+    """该 CLI 可绑定的 wire 协议（与 resolve_binding 的 allowed 一致）。
+    供死链告警/失败文案解释「为什么绑不上」：claude 只认 anthropic，
+    codex/dsh 只认 openai，其余开放双协议（含 wire 适配）。"""
+    if _deepseek_env_target(agent_kind_or_id) or agent_kind_or_id in ("codex-cli", "codex"):
+        return ("openai",)
+    if agent_kind_or_id in ("claude-code", "claude"):
+        return ("anthropic",)
+    return tuple(_BINDABLE_PROTOCOLS)
+
+
+def binding_dead_msg(cli_id):
+    """死链失败/告警文案（pipeline 死链闸门与本模块 sync 共用）：说明为什么
+    不回落本机默认 + 该 CLI 需要什么协议的供应商。"""
+    try:
+        protos = bindable_protocols(cli_id)
+    except Exception:
+        protos = ()
+    hint = ("该 CLI 仅接受 %s 协议的已启用供应商；" % "、".join(protos)) if protos else ""
+    return ("绑定链全部失效（链上供应商已停用/删除/无密钥，或模型已停用），"
+            "本步判失败、不回落 CLI 本机默认——%s请在「CLI 绑定」页为该 CLI 绑定已启用的供应商" % hint)
+
+
+def sync_binding_alerts():
+    """厂商/模型启停、删除、恢复后立即重评估各 CLI 绑定链的静态死链告警：
+    恢复的当场解除、新死的当场亮起——不必等下一次步骤执行才发现
+    （2026-09-17 起与死链硬失败闸门配套）。幂等：health 侧静默/恢复语义不变。"""
+    from . import health
+    try:
+        binds = bindings()
+    except Exception:
+        return
+    for cli_id in sorted(binds.keys()):
+        b = binds.get(cli_id) or {}
+        if not _binding_chain(b):
+            continue  # 没配过链的 CLI 不归静态告警管（执行层闸门在跑时兜）
+        try:
+            r = resolve_binding(cli_id)
+            ok = bool(r and r.get("call_chain"))
+        except Exception:
+            ok = False
+        try:
+            if ok:
+                health.report_binding_ok(cli_id)
+            else:
+                health.report_binding_dead(cli_id, binding_dead_msg(cli_id))
+        except Exception:
+            pass  # 告警是尽力而为的旁路：persist 失败（如目录不可用）不拖累操作本身
+
+
 def resolve_binding(agent_kind_or_id, difficulty="default"):
     """返回 {env:{}, model:..., model_fallbacks:[...], codex_provider:..., call_chain:[...]} 或 None。
 
@@ -1921,17 +2035,11 @@ def resolve_binding(agent_kind_or_id, difficulty="default"):
     provs = {p.get("id"): p for p in providers()}
     routing = bool(b.get("difficulty_routing"))
     tier = difficulty if difficulty in ("easy", "hard") else None
-    # dsh 走 DEEPSEEK_* env，端点必须是 OpenAI 兼容的 /chat/completions，
-    # anthropic 协议的网关注进去也调不通，直接判为不可绑定。
-    allowed = ("openai",) if _deepseek_env_target(agent_kind_or_id) else _BINDABLE_PROTOCOLS
-    # 2026-09-15 连载验收实测：CLI 与供应商协议必须匹配——codex 只吃 openai wire
-    # （codex_provider 机制），claude 只吃 anthropic wire。混着注入会产生
-    # 「codex 拿到 ANTHROPIC_* env 却缺 ORCH_API_KEY」这类必然失败的组合
-    # （症状：Missing environment variable: ORCH_API_KEY）。
-    if agent_kind_or_id in ("codex-cli", "codex"):
-        allowed = ("openai",)
-    elif agent_kind_or_id in ("claude-code", "claude"):
-        allowed = ("anthropic",)
+    # 协议必须匹配（bindable_protocols）：dsh 只吃 OpenAI 兼容端点，codex 只吃
+    # openai wire（codex_provider 机制），claude 只吃 anthropic wire——混着注入
+    # 会产生「codex 拿到 ANTHROPIC_* env 却缺 ORCH_API_KEY」这类必然失败的组合
+    # （2026-09-15 连载验收实测，症状：Missing environment variable: ORCH_API_KEY）。
+    allowed = bindable_protocols(agent_kind_or_id)
 
     if chain:
         entries = []
@@ -1958,6 +2066,10 @@ def resolve_binding(agent_kind_or_id, difficulty="default"):
                 continue  # 原生协议与适配过的 wire 都不匹配：跳过
             if prov.get("name") in down_set:
                 continue  # 健康监测判定 down：跳过，省掉无效等待
+            if _is_codex_target(agent_kind_or_id) and (ep[2] == "chat" or codex_wire_blocked(prov)):
+                continue  # codex 0.154+ 只讲 responses wire：chat-only 供应商在起跑前
+                          # 就剔除（此前撞了才冷却 30 分钟，每轮白烧一次注定失败的
+                          # 尝试——2026-09-17 续4 连载 c35 实测）
             if model and not _model_bindable(prov, model):
                 continue  # 模型被停用/删除：该条跳过（2026-09-15 告警弹框「禁用该模型」）
             # 多 KEY：同一厂商按 KEY 展开成多条，顺序即调用顺序。欠费的 KEY 被
@@ -1996,6 +2108,9 @@ def resolve_binding(agent_kind_or_id, difficulty="default"):
     ep = _entry_endpoint(prov, allowed)
     if not ep:
         return None  # google 只登记；dsh 只接受 OpenAI 兼容端点；未适配的不硬塞
+    if _is_codex_target(agent_kind_or_id) and (ep[2] == "chat" or codex_wire_blocked(prov)):
+        return None  # codex 0.154+ 只讲 responses：chat-only 供应商直接判不可绑
+                     # （解析为空 → 死链闸门/路由降权接手，不浪费 CLI 尝试）
     names = [m["name"] for m in _enabled_models(prov)]
     model = prov.get("model_" + tier) or "" if (routing and tier) else ""
     if not model:
@@ -2173,6 +2288,24 @@ def reorder_models(provider_id, ordered_names):
         return None
 
 
+def set_model_caps(provider_id, name, image_in):
+    """声明单个模型的模态能力（当前仅 image_in 图片输入）。返回错误文案或 None。"""
+    with _LOCK:
+        data = _load()
+        prov = next((p for p in data.get("providers", []) if p.get("id") == provider_id), None)
+        if not prov:
+            return "供应商不存在"
+        target = next((m for m in (prov.get("models") or [])
+                       if m.get("name") == name), None)
+        if target is None:
+            return "模型不存在: %s" % name
+        image_in = bool(image_in)
+        if bool(target.get("image_in")) != image_in:   # 有差异才落盘
+            target["image_in"] = image_in
+            _save(data)
+    return None
+
+
 def _post_json_http(url, headers, body, allow_private, timeout=20):
     """带 SSRF 防护的 POST。返回 (status, json_obj|None, err)。"""
     import urllib.parse