Grayscale/nanobot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(agent): remove legacy image mode prompt injection

Browse Source
This commit is contained in:
chengyongru
2026-07-12 00:35:17 +08:00
committed by Xubin Ren
parent 2eb7398f34
commit e206ee4e70
10 changed files with 1 additions and 198 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/utils/test_image_generation_intent.py
-25
View File
@@ -1,25 +0,0 @@
from nanobot.utils.image_generation_intent import image_generation_prompt
def test_image_generation_prompt_ignores_plain_messages() -> None:
assert image_generation_prompt("hello", {}) == "hello"
def test_image_generation_prompt_uses_auto_aspect_instruction() -> None:
prompt = image_generation_prompt(
"Draw a poster",
{"image_generation": {"enabled": True, "aspect_ratio": None}},
)
assert "Draw a poster" in prompt
assert "Use the generate_image tool" in prompt
assert "Choose the most suitable aspect_ratio yourself" in prompt
def test_image_generation_prompt_uses_selected_aspect_ratio() -> None:
prompt = image_generation_prompt(
"Draw a banner",
{"image_generation": {"enabled": True, "aspect_ratio": "16:9"}},
)
assert "aspect_ratio='16:9'" in prompt