diff --git a/api/tests/unit_tests/model_providers/test_chatglm_provider.py b/api/tests/unit_tests/model_providers/test_chatglm_provider.py index 9dfa1291f4..30e8be5b21 100644 --- a/api/tests/unit_tests/model_providers/test_chatglm_provider.py +++ b/api/tests/unit_tests/model_providers/test_chatglm_provider.py @@ -2,7 +2,9 @@ import pytest from unittest.mock import patch import json +import requests from langchain.schema import LLMResult, Generation, AIMessage, ChatResult, ChatGeneration +from requests import Response from core.model_providers.providers.base import CredentialsValidateFailedError from core.model_providers.providers.chatglm_provider import ChatGLMProvider @@ -26,8 +28,11 @@ def decrypt_side_effect(tenant_id, encrypted_key): def test_is_provider_credentials_valid_or_raise_valid(mocker): - mocker.patch('langchain.llms.chatglm.ChatGLM._call', - return_value="abc") + mock_response = Response() + mock_response.status_code = 200 + mock_response._content = json.dumps({'models': []}).encode('utf-8') + mocker.patch('requests.get', + return_value=mock_response) MODEL_PROVIDER_CLASS.is_provider_credentials_valid_or_raise(VALIDATE_CREDENTIAL)