From d13342f4a522d9cd8a87c2a4a94bd6e30bd69d64 Mon Sep 17 00:00:00 2001 From: shenchucheng Date: Fri, 26 Jul 2024 09:59:46 +0800 Subject: [PATCH] add Deepseek/Siliconflow LLMType --- metagpt/configs/llm_config.py | 2 ++ metagpt/provider/openai_api.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/metagpt/configs/llm_config.py b/metagpt/configs/llm_config.py index 39f6e61f1..48130eedc 100644 --- a/metagpt/configs/llm_config.py +++ b/metagpt/configs/llm_config.py @@ -32,6 +32,8 @@ class LLMType(Enum): MISTRAL = "mistral" YI = "yi" # lingyiwanwu OPEN_ROUTER = "open_router" + DEEPSEEK = "deepseek" + SILICONFLOW = "siliconflow" def __missing__(self, key): return self.OPENAI diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 47d7df6f1..fa689d54f 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -49,6 +49,8 @@ from metagpt.utils.token_counter import ( LLMType.MISTRAL, LLMType.YI, LLMType.OPEN_ROUTER, + LLMType.DEEPSEEK, + LLMType.SILICONFLOW, ] ) class OpenAILLM(BaseLLM):