fix: linting errors
This commit is contained in:
@@ -3,6 +3,50 @@ import re
|
||||
from typing import Any
|
||||
|
||||
|
||||
STRIX_API_BASE = "https://models.strix.ai/api/v1"
|
||||
|
||||
STRIX_PROVIDER_PREFIXES: dict[str, str] = {
|
||||
"claude-": "anthropic",
|
||||
"gpt-": "openai",
|
||||
"gemini-": "gemini",
|
||||
}
|
||||
|
||||
|
||||
def is_strix_model(model_name: str | None) -> bool:
|
||||
"""Check if model uses strix/ prefix."""
|
||||
return bool(model_name and model_name.startswith("strix/"))
|
||||
|
||||
|
||||
def get_strix_api_base(model_name: str | None) -> str | None:
|
||||
"""Return Strix API base URL if using strix/ model, None otherwise."""
|
||||
if is_strix_model(model_name):
|
||||
return STRIX_API_BASE
|
||||
return None
|
||||
|
||||
|
||||
def get_litellm_model_name(model_name: str | None) -> str | None:
|
||||
"""Convert strix/ prefixed model to litellm-compatible provider/model format.
|
||||
|
||||
Maps strix/ models to their corresponding litellm provider:
|
||||
- strix/claude-* -> anthropic/claude-*
|
||||
- strix/gpt-* -> openai/gpt-*
|
||||
- strix/gemini-* -> gemini/gemini-*
|
||||
- Other models -> openai/<model> (routed via Strix API)
|
||||
"""
|
||||
if not model_name:
|
||||
return model_name
|
||||
if not model_name.startswith("strix/"):
|
||||
return model_name
|
||||
|
||||
base_model = model_name[6:]
|
||||
|
||||
for prefix, provider in STRIX_PROVIDER_PREFIXES.items():
|
||||
if base_model.startswith(prefix):
|
||||
return f"{provider}/{base_model}"
|
||||
|
||||
return f"openai/{base_model}"
|
||||
|
||||
|
||||
def _truncate_to_first_function(content: str) -> str:
|
||||
if not content:
|
||||
return content
|
||||
|
||||
Reference in New Issue
Block a user