diff --git a/rag/llm/chat_model.py b/rag/llm/chat_model.py index 2d626d1af..ce65f7dd3 100644 --- a/rag/llm/chat_model.py +++ b/rag/llm/chat_model.py @@ -981,9 +981,9 @@ class SILICONFLOWChat(Base): class YiChat(Base): - def __init__(self, key, model_name, base_url="https://api.01.ai/v1"): + def __init__(self, key, model_name, base_url="https://api.lingyiwanwu.com/v1"): if not base_url: - base_url = "https://api.01.ai/v1" + base_url = "https://api.lingyiwanwu.com/v1" super().__init__(key, model_name, base_url)