diff --git a/tests/metagpt/provider/test_openai.py b/tests/metagpt/provider/test_openai.py index 1743fed92..77820a5f8 100644 --- a/tests/metagpt/provider/test_openai.py +++ b/tests/metagpt/provider/test_openai.py @@ -95,6 +95,10 @@ class TestOpenAI: ChatCompletionMessage( content='"""python\nprint(\'hello world\')"""', role="assistant", tool_calls=None ), + ChatCompletionMessage( + content="'''python\nprint(\"hello world\")'''", role="assistant", tool_calls=None + ), + ChatCompletionMessage(content="```python\nprint('hello world')```", role="assistant", tool_calls=None), ] ) choices = [