estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Feat/json mode (#2563)

Browse Source
This commit is contained in:
Yeuoly
2024-02-26 23:34:40 +08:00
committed by GitHub
parent 0620fa3094
commit 3e63abd335
31 changed files with 762 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
api/tests/integration_tests/model_runtime/wenxin/test_llm.py
View File

@@ -7,18 +7,18 @@ from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk,
from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
from core.model_runtime.entities.model_entities import AIModelEntity
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.wenxin.llm.llm import ErnieBotLarguageModel
from core.model_runtime.model_providers.wenxin.llm.llm import ErnieBotLargeLanguageModel
def test_predefined_models():
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
model_schemas = model.predefined_models()
assert len(model_schemas) >= 1
assert isinstance(model_schemas[0], AIModelEntity)
def test_validate_credentials_for_chat_model():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
with pytest.raises(CredentialsValidateFailedError):
model.validate_credentials(
@@ -39,7 +39,7 @@ def test_validate_credentials_for_chat_model():
def test_invoke_model_ernie_bot():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot',
@@ -67,7 +67,7 @@ def test_invoke_model_ernie_bot():
def test_invoke_model_ernie_bot_turbo():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot-turbo',
@@ -95,7 +95,7 @@ def test_invoke_model_ernie_bot_turbo():
def test_invoke_model_ernie_8k():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot-8k',
@@ -123,7 +123,7 @@ def test_invoke_model_ernie_8k():
def test_invoke_model_ernie_bot_4():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot-4',
@@ -151,7 +151,7 @@ def test_invoke_model_ernie_bot_4():
def test_invoke_stream_model():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot',
@@ -182,7 +182,7 @@ def test_invoke_stream_model():
def test_invoke_model_with_system():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot',
@@ -212,7 +212,7 @@ def test_invoke_model_with_system():
def test_invoke_with_search():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.invoke(
model='ernie-bot',
@@ -250,7 +250,7 @@ def test_invoke_with_search():
def test_get_num_tokens():
sleep(3)
model = ErnieBotLarguageModel()
model = ErnieBotLargeLanguageModel()
response = model.get_num_tokens(
model='ernie-bot',