ccoding 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

ccode/__init__.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-__version__ = "0.2.0"
+__version__ = "0.4.0"
 
 
 def main() -> None:

ccode/api.py

@@ -7,7 +7,7 @@ import urllib.error
 import urllib.request
 from typing import Any
 
-from .config import MODEL_KEYS, mask_secret
+from .config import MODEL_ENV_VARS, MODEL_KEYS, REQUIRED_MODEL_KEYS, mask_secret
 
 
 def _truncate_body(text: str, max_len: int = 200) -> str:
@@ -107,7 +107,7 @@ def build_env(config: dict[str, Any], masked: bool = False) -> dict[str, str]:
         entry = config["models"].get(key, {})
         model_id = entry.get("id")
         if model_id:
-            env[f"ANTHROPIC_DEFAULT_{key.upper()}_MODEL"] = model_id
+            env[MODEL_ENV_VARS[key]] = model_id
     for key, value in config.get("toggles", {}).items():
         env[key] = str(value)
     return env
@@ -118,7 +118,7 @@ def validate_launch_requirements(config: dict[str, Any]) -> str | None:
     api_key = config.get("api_key", "").strip()
     if not base_url or not api_key:
         return "Base URL and API key are required."
-    for key in MODEL_KEYS:
+    for key in REQUIRED_MODEL_KEYS:
         entry = config["models"].get(key, {})
         if not entry.get("owned_by") or not entry.get("id"):
             return "OPUS, SONNET, and HAIKU selections are required."

ccode/config.py

@@ -11,12 +11,20 @@ from typing import Any
 CONFIG_DIR = Path.home() / ".ccode"
 CONFIG_PATH = CONFIG_DIR / "config.json"
 DEFAULT_BASE_URL = "http://127.0.0.1:8317"
-MODEL_KEYS = ("opus", "sonnet", "haiku")
-MODEL_LABELS = {
-    "opus": "OPUS",
-    "sonnet": "SONNET",
-    "haiku": "HAIKU",
+# 模型 key 及其对应的环境变量，dict 顺序即主界面展示顺序。
+# auto 对应客户端 auto mode 分类器模型，其余沿用 ANTHROPIC_DEFAULT_<KEY>_MODEL。
+MODEL_ENV_VARS = {
+    "opus": "ANTHROPIC_DEFAULT_OPUS_MODEL",
+    "sonnet": "ANTHROPIC_DEFAULT_SONNET_MODEL",
+    "haiku": "ANTHROPIC_DEFAULT_HAIKU_MODEL",
+    "auto": "CLAUDE_CODE_AUTO_MODE_MODEL",
 }
+MODEL_KEYS = tuple(MODEL_ENV_VARS)
+MODEL_LABELS = {key: key.upper() for key in MODEL_KEYS}
+
+# auto 为可选项，其余模型启动时必选
+OPTIONAL_MODEL_KEYS = ("auto",)
+REQUIRED_MODEL_KEYS = tuple(key for key in MODEL_KEYS if key not in OPTIONAL_MODEL_KEYS)
 
 # toggles 的默认值：值统一用字符串存储
 DEFAULT_TOGGLES: dict[str, str] = {
@@ -85,11 +93,7 @@ def default_config() -> dict[str, Any]:
     return {
         "base_url": DEFAULT_BASE_URL,
         "api_key": "",
-        "models": {
-            "opus": {"owned_by": None, "id": None},
-            "sonnet": {"owned_by": None, "id": None},
-            "haiku": {"owned_by": None, "id": None},
-        },
+        "models": {key: {"owned_by": None, "id": None} for key in MODEL_KEYS},
         "toggles": DEFAULT_TOGGLES.copy(),
         "skip_permissions": False,
         "remote": copy.deepcopy(DEFAULT_REMOTE),