router-maestro 0.1.4__tar.gz → 0.1.5__tar.gz

{router_maestro-0.1.4 → router_maestro-0.1.5}/CLAUDE.md

@@ -57,7 +57,7 @@ Router-Maestro is a multi-model routing system that exposes both OpenAI-compatib
 - `schemas/` - Pydantic models for both API formats
 
 **CLI (`src/router_maestro/cli/`)**
-- Typer-based CLI with subcommands: `server`, `auth`, `model`, `context`, `config`, `stats`
+- Typer-based CLI with subcommands: `server`, `auth`, `model`, `context`, `config`
 - Each subcommand in its own module registered in `main.py`
 
 ### Data Flow
@@ -72,7 +72,7 @@ Router-Maestro is a multi-model routing system that exposes both OpenAI-compatib
 
 Configuration and data files follow XDG conventions:
 - **Config** (`~/.config/router-maestro/`): `providers.json`, `priorities.json`, `contexts.json`
-- **Data** (`~/.local/share/router-maestro/`): `auth.json`, `server.json`, `stats.db`
+- **Data** (`~/.local/share/router-maestro/`): `auth.json`, `server.json`
 
 ### Model Identification
 

{router_maestro-0.1.4 → router_maestro-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: router-maestro
-Version: 0.1.4
+Version: 0.1.5
 Summary: Multi-model routing and load balancing system with OpenAI-compatible API
 Author-email: Kanwen Li <likanwen@icloud.com>
 License-Expression: MIT

{router_maestro-0.1.4 → router_maestro-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "router-maestro"
-version = "0.1.4"
+version = "0.1.5"
 description = "Multi-model routing and load balancing system with OpenAI-compatible API"
 readme = "README.md"
 license = "MIT"

{router_maestro-0.1.4 → router_maestro-0.1.5}/src/router_maestro/__init__.py

@@ -1,3 +1,3 @@
 """Router-Maestro: Multi-model routing and load balancing system."""
 
-__version__ = "0.1.4"
+__version__ = "0.1.5"

{router_maestro-0.1.4 → router_maestro-0.1.5}/src/router_maestro/cli/config.py

@@ -139,19 +139,29 @@ def claude_code_config() -> None:
     )
     anthropic_url = f"{base_url}/api/anthropic"
 
-    config = {
-        "env": {
-            "ANTHROPIC_BASE_URL": anthropic_url,
-            "ANTHROPIC_AUTH_TOKEN": auth_token,
-            "ANTHROPIC_MODEL": main_model,
-            "ANTHROPIC_SMALL_FAST_MODEL": fast_model,
-            "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "1",
-        }
+    env_config = {
+        "ANTHROPIC_BASE_URL": anthropic_url,
+        "ANTHROPIC_AUTH_TOKEN": auth_token,
+        "ANTHROPIC_MODEL": main_model,
+        "ANTHROPIC_SMALL_FAST_MODEL": fast_model,
+        "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC": "1",
     }
 
+    # Load existing settings to preserve other sections (e.g., MCP servers)
+    existing_config: dict = {}
+    if settings_path.exists():
+        try:
+            with open(settings_path, encoding="utf-8") as f:
+                existing_config = json.load(f)
+        except (json.JSONDecodeError, OSError):
+            pass  # If file is corrupted, start fresh
+
+    # Merge: update env section while preserving other sections
+    existing_config["env"] = env_config
+
     settings_path.parent.mkdir(parents=True, exist_ok=True)
     with open(settings_path, "w", encoding="utf-8") as f:
-        json.dump(config, f, indent=2)
+        json.dump(existing_config, f, indent=2)
 
     console.print(
         Panel(

{router_maestro-0.1.4 → router_maestro-0.1.5}/src/router_maestro/server/routes/anthropic.py

@@ -26,6 +26,7 @@ from router_maestro.utils import (
     get_logger,
     map_openai_stop_reason_to_anthropic,
 )
+from router_maestro.utils.tokens import AnthropicStopReason
 
 logger = get_logger("server.routes.anthropic")
 
@@ -106,7 +107,7 @@ async def count_tokens(request: AnthropicCountTokensRequest):
 
     # Count messages
     for msg in request.messages:
-        content = msg.content if hasattr(msg, "content") else msg.get("content", "")
+        content = msg.content
         if isinstance(content, str):
             total_chars += len(content)
         elif isinstance(content, list):
@@ -115,12 +116,12 @@ async def count_tokens(request: AnthropicCountTokensRequest):
                     if block.get("type") == "text":
                         total_chars += len(block.get("text", ""))
                 elif hasattr(block, "text"):
-                    total_chars += len(block.text)
+                    total_chars += len(block.text)  # type: ignore[union-attr]
 
     return {"input_tokens": estimate_tokens_from_char_count(total_chars)}
 
 
-def _map_finish_reason(reason: str | None) -> str | None:
+def _map_finish_reason(reason: str | None) -> AnthropicStopReason | None:
     """Map OpenAI finish reason to Anthropic stop reason."""
     return map_openai_stop_reason_to_anthropic(reason)
 
@@ -144,7 +145,7 @@ def _estimate_input_tokens(request: AnthropicMessagesRequest) -> int:
 
     # Count messages
     for msg in request.messages:
-        content = msg.content if hasattr(msg, "content") else msg.get("content", "")
+        content = msg.content
         if isinstance(content, str):
             total_chars += len(content)
         elif isinstance(content, list):
@@ -161,7 +162,7 @@ def _estimate_input_tokens(request: AnthropicMessagesRequest) -> int:
                                 if isinstance(tc, dict) and tc.get("type") == "text":
                                     total_chars += len(tc.get("text", ""))
                 elif hasattr(block, "text"):
-                    total_chars += len(block.text)
+                    total_chars += len(block.text)  # type: ignore[union-attr]
 
     # Count tools definitions if present
     if request.tools:

{router_maestro-0.1.4 → router_maestro-0.1.5}/src/router_maestro/utils/tokens.py

@@ -1,8 +1,14 @@
 """Token estimation utilities."""
 
+from typing import Literal
+
 # Approximate characters per token for English text
 CHARS_PER_TOKEN = 4
 
+AnthropicStopReason = Literal[
+    "end_turn", "max_tokens", "stop_sequence", "tool_use", "pause_turn", "refusal"
+]
+
 
 def estimate_tokens(text: str) -> int:
     """Estimate token count from text.
@@ -31,7 +37,9 @@ def estimate_tokens_from_char_count(char_count: int) -> int:
     return char_count // CHARS_PER_TOKEN
 
 
-def map_openai_stop_reason_to_anthropic(openai_reason: str | None) -> str | None:
+def map_openai_stop_reason_to_anthropic(
+    openai_reason: str | None,
+) -> AnthropicStopReason | None:
     """Map OpenAI finish reason to Anthropic stop reason.
 
     Args:
@@ -42,7 +50,7 @@ def map_openai_stop_reason_to_anthropic(openai_reason: str | None) -> str | None
     """
     if openai_reason is None:
         return None
-    mapping = {
+    mapping: dict[str, AnthropicStopReason] = {
         "stop": "end_turn",
         "length": "max_tokens",
         "tool_calls": "tool_use",

{router_maestro-0.1.4 → router_maestro-0.1.5}/uv.lock

@@ -925,7 +925,7 @@ wheels = [
 
 [[package]]
 name = "router-maestro"
-version = "0.1.4"
+version = "0.1.5"
 source = { editable = "." }
 dependencies = [
     { name = "aiosqlite" },