Files
patbond-ai/app/providers/registry.py
T
lixi 4762358f54
CI / ai-gates (push) Failing after 29s
fix(security): 凭证扫描放行依赖注入传参 + 四仓口径更正
`api_key=volcengine.api_key` 这类**依赖注入传参**被 KEY-ASSIGN 判为疑似凭证。
这是规则的真缺口而非代码问题:行里没有任何秘密,而做 DI 的代码到处都是这种
写法——绕过一次就得在每个调用点绕过,那等于把规则废掉。故改规则。

新增 ALLOW_REFERENCE,刻意收得很紧以免开后门:
- 值必须是**未加引号**的小写起首标识符引用(真密钥要么带引号,要么不是标识符形状)
- 只允许**恰好一个点且随即行尾**
- 连字符不在字符集内

对抗性夹具验证(三真三假):
- 拦下 `api_key = "sk-T3Blbk…"`(OpenAI 形态)
- 拦下 `secret_key: "AKID…"`(腾讯云形态)
- 拦下 `auth_token=eyJ….eyJ….Sfl…`(**三段式 JWT——两个点,故不落进放行**)
- 放行 `api_key=lowercase.reference` / `api_key = settings.api_key` / `api_token: config.authToken`
原 5 形态泄漏夹具复验仍 8 条命中,一条不少。

另更正口径:脚本头与 git-workflow.md 仍写「三仓同构」,实际已是四仓——
patbond-ai 是唯一持有 AI 服务商凭证的仓,漏同步它风险最大。四仓 md5 已一致
(4945940093b4b8413b7d3eec8bf6a5ec),四仓 --all 全绿。

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
2026-09-20 12:18:18 +08:00

43 lines
1.6 KiB
Python

"""provider 注册表:按 `provider_code` 分发。
Java 侧按任务行冻结的 `provider_code_snapshot` 找适配器,**不按当前目录**——目录
可能被改指向或禁用,而在途任务必须仍由当初提交它的适配器轮询。用当前目录路由会
拿着别人的任务 ID 去问错的供应商。
"""
from __future__ import annotations
from app.providers.base import GenerationProvider
class ProviderRegistry:
"""`provider_code` → 适配器。"""
def __init__(self, providers: list[GenerationProvider] | None = None):
self._by_code: dict[str, GenerationProvider] = {}
for provider in providers or []:
self.register(provider)
def register(self, provider: GenerationProvider) -> None:
code = provider.provider_code
if code in self._by_code:
# 重复注册意味着装配出错。静默覆盖会让「到底哪个实例在处理任务」
# 取决于注册顺序,是最难查的一类问题。
raise ValueError(f"provider_code 重复注册:{code}")
self._by_code[code] = provider
def find(self, provider_code: str) -> GenerationProvider | None:
"""未知 code 返回 None。
对一个已持久化的任务而言,未知 code 意味着当初提交它的适配器已被移除。
调用方应当作**不可重试**的失败,而不是循环重试到预算耗尽。
"""
return self._by_code.get(provider_code)
@property
def codes(self) -> frozenset[str]:
return frozenset(self._by_code)
def __len__(self) -> int:
return len(self._by_code)