pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat stepfun (#1659)

Browse Source 
### What problem does this PR solve?

#1661
#1660

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

---------

Co-authored-by: lijianyong <lijianyong@stepfun.com>
This commit is contained in:
jianyongli
2024-07-24 10:49:37 +08:00
committed by GitHub
parent 80d703f9c2
commit 60356b52c6
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rag/llm/chat_model.py
View File

@ -85,7 +85,6 @@ class MoonshotChat(Base):
if not base_url: base_url="https://api.moonshot.cn/v1"
super().__init__(key, model_name, base_url)
class XinferenceChat(Base):
def __init__(self, key=None, model_name="", base_url=""):
key = "xxx"
@ -898,9 +897,9 @@ class OpenRouterChat(Base):
self.model_name = model_name
class StepFunChat(Base):
def __init__(self, key, model_name, base_url="https://api.stepfun.com/v1/chat/completions"):
def __init__(self, key, model_name, base_url="https://api.stepfun.com/v1"):
if not base_url:
base_url = "https://api.stepfun.com/v1/chat/completions"
base_url = "https://api.stepfun.com/v1"
super().__init__(key, model_name, base_url)