api-key-manager 2.2.0__tar.gz → 3.0.1__tar.gz

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: api-key-manager
-Version: 2.2.0
-Summary: Batch manage API keys for 44+ AI providers with CLI and Web interfaces
+Version: 3.0.1
+Summary: Batch manage API keys for 45+ AI providers with CLI and Web interfaces
 Author: Townrain
 License: MIT
 Project-URL: Homepage, https://github.com/Townrain/API-Key-Manager
@@ -85,7 +85,8 @@ Requires-Dist: ruff>=0.4.0; extra == "dev"
 
 | 服务商 | 前缀 | 显示名 |
 |--------|------|--------|
-| DashScope | `sk-sp-` | 阿里百炼 |
+| 阿里百炼 | `sk-ws-` / `sk-` |
+| 阿里百炼编程 | `sk-sp-` |
 | ModelScope | `ms-` | 魔搭 |
 | Zhipu GLM | `sk-` | 智谱 |
 | Kimi | `sk-` | 月之暗面 |
@@ -341,7 +342,7 @@ if best_score >= 200:
 ### 检测优先级
 
 | 优先级 | 方法 | 说明 |
-||------||------|
+|--------|------|------|
 | 1 | 模式匹配 | 唯一前缀，如 `sk-proj-` → OpenAI |
 | 2 | 格式匹配 | 特殊格式，如 `{id}.{secret}` → 智谱 |
 | 3 | 全并发探测 | 第一个返回200的服务商胜出 |
@@ -369,9 +370,6 @@ if best_score >= 200:
 - 基于关键词的模式匹配（authentication、expired、rate limit 等）
 - 长错误信息截断（超过100字符时截断并添加省略号）
 
-## 项目结构
-
-
 ## 模型检测
 
 ### 模型列表来源
@@ -718,6 +716,32 @@ const result = await client.checkSingleKey({ key: 'sk-xxx', provider: 'openai' }
 
 ## 更新日志
 
+
+### v3.0.0 (2026-06-15)
+
+- **精确匹配修复**: 能力检测从子串匹配改为 ^$ 锚定精确匹配，修复 o1 误匹配 o1-mini 等问题
+- **API Bug 修复**: 修复 /api/stats/chart 运行时崩溃、/api/keys 和 /api/keys/export 丢失 key 字段
+- **测试补充**: 新增 8 个端点的测试（test/token, test/concurrency, models/capabilities, models/check, progress/stream 等）
+- **前端缓存优化**: 模型类型切换使用本地缓存过滤，不再重复请求 API
+- **Cherry Studio 同步工作流修复**: 修复 GitHub Actions 路径错误和验证脚本
+- **Autofill 样式修复**: 修复浏览器 autofill 导致输入框底色变白
+
+### v2.2.2 (2026-06-14)
+
+- **模型检测二级窗口**: 新增模型检测弹窗，支持搜索、类型筛选、全选、批量检测
+- **模型类型图标**: 显示模型能力图标（视觉/推理/联网/工具/嵌入/重排/免费）
+- **模型能力数据更新**: 从 Cherry Studio 同步最新模型能力数据（3184 条记录）
+- **检测逻辑优化**: DeepSeek 欠费密钥正确识别，不再误判为其他服务商
+- **指定模型检测**: 自动检测服务商时也支持指定模型检测
+
+### v2.2.1 (2026-06-14)
+
+- **检测逻辑修复**: 自动识别后必须调用 check() 验证 key 有效性
+- **签名匹配修复**: 状态码不再硬编码 401，使用实际状态码
+- **不可靠服务商**: ppio、nvidia、modelscope 不参与加分
+- **模型选择器**: 新增模型下拉框，支持选择具体模型检测
+- **Cherry Studio 同步**: 修复 ownedBy 映射（alibaba→dashscope 等）
+- **测试补充**: 新增 test_parser_supplement、test_ssrf_supplement、test_validator_supplement
 ### v2.2.0 (2026-06-14)
 
 - **检测逻辑重构**: 添加三步检测逻辑（/v1/models → 并发测试）
@@ -726,7 +750,6 @@ const result = await client.checkSingleKey({ key: 'sk-xxx', provider: 'openai' }
 - **超时控制**: 所有网络请求 5 秒超时
 - **模型同步**: 从 Cherry Studio 同步模型数据，支持 ownedBy 映射
 - **新增服务商**: OpenCode Go、OpenCode Zen
-- **新增服务商**: OpenCode Go、OpenCode Zen
 
 ### v2.1.2 (2026-06-11)
 

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/README.md

@@ -52,7 +52,8 @@
 
 | 服务商 | 前缀 | 显示名 |
 |--------|------|--------|
-| DashScope | `sk-sp-` | 阿里百炼 |
+| 阿里百炼 | `sk-ws-` / `sk-` |
+| 阿里百炼编程 | `sk-sp-` |
 | ModelScope | `ms-` | 魔搭 |
 | Zhipu GLM | `sk-` | 智谱 |
 | Kimi | `sk-` | 月之暗面 |
@@ -308,7 +309,7 @@ if best_score >= 200:
 ### 检测优先级
 
 | 优先级 | 方法 | 说明 |
-||------||------|
+|--------|------|------|
 | 1 | 模式匹配 | 唯一前缀，如 `sk-proj-` → OpenAI |
 | 2 | 格式匹配 | 特殊格式，如 `{id}.{secret}` → 智谱 |
 | 3 | 全并发探测 | 第一个返回200的服务商胜出 |
@@ -336,9 +337,6 @@ if best_score >= 200:
 - 基于关键词的模式匹配（authentication、expired、rate limit 等）
 - 长错误信息截断（超过100字符时截断并添加省略号）
 
-## 项目结构
-
-
 ## 模型检测
 
 ### 模型列表来源
@@ -685,6 +683,32 @@ const result = await client.checkSingleKey({ key: 'sk-xxx', provider: 'openai' }
 
 ## 更新日志
 
+
+### v3.0.0 (2026-06-15)
+
+- **精确匹配修复**: 能力检测从子串匹配改为 ^$ 锚定精确匹配，修复 o1 误匹配 o1-mini 等问题
+- **API Bug 修复**: 修复 /api/stats/chart 运行时崩溃、/api/keys 和 /api/keys/export 丢失 key 字段
+- **测试补充**: 新增 8 个端点的测试（test/token, test/concurrency, models/capabilities, models/check, progress/stream 等）
+- **前端缓存优化**: 模型类型切换使用本地缓存过滤，不再重复请求 API
+- **Cherry Studio 同步工作流修复**: 修复 GitHub Actions 路径错误和验证脚本
+- **Autofill 样式修复**: 修复浏览器 autofill 导致输入框底色变白
+
+### v2.2.2 (2026-06-14)
+
+- **模型检测二级窗口**: 新增模型检测弹窗，支持搜索、类型筛选、全选、批量检测
+- **模型类型图标**: 显示模型能力图标（视觉/推理/联网/工具/嵌入/重排/免费）
+- **模型能力数据更新**: 从 Cherry Studio 同步最新模型能力数据（3184 条记录）
+- **检测逻辑优化**: DeepSeek 欠费密钥正确识别，不再误判为其他服务商
+- **指定模型检测**: 自动检测服务商时也支持指定模型检测
+
+### v2.2.1 (2026-06-14)
+
+- **检测逻辑修复**: 自动识别后必须调用 check() 验证 key 有效性
+- **签名匹配修复**: 状态码不再硬编码 401，使用实际状态码
+- **不可靠服务商**: ppio、nvidia、modelscope 不参与加分
+- **模型选择器**: 新增模型下拉框，支持选择具体模型检测
+- **Cherry Studio 同步**: 修复 ownedBy 映射（alibaba→dashscope 等）
+- **测试补充**: 新增 test_parser_supplement、test_ssrf_supplement、test_validator_supplement
 ### v2.2.0 (2026-06-14)
 
 - **检测逻辑重构**: 添加三步检测逻辑（/v1/models → 并发测试）
@@ -693,7 +717,6 @@ const result = await client.checkSingleKey({ key: 'sk-xxx', provider: 'openai' }
 - **超时控制**: 所有网络请求 5 秒超时
 - **模型同步**: 从 Cherry Studio 同步模型数据，支持 ownedBy 映射
 - **新增服务商**: OpenCode Go、OpenCode Zen
-- **新增服务商**: OpenCode Go、OpenCode Zen
 
 ### v2.1.2 (2026-06-11)
 

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/api_key_manager.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: api-key-manager
-Version: 2.2.0
-Summary: Batch manage API keys for 44+ AI providers with CLI and Web interfaces
+Version: 3.0.1
+Summary: Batch manage API keys for 45+ AI providers with CLI and Web interfaces
 Author: Townrain
 License: MIT
 Project-URL: Homepage, https://github.com/Townrain/API-Key-Manager
@@ -85,7 +85,8 @@ Requires-Dist: ruff>=0.4.0; extra == "dev"
 
 | 服务商 | 前缀 | 显示名 |
 |--------|------|--------|
-| DashScope | `sk-sp-` | 阿里百炼 |
+| 阿里百炼 | `sk-ws-` / `sk-` |
+| 阿里百炼编程 | `sk-sp-` |
 | ModelScope | `ms-` | 魔搭 |
 | Zhipu GLM | `sk-` | 智谱 |
 | Kimi | `sk-` | 月之暗面 |
@@ -341,7 +342,7 @@ if best_score >= 200:
 ### 检测优先级
 
 | 优先级 | 方法 | 说明 |
-||------||------|
+|--------|------|------|
 | 1 | 模式匹配 | 唯一前缀，如 `sk-proj-` → OpenAI |
 | 2 | 格式匹配 | 特殊格式，如 `{id}.{secret}` → 智谱 |
 | 3 | 全并发探测 | 第一个返回200的服务商胜出 |
@@ -369,9 +370,6 @@ if best_score >= 200:
 - 基于关键词的模式匹配（authentication、expired、rate limit 等）
 - 长错误信息截断（超过100字符时截断并添加省略号）
 
-## 项目结构
-
-
 ## 模型检测
 
 ### 模型列表来源
@@ -718,6 +716,32 @@ const result = await client.checkSingleKey({ key: 'sk-xxx', provider: 'openai' }
 
 ## 更新日志
 
+
+### v3.0.0 (2026-06-15)
+
+- **精确匹配修复**: 能力检测从子串匹配改为 ^$ 锚定精确匹配，修复 o1 误匹配 o1-mini 等问题
+- **API Bug 修复**: 修复 /api/stats/chart 运行时崩溃、/api/keys 和 /api/keys/export 丢失 key 字段
+- **测试补充**: 新增 8 个端点的测试（test/token, test/concurrency, models/capabilities, models/check, progress/stream 等）
+- **前端缓存优化**: 模型类型切换使用本地缓存过滤，不再重复请求 API
+- **Cherry Studio 同步工作流修复**: 修复 GitHub Actions 路径错误和验证脚本
+- **Autofill 样式修复**: 修复浏览器 autofill 导致输入框底色变白
+
+### v2.2.2 (2026-06-14)
+
+- **模型检测二级窗口**: 新增模型检测弹窗，支持搜索、类型筛选、全选、批量检测
+- **模型类型图标**: 显示模型能力图标（视觉/推理/联网/工具/嵌入/重排/免费）
+- **模型能力数据更新**: 从 Cherry Studio 同步最新模型能力数据（3184 条记录）
+- **检测逻辑优化**: DeepSeek 欠费密钥正确识别，不再误判为其他服务商
+- **指定模型检测**: 自动检测服务商时也支持指定模型检测
+
+### v2.2.1 (2026-06-14)
+
+- **检测逻辑修复**: 自动识别后必须调用 check() 验证 key 有效性
+- **签名匹配修复**: 状态码不再硬编码 401，使用实际状态码
+- **不可靠服务商**: ppio、nvidia、modelscope 不参与加分
+- **模型选择器**: 新增模型下拉框，支持选择具体模型检测
+- **Cherry Studio 同步**: 修复 ownedBy 映射（alibaba→dashscope 等）
+- **测试补充**: 新增 test_parser_supplement、test_ssrf_supplement、test_validator_supplement
 ### v2.2.0 (2026-06-14)
 
 - **检测逻辑重构**: 添加三步检测逻辑（/v1/models → 并发测试）
@@ -726,7 +750,6 @@ const result = await client.checkSingleKey({ key: 'sk-xxx', provider: 'openai' }
 - **超时控制**: 所有网络请求 5 秒超时
 - **模型同步**: 从 Cherry Studio 同步模型数据，支持 ownedBy 映射
 - **新增服务商**: OpenCode Go、OpenCode Zen
-- **新增服务商**: OpenCode Go、OpenCode Zen
 
 ### v2.1.2 (2026-06-11)
 

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/api_key_manager.egg-info/SOURCES.txt

@@ -89,12 +89,15 @@ tests/test_i18n.py
 tests/test_logger.py
 tests/test_openapi.py
 tests/test_parser.py
+tests/test_parser_supplement.py
 tests/test_provider_detection.py
 tests/test_providers.py
 tests/test_proxy.py
 tests/test_security.py
+tests/test_ssrf_supplement.py
 tests/test_storage.py
 tests/test_tester.py
 tests/test_validator.py
+tests/test_validator_supplement.py
 tests/test_web_fixes.py
 tests/test_webhook.py

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/key_manager/__init__.py

@@ -1,4 +1,4 @@
-"""Key Manager - Batch API key management for 44+ AI providers."""
+"""Key Manager - Batch API key management for 45+ AI providers."""
 from key_manager.core import KeyManager
 from key_manager.providers import PROVIDERS, get_display_name
 from key_manager.errors import KeyManagerError, ErrorCode
@@ -13,5 +13,5 @@ __all__ = [
     "KeyStore",
 ]
 
-__version__ = "2.2.0"
+__version__ = "3.0.0"
 

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/key_manager/api_models.py

@@ -71,6 +71,7 @@ class KeyListResponse(BaseModel):
 
 
 class KeyExportItem(BaseModel):
+    key_masked: str = Field(..., description="Masked key for display")
     provider: str = Field(..., description="Provider identifier")
     max_tokens: int | None = Field(None, description="Maximum token limit")
     max_concurrency: int | None = Field(None, description="Maximum concurrency limit")
@@ -111,11 +112,12 @@ class ImportResponse(BaseModel):
 class CheckSingleRequest(BaseModel):
     key: str = Field(..., min_length=1, description="API key to check")
     provider: str = Field("", description="Provider name (auto-detected if empty)")
+    model: str | None = Field(None, description="Specific model to test (tests all if empty)")
     custom_base_url: str | None = Field(None, description="Override provider base URL")
 
     model_config = {
         "json_schema_extra": {
-            "examples": [{"key": "sk-abc123", "provider": "openai"}]
+            "examples": [{"key": "sk-abc123", "provider": "openai", "model": "gpt-4o"}]
         }
     }
 
@@ -296,23 +298,26 @@ class StatsResponse(BaseModel):
     total: int = Field(0, description="Total key count")
 
 
-class StatsChartProviderEntry(BaseModel):
-    total: int = Field(0, description="Total keys for this provider")
-    valid: int = Field(0, description="Valid keys")
-    invalid: int = Field(0, description="Invalid keys")
-    error: int = Field(0, description="Errored keys")
-
-
 class StatsChartStatuses(BaseModel):
     valid: int = Field(0, description="Total valid keys")
     invalid: int = Field(0, description="Total invalid keys")
     error: int = Field(0, description="Total errored keys")
     unknown: int = Field(0, description="Total unknown-status keys")
 
+class StatsChartProviderEntry(BaseModel):
+    provider: str = Field(..., description="Provider identifier")
+    display_name: str = Field("", description="Human-readable provider name")
+    total: int = Field(0, description="Total keys for this provider")
+    valid: int = Field(0, description="Valid keys")
+    invalid: int = Field(0, description="Invalid keys")
+    error: int = Field(0, description="Errored keys")
+    statuses: StatsChartStatuses = Field(
+        default_factory=StatsChartStatuses, description="Status breakdown"
+    )
 
 class StatsChartResponse(BaseModel):
-    providers: dict[str, StatsChartProviderEntry] = Field(
-        default_factory=dict, description="Per-provider breakdown for charts"
+    providers: list[StatsChartProviderEntry] = Field(
+        default_factory=list, description="Per-provider breakdown for charts"
     )
     statuses: StatsChartStatuses = Field(
         default_factory=StatsChartStatuses, description="Global status counts"

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/key_manager/checker.py

@@ -1,14 +1,11 @@
 import asyncio
-import json
 from datetime import datetime
-from pathlib import Path
 
 import httpx
 
 from .providers import PROVIDERS
 from .validator import validate_keys
 
-
 async def run_check(keys_file: str = "./data/keys.json",
                     results_file: str = "./data/check_results.json",
                     logs_dir: str = "./data/logs",
@@ -16,14 +13,16 @@ async def run_check(keys_file: str = "./data/keys.json",
                     timeout: int = 30,
                     proxy: str = None,
                     retry_failed: bool = True,
-                    retry_count: int = 2) -> dict:
+                    retry_count: int = 2,
+                    config: dict = None) -> dict:
     results = await validate_keys(
         keys_file=keys_file,
         results_file=results_file,
         logs_dir=logs_dir,
         concurrency=concurrency,
         timeout=timeout,
-        proxy=proxy
+        proxy=proxy,
+        config=config
     )
 
     # Retry failed keys
@@ -40,12 +39,18 @@ async def run_check(keys_file: str = "./data/keys.json",
                 concurrency=concurrency,
                 timeout=timeout,
                 proxy=proxy,
-                status_filter="error"
+                status_filter="error",
+                config=config
             )
 
-            # Merge results
+            # Recompute summary from details to avoid double-counting
+            # The retry only re-checks error keys, so we need to update the summary
+            # based on what changed
+            results["summary"]["error"]["count"] = retry_results["summary"]["error"]["count"]
+            results["summary"]["error"]["keys"] = retry_results["summary"]["error"]["keys"]
             results["summary"]["valid"]["count"] += retry_results["summary"]["valid"]["count"]
+            results["summary"]["valid"]["keys"].extend(retry_results["summary"]["valid"]["keys"])
             results["summary"]["invalid"]["count"] += retry_results["summary"]["invalid"]["count"]
-            results["summary"]["error"]["count"] = retry_results["summary"]["error"]["count"]
+            results["summary"]["invalid"]["keys"].extend(retry_results["summary"]["invalid"]["keys"])
 
     return results

{api_key_manager-2.2.0 → api_key_manager-3.0.1}/key_manager/cli.py

@@ -2,7 +2,7 @@ import argparse
 import asyncio
 import json
 import sys
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from pathlib import Path
 
 from rich.console import Console
@@ -153,7 +153,7 @@ def cmd_report(args, config):
         return
 
     days = args.days or 7
-    cutoff = datetime.utcnow() - timedelta(days=days)
+    cutoff = datetime.now(timezone.utc) - timedelta(days=days)
 
     stats = {}
     for key, info in data["keys"].items():

api_key_manager-3.0.1/key_manager/config.py

@@ -0,0 +1,103 @@
+import copy
+import logging
+import yaml
+from pathlib import Path
+
+logger = logging.getLogger(__name__)
+
+DEFAULT_CONFIG = {
+    "scan": {
+        "directories": ["./data/input"],
+        "recursive": True,
+    },
+    "check": {
+        "interval_hours": 6,
+        "timeout_seconds": 30,
+        "concurrency": 100,
+        "retry_failed": True,
+        "retry_count": 2,
+    },
+    "test": {
+        "token_test": True,
+        "token_auto_detect": True,
+        "token_steps": [1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072, 262144, 524288, 1048576],
+        "token_max_manual": None,
+        "concurrency_test": True,
+        "concurrency_steps": [1, 5, 10, 20, 50, 100],
+        "concurrency_timeout_seconds": 120,
+    },
+    "storage": {
+        "keys_file": "./data/keys.json",
+        "check_results_file": "./data/check_results.json",
+        "test_results_file": "./data/test_results.json",
+        "logs_dir": "./data/logs",
+    },
+    "auth": {
+        "api_key": "",
+    },
+    "rate_limit": {
+        "requests_per_minute": 60,
+    },
+}
+
+
+def _deep_merge(base: dict, override: dict) -> dict:
+    result = base.copy()
+    for key, value in override.items():
+        if key in result and isinstance(result[key], dict) and isinstance(value, dict):
+            result[key] = _deep_merge(result[key], value)
+        else:
+            result[key] = value
+    return result
+
+
+def _validate_config(config: dict) -> dict:
+    """Validate config values and warn about unknown keys."""
+    valid_keys = set(DEFAULT_CONFIG.keys())
+    unknown_keys = set(config.keys()) - valid_keys
+    if unknown_keys:
+        logger.warning(f"Unknown config keys (ignored): {unknown_keys}")
+
+    # Validate numeric values
+    check_config = config.get("check", {})
+    if check_config.get("concurrency", 100) < 1:
+        logger.warning("check.concurrency must be >= 1, using default")
+        check_config["concurrency"] = 100
+    if check_config.get("timeout_seconds", 30) < 1:
+        logger.warning("check.timeout_seconds must be >= 1, using default")
+        check_config["timeout_seconds"] = 30
+
+    rate_limit = config.get("rate_limit", {})
+    if rate_limit.get("requests_per_minute", 60) < 0:
+        logger.warning("rate_limit.requests_per_minute must be >= 0, using default")
+        rate_limit["requests_per_minute"] = 60
+
+    return config
+
+
+def load_config(path: str = "config.yaml") -> dict:
+    config_path = Path(path)
+    if config_path.exists():
+        try:
+            with open(config_path, "r", encoding="utf-8") as f:
+                user_config = yaml.safe_load(f) or {}
+            if not isinstance(user_config, dict):
+                logger.warning(f"Config file {path} is not a dict, using defaults")
+                return copy.deepcopy(DEFAULT_CONFIG)
+            config = _deep_merge(DEFAULT_CONFIG, user_config)
+            return _validate_config(config)
+        except yaml.YAMLError as e:
+            logger.error(f"Failed to parse config file {path}: {e}")
+            return copy.deepcopy(DEFAULT_CONFIG)
+        except Exception as e:
+            logger.error(f"Failed to load config file {path}: {e}")
+            return copy.deepcopy(DEFAULT_CONFIG)
+    return copy.deepcopy(DEFAULT_CONFIG)
+
+
+def save_config(config: dict, path: str = "config.yaml"):
+    try:
+        with open(path, "w", encoding="utf-8") as f:
+            yaml.dump(config, f, default_flow_style=False, allow_unicode=True)
+    except Exception as e:
+        logger.error(f"Failed to save config file {path}: {e}")