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

api/libs/bearer_data_source.py

@@ -2,10 +2,10 @@
 from abc import abstractmethod
 
 import requests
-from api.models.source import DataSourceBearerBinding
 from flask_login import current_user
 
 from extensions.ext_database import db
+from models.source import DataSourceBearerBinding
 
 
 class BearerDataSource:

api/tests/integration_tests/model_runtime/__mock/huggingface_tei.py

@@ -1,5 +1,4 @@
-
-from api.core.model_runtime.model_providers.huggingface_tei.tei_helper import TeiModelExtraParameter
+from core.model_runtime.model_providers.huggingface_tei.tei_helper import TeiModelExtraParameter
 
 
 class MockTEIClass:
@@ -12,7 +11,7 @@ class MockTEIClass:
             model_type = 'embedding'
 
         return TeiModelExtraParameter(model_type=model_type, max_input_length=512, max_client_batch_size=1)
-    
+
     @staticmethod
     def invoke_tokenize(server_url: str, texts: list[str]) -> list[list[dict]]:
         # Use space as token separator, and split the text into tokens

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