diff --git a/src/cost_tracker.py b/src/cost_tracker.py index 60b9b15f..3b59720b 100644 --- a/src/cost_tracker.py +++ b/src/cost_tracker.py @@ -43,6 +43,13 @@ MODEL_PRICING = [ (r"claude-.*-sonnet", {"input_per_mtok": 3.0, "output_per_mtok": 15.0}), (r"claude-.*-opus", {"input_per_mtok": 15.0, "output_per_mtok": 75.0}), (r"deepseek-v3", {"input_per_mtok": 0.27, "output_per_mtok": 1.10}), + (r"qwen-turbo", {"input_per_mtok": 0.05, "output_per_mtok": 0.10}), + (r"qwen-plus", {"input_per_mtok": 0.40, "output_per_mtok": 1.20}), + (r"qwen-max", {"input_per_mtok": 2.00, "output_per_mtok": 6.00}), + (r"qwen-long", {"input_per_mtok": 0.07, "output_per_mtok": 0.28}), + (r"qwen-vl-plus", {"input_per_mtok": 0.21, "output_per_mtok": 0.63}), + (r"qwen-vl-max", {"input_per_mtok": 0.50, "output_per_mtok": 1.50}), + (r"qwen-audio", {"input_per_mtok": 0.10, "output_per_mtok": 0.30}), ] def estimate_cost(model: str, input_tokens: int, output_tokens: int) -> float: diff --git a/src/models.py b/src/models.py index ad96ddd2..0db0795f 100644 --- a/src/models.py +++ b/src/models.py @@ -53,7 +53,7 @@ from src.paths import get_config_path #region: Constants -PROVIDERS: List[str] = ["gemini", "anthropic", "gemini_cli", "deepseek", "minimax"] +PROVIDERS: List[str] = ["gemini", "anthropic", "gemini_cli", "deepseek", "minimax", "qwen"] AGENT_TOOL_NAMES: List[str] = [ "run_powershell",