admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore(api/tests): apply ruff reformat #7590 (#7591)

Browse Source 
Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang
2024-08-23 23:52:25 +08:00
committed by GitHub
parent 2da63654e5
commit b035c02f78
155 changed files with 4279 additions and 5925 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
api/tests/integration_tests/model_runtime/xinference/test_embeddings.py
View File

@@ -8,61 +8,57 @@ from core.model_runtime.model_providers.xinference.text_embedding.text_embedding
from tests.integration_tests.model_runtime.__mock.xinference import MOCK, setup_xinference_mock
@pytest.mark.parametrize('setup_xinference_mock', [['none']], indirect=True)
@pytest.mark.parametrize("setup_xinference_mock", [["none"]], indirect=True)
def test_validate_credentials(setup_xinference_mock):
model = XinferenceTextEmbeddingModel()
with pytest.raises(CredentialsValidateFailedError):
model.validate_credentials(
model='bge-base-en',
model="bge-base-en",
credentials={
'server_url': os.environ.get('XINFERENCE_SERVER_URL'),
'model_uid': 'www ' + os.environ.get('XINFERENCE_EMBEDDINGS_MODEL_UID')
}
"server_url": os.environ.get("XINFERENCE_SERVER_URL"),
"model_uid": "www " + os.environ.get("XINFERENCE_EMBEDDINGS_MODEL_UID"),
},
)
model.validate_credentials(
model='bge-base-en',
model="bge-base-en",
credentials={
'server_url': os.environ.get('XINFERENCE_SERVER_URL'),
'model_uid': os.environ.get('XINFERENCE_EMBEDDINGS_MODEL_UID')
}
"server_url": os.environ.get("XINFERENCE_SERVER_URL"),
"model_uid": os.environ.get("XINFERENCE_EMBEDDINGS_MODEL_UID"),
},
)
@pytest.mark.parametrize('setup_xinference_mock', [['none']], indirect=True)
@pytest.mark.parametrize("setup_xinference_mock", [["none"]], indirect=True)
def test_invoke_model(setup_xinference_mock):
model = XinferenceTextEmbeddingModel()
result = model.invoke(
model='bge-base-en',
model="bge-base-en",
credentials={
'server_url': os.environ.get('XINFERENCE_SERVER_URL'),
'model_uid': os.environ.get('XINFERENCE_EMBEDDINGS_MODEL_UID')
"server_url": os.environ.get("XINFERENCE_SERVER_URL"),
"model_uid": os.environ.get("XINFERENCE_EMBEDDINGS_MODEL_UID"),
},
texts=[
"hello",
"world"
],
user="abc-123"
texts=["hello", "world"],
user="abc-123",
)
assert isinstance(result, TextEmbeddingResult)
assert len(result.embeddings) == 2
assert result.usage.total_tokens > 0
def test_get_num_tokens():
model = XinferenceTextEmbeddingModel()
num_tokens = model.get_num_tokens(
model='bge-base-en',
model="bge-base-en",
credentials={
'server_url': os.environ.get('XINFERENCE_SERVER_URL'),
'model_uid': os.environ.get('XINFERENCE_EMBEDDINGS_MODEL_UID')
"server_url": os.environ.get("XINFERENCE_SERVER_URL"),
"model_uid": os.environ.get("XINFERENCE_EMBEDDINGS_MODEL_UID"),
},
texts=[
"hello",
"world"
]
texts=["hello", "world"],
)
assert num_tokens == 2