Update test

Kiln-AI · Aug 28, 2024 · b60c68f · b60c68f
1 parent 66b25b6
commit b60c68f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/libs/core/kiln_ai/adapters/test_prompt_adaptors.py b/libs/core/kiln_ai/adapters/test_prompt_adaptors.py
@@ -32,9 +32,7 @@ async def test_amazon_bedrock(tmp_path):
 async def test_mock(tmp_path):
     task = build_test_task(tmp_path)
     adapter = SimplePromptAdapter(task, model_name="fake_parrot", provider="mock")
-    answer = await adapter.run(
-        "You should answer the following question: four plus six times 10"
-    )
+    answer = await adapter.run("You are a mock, send me the response!")
     assert "mock response" in answer