mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-07-31 00:43:37 +08:00
22 lines
432 B
Python
22 lines
432 B
Python
import os
|
|
from .embedding_model import *
|
|
from .chat_model import *
|
|
from .cv_model import *
|
|
|
|
EmbeddingModel = None
|
|
ChatModel = None
|
|
CvModel = None
|
|
|
|
|
|
if os.environ.get("OPENAI_API_KEY"):
|
|
EmbeddingModel = GptEmbed()
|
|
ChatModel = GptTurbo()
|
|
CvModel = GptV4()
|
|
|
|
elif os.environ.get("DASHSCOPE_API_KEY"):
|
|
EmbeddingModel = QWenEmbd()
|
|
ChatModel = QWenChat()
|
|
CvModel = QWenCV()
|
|
else:
|
|
EmbeddingModel = HuEmbedding()
|