49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from backend.app.services.adaptive_rag_service import AdaptiveRAGService
|
|
|
|
|
|
def test_feishu_no_explicit_deep_should_not_force():
|
|
should = AdaptiveRAGService._should_force_deep_for_channel(
|
|
asked_via="feishu",
|
|
question="不是啊，我说的是 ai 引擎驱动方式",
|
|
effective_question="不是啊，我说的是 ai 引擎驱动方式",
|
|
reason="correction_turn",
|
|
)
|
|
assert not should
|
|
|
|
|
|
def test_feishu_explicit_deep_can_force():
|
|
should = AdaptiveRAGService._should_force_deep_for_channel(
|
|
asked_via="feishu",
|
|
question="请深入分析调用链并给证据链",
|
|
effective_question="请深入分析调用链并给证据链",
|
|
reason="engine_process_query",
|
|
)
|
|
assert should
|
|
|
|
|
|
def test_pmflow_force_deep_kept():
|
|
should = AdaptiveRAGService._should_force_deep_for_channel(
|
|
asked_via="pmflow",
|
|
question="不是啊，我说的是 ai 引擎驱动方式",
|
|
effective_question="不是啊，我说的是 ai 引擎驱动方式",
|
|
reason="correction_turn",
|
|
)
|
|
assert should
|
|
|
|
|
|
def test_feishu_auto_retry_requires_explicit_deep():
|
|
allow_plain = AdaptiveRAGService._allow_auto_deep_retry_for_channel(
|
|
asked_via="feishu",
|
|
question="这个问题不太对",
|
|
effective_question="这个问题不太对",
|
|
reason="snapshot_only",
|
|
)
|
|
allow_explicit = AdaptiveRAGService._allow_auto_deep_retry_for_channel(
|
|
asked_via="feishu",
|
|
question="请深度分析这个调用链",
|
|
effective_question="请深度分析这个调用链",
|
|
reason="snapshot_only",
|
|
)
|
|
assert not allow_plain
|
|
assert allow_explicit
|