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]>
- 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.
|
1037 |
if not base_url:
|
1038 |
-
base_url = "https://api.
|
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)
|