pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-02-06 10:35:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: update stepfun list (#12991)

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

Update stepfun list.

Add TTS and Sequence2Text functionalities.

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Yongteng Lei
2026-02-05 12:47:04 +08:00
committed by GitHub
parent 803b480f9c
commit 6361fc4b33
3 changed files with 122 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rag/llm/sequence2txt_model.py
View File

@ -59,6 +59,15 @@ class GPTSeq2txt(Base):
self.model_name = model_name
class StepFunSeq2txt(GPTSeq2txt):
_FACTORY_NAME = "StepFun"
def __init__(self, key, model_name="step-asr", lang="Chinese", base_url="https://api.stepfun.com/v1", **kwargs):
if not base_url:
base_url = "https://api.stepfun.com/v1"
super().__init__(key, model_name=model_name, base_url=base_url, **kwargs)
class QWenSeq2txt(Base):
_FACTORY_NAME = "Tongyi-Qianwen"