From a6533c0db79f0e49fbd809fce978f272817d725a Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Tue, 29 Apr 2025 14:57:41 +0800
Subject: [PATCH] feat: Support Alibaba Bailian supplier qwen3 model (#3020)

---
 .../impl/aliyun_bai_lian_model_provider/credential/llm.py   | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py b/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py
index f316a0c6d..5dce041ac 100644
--- a/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py
+++ b/apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py
@@ -47,7 +47,11 @@ class BaiLianLLMModelCredential(BaseForm, BaseModelCredential):
                     return False
         try:
             model = provider.get_model(model_type, model_name, model_credential, **model_params)
-            model.invoke([HumanMessage(content=gettext('Hello'))])
+            if model_params.get('stream'):
+                for res in model.stream([HumanMessage(content=gettext('Hello'))]):
+                    pass
+            else:
+                model.invoke([HumanMessage(content=gettext('Hello'))])
         except Exception as e:
             traceback.print_exc()
             if isinstance(e, AppApiException):