update mock llm aask_code

This commit is contained in:
yzlin 2024-01-19 00:09:58 +08:00
parent 3faa094248
commit 23fccdde67

View file

@ -69,7 +69,6 @@ class MockLLM(OriginalLLM):
A copy of metagpt.provider.openai_api.OpenAILLM.aask_code, we can't use super().aask because it will be mocked.
Since openai_api.OpenAILLM.aask_code is different from base_llm.BaseLLM.aask_code, we use the former.
"""
messages = self._process_message(messages)
rsp = await self._achat_completion_function(messages, **kwargs)
return self.get_choice_function_arguments(rsp)