From 945f14e1206db8acdd1d2bbf932c21fbd2d89de1 Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Thu, 27 Mar 2025 21:47:36 -0700
Subject: [PATCH] Test to close #854

---
 tests/conftest.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index cd053da..57be2ee 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -84,11 +84,11 @@ class MockModel(llm.Model):
 
 class EchoModel(llm.Model):
     model_id = "echo"
+    can_stream = True
 
     def execute(self, prompt, stream, response, conversation):
-        yield "system:\n{}\n\nprompt:\n{}".format(
-            prompt.system or "", prompt.prompt or ""
-        )
+        yield "system:\n{}\n\n".format(prompt.system or "")
+        yield "prompt:\n{}".format(prompt.prompt or "")
 
 
 class MockKeyModel(llm.KeyModel):