diff --git a/metagpt/configs/llm_config.py b/metagpt/configs/llm_config.py index 3f42cdc9d..a07a87cdf 100644 --- a/metagpt/configs/llm_config.py +++ b/metagpt/configs/llm_config.py @@ -27,7 +27,7 @@ class LLMType(Enum): QIANFAN = "qianfan" # Baidu BCE DASHSCOPE = "dashscope" # Aliyun LingJi DashScope MOONSHOT = "moonshot" - MISTRAL = 'mistral' + MISTRAL = "mistral" def __missing__(self, key): return self.OPENAI diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 0a423f210..e575232b0 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -50,7 +50,7 @@ See FAQ 5.8 raise retry_state.outcome.exception() -@register_provider([LLMType.OPENAI, LLMType.FIREWORKS, LLMType.OPEN_LLM, LLMType.MOONSHOT]) +@register_provider([LLMType.OPENAI, LLMType.FIREWORKS, LLMType.OPEN_LLM, LLMType.MOONSHOT, LLMType.MISTRAL]) class OpenAILLM(BaseLLM): """Check https://platform.openai.com/examples for examples"""