feat: optimize model when app create (#875)

2023-08-16 22:29:18 +08:00
parent cc2d71c253
commit b7c29ea1b6
5 changed files with 104 additions and 23 deletions
--- a/api/tests/unit_tests/model_providers/test_tongyi_provider.py
+++ b/api/tests/unit_tests/model_providers/test_tongyi_provider.py
@@ -26,7 +26,7 @@ def decrypt_side_effect(tenant_id, encrypted_key):


 def test_is_provider_credentials_valid_or_raise_valid(mocker):
-    mocker.patch('langchain.llms.tongyi.Tongyi._generate', return_value=LLMResult(generations=[[Generation(text="abc")]]))
+    mocker.patch('core.third_party.langchain.llms.tongyi_llm.EnhanceTongyi._generate', return_value=LLMResult(generations=[[Generation(text="abc")]]))

    MODEL_PROVIDER_CLASS.is_provider_credentials_valid_or_raise(VALIDATE_CREDENTIAL)