jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: support weixin ernie-bot-4 and chat mode (#1375)

Browse source
This commit is contained in:
takatost 2023-10-18 15:35:24 +08:00 • committed by GitHub
commit 7c9b585a47
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 174 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/tests/integration_tests/models/llm/test_wenxin_model.py
View file

@ -56,9 +56,8 @@ def test_run(mock_decrypt, mocker):
mocker.patch('core.model_providers.providers.base.BaseModelProvider.update_last_used', return_value=None)
model = get_mock_model('ernie-bot')
messages = [PromptMessage(content='Human: 1 + 1=? \nAssistant: Integer answer is:')]
messages = [PromptMessage(type=MessageType.USER, content='Human: 1 + 1=? \nAssistant: Integer answer is:')]
rst = model.run(
messages,
stop=['\nHuman:'],
messages
)
assert len(rst.content) > 0