AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 21:29:02 +08:00
Code Issues 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 from abc import abstractmethod
import requests import requests
from api.models.source import DataSourceBearerBinding
from flask_login import current_user from flask_login import current_user
from extensions.ext_database import db from extensions.ext_database import db
from models.source import DataSourceBearerBinding
class BearerDataSource: class BearerDataSource:

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

@ -1,5 +1,4 @@
from core.model_runtime.model_providers.huggingface_tei.tei_helper import TeiModelExtraParameter
from api.core.model_runtime.model_providers.huggingface_tei.tei_helper import TeiModelExtraParameter
class MockTEIClass: class MockTEIClass:

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

@ -1,12 +1,12 @@
import os import os
import pytest 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.entities.text_embedding_entities import TextEmbeddingResult
from core.model_runtime.errors.validate import CredentialsValidateFailedError from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.huggingface_tei.text_embedding.text_embedding import ( from core.model_runtime.model_providers.huggingface_tei.text_embedding.text_embedding import (
HuggingfaceTeiTextEmbeddingModel, HuggingfaceTeiTextEmbeddingModel,
TeiHelper,
) )
from tests.integration_tests.model_runtime.__mock.huggingface_tei import MockTEIClass from tests.integration_tests.model_runtime.__mock.huggingface_tei import MockTEIClass