Skip to content

Commit c0dae45

Browse files
committed
feat: add support for image model credential and update model info management
1 parent b18fb6e commit c0dae45

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

apps/models_provider/impl/siliconCloud_model_provider/siliconCloud_model_provider.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,15 @@
1313
ModelTypeConst, ModelInfoManage
1414
from models_provider.impl.siliconCloud_model_provider.credential.embedding import \
1515
SiliconCloudEmbeddingCredential
16+
from models_provider.impl.siliconCloud_model_provider.credential.image import SiliconCloudImageModelCredential
1617
from models_provider.impl.siliconCloud_model_provider.credential.llm import SiliconCloudLLMModelCredential
1718
from models_provider.impl.siliconCloud_model_provider.credential.reranker import SiliconCloudRerankerCredential
1819
from models_provider.impl.siliconCloud_model_provider.credential.stt import SiliconCloudSTTModelCredential
1920
from models_provider.impl.siliconCloud_model_provider.credential.tti import \
2021
SiliconCloudTextToImageModelCredential
2122
from models_provider.impl.siliconCloud_model_provider.credential.tts import SiliconCloudTTSModelCredential
2223
from models_provider.impl.siliconCloud_model_provider.model.embedding import SiliconCloudEmbeddingModel
24+
from models_provider.impl.siliconCloud_model_provider.model.image import SiliconCloudImage
2325
from models_provider.impl.siliconCloud_model_provider.model.llm import SiliconCloudChatModel
2426
from models_provider.impl.siliconCloud_model_provider.model.reranker import SiliconCloudReranker
2527
from models_provider.impl.siliconCloud_model_provider.model.stt import SiliconCloudSpeechToText
@@ -32,6 +34,7 @@
3234
openai_stt_model_credential = SiliconCloudSTTModelCredential()
3335
openai_reranker_model_credential = SiliconCloudRerankerCredential()
3436
openai_tti_model_credential = SiliconCloudTextToImageModelCredential()
37+
openai_image_model_credential = SiliconCloudImageModelCredential()
3538
openai_tts_model_credential = SiliconCloudTTSModelCredential()
3639
model_info_list = [
3740
ModelInfo('deepseek-ai/DeepSeek-R1-Distill-Llama-8B', '', ModelTypeConst.LLM,
@@ -112,6 +115,11 @@
112115
ModelTypeConst.TTS, openai_tts_model_credential,
113116
SiliconCloudTextToSpeech),
114117
]
118+
model_image_info_list = [
119+
ModelInfo('Qwen/Qwen3-VL-32B-Instruct', '',
120+
ModelTypeConst.IMAGE, openai_image_model_credential,
121+
SiliconCloudImage),
122+
]
115123
model_info_manage = (
116124
ModelInfoManage.builder()
117125
.append_model_info_list(model_info_list)
@@ -130,6 +138,8 @@
130138
.append_default_model_info(model_rerank_list[0])
131139
.append_model_info_list(model_tts_list)
132140
.append_default_model_info(model_tts_list[0])
141+
.append_model_info_list(model_image_info_list)
142+
.append_default_model_info(model_image_info_list[0])
133143

134144
.build()
135145
)

0 commit comments

Comments
 (0)