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

feat: enhance gemini models (#11497)

Browse Source
This commit is contained in:
非法操作
2024-12-17 12:05:13 +08:00
committed by GitHub
parent 56cfdce453
commit 74fdc16bd1
23 changed files with 138 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/tests/integration_tests/model_runtime/google/test_llm.py
View File

@@ -13,7 +13,7 @@ from core.model_runtime.entities.message_entities import (
)
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.google.llm.llm import GoogleLargeLanguageModel
from tests.integration_tests.model_runtime.__mock.google import setup_google_mock
from tests.integration_tests.model_runtime.__mock.google import setup_google_mock, setup_mock_redis
@pytest.mark.parametrize("setup_google_mock", [["none"]], indirect=True)
@@ -95,7 +95,7 @@ def test_invoke_stream_model(setup_google_mock):
@pytest.mark.parametrize("setup_google_mock", [["none"]], indirect=True)
def test_invoke_chat_model_with_vision(setup_google_mock):
def test_invoke_chat_model_with_vision(setup_google_mock, setup_mock_redis):
model = GoogleLargeLanguageModel()
result = model.invoke(
@@ -124,7 +124,7 @@ def test_invoke_chat_model_with_vision(setup_google_mock):
@pytest.mark.parametrize("setup_google_mock", [["none"]], indirect=True)
def test_invoke_chat_model_with_vision_multi_pics(setup_google_mock):
def test_invoke_chat_model_with_vision_multi_pics(setup_google_mock, setup_mock_redis):
model = GoogleLargeLanguageModel()
result = model.invoke(