61db82cb9b
人物图片功能重做: bbox 随核心视频分析那一次 Gemini 调用一并产出(prompts.py 加
person_appearances.bbox 字段, [ymin,xmin,ymax,xmax] 0-1000 归一化), frame_service
直接用存好的 bbox 裁剪头像/事件缩略图, 删除原来"展示时额外调用 Gemini 定位人物"的
整套逻辑(locate_person_bbox/VLM 校验/熔断), 从架构上消除与核心视频分析共抢配额的
问题; 用真实数据验证裁剪结果正确框住人物本体。
NVIDIA 模型修复: 实测原配置的 3 个模型均不可用(asset_id 引用 500/400, 不支持视频),
改用 nemotron-3-nano-omni 的 base64 内嵌视频方式(唯一实测打通), 加 max_base64_mb
防止对大文件做注定失败的编码。
Gemini 多 Key 轮换: 支持 extra_api_keys 配置多个独立项目的 key, 配额用尽时依次
换 key 重试(每换 key 需重新上传, Files API 按项目隔离)。
稳定性加固: CircuitBreaker HALF_OPEN 清空旧失败计数(修复探测一失败就重新 OPEN 的
bug); chat() 统一接入熔断器(原来只有视频分析路径检查); NVIDIA 适配器改用共享
json_parser(原来自己重复实现且不做 schema 校验); Gemini Files API 上传超时也尝试
清理远程孤儿文件; video_processor/video_queue 里直接操作 OracleDB._conn 的裸 SQL
改走新增的 set_event_start_time/mark_video_invalid/reset_video_to_pending 方法;
/health 加入队列线程存活状态; 密钥改用 ${ENV_VAR} 引用(.env 已支持自动加载),
不再明文写入 config.yaml。
工程质量: 新增 fam-edge/tests(32 个单元测试, 覆盖熔断器状态机/JSON 解析容错/
时间戳解析/bbox 坐标换算/多 key 解析), 新增 scripts/smoke_test.py(发版前接口
稳定性检查); 清理死代码(OllamaAdapter.analyze_frames、get_sync_delta 死分支、
未使用的 vision_timeout/max_concurrent_tasks 配置项); 修正 get_events_for_label
排序(改最近优先 + 过滤畸形历史时间戳)。
已部署 Oracle 并跑通 smoke test 全部 6 项检查。
Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
import time
|
||
|
||
from fam_edge.model_adapters.circuit_breaker import CircuitBreaker
|
||
|
||
|
||
def test_closed_stays_closed_below_threshold():
|
||
cb = CircuitBreaker(threshold=3, cooldown=1)
|
||
cb.record_failure()
|
||
cb.record_failure()
|
||
assert not cb.is_open()
|
||
assert cb.state == 'CLOSED'
|
||
|
||
|
||
def test_opens_at_threshold():
|
||
cb = CircuitBreaker(threshold=3, cooldown=1)
|
||
for _ in range(3):
|
||
cb.record_failure()
|
||
assert cb.is_open()
|
||
assert cb.state == 'OPEN'
|
||
|
||
|
||
def test_half_open_after_cooldown_and_clears_stale_failures():
|
||
"""进入 HALF_OPEN 时应清空旧失败计数，探测调用失败一次不该立刻把它"凑数"重新 OPEN。"""
|
||
cb = CircuitBreaker(threshold=3, cooldown=0.05)
|
||
for _ in range(3):
|
||
cb.record_failure()
|
||
assert cb.is_open()
|
||
time.sleep(0.1)
|
||
# 冷却期已过，第一次 is_open() 调用把状态转为 HALF_OPEN 并放行一次探测
|
||
assert cb.is_open() is False
|
||
assert cb.state == 'HALF_OPEN'
|
||
assert len(cb.failures) == 0
|
||
# 探测失败一次：由于 deque 已清空，不应该只凭这一次失败就再次 OPEN
|
||
# （state 仍是 HALF_OPEN，不是 OPEN，is_open() 只在 state=='OPEN' 时为 True）
|
||
cb.record_failure()
|
||
assert cb.state != 'OPEN'
|
||
assert not cb.is_open()
|
||
|
||
|
||
def test_half_open_probe_success_closes():
|
||
cb = CircuitBreaker(threshold=2, cooldown=0.05)
|
||
cb.record_failure()
|
||
cb.record_failure()
|
||
time.sleep(0.1)
|
||
assert cb.is_open() is False # 触发 HALF_OPEN 转换
|
||
cb.record_success()
|
||
assert cb.state == 'CLOSED'
|
||
assert len(cb.failures) == 0
|
||
|
||
|
||
def test_disabled_never_opens():
|
||
cb = CircuitBreaker(threshold=1, cooldown=1, enabled=False)
|
||
cb.record_failure()
|
||
cb.record_failure()
|
||
assert not cb.is_open()
|