From 31813f2512b176b1838c13a703cc640846362da6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Tue, 23 Jan 2024 17:16:15 +0800 Subject: [PATCH] add new test for tool_calls_rsp. --- tests/metagpt/provider/test_openai.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/metagpt/provider/test_openai.py b/tests/metagpt/provider/test_openai.py index 2e5799475..7a771bcac 100644 --- a/tests/metagpt/provider/test_openai.py +++ b/tests/metagpt/provider/test_openai.py @@ -87,6 +87,16 @@ class TestOpenAI: messages.append( ChatCompletionMessage(content="Completed a python code for hello world!", role="assistant", tool_calls=None) ) + # 添加 openai tool calls respond bug, code 出现在ChatCompletionMessage.content中 + messages.extend( + [ + 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 = [ Choice(finish_reason="tool_calls", logprobs=None, index=i, message=msg) for i, msg in enumerate(messages) ]