Spaces:
Sleeping
Sleeping
fix: Update agent functionality
Browse files- src/agent/image_agent.py +1 -1
- src/agent/music_agent.py +1 -1
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,
|