From d6e87639148285676bdaceaea4abe97e90aa032b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 May 2024 15:50:38 -0500 Subject: [PATCH] consistently look up client by id --- adventure/server.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/adventure/server.py b/adventure/server.py index dbc15f7..e0a25c1 100644 --- a/adventure/server.py +++ b/adventure/server.py @@ -40,7 +40,7 @@ async def handler(websocket): ) def sync_turn(character: str, prompt: str) -> bool: - if websocket not in characters: + if id not in characters: return False asyncio.run(next_turn(character, prompt)) @@ -67,7 +67,7 @@ async def handler(websocket): data = loads(message) message_type = data.get("type", None) if message_type == "player": - character = characters.get(websocket) + character = characters.get(id) if character: del characters[id] @@ -86,7 +86,7 @@ async def handler(websocket): # player_name = data["player"] player = RemotePlayer(actor.name, actor.backstory, sync_turn, fallback_agent=llm_agent) characters[id] = player - logger.info(f"Client {websocket} is now character {character_name}") + logger.info(f"Client {id} is now character {character_name}") # swap out the LLM agent set_actor_agent_for_name(actor.name, actor, player) @@ -108,7 +108,7 @@ async def handler(websocket): connected.remove(websocket) # swap out the character for the original agent when they disconnect - if websocket in characters: + if id in characters: player = characters[id] del characters[id]