Jofthomas's picture
Jofthomas HF staff
bulk
841082c
raw
history blame
5.91 kB
import { v, Infer, ObjectType } from 'convex/values';
import { Game } from './game';
import {
DAY_DURATION,
NIGHT_DURATION,
WWOLF_VOTE_DURATION,
PLAYER_KILL_VOTE_DURATION,
} from '../constants';
import { processVotes } from './voting';
import { parseLLMVotingResult } from './voting';
import { LLmvotingCallWerewolf } from './voting';
import { GameId } from './ids';
import { Player } from './player';
export type CycleState = 'Day' | 'Night' | 'WerewolfVoting' | 'PlayerKillVoting' | 'EndGame' | 'LobbyState'
const stateDurations: { [key in CycleState]: number } = {
Day: DAY_DURATION,
Night: NIGHT_DURATION,
WerewolfVoting: WWOLF_VOTE_DURATION,
PlayerKillVoting: PLAYER_KILL_VOTE_DURATION,
EndGame: Infinity,
LobbyState: Infinity
};
const normalCycle: CycleState[] = [
'Day',
'Night',
'PlayerKillVoting',
'WerewolfVoting',
];
const pushToGist = (averageCorrectVotes: number[]) => {
const token = process.env.GITHUB_TOKEN; // Read GitHub token from environment variables
const data = {
description: "Average Correct Votes",
public: true,
files: {
"averageCorrectVotes.json": {
content: JSON.stringify(averageCorrectVotes)
}
}
};
const headers = {
"Accept": "application/vnd.github+json",
"Authorization": `Bearer ${token}`,
"X-GitHub-Api-Version": "2022-11-28"
};
fetch('https://api.github.com/gists', {
method: 'POST',
headers: headers,
body: JSON.stringify(data)
})
.then(response => response.json())
.then(data => console.log('Gist created:', data.html_url))
.catch(error => console.error('Error creating Gist:', error));
}
const getCorrectVotesPercentage = (game: Game, playerId: GameId<'players'>, llms: Player[] ) => {
const playerVotes = game.world.llmVotes.filter((vote) => vote.voter === playerId);
if (playerVotes.length === 0) {
return 0;
}
const correctVotes = playerVotes[0].playerIds.filter((id) => llms.map((llm) => llm.id).includes(id));
return correctVotes.length / llms.length;
}
export const gameCycleSchema = {
currentTime: v.number(),
cycleState: v.union(
v.literal('Day'),
v.literal('Night'),
v.literal('WerewolfVoting'),
v.literal('PlayerKillVoting'),
v.literal('EndGame'),
v.literal('LobbyState'),
),
cycleIndex: v.number(),
};
export type SerializedGameCycle = ObjectType<typeof gameCycleSchema>;
const onStateChange = (prevState: CycleState, newState: CycleState, game: Game, now: number) => {
console.log(`state changed: ${ prevState } -> ${ newState }`);
console.log("newState is :",newState)
if(newState ==="WerewolfVoting"){
console.log('players are : ', game.playerDescriptions);
const allVillagers = [...game.world.players.values()]
const villagers = [...game.playerDescriptions.values()].filter(player =>
player.type === 'villager'
);
// TODO: You should't vote for yourelf
allVillagers.forEach((villager) => {
LLmvotingCallWerewolf(villager, villagers).then(result => {
parseLLMVotingResult(villager, result, game)
})
})
};
if(newState ==="PlayerKillVoting"){
const werewolves = [...game.world.players.values()].filter((were) => {
game.playerDescriptions.get(were.id)?.type === 'werewolf'
})
const villagers = [...game.playerDescriptions.values()]
werewolves.forEach((were) => {
LLmvotingCallWerewolf(were, villagers).then(result => {
parseLLMVotingResult(were, result, game)
})
})
};
if (prevState === 'PlayerKillVoting') {
const mostVotedPlayer = processVotes(game.world.gameVotes, [...game.world.players.values()])[0];
const playerToKill = game.world.players.get(mostVotedPlayer.playerId);
console.log(`killing: ${playerToKill?.id}, with ${game.world.gameVotes.length} votes`)
if (playerToKill) {
playerToKill.kill(game, now);
}
game.world.gameVotes = [];
}
if (prevState === 'WerewolfVoting') {
const mostVotedPlayer = processVotes(game.world.gameVotes, [...game.world.players.values()])[0];
const suspect = game.world.players.get(mostVotedPlayer.playerId);
console.log(`suspect: ${suspect?.id}, with ${game.world.gameVotes.length} votes`)
if (suspect?.playerType(game) === 'werewolf') {
suspect?.kill(game, now)
}
game.world.gameVotes = [];
}
if (newState === 'EndGame') {
const llms = [...game.world.playersInit.values()].filter((player) => {
!player.human
})
const averageCorrectVotes = game.world.llmVotes.map((votes) => {
return getCorrectVotesPercentage(game, votes.voter, llms);
})
// Push to gist
pushToGist(averageCorrectVotes);
}
};
export class GameCycle {
currentTime: number;
cycleState: CycleState;
cycleIndex: number;
constructor(serialized: SerializedGameCycle) {
const { currentTime, cycleState, cycleIndex } = serialized;
this.currentTime = currentTime;
this.cycleState = cycleState;
this.cycleIndex = cycleIndex;
}
endgame(game: Game) {
this.currentTime = 0;
onStateChange(this.cycleState, 'EndGame', game, 0);
this.cycleState = 'EndGame';
this.cycleIndex = -1;
console.log('EndGame reached')
}
// Tick method to increment the counter
tick(game: Game, tickDuration: number) {
console.log(process.env.GITHUB_TOKEN)
this.currentTime += tickDuration;
if (this.currentTime >= stateDurations[this.cycleState]) {
const prevState = this.cycleState;
this.currentTime = 0;
this.cycleIndex = (this.cycleIndex + 1) % normalCycle.length;
this.cycleState = normalCycle[this.cycleIndex];
onStateChange(prevState, this.cycleState, game, tickDuration);
}
}
serialize(): SerializedGameCycle {
const { currentTime, cycleState, cycleIndex } = this;
return {
currentTime,
cycleState,
cycleIndex,
};
}
}