diff --git a/apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py b/apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py index 6011aee751c..5d6bd90d24a 100644 --- a/apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py +++ b/apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py @@ -65,12 +65,12 @@ def is_valid( return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) if model_params.get("stream"): - for res in model.stream([HumanMessage(content=gettext("Hello"))]): + for res in model.stream([HumanMessage(content="1")]): pass else: - model.invoke([HumanMessage(content=gettext("Hello"))]) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/anthropic_model_provider/credential/llm.py b/apps/models_provider/impl/anthropic_model_provider/credential/llm.py index 03e07ab001a..5dab629b138 100644 --- a/apps/models_provider/impl/anthropic_model_provider/credential/llm.py +++ b/apps/models_provider/impl/anthropic_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/aws_bedrock_model_provider/credential/image.py b/apps/models_provider/impl/aws_bedrock_model_provider/credential/image.py index 437c68abb0a..d4ca8c9aaa4 100644 --- a/apps/models_provider/impl/aws_bedrock_model_provider/credential/image.py +++ b/apps/models_provider/impl/aws_bedrock_model_provider/credential/image.py @@ -67,7 +67,7 @@ def is_valid( try: model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model.invoke([HumanMessage(content="1")]) except AppApiException: raise except Exception as e: diff --git a/apps/models_provider/impl/aws_bedrock_model_provider/credential/llm.py b/apps/models_provider/impl/aws_bedrock_model_provider/credential/llm.py index 290093cd53f..2b123b6e6d6 100644 --- a/apps/models_provider/impl/aws_bedrock_model_provider/credential/llm.py +++ b/apps/models_provider/impl/aws_bedrock_model_provider/credential/llm.py @@ -66,8 +66,8 @@ def is_valid( return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except AppApiException: raise except Exception as e: diff --git a/apps/models_provider/impl/azure_model_provider/credential/llm.py b/apps/models_provider/impl/azure_model_provider/credential/llm.py index 823a46ac27c..8e0da23a44c 100644 --- a/apps/models_provider/impl/azure_model_provider/credential/llm.py +++ b/apps/models_provider/impl/azure_model_provider/credential/llm.py @@ -85,8 +85,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException) or isinstance(e, BadRequestError): diff --git a/apps/models_provider/impl/deepseek_model_provider/credential/llm.py b/apps/models_provider/impl/deepseek_model_provider/credential/llm.py index 12f0c5a937c..79e42085556 100644 --- a/apps/models_provider/impl/deepseek_model_provider/credential/llm.py +++ b/apps/models_provider/impl/deepseek_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/docker_ai_model_provider/credential/llm.py b/apps/models_provider/impl/docker_ai_model_provider/credential/llm.py index 779ce5a5bc3..9c7f7387408 100644 --- a/apps/models_provider/impl/docker_ai_model_provider/credential/llm.py +++ b/apps/models_provider/impl/docker_ai_model_provider/credential/llm.py @@ -71,8 +71,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException) or isinstance(e, BadRequestError): diff --git a/apps/models_provider/impl/gemini_model_provider/credential/llm.py b/apps/models_provider/impl/gemini_model_provider/credential/llm.py index b09340d525f..a5f479613a4 100644 --- a/apps/models_provider/impl/gemini_model_provider/credential/llm.py +++ b/apps/models_provider/impl/gemini_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - res = model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + res = model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/kimi_model_provider/credential/llm.py b/apps/models_provider/impl/kimi_model_provider/credential/llm.py index b6f0b214e16..e8581ccd5e2 100644 --- a/apps/models_provider/impl/kimi_model_provider/credential/llm.py +++ b/apps/models_provider/impl/kimi_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/minimax_model_provider/credential/llm.py b/apps/models_provider/impl/minimax_model_provider/credential/llm.py index f6910a33351..807e05edd59 100644 --- a/apps/models_provider/impl/minimax_model_provider/credential/llm.py +++ b/apps/models_provider/impl/minimax_model_provider/credential/llm.py @@ -62,8 +62,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/openai_model_provider/credential/llm.py b/apps/models_provider/impl/openai_model_provider/credential/llm.py index be6e997411f..f52b5f43307 100644 --- a/apps/models_provider/impl/openai_model_provider/credential/llm.py +++ b/apps/models_provider/impl/openai_model_provider/credential/llm.py @@ -71,8 +71,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException) or isinstance(e, BadRequestError): diff --git a/apps/models_provider/impl/regolo_model_provider/credential/llm.py b/apps/models_provider/impl/regolo_model_provider/credential/llm.py index 67efdf6421e..f3c648a9170 100644 --- a/apps/models_provider/impl/regolo_model_provider/credential/llm.py +++ b/apps/models_provider/impl/regolo_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/siliconCloud_model_provider/credential/llm.py b/apps/models_provider/impl/siliconCloud_model_provider/credential/llm.py index 6cea7f1e68f..4fd82923c00 100644 --- a/apps/models_provider/impl/siliconCloud_model_provider/credential/llm.py +++ b/apps/models_provider/impl/siliconCloud_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/tencent_model_provider/credential/llm.py b/apps/models_provider/impl/tencent_model_provider/credential/llm.py index 931105203e0..534bc2d5d73 100644 --- a/apps/models_provider/impl/tencent_model_provider/credential/llm.py +++ b/apps/models_provider/impl/tencent_model_provider/credential/llm.py @@ -63,8 +63,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/vllm_model_provider/credential/llm.py b/apps/models_provider/impl/vllm_model_provider/credential/llm.py index b3b8265bbb3..f4dfd23563c 100644 --- a/apps/models_provider/impl/vllm_model_provider/credential/llm.py +++ b/apps/models_provider/impl/vllm_model_provider/credential/llm.py @@ -63,9 +63,9 @@ def is_valid( raise AppApiException( ValidCode.valid_error.value, gettext("The model does not exist, please download the model first") ) - model = provider.get_model(model_type, model_name, model_credential, **model_params) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) try: - res = model.invoke([HumanMessage(content=gettext("Hello"))]) + res = model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) raise AppApiException( diff --git a/apps/models_provider/impl/volcanic_engine_model_provider/credential/llm.py b/apps/models_provider/impl/volcanic_engine_model_provider/credential/llm.py index dc301de0b9c..557d73a0d4e 100644 --- a/apps/models_provider/impl/volcanic_engine_model_provider/credential/llm.py +++ b/apps/models_provider/impl/volcanic_engine_model_provider/credential/llm.py @@ -70,8 +70,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - res = model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + res = model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/wenxin_model_provider/credential/llm.py b/apps/models_provider/impl/wenxin_model_provider/credential/llm.py index ce3e3447d4f..d63a94efe72 100644 --- a/apps/models_provider/impl/wenxin_model_provider/credential/llm.py +++ b/apps/models_provider/impl/wenxin_model_provider/credential/llm.py @@ -58,7 +58,7 @@ def is_valid( ): # 根据api_version检查必需字段 api_version = model_credential.get("api_version", "v1") - model = provider.get_model(model_type, model_name, model_credential, **model_params) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) if api_version == "v1": model_type_list = provider.get_model_type_list() if not any(list(filter(lambda mt: mt.get("value") == model_type, model_type_list))): @@ -83,7 +83,7 @@ def is_valid( else: return False try: - model.invoke([HumanMessage(content=gettext("Hello"))]) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) raise e diff --git a/apps/models_provider/impl/xf_model_provider/credential/llm.py b/apps/models_provider/impl/xf_model_provider/credential/llm.py index f18a7bdc5fb..b23077c4815 100644 --- a/apps/models_provider/impl/xf_model_provider/credential/llm.py +++ b/apps/models_provider/impl/xf_model_provider/credential/llm.py @@ -97,8 +97,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/xinference_model_provider/credential/llm.py b/apps/models_provider/impl/xinference_model_provider/credential/llm.py index 61db84eb1d1..378dec8d557 100644 --- a/apps/models_provider/impl/xinference_model_provider/credential/llm.py +++ b/apps/models_provider/impl/xinference_model_provider/credential/llm.py @@ -65,8 +65,8 @@ def is_valid( raise AppApiException( ValidCode.valid_error.value, gettext("The model does not exist, please download the model first") ) - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) return True def encryption_dict(self, model_info: Dict[str, object]): diff --git a/apps/models_provider/impl/zhipu_model_provider/credential/llm.py b/apps/models_provider/impl/zhipu_model_provider/credential/llm.py index 2e3910e8e95..46f5197a86f 100644 --- a/apps/models_provider/impl/zhipu_model_provider/credential/llm.py +++ b/apps/models_provider/impl/zhipu_model_provider/credential/llm.py @@ -69,8 +69,8 @@ def is_valid( else: return False try: - model = provider.get_model(model_type, model_name, model_credential, **model_params) - model.invoke([HumanMessage(content=gettext("Hello"))]) + model = provider.get_model(model_type, model_name, model_credential, **{**model_params, "max_tokens": 1}) + model.invoke([HumanMessage(content="1")]) except Exception as e: maxkb_logger.error(f"Exception: {e}", exc_info=True) if isinstance(e, AppApiException):