test(*): Avoid import from api in tests. (#7251)

2024-08-14 14:09:26 +08:00
parent 52383d0161
commit b5d472fad7
3 changed files with 4 additions and 5 deletions
--- a/api/tests/integration_tests/model_runtime/huggingface_tei/test_embeddings.py
+++ b/api/tests/integration_tests/model_runtime/huggingface_tei/test_embeddings.py
@@ -1,12 +1,12 @@
 import os

 import pytest
-from api.core.model_runtime.model_providers.huggingface_tei.text_embedding.text_embedding import TeiHelper

 from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.huggingface_tei.text_embedding.text_embedding import (
    HuggingfaceTeiTextEmbeddingModel,
+    TeiHelper,
 )
 from tests.integration_tests.model_runtime.__mock.huggingface_tei import MockTEIClass