mirror of
https://github.com/HKUDS/nanobot.git
synced 2026-08-04 08:28:36 +00:00
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
"""Helpers for WebUI image-generation intent metadata."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
IMAGE_GENERATION_METADATA_KEY = "image_generation"
|
|
|
|
|
|
def image_generation_prompt(content: str, metadata: dict[str, Any] | None) -> str:
|
|
"""Decorate a user prompt when WebUI image mode is enabled."""
|
|
runtime_context = image_generation_runtime_context(metadata)
|
|
return f"{content}\n\n{runtime_context}" if runtime_context else content
|
|
|
|
|
|
def image_generation_runtime_context(metadata: dict[str, Any] | None) -> str:
|
|
"""Return the model-only instruction for WebUI image generation mode."""
|
|
raw = (metadata or {}).get(IMAGE_GENERATION_METADATA_KEY)
|
|
if not isinstance(raw, dict) or raw.get("enabled") is not True:
|
|
return ""
|
|
|
|
aspect_ratio = raw.get("aspect_ratio")
|
|
if isinstance(aspect_ratio, str) and aspect_ratio.strip():
|
|
instruction = (
|
|
"The user selected WebUI image generation mode. Use the generate_image tool. "
|
|
f"When calling generate_image, pass aspect_ratio={aspect_ratio!r}."
|
|
)
|
|
else:
|
|
instruction = (
|
|
"The user selected WebUI image generation mode. Use the generate_image tool. "
|
|
"Choose the most suitable aspect_ratio yourself from the prompt and intended use."
|
|
)
|
|
return f"[WebUI image generation instruction: {instruction}]"
|