jketreno/backstory
1
0
Fork 0
Code Issues 12 Actions Releases Activity

Provide logging info

Browse Source
This commit is contained in:
James Ketr 2025-05-21 17:29:02 -07:00
parent c7bdc10cb6
commit 53c2496d55
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/utils/agents/persona.py
View File

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