readded message logging

.gitignore

@@ -2,4 +2,5 @@
 /experiments/
 /test/
 /.idea/
-*/__pycache__/
+*/__pycache__/
+/data/

src/agent_interfaces.py

@@ -6,6 +6,7 @@ from colorama import Fore, Style
 from pydantic import BaseModel, ValidationError
 
 from message import Message, AgentMessage
+from data_collection import save
 
 FORMAT_INSTRUCTIONS = """The output should be reformatted as a JSON instance that conforms to the JSON schema below.
 Here is the output schema:
@@ -33,6 +34,7 @@ class BaseAgentInterface:
     def add_message(self, message: Message):
         """Adds a message to the message history, without generating a response."""
         bound_message = AgentMessage.from_message(message, self.id, len(self.messages))
+        save(bound_message)
         self.messages.append(bound_message)
 
     def respond_to(self, message: Message) -> Message:

src/data_collection.py

@@ -0,0 +1,32 @@
+import os
+from typing import NewType
+
+import pydantic
+
+import message
+import player
+
+from pydantic import BaseModel
+
+Model = NewType("Model", BaseModel)
+
+
+data_dir = os.path.join(os.pardir, "data")
+
+
+def save(log_object: Model):
+    log_file = get_log_file(log_object)
+
+    with open(log_file, "a+") as f:
+        f.write(log_object.model_dump_json() + "\n")
+
+
+def get_log_file(log_object: Model) -> str:
+    match type(log_object):
+        case message.AgentMessage:
+            log_file = "messages.jsonl"
+        # ...
+        case _:
+            raise ValueError(f"Unknown log object type: {type(log_object)}")
+
+    return os.path.join(data_dir, log_file)