浏览代码

Feat: add GiteeAI as a llm provider. (#8572)

### What problem does this PR solve?

#1853

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
tags/v0.20.0
Kevin Hu 4 个月前
父节点
当前提交
d46c24045f
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 9 次插入3 次删除
  1. 5
    3
      rag/llm/embedding_model.py
  2. 4
    0
      rag/llm/rerank_model.py

+ 5
- 3
rag/llm/embedding_model.py 查看文件

@@ -722,9 +722,7 @@ class UpstageEmbed(OpenAIEmbed):


class SILICONFLOWEmbed(Base):
def __init__(
self, key, model_name, base_url="https://api.siliconflow.cn/v1/embeddings"
):
def __init__(self, key, model_name, base_url="https://api.siliconflow.cn/v1/embeddings"):
if not base_url:
base_url = "https://api.siliconflow.cn/v1/embeddings"
self.headers = {
@@ -911,9 +909,13 @@ class GPUStackEmbed(OpenAIEmbed):

class NovitaEmbed(SILICONFLOWEmbed):
def __init__(self, key, model_name, base_url="https://api.novita.ai/v3/openai/embeddings"):
if not base_url:
base_url = "https://api.novita.ai/v3/openai/embeddings"
super().__init__(key, model_name, base_url)


class GiteeEmbed(SILICONFLOWEmbed):
def __init__(self, key, model_name, base_url="https://ai.gitee.com/v1/embeddings"):
if not base_url:
base_url = "https://ai.gitee.com/v1/embeddings"
super().__init__(key, model_name, base_url)

+ 4
- 0
rag/llm/rerank_model.py 查看文件

@@ -629,9 +629,13 @@ class GPUStackRerank(Base):

class NovitaRerank(JinaRerank):
def __init__(self, key, model_name, base_url="https://api.novita.ai/v3/openai/rerank"):
if not base_url:
base_url = "https://api.novita.ai/v3/openai/rerank"
super().__init__(key, model_name, base_url)


class GiteeRerank(JinaRerank):
def __init__(self, key, model_name, base_url="https://ai.gitee.com/v1/rerank"):
if not base_url:
base_url = "https://ai.gitee.com/v1/rerank"
super().__init__(key, model_name, base_url)

正在加载...
取消
保存