Unverified Commit e1d22033 authored by takatost's avatar takatost Committed by GitHub

fix: provider chatglm tests error (#1618)

parent 93467cb3
...@@ -2,7 +2,9 @@ import pytest ...@@ -2,7 +2,9 @@ import pytest
from unittest.mock import patch from unittest.mock import patch
import json import json
import requests
from langchain.schema import LLMResult, Generation, AIMessage, ChatResult, ChatGeneration 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.base import CredentialsValidateFailedError
from core.model_providers.providers.chatglm_provider import ChatGLMProvider from core.model_providers.providers.chatglm_provider import ChatGLMProvider
...@@ -26,8 +28,11 @@ def decrypt_side_effect(tenant_id, encrypted_key): ...@@ -26,8 +28,11 @@ def decrypt_side_effect(tenant_id, encrypted_key):
def test_is_provider_credentials_valid_or_raise_valid(mocker): def test_is_provider_credentials_valid_or_raise_valid(mocker):
mocker.patch('langchain.llms.chatglm.ChatGLM._call', mock_response = Response()
return_value="abc") 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) MODEL_PROVIDER_CLASS.is_provider_credentials_valid_or_raise(VALIDATE_CREDENTIAL)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment