fix example test

h0rv · h0rv · commit 95aa687b75e8 · 2025-12-12T16:20:08.000-05:00
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -698,6 +698,12 @@ async def model_logic(  # noqa: C901
                     FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='wide-axolotl-city')),
                 ]
             )
+        elif m.content == 'Generate a high-resolution wide landscape illustration of an axolotl.':
+            return ModelResponse(
+                parts=[
+                    FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='high-res-axolotl')),
+                ]
+            )
         elif m.content == 'Generate a chart of y=x^2 for x=-5 to 5.':
             return ModelResponse(
                 parts=[

Original file line number	Diff line number	Diff line change
`@@ -698,6 +698,12 @@ async def model_logic( # noqa: C901`
`698`	`698`	`FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='wide-axolotl-city')),`
`699`	`699`	`]`
`700`	`700`	`)`
	`701`	`+ elif m.content == 'Generate a high-resolution wide landscape illustration of an axolotl.':`
	`702`	`+ return ModelResponse(`
	`703`	`+ parts=[`
	`704`	`+ FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='high-res-axolotl')),`
	`705`	`+ ]`
	`706`	`+ )`
`701`	`707`	`elif m.content == 'Generate a chart of y=x^2 for x=-5 to 5.':`
`702`	`708`	`return ModelResponse(`
`703`	`709`	`parts=[`