We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2de1593 commit 5f3e83dCopy full SHA for 5f3e83d
tests/_utils/llm.py
@@ -89,13 +89,11 @@ def mk_mock_openai_response(messages: list[str]) -> mock.MagicMock:
89
response = mock.MagicMock()
90
91
response.choices = [_mk_mock_choice(message) for message in messages]
92
- response.dict.return_value = {
93
- "choices": [{
94
- 'message': {
95
- 'role': 'assistant', 'content': message
96
- }
97
- } for message in messages]
98
+
+ response_dict = {"choices": [{'message': {'role': 'assistant', 'content': message}} for message in messages]}
+ response.dict.return_value = response_dict
+ response.model_dump.return_value = response_dict
99
100
return response
101
0 commit comments