Skip to content

Commit 49668ea

Browse files
feat: Add SiliconFlow model support, including Deepseek, InternLM, Qw… (#2213)
Co-authored-by: JINO ROHIT <[email protected]> Co-authored-by: waleedalzarooni
1 parent d61d803 commit 49668ea

File tree

1 file changed

+68
-0
lines changed

1 file changed

+68
-0
lines changed

camel/types/enums.py

Lines changed: 68 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -332,16 +332,44 @@ class ModelType(UnifiedModelType, Enum):
332332
# SiliconFlow models support tool calling
333333
SILICONFLOW_DEEPSEEK_V2_5 = "deepseek-ai/DeepSeek-V2.5"
334334
SILICONFLOW_DEEPSEEK_V3 = "deepseek-ai/DeepSeek-V3"
335+
SILICONFLOW_PRO_DEEPSEEK_V3 = "Pro/deepseek-ai/DeepSeek-V3"
336+
SILICONFLOW_DEEPSEEK_R1 = "deepseek-ai/DeepSeek-R1"
337+
SILICONFLOW_PRO_DEEPSEEK_R1 = "Pro/deepseek-ai/DeepSeek-R1"
338+
SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_32B = (
339+
"deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
340+
)
341+
SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_14B = (
342+
"deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
343+
)
344+
SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_7B = (
345+
"deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"
346+
)
347+
SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_1_5B = (
348+
"deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
349+
)
350+
SILICONFLOW_PRO_DEEPSEEK_R1_DISTILL_QWEN_7B = (
351+
"Pro/deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"
352+
)
353+
SILICONFLOW_PRO_DEEPSEEK_R1_DISTILL_QWEN_1_5B = (
354+
"Pro/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
355+
)
335356
SILICONFLOW_INTERN_LM2_5_20B_CHAT = "internlm/internlm2_5-20b-chat"
336357
SILICONFLOW_INTERN_LM2_5_7B_CHAT = "internlm/internlm2_5-7b-chat"
337358
SILICONFLOW_PRO_INTERN_LM2_5_7B_CHAT = "Pro/internlm/internlm2_5-7b-chat"
359+
SILICONFLOW_QWQ_32B = "Qwen/QwQ-32B"
338360
SILICONFLOW_QWEN2_5_72B_INSTRUCT = "Qwen/Qwen2.5-72B-Instruct"
339361
SILICONFLOW_QWEN2_5_32B_INSTRUCT = "Qwen/Qwen2.5-32B-Instruct"
340362
SILICONFLOW_QWEN2_5_14B_INSTRUCT = "Qwen/Qwen2.5-14B-Instruct"
341363
SILICONFLOW_QWEN2_5_7B_INSTRUCT = "Qwen/Qwen2.5-7B-Instruct"
342364
SILICONFLOW_PRO_QWEN2_5_7B_INSTRUCT = "Pro/Qwen/Qwen2.5-7B-Instruct"
343365
SILICONFLOW_THUDM_GLM_4_9B_CHAT = "THUDM/glm-4-9b-chat"
344366
SILICONFLOW_PRO_THUDM_GLM_4_9B_CHAT = "Pro/THUDM/glm-4-9b-chat"
367+
SILICONFLOW_THUDM_GLM_Z1_32B_0414 = "THUDM/GLM-Z1-32B-0414"
368+
SILICONFLOW_THUDM_GLM_4_32B_0414 = "THUDM/GLM-4-32B-0414"
369+
SILICONFLOW_THUDM_GLM_Z1_RUMINATION_32B_0414 = (
370+
"THUDM/GLM-Z1-Rumination-32B-0414"
371+
)
372+
SILICONFLOW_THUDM_GLM_4_9B_0414 = "THUDM/GLM-4-9B-0414"
345373

346374
# AIML models support tool calling
347375
AIML_MIXTRAL_8X7B = "mistralai/Mixtral-8x7B-Instruct-v0.1"
@@ -983,16 +1011,30 @@ def is_siliconflow(self) -> bool:
9831011
return self in {
9841012
ModelType.SILICONFLOW_DEEPSEEK_V2_5,
9851013
ModelType.SILICONFLOW_DEEPSEEK_V3,
1014+
ModelType.SILICONFLOW_PRO_DEEPSEEK_V3,
1015+
ModelType.SILICONFLOW_DEEPSEEK_R1,
1016+
ModelType.SILICONFLOW_PRO_DEEPSEEK_R1,
1017+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_32B,
1018+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_14B,
1019+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_7B,
1020+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_1_5B,
1021+
ModelType.SILICONFLOW_PRO_DEEPSEEK_R1_DISTILL_QWEN_7B,
1022+
ModelType.SILICONFLOW_PRO_DEEPSEEK_R1_DISTILL_QWEN_1_5B,
9861023
ModelType.SILICONFLOW_INTERN_LM2_5_20B_CHAT,
9871024
ModelType.SILICONFLOW_INTERN_LM2_5_7B_CHAT,
9881025
ModelType.SILICONFLOW_PRO_INTERN_LM2_5_7B_CHAT,
1026+
ModelType.SILICONFLOW_QWQ_32B,
9891027
ModelType.SILICONFLOW_QWEN2_5_72B_INSTRUCT,
9901028
ModelType.SILICONFLOW_QWEN2_5_32B_INSTRUCT,
9911029
ModelType.SILICONFLOW_QWEN2_5_14B_INSTRUCT,
9921030
ModelType.SILICONFLOW_QWEN2_5_7B_INSTRUCT,
9931031
ModelType.SILICONFLOW_PRO_QWEN2_5_7B_INSTRUCT,
9941032
ModelType.SILICONFLOW_THUDM_GLM_4_9B_CHAT,
9951033
ModelType.SILICONFLOW_PRO_THUDM_GLM_4_9B_CHAT,
1034+
ModelType.SILICONFLOW_THUDM_GLM_Z1_32B_0414,
1035+
ModelType.SILICONFLOW_THUDM_GLM_4_32B_0414,
1036+
ModelType.SILICONFLOW_THUDM_GLM_Z1_RUMINATION_32B_0414,
1037+
ModelType.SILICONFLOW_THUDM_GLM_4_9B_0414,
9961038
}
9971039

9981040
@property
@@ -1250,6 +1292,15 @@ def token_limit(self) -> int:
12501292
ModelType.NOVITA_LLAMA_3_2_11B_VISION,
12511293
ModelType.NOVITA_LLAMA_3_2_3B,
12521294
ModelType.NEBIUS_MISTRAL_7B_INSTRUCT,
1295+
ModelType.SILICONFLOW_INTERN_LM2_5_20B_CHAT,
1296+
ModelType.SILICONFLOW_INTERN_LM2_5_7B_CHAT,
1297+
ModelType.SILICONFLOW_PRO_INTERN_LM2_5_7B_CHAT,
1298+
ModelType.SILICONFLOW_QWQ_32B,
1299+
ModelType.SILICONFLOW_QWEN2_5_72B_INSTRUCT,
1300+
ModelType.SILICONFLOW_QWEN2_5_32B_INSTRUCT,
1301+
ModelType.SILICONFLOW_QWEN2_5_14B_INSTRUCT,
1302+
ModelType.SILICONFLOW_QWEN2_5_7B_INSTRUCT,
1303+
ModelType.SILICONFLOW_PRO_QWEN2_5_7B_INSTRUCT,
12531304
}:
12541305
return 32_768
12551306
elif self in {
@@ -1270,6 +1321,17 @@ def token_limit(self) -> int:
12701321
ModelType.NOVITA_DEEPSEEK_R1_DISTILL_QWEN_14B,
12711322
ModelType.NOVITA_DEEPSEEK_R1_DISTILL_QWEN_32B,
12721323
ModelType.NOVITA_DEEPSEEK_R1,
1324+
ModelType.SILICONFLOW_DEEPSEEK_V2_5,
1325+
ModelType.SILICONFLOW_DEEPSEEK_V3,
1326+
ModelType.SILICONFLOW_PRO_DEEPSEEK_V3,
1327+
ModelType.SILICONFLOW_DEEPSEEK_R1,
1328+
ModelType.SILICONFLOW_PRO_DEEPSEEK_R1,
1329+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_32B,
1330+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_14B,
1331+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_7B,
1332+
ModelType.SILICONFLOW_DEEPSEEK_R1_DISTILL_QWEN_1_5B,
1333+
ModelType.SILICONFLOW_PRO_DEEPSEEK_R1_DISTILL_QWEN_7B,
1334+
ModelType.SILICONFLOW_PRO_DEEPSEEK_R1_DISTILL_QWEN_1_5B,
12731335
}:
12741336
return 64_000
12751337
elif self in {
@@ -1376,6 +1438,12 @@ def token_limit(self) -> int:
13761438
ModelType.COMETAPI_DEEPSEEK_REASONER,
13771439
ModelType.COMETAPI_QWEN3_30B_A3B,
13781440
ModelType.COMETAPI_QWEN3_CODER_PLUS_2025_07_22,
1441+
ModelType.SILICONFLOW_THUDM_GLM_4_9B_CHAT,
1442+
ModelType.SILICONFLOW_PRO_THUDM_GLM_4_9B_CHAT,
1443+
ModelType.SILICONFLOW_THUDM_GLM_Z1_32B_0414,
1444+
ModelType.SILICONFLOW_THUDM_GLM_4_32B_0414,
1445+
ModelType.SILICONFLOW_THUDM_GLM_Z1_RUMINATION_32B_0414,
1446+
ModelType.SILICONFLOW_THUDM_GLM_4_9B_0414,
13791447
}:
13801448
return 128_000
13811449
elif self in {

0 commit comments

Comments
 (0)