Merge pull request #953 from better629/main

Feat fix missing mistral
This commit is contained in:
better629 2024-03-02 22:18:48 +08:00 committed by GitHub
commit 1012896afd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View file

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

View file

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