jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
commit b035c02f78
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
155 changed files with 4279 additions and 5925 deletions
Download patch file Download diff file Expand all files Collapse all files

12
api/tests/integration_tests/model_runtime/siliconflow/test_rerank.py
View file

@ -13,9 +13,7 @@ def test_validate_credentials():
with pytest.raises(CredentialsValidateFailedError):
model.validate_credentials(
model="BAAI/bge-reranker-v2-m3",
credentials={
"api_key": "invalid_key"
},
credentials={"api_key": "invalid_key"},
)
model.validate_credentials(
@ -30,17 +28,17 @@ def test_invoke_model():
model = SiliconflowRerankModel()
result = model.invoke(
model='BAAI/bge-reranker-v2-m3',
model="BAAI/bge-reranker-v2-m3",
credentials={
"api_key": os.environ.get("API_KEY"),
},
query="Who is Kasumi?",
docs=[
"Kasumi is a girl's name of Japanese origin meaning \"mist\".",
'Kasumi is a girl\'s name of Japanese origin meaning "mist".',
"Her music is a kawaii bass, a mix of future bass, pop, and kawaii music ",
"and she leads a team named PopiParty."
"and she leads a team named PopiParty.",
],
score_threshold=0.8
score_threshold=0.8,
)
assert isinstance(result, RerankResult)