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

44
api/tests/integration_tests/model_runtime/wenxin/test_embedding.py
View file

@ -10,13 +10,10 @@ def test_invoke_embedding_v1():
model = WenxinTextEmbeddingModel()
response = model.invoke(
model='embedding-v1',
credentials={
'api_key': os.environ.get('WENXIN_API_KEY'),
'secret_key': os.environ.get('WENXIN_SECRET_KEY')
},
texts=['hello', '你好', 'xxxxx'],
user="abc-123"
model="embedding-v1",
credentials={"api_key": os.environ.get("WENXIN_API_KEY"), "secret_key": os.environ.get("WENXIN_SECRET_KEY")},
texts=["hello", "你好", "xxxxx"],
user="abc-123",
)
assert isinstance(response, TextEmbeddingResult)
@ -29,13 +26,10 @@ def test_invoke_embedding_bge_large_en():
model = WenxinTextEmbeddingModel()
response = model.invoke(
model='bge-large-en',
credentials={
'api_key': os.environ.get('WENXIN_API_KEY'),
'secret_key': os.environ.get('WENXIN_SECRET_KEY')
},
texts=['hello', '你好', 'xxxxx'],
user="abc-123"
model="bge-large-en",
credentials={"api_key": os.environ.get("WENXIN_API_KEY"), "secret_key": os.environ.get("WENXIN_SECRET_KEY")},
texts=["hello", "你好", "xxxxx"],
user="abc-123",
)
assert isinstance(response, TextEmbeddingResult)
@ -48,13 +42,10 @@ def test_invoke_embedding_bge_large_zh():
model = WenxinTextEmbeddingModel()
response = model.invoke(
model='bge-large-zh',
credentials={
'api_key': os.environ.get('WENXIN_API_KEY'),
'secret_key': os.environ.get('WENXIN_SECRET_KEY')
},
texts=['hello', '你好', 'xxxxx'],
user="abc-123"
model="bge-large-zh",
credentials={"api_key": os.environ.get("WENXIN_API_KEY"), "secret_key": os.environ.get("WENXIN_SECRET_KEY")},
texts=["hello", "你好", "xxxxx"],
user="abc-123",
)
assert isinstance(response, TextEmbeddingResult)
@ -67,13 +58,10 @@ def test_invoke_embedding_tao_8k():
model = WenxinTextEmbeddingModel()
response = model.invoke(
model='tao-8k',
credentials={
'api_key': os.environ.get('WENXIN_API_KEY'),
'secret_key': os.environ.get('WENXIN_SECRET_KEY')
},
texts=['hello', '你好', 'xxxxx'],
user="abc-123"
model="tao-8k",
credentials={"api_key": os.environ.get("WENXIN_API_KEY"), "secret_key": os.environ.get("WENXIN_SECRET_KEY")},
texts=["hello", "你好", "xxxxx"],
user="abc-123",
)
assert isinstance(response, TextEmbeddingResult)