add Deepseek/Siliconflow LLMType

This commit is contained in:
shenchucheng 2024-07-26 09:59:46 +08:00
parent 8483d5ae1e
commit d13342f4a5
2 changed files with 4 additions and 0 deletions

View file

@ -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

View file

@ -49,6 +49,8 @@ from metagpt.utils.token_counter import (
LLMType.MISTRAL,
LLMType.YI,
LLMType.OPEN_ROUTER,
LLMType.DEEPSEEK,
LLMType.SILICONFLOW,
]
)
class OpenAILLM(BaseLLM):