feat: lambda handler

Author: luisds95

Diff for: .gitignore

@@ -1,3 +1,5 @@
+tmp/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

Diff for: poetry.lock

@@ -4,11 +4,13 @@ version = "0.1.0"
 description = "TicTacToe, solved with Reinforcement Learning"
 authors = ["Luis Da Silva <[email protected]>"]
 license = "MIT"
+packages = [
+    { include = "tictactoe" },
+]
 
 [tool.poetry.dependencies]
 python = "^3.8"
 click = "^8.0.3"
-numpy = "^1.22.1"
 Flask = "^2.0.2"
 gunicorn = "^20.1.0"
 

Diff for: pyproject.toml

@@ -1,7 +1,5 @@
 from typing import Dict, Optional, Union
 
-import numpy as np
-
 from tictactoe.agent.base import TrainableAgent
 from tictactoe.agent.enums import AgentTypes
 from tictactoe.database import Database
@@ -28,6 +26,7 @@ def __init__(
 
         self.set_player_number(player_number)
 
+    @property
     def agent_type(self) -> AgentTypes:
         return AgentTypes.searcher
 
@@ -66,9 +65,9 @@ def _evaluate_moves(self, board: Board, current_depth: int) -> Dict[int, int]:
                         ).values()
                     ]
                     values[move] = (
-                        np.max(future_values)
+                        max(future_values)
                         if new_board.next_player == self.player_number.value
-                        else np.min(future_values)
+                        else min(future_values)
                     )
             elif new_board.outcome == GameOutcome.DRAW:
                 values[move] = self.DRAW_REWARD

Diff for: tictactoe/agent/searcher.py

@@ -2,14 +2,9 @@
 
 from flask import Flask, request
 
-from tictactoe.agent import ExhaustiveSearchAgent
 from tictactoe.database import DictDatabase
-from tictactoe.environment import (
-    DICT_DATABASE_FILE,
-    Board,
-    GameOutcome,
-    InvalidBoardError,
-)
+from tictactoe.environment import DICT_DATABASE_FILE, Board, InvalidBoardError
+from tictactoe.game import get_move_response
 
 app = Flask(__name__)
 database = DictDatabase(DICT_DATABASE_FILE)
@@ -32,18 +27,12 @@ def get_action() -> str:
         {"action": 2, "result": "NA"}
     """
     args = request.args.to_dict()
+    board_str = args.get("board", "0")
 
     try:
-        board = Board(args.get("board", "0"))
+        board = Board(board_str)
     except InvalidBoardError as e:
         return json.dumps({"error": str(e)})
 
-    if board.outcome == GameOutcome.NA:
-        agent = ExhaustiveSearchAgent(db=database, player_number=board.next_player)
-        action = agent.get_action(board)
-        board.make_move(action)
-        response = {"action": action, "result": board.outcome.name}
-    else:
-        response = {"result": board.outcome.name}
-
+    response = get_move_response(board, database)
     return json.dumps(response)

Diff for: tictactoe/app.py

@@ -132,7 +132,7 @@ def search_for_winner_in_columns(self) -> GameOutcome:
         return GameOutcome.NA
 
     def search_for_winner_in_rows(self) -> GameOutcome:
-        for row in range(0, N_ROWS ** 2, N_ROWS):
+        for row in range(0, N_ROWS**2, N_ROWS):
             if (
                 self.state[row] != "0"
                 and self.state[row] == self.state[row + 1] == self.state[row + 2]

Diff for: tictactoe/environment/board.py

@@ -1 +1,2 @@
+from tictactoe.game.move import get_move_response
 from tictactoe.game.play import Game

Diff for: tictactoe/game/__init__.py

@@ -0,0 +1,16 @@
+from typing import Any, Dict
+
+from tictactoe.agent import ExhaustiveSearchAgent
+from tictactoe.database import Database
+from tictactoe.environment import Board, GameOutcome
+
+
+def get_move_response(board: Board, database: Database) -> Dict[str, Any]:
+    if board.outcome == GameOutcome.NA:
+        agent = ExhaustiveSearchAgent(db=database, player_number=board.next_player)
+        action = agent.get_action(board)
+        board.make_move(action)
+        response = {"action": action, "result": board.outcome.name}
+    else:
+        response = {"result": board.outcome.name}
+    return response

Diff for: tictactoe/game/move.py

@@ -0,0 +1,22 @@
+import json
+from typing import Any, Dict
+
+from tictactoe.database import DictDatabase
+from tictactoe.environment import DICT_DATABASE_FILE, Board, InvalidBoardError
+from tictactoe.game import get_move_response
+from pathlib import Path
+
+
+def move_handler(event: Dict[str, Any], context: Dict[str, Any]) -> str:
+    board_str = event.get("board", "0")
+
+    try:
+        board = Board(board_str)
+    except InvalidBoardError as e:
+        return json.dumps({"error": str(e)})
+
+    database = DictDatabase(data_path=DICT_DATABASE_FILE)
+    database.connect()
+
+    response = get_move_response(board, database)
+    return json.dumps(response)