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/support 302ai provider (#8742)

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

Support 302.AI provider.

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
JI4JUN
2025-07-31 14:48:30 +08:00
committed by GitHub
parent 46ded9d329
commit aeaeb169e4
10 changed files with 205 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rag/llm/rerank_model.py
View File

@ -617,3 +617,12 @@ class GiteeRerank(JinaRerank):
if not base_url:
base_url = "https://ai.gitee.com/v1/rerank"
super().__init__(key, model_name, base_url)
class Ai302Rerank(Base):
_FACTORY_NAME = "302.AI"
def __init__(self, key, model_name, base_url="https://api.302.ai/v1/rerank"):
if not base_url:
base_url = "https://api.302.ai/v1/rerank"
super().__init__(key, model_name, base_url)