diff --git a/tests/test_builder.py b/tests/test_builder.py index 4ba4c0e1..908b7fde 100644 --- a/tests/test_builder.py +++ b/tests/test_builder.py @@ -29,8 +29,7 @@ class TestPromptBuilder: result = builder.build(messages) assert len(result) == 2 - assert "pirate" in result[0].content - + builder = PromptBuilder(config=PromptConfig(system_template="You are a pirate.")) def test_build_with_tools(self): messages = [Message(role=Role.USER, content="Search for something")] tools = [