Exposes number of guesses for the guesser

codenames/challenge_runner.py

@@ -18,8 +18,9 @@ def play_challenge(
     Play a challenge game with LLM spymaster and student guesser code.
     
     The guesser_fn should have signature:
-        guesser_fn(clue: str, board_state: list[str]) -> Optional[str]
-    where board_state is a list of unrevealed word strings.
+        guesser_fn(clue: str, board_state: list[str], number: int) -> Optional[str]
+    where board_state is a list of unrevealed word strings and number is the
+    spymaster's clue number (max guesses = number + 1).
     """
     game = ChallengeGame(dictionary=dictionary, seed=seed)
     master = MasterPlayer(dictionary, model_name=model_name)
@@ -57,13 +58,15 @@ def play_challenge(
         while not game.won and not game.lost and game.clue_word is not None:
             unrevealed = [c.word for c in game.board.cards if not c.revealed]
             try:
-                guess_word = guesser_fn(game.clue_word, unrevealed)
+                guess_word = guesser_fn(game.clue_word, unrevealed, game.clue_number)
             except Exception as e:
                 if verbose:
                     print(f"Guesser error: {e}")
+                game.end_round()
                 break
 
             if guess_word is None:
+                game.end_round()
                 break
 
             role, msg = game.guess(guess_word)

my_guesser.py

@@ -9,9 +9,10 @@ _dictionary = load_dictionary()
 _index = EmbeddingIndex(_dictionary)
 
 
-def guesser(clue: str, board_state: list[str]) -> Optional[str]:
+def guesser(clue: str, board_state: list[str], number: int) -> Optional[str]:
     """
-    Given a clue word and list of unrevealed board words,
-    return a word.
+    Given a clue word, list of unrevealed board words, and the spymaster's
+    number, return a word to guess, or None to stop guessing.
+    You can make up to (number + 1) guesses per round.
     """
     return board_state[0]