From 9729ca2aed2c971609db036c45e16e70e1097b3f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=BB=84=E8=85=BE?= <101850389+hangters@users.noreply.github.com> Date: Fri, 16 Aug 2024 16:35:40 +0800 Subject: [PATCH] fix 01.ai url error (#1977) ### What problem does this PR solve? #1976 fix 01.ai url error ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --------- Co-authored-by: Zhedong Cen --- rag/llm/chat_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rag/llm/chat_model.py b/rag/llm/chat_model.py index 6a56026b6..1c52b32ee 100644 --- a/rag/llm/chat_model.py +++ b/rag/llm/chat_model.py @@ -1033,7 +1033,7 @@ class SILICONFLOWChat(Base): class YiChat(Base): - def __init__(self, key, model_name, base_url="https://api.lingyiwanwu.com/v1"): + def __init__(self, key, model_name, base_url="https://api.01.ai/v1"): if not base_url: - base_url = "https://api.lingyiwanwu.com/v1" + base_url = "https://api.01.ai/v1" super().__init__(key, model_name, base_url) \ No newline at end of file