18 lines
527 B
Python
18 lines
527 B
Python
from app.chat_qa.qa_state import QAState
|
||
|
||
|
||
# ==================== 条件路由 ====================
|
||
def route_by_quality(state: QAState) -> str:
|
||
"""根据质检结果和重试次数决定路由"""
|
||
is_valid = state.get("is_valid", False)
|
||
retry = state["retry_count"]
|
||
qas = state.get("current_qas", [])
|
||
|
||
# 质检通过：有有效QA且评分足够
|
||
if is_valid :
|
||
return "pass"
|
||
|
||
# 质检不通过，但还有重试机会（最多3次）
|
||
if retry < 3:
|
||
return "retry"
|
||
return "fail" |