From 1793a5fec64cdb624b4b425f7c6798ea7a5627af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Mon, 22 Jan 2024 16:09:23 +0800 Subject: [PATCH] update function_rsps. --- tests/metagpt/provider/test_openai.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/metagpt/provider/test_openai.py b/tests/metagpt/provider/test_openai.py index 7af2f6892..2e5799475 100644 --- a/tests/metagpt/provider/test_openai.py +++ b/tests/metagpt/provider/test_openai.py @@ -66,11 +66,17 @@ class TestOpenAI: @pytest.fixture def tool_calls_rsp(self): function_rsps = [ - Function(arguments='{\n"language": "python",\n"code": "print(\'hello world\')"', name="execute"), - Function(arguments='{\n"language": "python",\n"code": ```print("hello world")```', name="execute"), - Function(arguments='{\n"language": "python",\n"code": \'print("hello world")\'', name="execute"), - Function(arguments='{\n"language": \'python\',\n"code": "print(\'hello world\')"', name="execute"), + Function(arguments='{\n"language": "python",\n"code": "print(\'hello world\')"}', name="execute"), + Function(arguments='{\n"language": "python",\n"code": \'print("hello world")\'}', name="execute"), + Function(arguments='{\n"language": \'python\',\n"code": "print(\'hello world\')"}', name="execute"), + Function(arguments='{\n"language": "python",\n"code": "print(\'hello world\')"}', name="execute"), + Function(arguments='{\n"language": "python",\n"code": ```print("hello world")```}', name="execute"), + Function(arguments='{\n"language": "python",\n"code": """print("hello world")"""}', name="execute"), Function(arguments='\nprint("hello world")\\n', name="execute"), + # only `{` in arguments + Function(arguments='{\n"language": "python",\n"code": "print(\'hello world\')"', name="execute"), + # no `{`, `}` in arguments + Function(arguments='\n"language": "python",\n"code": "print(\'hello world\')"', name="execute"), ] tool_calls = [ ChatCompletionMessageToolCall(type="function", id=f"call_{i}", function=f)