hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-22 01:06:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add nomic embedding model provider (#8640)

Browse Source
This commit is contained in:
ice yao
2024-09-23 19:57:21 +08:00
committed by GitHub
parent 4f69adc8ab
commit d7aada38a1
17 changed files with 506 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
api/tests/integration_tests/model_runtime/nomic/test_provider.py Normal file
View File

@@ -0,0 +1,22 @@
import os
import pytest
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.nomic.nomic import NomicAtlasProvider
from core.model_runtime.model_providers.nomic.text_embedding.text_embedding import NomicTextEmbeddingModel
from tests.integration_tests.model_runtime.__mock.nomic_embeddings import setup_nomic_mock
@pytest.mark.parametrize("setup_nomic_mock", [["text_embedding"]], indirect=True)
def test_validate_provider_credentials(setup_nomic_mock):
provider = NomicAtlasProvider()
with pytest.raises(CredentialsValidateFailedError):
provider.validate_provider_credentials(credentials={})
provider.validate_provider_credentials(
credentials={
"nomic_api_key": os.environ.get("NOMIC_API_KEY"),
},
)