mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 01:10:23 +08:00
ab82841426
将 Rust 版上游可用模型抓取逻辑收敛到 Python 版行为，统一后台自动抓模 与管理员 provider-query 的模型发现路径，消除标准 /models、固定模型目录、 Antigravity、Vertex AI 等 provider 在两端实现上的分叉。 核心变更： - 在 aether-model-fetch 中引入统一抓模策略层 - 覆盖标准 /models、Vertex API Key、Vertex Service Account、 Antigravity fetchAvailableModels、固定模型目录五类抓模路径 - 将 provider-query 与后台自动抓模都切换到共享抓模入口，避免重复拼接 URL、headers 和 provider 特判逻辑 标准模型抓取对齐： - 按 Python 语义调整抓模优先级： openai:chat > openai:cli > openai:compact claude:chat > claude:cli gemini:chat > gemini:cli - 从抓模候选中移除 openai:responses - 为 openai:cli/openai:compact、claude:cli、gemini:* 补齐 Python 同款 User-Agent / 浏览器指纹请求头 - Claude 抓模保留 after_id 分页语义 - Gemini 抓模统一为 v1beta/models?key=... 语义 provider-query 对齐： - 返回结果改为按 model id 聚合，并合并/排序 api_formats - 最终模型列表按 model id 排序，行为与 Python 保持一致 - 固定目录 provider（codex/kiro/claude_code/gemini_cli）不再依赖活跃 endpoint，即使无 endpoint 也能返回预设模型目录 - Antigravity 多 key 查询改为按账户可用性 + tier 排序，首个成功结果即 停止，并接入 provider 级缓存 - 仅配置 openai:responses 的 provider 不再被视为抓模成功路径 自动抓模对齐： - 自动抓模成功时写入 allowed_models、upstream_models cache，并同步 upstream_metadata - upstream_metadata 合并逻辑对齐 Python，对 quota_by_model 做模型级合并， 并保留已有 reset_time - 自动抓模失败时不覆盖已有 allowed_models - 固定目录 provider 在无 endpoint 场景下也可成功更新 allowed_models Antigravity 对齐： - 使用 POST /v1internal:fetchAvailableModels 抓取可用模型 - 按 Python 规则处理 URL fallback 和 429/404/408/5xx fallback 状态 - 强制要求 auth_config.project_id - 过滤 Python 黑名单模型 - 解析并持久化 upstream_metadata.antigravity.quota_by_model Vertex AI 对齐： - API Key 模式仅抓取 publishers/google/models - Service Account 模式新增 JWT token exchange，并按 Python region 顺序 抓取 google + anthropic publishers - 模型 owned_by / display_name / api_format 推断与 Python 对齐 - 软 404 处理行为与 Python 收敛 Gemini CLI / 固定目录对齐： - Gemini CLI 改为返回 Python 预设模型目录 - 在可用时通过 loadCodeAssist 补充 plan_type/project_id 元数据 - Codex/Kiro/Claude Code 改为共享固定模型目录实现 测试： - 扩展 aether-model-fetch 单元测试，覆盖格式优先级、openai:responses 排除、 请求头、Claude 分页、Gemini query auth、固定目录与 metadata 合并 - 调整 provider-query 控制面测试到 Python 语义 - 新增自动抓模运行时测试，覆盖固定目录成功、Antigravity metadata 合并、 失败保留旧 allowed_models 验证： - cargo nextest run -p aether-model-fetch --lib - cargo nextest run -p aether-gateway control::admin::provider_query model_fetch::runtime::tests
23 lines
537 B
TOML
23 lines
537 B
TOML
[package]
|
|
name = "aether-model-fetch"
|
|
version = "0.1.0"
|
|
edition.workspace = true
|
|
license.workspace = true
|
|
repository.workspace = true
|
|
|
|
[dependencies]
|
|
aether-contracts.workspace = true
|
|
aether-data-contracts.workspace = true
|
|
aether-provider-transport.workspace = true
|
|
aether-scheduler-core.workspace = true
|
|
async-trait.workspace = true
|
|
base64.workspace = true
|
|
regex.workspace = true
|
|
rsa = "0.9.10"
|
|
serde_json.workspace = true
|
|
sha2 = { workspace = true, features = ["oid"] }
|
|
uuid.workspace = true
|
|
|
|
[dev-dependencies]
|
|
tokio.workspace = true
|