added chameleon guessing logic

src/game.py

@@ -8,16 +8,16 @@ from parser import ParserKani
 # Default Values
 NUMBER_OF_PLAYERS = 5
 
-class Game:
 
+class Game:
     log_dir = os.path.join(os.pardir, "experiments")
     player_log_file = "{game_id}-{role}-{player_num}.jsonl"
     parser_log_file = "{game_id}-parser.jsonl"
 
     def __init__(self,
-            human_name: str = None,
-            number_of_players: int = NUMBER_OF_PLAYERS
-        ):
+                 human_name: str = None,
+                 number_of_players: int = NUMBER_OF_PLAYERS
+                 ):
 
         # Game ID
         self.game_id = game_id()
@@ -47,7 +47,8 @@ class Game:
             else:
                 role = "herd"
 
-            log_path = os.path.join(self.log_dir, self.player_log_file.format(game_id=self.game_id, role=role, player_num=i))
+            log_path = os.path.join(self.log_dir,
+                                    self.player_log_file.format(game_id=self.game_id, role=role, player_num=i))
 
             self.players.append(Player(name, controller, role, log_filepath=log_path))
 
@@ -93,13 +94,31 @@ class Game:
                 self.player_responses.append({"sender": player.name, "response": output.description})
 
             # Phase II: Chameleon Decides if they want to guess the animal (secretly)
-            chameleon_will_guess = False
+            prompt = fetch_prompt("chameleon_guess_decision")
+
+            response = await self.players[self.chameleon_index].respond_to(prompt)
+            output = await self.parser.parse(prompt, response, ChameleonGuessDecisionModel)
+
+            if output.decision == "guess":
+                chameleon_will_guess = True
+            else:
+                chameleon_will_guess = False
 
             # Phase III: Chameleon Guesses Animal or All Players Vote for Chameleon
             if chameleon_will_guess:
                 # Chameleon Guesses Animal
-                # TODO: Add Chameleon Guessing Logic
-                pass
+                prompt = fetch_prompt("chameleon_guess_animal")
+
+                response = await self.players[self.chameleon_index].respond_to(prompt)
+                output = await self.parser.parse(prompt, response, ChameleonGuessAnimalModel)
+
+                if output.animal == herd_animal:
+                    game_over = True
+                    winner = "chameleon"
+                else:
+                    game_over = True
+                    winner = "herd"
+
             else:
                 # All Players Vote for Chameleon
                 player_votes = []
@@ -137,5 +156,3 @@ class Game:
         # Herd Wins by Correctly Guessing Chameleon - 2 points (each)
 
         # Log Game Info
-
-

src/game_utils.py

@@ -73,6 +73,16 @@ Previously Mentioned Descriptions:
 {player_responses}
 """
 
+_chameleon_guess_decision = """\
+You now have the opportunity to guess what animal the other players are pretending to be.
+If you guess correctly you will WIN, if you guess incorrectly you will LOSE.
+If you believe you know what animal the other players are pretending to be make choose to GUESS, otherwise choose to PASS.
+"""
+
+_chameleon_guess_animal = """\
+What animal do you think the other players are pretending to be?
+"""
+
 _vote_prompt = """\
 Below are the responses from all players. Now it is time to vote. Choose from the players below who you think the Chameleon is.
 {player_responses}
@@ -81,6 +91,8 @@ Below are the responses from all players. Now it is time to vote. Choose from th
 prompts = {
     "herd_animal": _game_rules + _herd_animal,
     "chameleon_animal": _game_rules + _chameleon_animal,
+    "chameleon_guess_decision": _chameleon_guess_decision,
+    "chameleon_guess_animal": _chameleon_guess_animal,
     "vote": _vote_prompt
 }
 

src/models.py

@@ -1,3 +1,5 @@
+from typing import Annotated
+
 from pydantic import BaseModel, field_validator
 
 MAX_DESCRIPTION_LEN = 10
@@ -31,6 +33,21 @@ class AnimalGuessModel(BaseModel):
     animal_name: str
 
 
+class ChameleonGuessDecisionModel(BaseModel):
+    decision: Annotated[str, "Must be one of: ['guess', 'pass']"]
+
+    @field_validator('decision')
+    @classmethod
+    def check_decision(cls, v) -> str:
+        if v.lower() not in ['guess', 'pass']:
+            raise ValueError("Decision must be one of: ['guess', 'pass']")
+        return v
+
+
+class ChameleonGuessAnimalModel(BaseModel):
+    animal: str
+
+
 class VoteModel(BaseModel):
     vote: str