From e44e3a67b08238b99b0416692351fc11bc7337c6 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 14 Nov 2024 10:18:25 +0800 Subject: [PATCH] adapt to lower case cohere (#3392) ### What problem does this PR solve? #3384 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/db/init_data.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/db/init_data.py b/api/db/init_data.py index 7ecc54381..20e42aab6 100644 --- a/api/db/init_data.py +++ b/api/db/init_data.py @@ -97,6 +97,8 @@ def init_superuser(): def init_llm_factory(): try: LLMService.filter_delete([(LLM.fid == "MiniMax" or LLM.fid == "Minimax")]) + LLMService.filter_delete([(LLM.fid == "cohere")]) + LLMFactoriesService.filter_delete([LLMFactories.name == "cohere"]) except Exception: pass @@ -128,6 +130,7 @@ def init_llm_factory(): LLMFactoriesService.filter_delete([LLMFactoriesService.model.name == "QAnything"]) LLMService.filter_delete([LLMService.model.fid == "QAnything"]) TenantLLMService.filter_update([TenantLLMService.model.llm_factory == "QAnything"], {"llm_factory": "Youdao"}) + TenantLLMService.filter_update([TenantLLMService.model.llm_factory == "cohere"], {"llm_factory": "Cohere"}) TenantService.filter_update([1 == 1], { "parser_ids": "naive:General,qa:Q&A,resume:Resume,manual:Manual,table:Table,paper:Paper,book:Book,laws:Laws,presentation:Presentation,picture:Picture,one:One,audio:Audio,knowledge_graph:Knowledge Graph,email:Email"}) ## insert openai two embedding models to the current openai user.