update wenxin llm (#2929)

2024-03-26 20:36:21 -07:00 · 2024-03-26 20:36:21 -07:00 · 5becb4c43a
commit 5becb4c43a
parent 13694293e3
16 changed files with 357 additions and 3 deletions
--- a/api/tests/integration_tests/model_runtime/wenxin/test_llm.py
+++ b/api/tests/integration_tests/model_runtime/wenxin/test_llm.py
@ -154,7 +154,7 @@ def test_invoke_stream_model():
    model = ErnieBotLargeLanguageModel()

    response = model.invoke(
-        model='ernie-bot',
+        model='ernie-3.5-8k',
        credentials={
            'api_key': os.environ.get('WENXIN_API_KEY'),
            'secret_key': os.environ.get('WENXIN_SECRET_KEY')