fix error when use farui-plus model (#7316)

Co-authored-by: 雪风 <xuefeng@shifaedu.cn>
langgenius · Aug 15, 2024 · 6fdbc7d · 6fdbc7d
1 parent d1a6702
commit 6fdbc7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/core/model_runtime/model_providers/tongyi/llm/llm.py b/api/core/model_runtime/model_providers/tongyi/llm/llm.py
@@ -159,6 +159,8 @@ def get_num_tokens(self, model: str, credentials: dict, prompt_messages: list[Pr
         """
         if model in ['qwen-turbo-chat', 'qwen-plus-chat']:
             model = model.replace('-chat', '')
+        if model == 'farui-plus':
+            model = 'qwen-farui-plus'
 
         if model in self.tokenizers:
             tokenizer = self.tokenizers[model]