From 748721caeb7a8033bb3db226f0a1060ce1210170 Mon Sep 17 00:00:00 2001 From: better629 Date: Sat, 2 Mar 2024 22:17:00 +0800 Subject: [PATCH] fix missing mistral --- metagpt/configs/llm_config.py | 2 +- metagpt/provider/openai_api.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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"""