gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add embeddings

Browse source
This commit is contained in:
michaelmanley 2025-12-15 19:34:36 +10:00
parent 523962d4aa
commit 526fd07858
2 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docs/references/supported_models.mdx
View file

@ -15,7 +15,7 @@ A complete list of models supported by RAGFlow, which will continue to expand.
| Provider | Chat | Embedding | Rerank | Img2txt | Speech2txt | TTS |
| --------------------- | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ | ------------------ |
| AI Badgr | :heavy_check_mark: | | | | | |
| AI Badgr | :heavy_check_mark: | :heavy_check_mark: | | | | |
| Anthropic | :heavy_check_mark: | | | | | |
| Azure-OpenAI | :heavy_check_mark: | :heavy_check_mark: | | :heavy_check_mark: | :heavy_check_mark: | |
| BAAI | | :heavy_check_mark: | :heavy_check_mark: | | | |

9
rag/llm/embedding_model.py
View file

@ -911,6 +911,15 @@ class GiteeEmbed(SILICONFLOWEmbed):
base_url = "https://ai.gitee.com/v1/embeddings"
super().__init__(key, model_name, base_url)
class AiBadgrEmbed(OpenAIEmbed):
_FACTORY_NAME = "AI Badgr"
def __init__(self, key, model_name, base_url="https://aibadgr.com/api/v1"):
if not base_url:
base_url = "https://aibadgr.com/api/v1"
super().__init__(key, model_name, base_url)
class DeepInfraEmbed(OpenAIEmbed):
_FACTORY_NAME = "DeepInfra"