bone-agent 1.3.2 → 1.4.0

package/src/llm/config.py

@@ -32,6 +32,7 @@ ENV_API_KEYS = {
     'MINIMAX_PLAN_API_KEY': os.environ.get('MINIMAX_PLAN_API_KEY'),
     'MINIMAX_API_KEY': os.environ.get('MINIMAX_API_KEY'),
     'BONE_PROXY_API_KEY': os.environ.get('BONE_PROXY_API_KEY'),
+    'CODEX_PLAN_API_KEY': os.environ.get('CODEX_PLAN_API_KEY'),
 }
 
 # Detect platform for llama.cpp paths
@@ -73,6 +74,24 @@ def _load_config():
 
 _CONFIG = _load_config()
 
+
+def _get_codex_token() -> str:
+    """Read access token from Codex CLI's cached auth (~/.codex/auth.json).
+
+    Returns the access_token if available, empty string otherwise.
+    Codex CLI stores OAuth tokens here after `codex login`.
+    """
+    try:
+        auth_path = Path.home() / ".codex" / "auth.json"
+        if not auth_path.exists():
+            return ""
+        import json
+        data = json.loads(auth_path.read_text(encoding="utf-8"))
+        return data.get("tokens", {}).get("access_token", "")
+    except Exception:
+        return ""
+
+
 # Cache for provider registry (built once at module load)
 _provider_registry_cache = None
 _cached_provider = None
@@ -134,8 +153,8 @@ def _get_provider_registry():
             },
             "default_temperature": 0.1,
             "default_top_p": 0.9,
-            "allow_top_p": True,
-            "allow_temperature": True,
+            "allow_top_p": False,
+            "allow_temperature": False,
             "cost_in": 0.0,
             "cost_out": 0.0
         },
@@ -311,6 +330,22 @@ def _get_provider_registry():
             "allow_temperature": True,
             **_model_cost("KIMI_MODEL"),
         },
+        "codex": {
+            "type": "api",
+            "api_key": _CONFIG.get("CODEX_PLAN_API_KEY", "") or _get_codex_token(),
+            "model": _CONFIG.get("CODEX_PLAN_MODEL", "gpt-5.4-mini"),
+            "api_base": _CONFIG.get("CODEX_PLAN_API_BASE", "https://chatgpt.com/backend-api/codex"),
+            "endpoint": "/responses",
+            "error_prefix": "Codex",
+            "config_keys": {
+                "CODEX_PLAN_API_KEY": "",
+                "CODEX_PLAN_MODEL": "",
+                "CODEX_PLAN_API_BASE": "https://chatgpt.com/backend-api/codex",
+            },
+            "allow_temperature": False,
+            "allow_top_p": False,
+            **_model_cost("CODEX_PLAN_MODEL"),
+        },
         "bone": {
             "type": "api",
             "api_key": _CONFIG.get("BONE_PROXY_API_KEY", ""),
@@ -354,7 +389,7 @@ def reload_config():
     
     Note: This is a manual operation - call after config changes.
     """
-    global _CONFIG, _provider_registry_cache, _cached_provider, PROVIDER_REGISTRY, LLM_PROVIDER, STATUS_BAR_SETTINGS
+    global _CONFIG, _provider_registry_cache, _cached_provider, PROVIDER_REGISTRY, LLM_PROVIDER, STATUS_BAR_SETTINGS, MEMORY_SETTINGS
     _CONFIG = _load_config()
     _provider_registry_cache = None
     _cached_provider = None
@@ -363,6 +398,8 @@ def reload_config():
     LLM_PROVIDER = _get_provider()
     # Rebuild status bar settings
     STATUS_BAR_SETTINGS = _build_status_bar_settings()
+    # Rebuild memory settings
+    MEMORY_SETTINGS = _build_memory_settings()
 
 
 def _build_status_bar_settings():
@@ -377,8 +414,27 @@ def _build_status_bar_settings():
     }
 
 
+def _build_memory_settings():
+    """Build MEMORY_SETTINGS dict from current _CONFIG."""
+    ms = _CONFIG.get("MEMORY_SETTINGS", {})
+    return {
+        "enabled": ms.get("enabled", True),
+    }
+
+
+def update_memory_settings(settings_dict):
+    """Update MEMORY_SETTINGS at runtime.
+
+    Returns:
+        Updated MEMORY_SETTINGS dict
+    """
+    global MEMORY_SETTINGS
+    MEMORY_SETTINGS.update(settings_dict)
+    return MEMORY_SETTINGS
+
+
 def update_status_bar_settings(settings_dict):
-    """Update STATUS_BAR_SETTINGS at runtime and persist to config.
+    """Update STATUS_BAR_SETTINGS at runtime.
 
     Args:
         settings_dict: Dict of settings to update (e.g., {"show_cost": False})
@@ -400,6 +456,14 @@ def get_providers():
     return list(PROVIDER_REGISTRY.keys())
 
 
+def get_provider_display_name(provider: str) -> str:
+    """Return the user-facing provider name for a provider key."""
+    display_names = {
+        "codex": "Codex",
+    }
+    return display_names.get(provider, provider.replace("_", " ").title())
+
+
 # ============================================================================
 # PROVIDER REGISTRY - Centralized provider configuration
 # ============================================================================
@@ -413,6 +477,7 @@ __all__ = [
     "CONFIG_PATH",
     "PROVIDER_REGISTRY",
     "get_providers",
+    "get_provider_display_name",
     "get_model_cost",
     "LLM_PROVIDER",
     "TOOLS_ENABLED",
@@ -426,6 +491,8 @@ __all__ = [
     "reload_config",
     "STATUS_BAR_SETTINGS",
     "update_status_bar_settings",
+    "MEMORY_SETTINGS",
+    "update_memory_settings",
 ]
 
 
@@ -445,6 +512,9 @@ WEB_SEARCH_REQUIRE_CONFIRMATION = False
 # Status bar configuration
 STATUS_BAR_SETTINGS = _build_status_bar_settings()
 
+# Memory configuration
+MEMORY_SETTINGS = _build_memory_settings()
+
 # Tool approval modes
 APPROVE_MODES = ("safe", "accept_edits", "danger")
 CYCLEABLE_APPROVE_MODES = ("safe", "accept_edits")