test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-02 19:27:48 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
-LAN- 2024-08-14 14:09:26 +08:00 committed by GitHub
parent 52383d0161
commit b5d472fad7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/libs/bearer_data_source.py
View File

@ -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:

3
api/tests/integration_tests/model_runtime/__mock/huggingface_tei.py
View File

@ -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:

2
api/tests/integration_tests/model_runtime/huggingface_tei/test_embeddings.py
View File

@ -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