Upload stratego/utils/game_move_tracker.py with huggingface_hub

stratego/utils/game_move_tracker.py

@@ -0,0 +1,104 @@
+# stratego/memory_tracker.py
+from __future__ import annotations
+import datetime
+from typing import List, Dict, Any, Optional
+
+class GameMoveTracker:
+    """
+    In-memory tracker of all moves (for both players).
+    Does NOT reveal piece identities, only events like capture/bomb/flag.
+    Produces a clean text summary suitable for LLM prompt injection.
+    """
+
+    def __init__(self):
+        self.turn: int = 0
+        self.history: List[Dict[str, Any]] = []   # global chronological history
+
+    # ---------------------------------------------------------
+    # RECORDING MOVES
+    # ---------------------------------------------------------
+    def record(
+        self,
+        player: int,
+        move: str,
+        event: Optional[str] = None,   # "capture", "bomb", "flag", etc.
+        extra: Optional[str] = None    # captured piece? "explosion"? etc.
+    ):
+        """
+        Store one full move entry, but without revealing piece identities.
+        """
+        entry = {
+            "turn": self.turn,
+            "player": player,
+            "move": move,   # e.g. "E4 F4"
+            "event": event, # optional special event
+            "extra": extra, # optional detail
+            "timestamp": datetime.datetime.now().isoformat(timespec="seconds"),
+        }
+
+        self.history.append(entry)
+        self.turn += 1
+
+    # ---------------------------------------------------------
+    # ACCESSORS
+    # ---------------------------------------------------------
+    def get_player_moves(self, pid: int) -> List[Dict[str, Any]]:
+        """Moves made by a specific player."""
+        return [h for h in self.history if h["player"] == pid]
+
+    def get_opponent_moves(self, pid: int) -> List[Dict[str, Any]]:
+        """Moves made by the opponent."""
+        return [h for h in self.history if h["player"] != pid]
+
+    def last_move(self) -> Optional[Dict[str, Any]]:
+        """Most recent move."""
+        return self.history[-1] if self.history else None
+
+    # ---------------------------------------------------------
+    # GENERATE STRING FOR PROMPT
+    # ---------------------------------------------------------
+    def to_prompt_string(self, player_id: int) -> str:
+        """
+        Return a clean string that summarizes the match so far.
+        Keeps ONLY the last 20 moves to avoid loops.
+        """
+
+        if not self.history:
+            return "No previous moves have been played.\n"
+
+        # ➤ KEEP ONLY LAST 20 ENTRIES
+        MAX_HISTORY = 20
+        if len(self.history) > MAX_HISTORY:
+            # delete oldest until only 20 remain
+            self.history = self.history[-MAX_HISTORY:]
+
+        lines = ["Game History (most recent last):"]
+
+        for entry in self.history:
+            turn = entry["turn"]
+            pid  = entry["player"]
+            move = entry["move"]
+            event = entry["event"]
+            extra = entry["extra"]
+
+            if pid == player_id:
+                prefix = f"Turn {turn}: You played {move}"
+            else:
+                prefix = f"Turn {turn}: Opponent played {move}"
+
+            if event == "capture":
+                prefix += " (capture occurred)"
+            elif event == "bomb":
+                prefix += " (bomb explosion)"
+            elif event == "flag":
+                prefix += " (FLAG CAPTURE — game-ending)"
+            elif event == "invalid":
+                prefix += " (invalid move?)"
+
+            if extra:
+                prefix += f" — {extra}"
+
+            lines.append(prefix)
+
+        return "\n".join(lines) + "\n"
+