kikikita commited on
Commit
7165472
·
1 Parent(s): 5287ab1

feat: refactor image generation call in process_step function

Browse files
Files changed (1) hide show
  1. 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
- user_hash=user_hash,
60
- scene_id=f"ending_{uuid.uuid4()}",
61
- change_scene=change_scene,
 
 
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