Improved board representation, misc cleanup

arena/board.py

@@ -6,7 +6,7 @@ YELLOW = -1
 EMPTY = 0
 show = {EMPTY: "⚪️", RED: "🔴", YELLOW: "🟡"}
 pieces = {EMPTY: "", RED: "red", YELLOW: "yellow"}
-simple = {EMPTY: ".", RED: "R", YELLOW: "Y"}
+simple = {EMPTY: "_", RED: "R", YELLOW: "Y"}
 cols = "ABCDEFG"
 
 
@@ -85,10 +85,10 @@ class Board:
         """
         An alternative representation, used in prompting so that the LLM sees this 2 ways
         """
-        result = "ABCDEFG\n"
+        result = " A B C D E F G\n"
         for y in range(6):
             for x in range(7):
-                result += simple[self.cells[5 - y][x]]
+                result += " " + simple[self.cells[5 - y][x]]
             result += "\n"
         return result
 

arena/llm.py

@@ -34,13 +34,8 @@ class LLM(ABC):
         :param max_tokens: max number of tokens to generate
         :return: the response from the AI
         """
-        print("_____")
-        print(f"Calling {self.model_name}")
-        print("System prompt:\n" + system)
-        print("User prompt:\n" + user)
+
         result = self.protected_send(system, user, max_tokens)
-        print("Response:\n" + result)
-        print("_____")
         left = result.find("{")
         right = result.rfind("}")
         if left > -1 and right > -1:
@@ -58,9 +53,9 @@ class LLM(ABC):
             try:
                 return self._send(system, user, max_tokens)
             except Exception as e:
-                print(f"Exception on calling LLM of {e}")
+                logging.error(f"Exception on calling LLM of {e}")
                 if retries:
-                    print("Waiting 2s and retrying")
+                    logging.warning("Waiting 2s and retrying")
                     time.sleep(2)
         return "{}"
 
@@ -301,7 +296,9 @@ class Ollama(LLM):
         )
         reply = response.choices[0].message.content
         if "</think>" in reply:
-            print("Thoughts:\n" + reply.split("</think>")[0].replace("<think>", ""))
+            logging.info(
+                "Thoughts:\n" + reply.split("</think>")[0].replace("<think>", "")
+            )
             reply = reply.split("</think>")[1]
         return reply
 
@@ -357,7 +354,9 @@ class DeepSeekLocal(LLM):
         )
         reply = response.choices[0].message.content
         if "</think>" in reply:
-            print("Thoughts:\n" + reply.split("</think>")[0].replace("<think>", ""))
+            logging.info(
+                "Thoughts:\n" + reply.split("</think>")[0].replace("<think>", "")
+            )
             reply = reply.split("</think>")[1]
         return reply
 

arena/player.py

@@ -2,6 +2,7 @@ from arena.llm import LLM
 from arena.board import pieces, cols
 import json
 import random
+import logging
 
 
 class Player:
@@ -52,7 +53,7 @@ Here is the current board, with row 1 at the bottom of the board:
 
 {board.json()}
 
-Here's another way of looking at the board visually, where R represents a red counter and Y for a yellow counter.
+Here's another way of looking at the board visually, where R represents a red counter, Y for a yellow counter, and _ represents an empty square.
 
 {board.alternative()}
 
@@ -110,7 +111,8 @@ You must pick one of these letters for your move_column: {legal_moves}{illegal_m
             self.opportunities = result.get("opportunities") or ""
             self.strategy = result.get("strategy") or ""
         except Exception as e:
-            print(f"Exception {e}")
+            logging.error(f"Exception {e}")
+            logging.exception(e)
             board.forfeit = True
             board.winner = -1 * board.player