|
@@ -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)
|
|
|
|