added game and player level logging

src/game.py

@@ -11,13 +11,15 @@ NUMBER_OF_PLAYERS = 5
 
 class Game:
     log_dir = os.path.join(os.pardir, "experiments")
-    player_log_file = "{game_id}-{role}-{player_num}.jsonl"
+    player_log_file = "{player_id}.jsonl"
     parser_log_file = "{game_id}-parser.jsonl"
+    game_log_file = "{game_id}-game.jsonl"
 
-    def __init__(self,
-                 human_name: str = None,
-                 number_of_players: int = NUMBER_OF_PLAYERS
-                 ):
+    def __init__(
+            self,
+            human_name: str = None,
+            number_of_players: int = NUMBER_OF_PLAYERS
+    ):
 
         # Game ID
         self.game_id = game_id()
@@ -28,6 +30,7 @@ class Game:
             self.human_index = random_index(number_of_players)
         else:
             ai_names = random_names(number_of_players)
+            self.human_index = None
 
         # Choose Chameleon
         self.chameleon_index = random_index(number_of_players)
@@ -47,10 +50,14 @@ 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))
+            player_id = f"{self.game_id}-{i + 1}-{role}"
 
-            self.players.append(Player(name, controller, role, log_filepath=log_path))
+            log_path = os.path.join(
+                self.log_dir,
+                self.player_log_file.format(player_id=player_id)
+            )
+
+            self.players.append(Player(name, controller, role, player_id, log_filepath=log_path))
 
         # Game State
         self.player_responses = []
@@ -74,9 +81,9 @@ class Game:
         self.player_responses = []
         herd_animal = random_animal()
 
-        game_over = False
+        winner = None
 
-        while not game_over:
+        while not winner:
             # Phase I: Collect Player Animal Descriptions
             for player in self.players:
                 if player.role == "chameleon":
@@ -113,10 +120,8 @@ class Game:
                 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:
@@ -142,7 +147,6 @@ class Game:
                 accused_player = count_chameleon_votes(player_votes)
 
                 if accused_player:
-                    game_over = True
                     if accused_player == self.players[self.chameleon_index].name:
                         winner = "herd"
                     else:
@@ -156,3 +160,14 @@ class Game:
         # Herd Wins by Correctly Guessing Chameleon - 2 points (each)
 
         # Log Game Info
+        game_log = {
+            "game_id": self.game_id,
+            "herd_animal": herd_animal,
+            "number_of_players": len(self.players),
+            "human_player": self.players[self.human_index].id if self.human_index else "None",
+            "chameleon": self.players[self.chameleon_index].id,
+            "winner": winner
+        }
+        game_log_path = os.path.join(self.log_dir, self.game_log_file.format(game_id=self.game_id))
+
+        log(game_log, game_log_path)

src/game_utils.py

@@ -3,6 +3,7 @@ Utilities for the game including random selections and prompts.
 """
 import random
 import string
+import json
 from collections import Counter
 
 ALPHABET = string.ascii_lowercase + string.digits
@@ -46,6 +47,12 @@ def count_chameleon_votes(player_votes: list[str]) -> str | None:
         return None
 
 
+def log(log_object, log_file):
+    with open(log_file, "a+") as f:
+        f.write(json.dumps(log_object) + "\n")
+
+
+# Prompt Stuff
 def fetch_prompt(prompt_name):
     return prompts[prompt_name]
 

src/main.py

@@ -8,6 +8,8 @@ def main():
 
     game = Game(human_name=name)
 
+    # game = Game()
+
     asyncio.run(game.start())
 
 

src/player.py

@@ -1,10 +1,12 @@
 import os
+import json
 import asyncio
 
 import openai
 from agents import LogMessagesKani
 from kani.engines.openai import OpenAIEngine
 
+from game_utils import log
 
 # Using TGI Inference Endpoints from Hugging Face
 # api_type = "tgi"
@@ -24,8 +26,9 @@ openai_engine = OpenAIEngine(model="gpt-3.5-turbo")
 
 
 class Player:
-    def __init__(self, name: str, controller_type: str, role: str, log_filepath: str = None):
+    def __init__(self, name: str, controller_type: str, role: str, id: str = None, log_filepath: str = None):
         self.name = name
+        self.id = id
         self.controller = controller_type
         if controller_type == "ai":
             self.kani = LogMessagesKani(openai_engine, log_filepath=log_filepath)
@@ -33,6 +36,15 @@ class Player:
         self.role = role
         self.messages = []
 
+        if log_filepath:
+            player_info = {
+                "id": self.id,
+                "name": self.name,
+                "role": self.role,
+                "controller": controller_type,
+            }
+            log(player_info, log_filepath)
+
     async def respond_to(self, prompt: str) -> str:
         """Makes the player respond to a prompt. Returns the response."""
         # Generate a response from the controller