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/__mock/huggingface_tei.py
+++ b/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