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