Spaces:
Sleeping
Sleeping
feat: refactor image generation call in process_step function
Browse files- src/agent/runner.py +6 -4
src/agent/runner.py
CHANGED
|
@@ -55,10 +55,12 @@ async def process_step(
|
|
| 55 |
|
| 56 |
ending_desc = ending_info.get("description") or ending_info.get("condition", "")
|
| 57 |
change_scene = await generate_image_prompt(ending_desc, user_hash)
|
| 58 |
-
image_path = await generate_scene_image(
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
|
|
|
|
|
|
| 62 |
)
|
| 63 |
|
| 64 |
response["ending"] = ending_info
|
|
|
|
| 55 |
|
| 56 |
ending_desc = ending_info.get("description") or ending_info.get("condition", "")
|
| 57 |
change_scene = await generate_image_prompt(ending_desc, user_hash)
|
| 58 |
+
image_path = await generate_scene_image.ainvoke(
|
| 59 |
+
{
|
| 60 |
+
"user_hash": user_hash,
|
| 61 |
+
"scene_id": f"ending_{uuid.uuid4()}",
|
| 62 |
+
"change_scene": change_scene,
|
| 63 |
+
}
|
| 64 |
)
|
| 65 |
|
| 66 |
response["ending"] = ending_info
|