ssube
/
taleweave-ai
1
0
Fork 0
Code Actions

consistently look up client by id

This commit is contained in:
Sean Sube 2024-05-05 15:50:38 -05:00
parent e2a8e09c14
commit d6e8763914
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
adventure/server.py
View File

@ -40,7 +40,7 @@ async def handler(websocket):
) )
def sync_turn(character: str, prompt: str) -> bool: def sync_turn(character: str, prompt: str) -> bool:
if websocket not in characters: if id not in characters:
return False return False
asyncio.run(next_turn(character, prompt)) asyncio.run(next_turn(character, prompt))
@ -67,7 +67,7 @@ async def handler(websocket):
data = loads(message) data = loads(message)
message_type = data.get("type", None) message_type = data.get("type", None)
if message_type == "player": if message_type == "player":
character = characters.get(websocket) character = characters.get(id)
if character: if character:
del characters[id] del characters[id]
@ -86,7 +86,7 @@ async def handler(websocket):
# player_name = data["player"] # player_name = data["player"]
player = RemotePlayer(actor.name, actor.backstory, sync_turn, fallback_agent=llm_agent) player = RemotePlayer(actor.name, actor.backstory, sync_turn, fallback_agent=llm_agent)
characters[id] = player 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 # swap out the LLM agent
set_actor_agent_for_name(actor.name, actor, player) set_actor_agent_for_name(actor.name, actor, player)
@ -108,7 +108,7 @@ async def handler(websocket):
connected.remove(websocket) connected.remove(websocket)
# swap out the character for the original agent when they disconnect # swap out the character for the original agent when they disconnect
if websocket in characters: if id in characters:
player = characters[id] player = characters[id]
del characters[id] del characters[id]