mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-05-24 10:32:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(image-generation): keep image presence helper stable

Browse Source
This commit is contained in:
Xubin Ren 2026-05-22 22:19:32 +08:00
parent 143224e25a
commit c1073f2986
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
nanobot/providers/image_generation.py
View File

@ -219,7 +219,7 @@ class ImageGenerationProvider(ABC):
image_size: str | None = None, image_size: str | None = None,
) -> GeneratedImageResponse: ... ) -> GeneratedImageResponse: ...
def _ensure_images(self, images: list[str], data: dict[str, Any]) -> None: def _require_images(self, images: list[str], data: dict[str, Any]) -> None:
if images: if images:
return return
provider_error = data.get("error") if isinstance(data, dict) else None provider_error = data.get("error") if isinstance(data, dict) else None
@ -326,7 +326,7 @@ class OpenRouterImageGenerationClient(ImageGenerationProvider):
if isinstance(url_value, str) and url_value.startswith("data:image/"): if isinstance(url_value, str) and url_value.startswith("data:image/"):
images.append(url_value) images.append(url_value)
self._ensure_images(images, data) self._require_images(images, data)
return GeneratedImageResponse( return GeneratedImageResponse(
images=images, images=images,
@ -428,7 +428,7 @@ class AIHubMixImageGenerationClient(ImageGenerationProvider):
payload = response.json() payload = response.json()
images = await _aihubmix_images_from_payload(client, payload) images = await _aihubmix_images_from_payload(client, payload)
self._ensure_images(images, payload) self._require_images(images, payload)
return GeneratedImageResponse(images=images, content="", raw=payload) return GeneratedImageResponse(images=images, content="", raw=payload)
@ -670,7 +670,7 @@ class GeminiImageGenerationClient(ImageGenerationProvider):
if isinstance(b64, str) and b64: if isinstance(b64, str) and b64:
images.append(f"data:{mime};base64,{b64}") images.append(f"data:{mime};base64,{b64}")
self._ensure_images(images, data) self._require_images(images, data)
return GeneratedImageResponse(images=images, content="", raw=data) return GeneratedImageResponse(images=images, content="", raw=data)
@ -728,7 +728,7 @@ class GeminiImageGenerationClient(ImageGenerationProvider):
if b64: if b64:
images.append(f"data:{mime};base64,{b64}") images.append(f"data:{mime};base64,{b64}")
self._ensure_images(images, data) self._require_images(images, data)
return GeneratedImageResponse( return GeneratedImageResponse(
images=images, images=images,
@ -882,7 +882,7 @@ class MiniMaxImageGenerationClient(ImageGenerationProvider):
payload = response.json() payload = response.json()
images = _minimax_images_from_payload(payload) images = _minimax_images_from_payload(payload)
self._ensure_images(images, payload) self._require_images(images, payload)
return GeneratedImageResponse(images=images, content="", raw=payload) return GeneratedImageResponse(images=images, content="", raw=payload)
@ -1408,7 +1408,7 @@ class StepFunImageGenerationClient(ImageGenerationProvider):
payload = response.json() payload = response.json()
images = _stepfun_images_from_payload(payload) images = _stepfun_images_from_payload(payload)
self._ensure_images(images, payload) self._require_images(images, payload)
return GeneratedImageResponse(images=images, content="", raw=payload) return GeneratedImageResponse(images=images, content="", raw=payload)