Provide logging info

This commit is contained in:
James Ketr 2025-05-21 17:29:02 -07:00
parent c7bdc10cb6
commit 53c2496d55

View File

@ -324,15 +324,18 @@ class PersonaGenerator(Agent):
message.response = prompt
yield message
logger.info("Beginning image generation...")
request = ImageRequest(filepath=os.path.join(defines.user_dir, persona["username"], f"profile.png"), prompt=prompt)
placeholder = Message(prompt=prompt)
async for placeholder in generate_image_status(
message=placeholder,
**request.model_dump()
):
logger.info("Image generation continue...")
if placeholder.status != "done":
placeholder.response = placeholder.response
yield message
yield placeholder
logger.info("Image generation done...")
persona["has_profile"] = True
#