fix and test

Author: Luncenok

scripts/random_game.py

@@ -56,7 +56,8 @@ def main(reset=False, always_kill=False, remove_last_n=0):
                 try:
                     # Here, you would insert your custom LLM call and log probability logic.
                     action_taken = random.choice(actions_pd)
-                    llm_response, cot = action_taken.text, "<think>Was thinking about this option: " + action_taken.text + "</think>"
+                    action_taken.set_stories()
+                    llm_response, cot = action_taken.command_perspective, "<think>Was thinking about this option: " + action_taken.command_perspective + "</think>"
                 except KeyboardInterrupt:
                     action_idx, llm_response, cot, _ = prompt_manual_fallback_action(actions_pd)
 
@@ -87,7 +88,7 @@ def main(reset=False, always_kill=False, remove_last_n=0):
                 action_taken = random.choice(kill_actions)
             elif current_player.role == PlayerRole.CREWMATE and task_actions:
                 action_taken = random.choice(task_actions)
-            llm_response, cot = action_taken.text, "<think>Was thinking about this option: " + action_taken.text + "</think>"
+            llm_response, cot = action_taken.command_perspective, "<think>Was thinking about this option: " + action_taken.command_perspective + "</think>"
         except KeyboardInterrupt:
             action_idx, llm_response, cot, _ = prompt_manual_fallback_action(actions_player_can_take)
 

src/among_them/models/action.py

@@ -26,7 +26,7 @@ def __init__(
     def set_stories(self):
         if self.type == ActionType.MOVE:
             self.command_perspective = f"move to {self.target_location.value}"
-            self.agent_perspective = f"You are in {self.target_location.value}."
+            self.agent_perspective = f"You moved to {self.target_location.value}."
             self.observer_perspective = f"You saw {self.player_name} move to {self.target_location.value}."
             self.global_perspective = f"{self.player_name} moved to {self.target_location.value}."
         elif self.type == ActionType.WAIT:

src/among_them/utils/player_utils.py

@@ -39,7 +39,7 @@ def get_last_player_action(history: List[History], player: Player) -> History:
     for i in range(len(history) - 1, -1, -1):
         if history[i].action_taken.player_name == player.name:
             return history[i]
-        elif history[i].action_taken.target_message and history[i].action_taken.target_message.startswith("Everyone is in the cafeteria and start from there"):
+        elif getattr(history[i].action_taken, "target_message", None) and history[i].action_taken.target_message.startswith("Everyone is in the cafeteria and start from there"):
             return history[i]
     return history[0]
 

src/among_them/utils/ui_utils.py

@@ -35,7 +35,7 @@ def prompt_manual_fallback_action(actions: List[Action]) -> Tuple[int, str, str,
     """Handles manual action selection when AI generation is interrupted."""
     print("\nAI action generation interrupted. Please choose an action manually:")
     for i, action in enumerate(actions):
-        print(f"{i + 1}. {action.text}")
+        print(f"{i + 1}. {action.set_stories().command_perspective}")
 
     if actions[0].type == ActionType.SPEAK:
         try:
@@ -49,10 +49,10 @@ def prompt_manual_fallback_action(actions: List[Action]) -> Tuple[int, str, str,
             choice_idx = int(choice) - 1
             if 0 <= choice_idx < len(actions):
                 selected_action = actions[choice_idx]
-                print(f"You chose: {selected_action.text}")
+                print(f"You chose: {selected_action.command_perspective}")
                 return (
                     choice_idx,
-                    selected_action.text,
+                    selected_action.command_perspective,
                     "",
                     {},
                 )
@@ -64,7 +64,7 @@ def prompt_manual_fallback_action(actions: List[Action]) -> Tuple[int, str, str,
             print("\nInput stream closed. Defaulting to first action.")
             return (
                 0,
-                actions[0].text,
+                actions[0].command_perspective,
                 "",
                 {},
             )