0c05463617
- 语音：序数解析（第一个/第二个等）、解析失败计数与 API detail.retry_remaining； 百度 ASR 固定 dev_pid 为普通话；SurgeryPipelineError 支持 extra 并入 HTTP detail。 - Demo：demo 路由与假 RTSP、客户端 index 与 README；BackendResolver 与配置调整。 - 可观测：消耗 TSV 日志、语音文件日志、终端 Markdown 辅助；相关测试与依赖更新。 - 注意：.env 仍被 gitignore，本地密钥不会进入本提交。 Made-with: Cursor
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
from app.services.voice_confirm import (
|
|
build_prompt_text,
|
|
match_voice_choice_against_candidates,
|
|
parse_voice_choice,
|
|
)
|
|
|
|
|
|
def test_parse_voice_choice_substring() -> None:
|
|
assert parse_voice_choice("用的是纱布对吧", ["纱布", "缝线"]) == "纱布"
|
|
|
|
|
|
def test_parse_voice_choice_numeric() -> None:
|
|
assert parse_voice_choice("第2个", ["纱布", "缝线", "钳子"]) == "缝线"
|
|
|
|
|
|
def test_parse_voice_choice_ordinal_chinese() -> None:
|
|
opts = ["纱布", "缝线", "钳子"]
|
|
assert parse_voice_choice("第一个", opts) == "纱布"
|
|
assert parse_voice_choice("第一个。", opts) == "纱布"
|
|
assert parse_voice_choice("第2个", opts) == "缝线"
|
|
assert parse_voice_choice("第二", opts) == "缝线"
|
|
assert parse_voice_choice("选3", opts) == "钳子"
|
|
assert parse_voice_choice("选项2", ["纱布", "缝线"]) == "缝线"
|
|
|
|
|
|
def test_parse_voice_choice_single_chinese_digit_with_few_options() -> None:
|
|
assert parse_voice_choice("一", ["纱布", "缝线"]) == "纱布"
|
|
assert parse_voice_choice("两", ["纱布", "缝线"]) == "缝线"
|
|
|
|
|
|
def test_parse_voice_choice_negative() -> None:
|
|
assert parse_voice_choice("不是", ["纱布", "缝线"]) is None
|
|
|
|
|
|
def test_build_prompt_contains_options() -> None:
|
|
text = build_prompt_text([("纱布", 0.4), ("缝线", 0.3)])
|
|
assert "纱布" in text
|
|
assert "缝线" in text
|
|
|
|
|
|
def test_match_voice_against_full_candidate_list() -> None:
|
|
assert (
|
|
match_voice_choice_against_candidates(
|
|
"刚才用的是止血钳",
|
|
["纱布", "缝线", "止血钳"],
|
|
)
|
|
== "止血钳"
|
|
)
|
|
|
|
|
|
def test_match_voice_longest_candidate_first() -> None:
|
|
assert (
|
|
match_voice_choice_against_candidates(
|
|
"拿的一次性止血钳",
|
|
["止血钳", "一次性止血钳"],
|
|
)
|
|
== "一次性止血钳"
|
|
)
|