mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-08 20:42:30 +08:00
@ -1,2 +1,21 @@
|
||||
from .embedding_model import HuEmbedding
|
||||
from .chat_model import GptTurbo
|
||||
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()
|
||||
|
||||
Reference in New Issue
Block a user