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

2023-10-18 15:35:24 +08:00
parent c039f4af83
commit 7c9b585a47
6 changed files with 174 additions and 77 deletions
--- a/api/tests/integration_tests/models/llm/test_wenxin_model.py
+++ b/api/tests/integration_tests/models/llm/test_wenxin_model.py
@@ -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