From 21d0b336c0f496e9fd84381d46f43d9e3cacd21d Mon Sep 17 00:00:00 2001 From: Dave Kolas <4643013+dkolas@users.noreply.github.com> Date: Tue, 7 Nov 2023 11:51:49 -0600 Subject: [PATCH] change DALL-E test prompts (#594) --- tests/steamship_tests/agents/tools/test_dalle_tool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/steamship_tests/agents/tools/test_dalle_tool.py b/tests/steamship_tests/agents/tools/test_dalle_tool.py index e036fa64e..4d10ee3e4 100644 --- a/tests/steamship_tests/agents/tools/test_dalle_tool.py +++ b/tests/steamship_tests/agents/tools/test_dalle_tool.py @@ -16,7 +16,7 @@ def test_dalle_tool(client: Steamship): OpenAI(client=client), AgentContext.get_or_create(client=client, context_keys={"id": "test"}), ) - res = tool.run([Block(text="Mario jumping over a fire flower")], context) + res = tool.run([Block(text="a man jumping over a rock")], context) assert len(res) == 1 assert res[0].public_data is True @@ -34,6 +34,6 @@ def test_dalle_tool_private_block(client: Steamship): OpenAI(client=client), AgentContext.get_or_create(client=client, context_keys={"id": "test"}), ) - res = tool.run([Block(text="Mario jumping over a fire flower")], context) + res = tool.run([Block(text="a man jumping over a rock")], context) assert len(res) == 1 assert res[0].public_data is False