feat: add client-side maia analysis in alternate /analyze page

Author: kevinjosethomas

package-lock.json

@@ -25,7 +25,7 @@
     "lila-stockfish-web": "^0.0.7",
     "next": "^13.4.16",
     "next-transpile-modules": "^10.0.1",
-    "onnxruntime-web": "^1.20.1",
+    "onnxruntime-web": "^1.21.0-dev.20250114-228dd16893",
     "react": "^18.2.0",
     "react-chartjs-2": "^5.2.0",
     "react-dom": "^18.2.0",

package.json

@@ -0,0 +1,307 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import {
+  Player,
+  MoveMap,
+  MaiaEvaluation,
+  ClientAnalyzedGame,
+  PositionEvaluation,
+  StockfishEvaluation,
+} from 'src/types'
+import { buildUrl } from '../utils'
+import { AvailableMoves } from 'src/types/training'
+
+export const getClientAnalyzedTournamentGame = async (
+  gameId = ['FkgYSri1'],
+) => {
+  const res = await fetch(
+    buildUrl(`analysis/analysis_list/${gameId.join('/')}`),
+  )
+
+  if (res.status === 401) {
+    throw new Error('Unauthorized')
+  }
+
+  const data = await res.json()
+  const id = data['id']
+  const termination = {
+    ...data['termination'],
+    condition: 'Normal',
+  }
+
+  const gameType = 'blitz'
+  const blackPlayer = data['black_player'] as Player
+  const whitePlayer = data['white_player'] as Player
+
+  const maiaEvals: { [model: string]: MoveMap[] } = {}
+  const stockfishEvaluations: MoveMap[] = data['stockfish_evals']
+
+  const availableMoves: AvailableMoves[] = []
+
+  for (const model of data['maia_versions']) {
+    maiaEvals[model] = data['maia_evals'][model]
+  }
+
+  for (const position of data['move_maps']) {
+    const moves: AvailableMoves = {}
+    for (const move of position) {
+      const fromTo = move.move.join('')
+      const san = move['move_san']
+      const { check, fen } = move
+
+      moves[fromTo] = {
+        board: fen,
+        check,
+        san,
+        lastMove: move.move,
+      }
+    }
+    availableMoves.push(moves)
+  }
+
+  const gameStates = data['game_states']
+
+  const moves = gameStates.map((gameState: any) => {
+    const {
+      last_move: lastMove,
+      fen,
+      check,
+      last_move_san: san,
+      evaluations: maia_values,
+    } = gameState
+
+    return {
+      board: fen,
+      lastMove,
+      san,
+      check,
+      maia_values,
+    }
+  })
+
+  const maiaEvaluations = [] as { [rating: number]: MaiaEvaluation }[]
+
+  return {
+    id,
+    blackPlayer,
+    whitePlayer,
+    moves,
+    maiaEvaluations,
+    stockfishEvaluations,
+    availableMoves,
+    gameType,
+    termination,
+  } as any as ClientAnalyzedGame
+}
+
+export const getClientAnalyzedLichessGame = async (
+  id: string,
+  pgn: string,
+  maia_model = 'maia_kdd_1500',
+) => {
+  const res = await fetch(
+    buildUrl(
+      'analysis/analyze_user_game?' +
+        new URLSearchParams({
+          maia_model,
+        }),
+    ),
+    {
+      method: 'POST',
+      body: pgn,
+      headers: {
+        'Content-Type': 'text/plain',
+      },
+    },
+  )
+
+  if (res.status === 401) {
+    throw new Error('Unauthorized')
+  }
+
+  const data = await res.json()
+
+  const termination = {
+    ...data['termination'],
+    condition: 'Normal',
+  }
+
+  const gameType = 'blitz'
+  const blackPlayer = data['black_player'] as Player
+  const whitePlayer = data['white_player'] as Player
+
+  const maiaEvals: { [model: string]: MoveMap[] } = {}
+  const positionEvaluations: { [model: string]: PositionEvaluation[] } = {}
+  const availableMoves: AvailableMoves[] = []
+
+  for (const model of data['maia_versions']) {
+    maiaEvals[model] = data['maia_evals'][model]
+    positionEvaluations[model] = Object.keys(data['maia_evals'][model]).map(
+      () => ({
+        trickiness: 1,
+        performance: 1,
+      }),
+    )
+  }
+
+  for (const position of data['move_maps']) {
+    const moves: AvailableMoves = {}
+    for (const move of position) {
+      const fromTo = move.move.join('')
+      const san = move['move_san']
+      const { check, fen } = move
+
+      moves[fromTo] = {
+        board: fen,
+        check,
+        san,
+        lastMove: move.move,
+      }
+    }
+    availableMoves.push(moves)
+  }
+
+  const gameStates = data['game_states']
+
+  const moves = gameStates.map((gameState: any) => {
+    const {
+      last_move: lastMove,
+      fen,
+      check,
+      last_move_san: san,
+      evaluations: maia_values,
+    } = gameState
+
+    return {
+      board: fen,
+      lastMove,
+      san,
+      check,
+      maia_values,
+    }
+  })
+
+  const maiaEvaluations = [] as { [rating: number]: MaiaEvaluation }[]
+  const stockfishEvaluations: StockfishEvaluation[] = []
+
+  return {
+    id,
+    blackPlayer,
+    whitePlayer,
+    moves,
+    availableMoves,
+    gameType,
+    termination,
+    maiaEvaluations,
+    stockfishEvaluations,
+    type: 'brain',
+    pgn,
+  } as ClientAnalyzedGame
+}
+
+export const getClientAnalyzedUserGame = async (
+  id: string,
+  game_type: 'play' | 'hand' | 'brain',
+  maia_model = 'maia_kdd_1500',
+) => {
+  const res = await fetch(
+    buildUrl(
+      `analysis/user/analyze_user_maia_game/${id}?` +
+        new URLSearchParams({
+          game_type,
+          maia_model,
+        }),
+    ),
+    {
+      method: 'GET',
+      headers: {
+        'Content-Type': 'text/plain',
+      },
+    },
+  )
+
+  if (res.status === 401) {
+    throw new Error('Unauthorized')
+  }
+
+  const data = await res.json()
+
+  const termination = {
+    ...data['termination'],
+    condition: 'Normal',
+  }
+
+  const gameType = 'blitz'
+  const blackPlayer = data['black_player'] as Player
+  const whitePlayer = data['white_player'] as Player
+
+  const maiaPattern = /maia_kdd_1\d00/
+
+  if (maiaPattern.test(blackPlayer.name)) {
+    blackPlayer.name = blackPlayer.name.replace('maia_kdd_', 'Maia ')
+  }
+
+  if (maiaPattern.test(whitePlayer.name)) {
+    whitePlayer.name = whitePlayer.name.replace('maia_kdd_', 'Maia ')
+  }
+
+  const maiaEvals: { [model: string]: MoveMap[] } = {}
+
+  const availableMoves: AvailableMoves[] = []
+
+  for (const model of data['maia_versions']) {
+    maiaEvals[model] = data['maia_evals'][model]
+  }
+
+  for (const position of data['move_maps']) {
+    const moves: AvailableMoves = {}
+    for (const move of position) {
+      const fromTo = move.move.join('')
+      const san = move['move_san']
+      const { check, fen } = move
+
+      moves[fromTo] = {
+        board: fen,
+        check,
+        san,
+        lastMove: move.move,
+      }
+    }
+    availableMoves.push(moves)
+  }
+
+  const gameStates = data['game_states']
+
+  const moves = gameStates.map((gameState: any) => {
+    const {
+      last_move: lastMove,
+      fen,
+      check,
+      last_move_san: san,
+      evaluations: maia_values,
+    } = gameState
+
+    return {
+      board: fen,
+      lastMove,
+      san,
+      check,
+      maia_values,
+    }
+  })
+
+  const maiaEvaluations = [] as { [rating: number]: MaiaEvaluation }[]
+  const stockfishEvaluations: StockfishEvaluation[] = []
+
+  return {
+    id,
+    blackPlayer,
+    whitePlayer,
+    moves,
+    availableMoves,
+    gameType,
+    termination,
+    maiaEvaluations,
+    stockfishEvaluations,
+    type: 'brain',
+  } as ClientAnalyzedGame
+}

public/maia2/maia_rapid.onnx

@@ -1 +1,2 @@
 export * from './analysis'
+export * from './clientAnalysis'

src/api/analysis/clientAnalysis.ts

@@ -7,3 +7,5 @@ export * from './useAnalysisController'
 export * from './useTuringController'
 export * from './usePlayController'
 export * from './useStockfishEngine'
+export * from './useMaiaEngine'
+export * from './useClientAnalysisController'

src/api/analysis/index.ts

@@ -75,6 +75,8 @@ export const useAnalysisController = (
       })
     }
 
+    console.log(moveMap)
+
     return moveMap
   }, [controller.currentIndex, game.maiaEvaluations, maiaModels])
 

src/hooks/index.ts

@@ -0,0 +1 @@
+export * from './useClientAnalysisController'