From d46c24045f88c9bb98fd08604b4002be1ed64077 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Mon, 30 Jun 2025 11:22:11 +0800 Subject: [PATCH] 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) --- rag/llm/embedding_model.py | 8 +++++--- rag/llm/rerank_model.py | 4 ++++ 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index 6fb815919..f53d492b9 100644 --- a/rag/llm/embedding_model.py +++ b/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) \ No newline at end of file diff --git a/rag/llm/rerank_model.py b/rag/llm/rerank_model.py index 03d9fac5a..fafab7ee0 100644 --- a/rag/llm/rerank_model.py +++ b/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) \ No newline at end of file