AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 03:19:01 +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:

5
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:
@ -12,7 +11,7 @@ class MockTEIClass:
model_type = 'embedding' model_type = 'embedding'
return TeiModelExtraParameter(model_type=model_type, max_input_length=512, max_client_batch_size=1) return TeiModelExtraParameter(model_type=model_type, max_input_length=512, max_client_batch_size=1)
@staticmethod @staticmethod
def invoke_tokenize(server_url: str, texts: list[str]) -> list[list[dict]]: def invoke_tokenize(server_url: str, texts: list[str]) -> list[list[dict]]:
# Use space as token separator, and split the text into tokens # Use space as token separator, and split the text into tokens

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