Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-04-14 18:21:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Test to close #854

Browse source
This commit is contained in:
Simon Willison 2025-03-27 21:47:36 -07:00
parent 9a24605996
commit 945f14e120
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/conftest.py
View file

@ -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):