From ef59c5bab979cb638f5ed046068cb75ab0ba80ce Mon Sep 17 00:00:00 2001 From: TensorNull <129579691+TensorNull@users.noreply.github.com> Date: Fri, 26 Sep 2025 10:50:56 +0800 Subject: [PATCH] FIX: Rename the CometEmbed and CometSeq2txt classes to CometAPIEmbed and CometAPISeq2txt, and correct supported_models.mdx. (#10298) ### What problem does this PR solve? Rename the CometEmbed and CometSeq2txt classes to CometAPIEmbed and CometAPISeq2txt, and correct supported_models.mdx. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- docs/references/supported_models.mdx | 2 +- rag/llm/embedding_model.py | 2 +- rag/llm/sequence2txt_model.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/references/supported_models.mdx b/docs/references/supported_models.mdx index 8302bd308..d9587f863 100644 --- a/docs/references/supported_models.mdx +++ b/docs/references/supported_models.mdx @@ -65,7 +65,7 @@ A complete list of models supported by RAGFlow, which will continue to expand. | 01.AI | :heavy_check_mark: | | | | | | | DeepInfra | :heavy_check_mark: | :heavy_check_mark: | | | :heavy_check_mark: | :heavy_check_mark: | | 302.AI | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | | | -| CometAPI | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | | | +| CometAPI | :heavy_check_mark: | :heavy_check_mark: | | | | | ```mdx-code-block diff --git a/rag/llm/embedding_model.py b/rag/llm/embedding_model.py index 4fc17eb31..d518ddec7 100644 --- a/rag/llm/embedding_model.py +++ b/rag/llm/embedding_model.py @@ -956,7 +956,7 @@ class Ai302Embed(Base): super().__init__(key, model_name, base_url) -class CometEmbed(OpenAIEmbed): +class CometAPIEmbed(OpenAIEmbed): _FACTORY_NAME = "CometAPI" def __init__(self, key, model_name, base_url="https://api.cometapi.com/v1"): diff --git a/rag/llm/sequence2txt_model.py b/rag/llm/sequence2txt_model.py index b2d1a5aaa..3a7bcf72c 100644 --- a/rag/llm/sequence2txt_model.py +++ b/rag/llm/sequence2txt_model.py @@ -236,7 +236,7 @@ class DeepInfraSeq2txt(Base): self.model_name = model_name -class CometSeq2txt(Base): +class CometAPISeq2txt(Base): _FACTORY_NAME = "CometAPI" def __init__(self, key, model_name="whisper-1", base_url="https://api.cometapi.com/v1", **kwargs):