黄腾 aopstudio commited on
Commit
f511e07
·
1 Parent(s): 6bf835b

fix 01.ai url error (#1977)

Browse files

### 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 <[email protected]>

Files changed (1) hide show
  1. rag/llm/chat_model.py +2 -2
rag/llm/chat_model.py CHANGED
@@ -1033,7 +1033,7 @@ class SILICONFLOWChat(Base):
1033
 
1034
 
1035
  class YiChat(Base):
1036
- def __init__(self, key, model_name, base_url="https://api.lingyiwanwu.com/v1"):
1037
  if not base_url:
1038
- base_url = "https://api.lingyiwanwu.com/v1"
1039
  super().__init__(key, model_name, base_url)
 
1033
 
1034
 
1035
  class YiChat(Base):
1036
+ def __init__(self, key, model_name, base_url="https://api.01.ai/v1"):
1037
  if not base_url:
1038
+ base_url = "https://api.01.ai/v1"
1039
  super().__init__(key, model_name, base_url)