| from typing import Optional |
|
|
| from agent_build_sdk.model.werewolf_model import ( |
| STATUS_DISCUSS, |
| STATUS_SHERIFF, |
| STATUS_SHERIFF_ELECTION, |
| STATUS_SHERIFF_SPEECH, |
| STATUS_SHERIFF_PK, |
| STATUS_SHERIFF_SPEECH_ORDER, |
| STATUS_SHERIFF_VOTE, |
| STATUS_VOTE, |
| STATUS_SKILL, |
| STATUS_WOLF_SPEECH, |
| ) |
|
|
| from core import ruleset |
| from core.actions import ( |
| DiscussAction, |
| SheriffRunAction, |
| SheriffSpeechOrderAction, |
| SheriffTransferAction, |
| SheriffVoteAction, |
| SkillAction, |
| VoteAction, |
| ) |
| from core.output_guard import normalize_llm_text |
|
|
|
|
| def parse_action(status: str, text: Optional[str]): |
| s = normalize_llm_text(text) |
| if status in {STATUS_DISCUSS, STATUS_WOLF_SPEECH, STATUS_SHERIFF_SPEECH, STATUS_SHERIFF_PK}: |
| return DiscussAction(text=s) |
| if status == STATUS_VOTE: |
| return VoteAction(target=s) |
| if status == STATUS_SKILL: |
| if s == ruleset.NO_SHOOT or s == ruleset.WITCH_NO_USE: |
| return SkillAction(target=None) |
| return SkillAction(target=s) |
| if status == STATUS_SHERIFF_VOTE: |
| return SheriffVoteAction(target=s) |
| if status == STATUS_SHERIFF: |
| return SheriffTransferAction(target=s) |
| if status == STATUS_SHERIFF_ELECTION: |
| if s == ruleset.SHERIFF_RUN: |
| return SheriffRunAction(join=True) |
| if s == ruleset.SHERIFF_NOT_RUN: |
| return SheriffRunAction(join=False) |
| return None |
| if status == STATUS_SHERIFF_SPEECH_ORDER: |
| return SheriffSpeechOrderAction(order=s) |
| return None |
|
|
