add new test for tool_calls_rsp.

This commit is contained in:
刘棒棒 2024-01-23 17:16:15 +08:00
parent 987eb6d38b
commit 31813f2512

View file

@ -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)
]