router-maestro 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

router_maestro/__init__.py

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

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/server/routes/anthropic.py

@@ -3,6 +3,7 @@
 import json
 import uuid
 from collections.abc import AsyncGenerator
+from datetime import UTC, datetime
 
 from fastapi import APIRouter, HTTPException
 from fastapi.responses import StreamingResponse
@@ -13,6 +14,8 @@ from router_maestro.server.schemas.anthropic import (
     AnthropicCountTokensRequest,
     AnthropicMessagesRequest,
     AnthropicMessagesResponse,
+    AnthropicModelInfo,
+    AnthropicModelList,
     AnthropicStreamState,
     AnthropicTextBlock,
     AnthropicUsage,
@@ -26,6 +29,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 +110,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 +119,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 +148,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 +165,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:
@@ -226,3 +230,93 @@ async def stream_response(
             },
         }
         yield f"event: error\ndata: {json.dumps(error_event)}\n\n"
+
+
+def _generate_display_name(model_id: str) -> str:
+    """Generate a human-readable display name from model ID.
+
+    Transforms model IDs like 'github-copilot/claude-sonnet-4' into
+    'Claude Sonnet 4 (github-copilot)'.
+    """
+    if "/" in model_id:
+        provider, model_name = model_id.split("/", 1)
+    else:
+        provider = ""
+        model_name = model_id
+
+    # Capitalize words and handle common patterns
+    words = model_name.replace("-", " ").replace("_", " ").split()
+    display_words = []
+    for word in words:
+        # Keep version numbers as-is
+        if word.replace(".", "").isdigit():
+            display_words.append(word)
+        else:
+            display_words.append(word.capitalize())
+
+    display_name = " ".join(display_words)
+    if provider:
+        display_name = f"{display_name} ({provider})"
+
+    return display_name
+
+
+@router.get("/api/anthropic/v1/models")
+async def list_models(
+    limit: int = 20,
+    after_id: str | None = None,
+    before_id: str | None = None,
+) -> AnthropicModelList:
+    """List available models in Anthropic format.
+
+    Args:
+        limit: Maximum number of models to return (default 20)
+        after_id: Return models after this ID (for forward pagination)
+        before_id: Return models before this ID (for backward pagination)
+    """
+    model_router = get_router()
+    models = await model_router.list_models()
+
+    # Generate ISO 8601 timestamp for created_at
+    # Using current time since actual creation dates aren't tracked
+    created_at = datetime.now(UTC).strftime("%Y-%m-%dT%H:%M:%SZ")
+
+    # Convert to Anthropic format
+    anthropic_models = [
+        AnthropicModelInfo(
+            id=model.id,
+            created_at=created_at,
+            display_name=_generate_display_name(model.id),
+            type="model",
+        )
+        for model in models
+    ]
+
+    # Handle pagination
+    start_idx = 0
+    if after_id:
+        for i, model in enumerate(anthropic_models):
+            if model.id == after_id:
+                start_idx = i + 1
+                break
+
+    end_idx = len(anthropic_models)
+    if before_id:
+        for i, model in enumerate(anthropic_models):
+            if model.id == before_id:
+                end_idx = i
+                break
+
+    # Apply limit
+    paginated = anthropic_models[start_idx : min(start_idx + limit, end_idx)]
+
+    first_id = paginated[0].id if paginated else None
+    last_id = paginated[-1].id if paginated else None
+    has_more = (start_idx + limit) < end_idx
+
+    return AnthropicModelList(
+        data=paginated,
+        first_id=first_id,
+        last_id=last_id,
+        has_more=has_more,
+    )

router_maestro/server/schemas/anthropic.py

@@ -244,3 +244,24 @@ class AnthropicStreamState(BaseModel):
     estimated_input_tokens: int = 0  # Estimated input tokens from request
     last_usage: dict | None = None  # Track the latest usage from stream chunks
     message_complete: bool = False  # Track if message_stop was sent
+
+
+# Models API types
+
+
+class AnthropicModelInfo(BaseModel):
+    """Anthropic model object."""
+
+    id: str
+    created_at: str  # ISO 8601 datetime
+    display_name: str
+    type: Literal["model"] = "model"
+
+
+class AnthropicModelList(BaseModel):
+    """Anthropic models list response with pagination."""
+
+    data: list[AnthropicModelInfo]
+    first_id: str | None = None
+    last_id: str | None = None
+    has_more: bool = False

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.dist-info → router_maestro-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: router-maestro
-Version: 0.1.4
+Version: 0.1.6
 Summary: Multi-model routing and load balancing system with OpenAI-compatible API
 Author-email: Kanwen Li <likanwen@icloud.com>
 License-Expression: MIT
@@ -37,6 +37,9 @@ Description-Content-Type: text/markdown
 
 # Router-Maestro
 
+[![CI](https://github.com/MadSkittles/Router-Maestro/actions/workflows/ci.yml/badge.svg)](https://github.com/MadSkittles/Router-Maestro/actions/workflows/ci.yml)
+[![Release](https://github.com/MadSkittles/Router-Maestro/actions/workflows/release.yml/badge.svg)](https://github.com/MadSkittles/Router-Maestro/actions/workflows/release.yml)
+
 Multi-model routing router with OpenAI-compatible and Anthropic-compatible APIs. Route LLM requests across GitHub Copilot, OpenAI, Anthropic, and custom providers with intelligent fallback and priority-based selection.
 
 ## TL;DR

{router_maestro-0.1.4.dist-info → router_maestro-0.1.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-router_maestro/__init__.py,sha256=dQ7VGwTEZyoTDYi_HcMlVOwL7DKJ75prK_l_Qa9eN_4,92
+router_maestro/__init__.py,sha256=HJQ2eE5cH3zVHfskAha2jYgacsEZHU9FmRfQTzraiAE,92
 router_maestro/__main__.py,sha256=cUHr8B7JBiv5HhnN6l2iayDkGSBpI5Kf4I3jv9I_I3o,121
 router_maestro/auth/__init__.py,sha256=0JgD1w2gtGSkj809kgSKQanYYkncg6eF-hHoz-jQPgo,353
 router_maestro/auth/github_oauth.py,sha256=acQlAA2Zh6c8KQYdzXbC4ww0EJ41AgvbI5ixpFuNoRg,5060
@@ -7,7 +7,7 @@ router_maestro/auth/storage.py,sha256=TCLxgQ1lWcWD4xJXJzx5OMpvuAun_LSRItK0zhR6H0
 router_maestro/cli/__init__.py,sha256=yIAshaHpLL0WrDFmRpoMRM2EUe75x0wmM5NlGW3C89s,37
 router_maestro/cli/auth.py,sha256=eq5LBUohbMnHS4dZeyvq4OQAjzdrJ-StP2FGuUhkKa0,5940
 router_maestro/cli/client.py,sha256=mRzpsA_Dxn-Xq7W1_t6EiyddMI0a3cvuTL6-2JuV4mE,9383
-router_maestro/cli/config.py,sha256=9e_O0J-cxX2FCqHl8bKFKh897PK9mErTp31SXYmmU5c,5667
+router_maestro/cli/config.py,sha256=SGrWdd59eX1F8KzhHMJ_u2AkYfFuYlyLl6Nfz6u_rNk,6100
 router_maestro/cli/context.py,sha256=EPbT7fReIW17veU76CSAcv8QjzMsCIPm1QDBlGsV8fQ,4549
 router_maestro/cli/main.py,sha256=5yiK4Q149goSB2KKzgMuF5EpcC8FBzOUCkEt8wY5NAU,1314
 router_maestro/cli/model.py,sha256=2IG3IpQWh8Ejdv5Htcgr90O2v2UAa80TU15oOniPdvk,9054
@@ -35,18 +35,18 @@ router_maestro/server/middleware/__init__.py,sha256=PhtP2E04wApnOUBLE76mrOa0sSHp
 router_maestro/server/middleware/auth.py,sha256=Ak3k5cC8m4qPGUIheuOB--QiFvs6GIAcTRJqtCGCjAA,2018
 router_maestro/server/routes/__init__.py,sha256=eGEpNCnSRVQC1pFL7_evDmZfkMrviuI-n1okAS-YnhM,397
 router_maestro/server/routes/admin.py,sha256=oub4hDrYaytuorXkJzmz0YZ4Z2rcyNuwKcK_4IGvcDY,8942
-router_maestro/server/routes/anthropic.py,sha256=T5-rHBPDyPxP4Cs0yzm7Kvvn-zgV6jspnZdoSVDeH2w,8041
+router_maestro/server/routes/anthropic.py,sha256=r9GKXdb5yg0G6BFvbKx2-hIGlCM736NgEIESXbSomgE,10794
 router_maestro/server/routes/chat.py,sha256=vyYX1ILhgAb9HYD87h1U3c5btpplqkTaejA81pWg4Oo,4752
 router_maestro/server/routes/models.py,sha256=PTSXojNFN9j90Bke74ZO6sEsfIc8u_4A69eW1QzFIbc,716
 router_maestro/server/schemas/__init__.py,sha256=VmJZoTMLb-bF33m79urhbejVdLfjDGMqCJP5QvWbHsU,1176
 router_maestro/server/schemas/admin.py,sha256=DuUojkCcq9n8pDhWG6L0SpzQooh91lmHjCRzgZ4AMwk,2369
-router_maestro/server/schemas/anthropic.py,sha256=hNl6rZ7AX-HdLxtsd0cWpZjpIyK1AkEBcuiQpZQqPYc,6136
+router_maestro/server/schemas/anthropic.py,sha256=S5TFYDd8Iw7Oxjki6ng84DGVB90G0-mOza5D5r3rwOY,6566
 router_maestro/server/schemas/openai.py,sha256=s2487RYIn1h-CIaUpLue9BScDaTsafbVg5yc-kKhfME,2141
 router_maestro/utils/__init__.py,sha256=oSQyV--FueMPggRfjWWVnAKtjkcZWFOm9hCTymu0oZU,409
 router_maestro/utils/logging.py,sha256=gJWoRYibAxCWn4VmTmnrwpBRzQ7Uu5YIEk5zDiF9X_k,2393
-router_maestro/utils/tokens.py,sha256=t2E5BrrE5X3VCgw-rYFMkic7heJ0huj9rrOXAIlKq8o,1330
-router_maestro-0.1.4.dist-info/METADATA,sha256=0f158gqfAjoReykKlaeBee6eraI8fwKqmAW9WifpKNo,12161
-router_maestro-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-router_maestro-0.1.4.dist-info/entry_points.txt,sha256=zoFUxxvNcFe0nTgpRbIdygIDEOla3KbvW6HbOCOlgv4,63
-router_maestro-0.1.4.dist-info/licenses/LICENSE,sha256=Ea86BSGu7_tpLAuzif_JmM9zjMoKQEf95VVF9sZw3Jo,1084
-router_maestro-0.1.4.dist-info/RECORD,,
+router_maestro/utils/tokens.py,sha256=U5PXJv_6ba5xgMBG0c5qB96Yu6uLscSUjMWYTdNests,1530
+router_maestro-0.1.6.dist-info/METADATA,sha256=5NS_Z6XU27HDauDX1_ZIdPTD_RrXLnFCgTPVTQ-FSAM,12501
+router_maestro-0.1.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+router_maestro-0.1.6.dist-info/entry_points.txt,sha256=zoFUxxvNcFe0nTgpRbIdygIDEOla3KbvW6HbOCOlgv4,63
+router_maestro-0.1.6.dist-info/licenses/LICENSE,sha256=Ea86BSGu7_tpLAuzif_JmM9zjMoKQEf95VVF9sZw3Jo,1084
+router_maestro-0.1.6.dist-info/RECORD,,