Quellcode durchsuchen

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 vor 4 Monaten
Ursprung
Commit
d46c24045f
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
2 geänderte Dateien mit 9 neuen und 3 gelöschten Zeilen
  1. 5
    3
      rag/llm/embedding_model.py
  2. 4
    0
      rag/llm/rerank_model.py

+ 5
- 3
rag/llm/embedding_model.py Datei anzeigen

@@ -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 Datei anzeigen

@@ -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)

Laden…
Abbrechen
Speichern