pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

FIX: Rename the CometEmbed and CometSeq2txt classes to CometAPIEmbed and CometAPISeq2txt, and correct supported_models.mdx. (#10298)

Browse Source 
### 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)
This commit is contained in:
TensorNull
2025-09-26 10:50:56 +08:00
committed by GitHub
parent 62b7c655c5
commit ef59c5bab9
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/embedding_model.py
View File

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

2
rag/llm/sequence2txt_model.py
View File

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