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