kikikita commited on
Commit
f31c662
·
1 Parent(s): 70bca69

fix: make get_user_state calls asynchronous in image_agent and music_agent

Browse files
src/agent/image_agent.py CHANGED
@@ -75,7 +75,7 @@ async def generate_image_prompt(user_hash: str, scene_description: str, last_cho
75
  """
76
  logger.info(f"Generating image prompt for the current scene: {scene_description}")
77
 
78
- state = get_user_state(user_hash)
79
  scene = GAME_STATE_PROMPT.format(
80
  lore=state.story_frame.lore,
81
  goal=state.story_frame.goal,
 
75
  """
76
  logger.info(f"Generating image prompt for the current scene: {scene_description}")
77
 
78
+ state = await get_user_state(user_hash)
79
  scene = GAME_STATE_PROMPT.format(
80
  lore=state.story_frame.lore,
81
  goal=state.story_frame.goal,
src/agent/music_agent.py CHANGED
@@ -43,7 +43,7 @@ llm = create_light_llm(0.1).with_structured_output(MusicPrompt)
43
  async def generate_music_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> str:
44
  logger.info(f"Generating music prompt for the current scene: {scene_description}")
45
 
46
- state = get_user_state(user_hash)
47
  scene = GAME_STATE_PROMPT.format(
48
  lore=state.story_frame.lore,
49
  goal=state.story_frame.goal,
 
43
  async def generate_music_prompt(user_hash: str, scene_description: str, last_choice = "No choice yet") -> str:
44
  logger.info(f"Generating music prompt for the current scene: {scene_description}")
45
 
46
+ state = await get_user_state(user_hash)
47
  scene = GAME_STATE_PROMPT.format(
48
  lore=state.story_frame.lore,
49
  goal=state.story_frame.goal,