Provide logging info
This commit is contained in:
parent
c7bdc10cb6
commit
53c2496d55
@ -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
|
||||
|
||||
#
|
||||
|
Loading…
x
Reference in New Issue
Block a user