From f9d099be1b788d82dd6ef2dfa809e07ed5f5a096 Mon Sep 17 00:00:00 2001 From: jimmyzhuu Date: Wed, 6 May 2026 13:23:04 +0800 Subject: [PATCH] feat: add qianfan vision model constants --- common/const.py | 3 +++ tests/test_qianfan_provider.py | 10 ++++++++++ 2 files changed, 13 insertions(+) diff --git a/common/const.py b/common/const.py index 9742af9c..eae1bf92 100644 --- a/common/const.py +++ b/common/const.py @@ -92,6 +92,8 @@ ERNIE_45_TURBO_128K = "ernie-4.5-turbo-128k" ERNIE_45_TURBO_32K = "ernie-4.5-turbo-32k" ERNIE_X1_TURBO_32K = "ernie-x1-turbo-32k" ERNIE_4_TURBO_8K = "ERNIE-4.0-Turbo-8K" +ERNIE_45_TURBO_VL_PREVIEW = "ernie-4.5-turbo-vl-preview" +ERNIE_45_VL_28B_A3B = "ernie-4.5-vl-28b-a3b" # Qwen (通义千问 - 阿里云 DashScope) QWEN_TURBO = "qwen-turbo" @@ -169,6 +171,7 @@ MODEL_LIST = [ # Baidu Qianfan / ERNIE QIANFAN, ERNIE_5, ERNIE_45_TURBO_128K, ERNIE_45_TURBO_32K, ERNIE_X1_TURBO_32K, ERNIE_4_TURBO_8K, + ERNIE_45_TURBO_VL_PREVIEW, ERNIE_45_VL_28B_A3B, # MiniMax MiniMax, MINIMAX_M2_7, MINIMAX_M2_7_HIGHSPEED, MINIMAX_M2_5, MINIMAX_M2_1, MINIMAX_M2_1_LIGHTNING, MINIMAX_M2, MINIMAX_ABAB6_5, diff --git a/tests/test_qianfan_provider.py b/tests/test_qianfan_provider.py index 7f638e9e..51e01ff9 100644 --- a/tests/test_qianfan_provider.py +++ b/tests/test_qianfan_provider.py @@ -19,10 +19,20 @@ class TestQianfanConstantsAndRouting(unittest.TestCase): self.assertEqual(const.ERNIE_45_TURBO_128K, "ernie-4.5-turbo-128k") self.assertEqual(const.ERNIE_45_TURBO_32K, "ernie-4.5-turbo-32k") self.assertEqual(const.ERNIE_X1_TURBO_32K, "ernie-x1-turbo-32k") + self.assertEqual( + const.ERNIE_45_TURBO_VL_PREVIEW, + "ernie-4.5-turbo-vl-preview", + ) + self.assertEqual( + const.ERNIE_45_VL_28B_A3B, + "ernie-4.5-vl-28b-a3b", + ) self.assertIn(const.QIANFAN, const.MODEL_LIST) self.assertIn(const.ERNIE_45_TURBO_128K, const.MODEL_LIST) self.assertIn(const.ERNIE_45_TURBO_32K, const.MODEL_LIST) self.assertIn(const.ERNIE_X1_TURBO_32K, const.MODEL_LIST) + self.assertIn(const.ERNIE_45_TURBO_VL_PREVIEW, const.MODEL_LIST) + self.assertIn(const.ERNIE_45_VL_28B_A3B, const.MODEL_LIST) def test_qianfan_config_keys_are_available(self): import config