consistently look up client by id
This commit is contained in:
parent
e2a8e09c14
commit
d6e8763914
|
@ -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]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue