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

190
api/tests/integration_tests/model_runtime/ollama/test_llm.py
View File

File diff suppressed because one or more lines are too long

48
api/tests/integration_tests/model_runtime/ollama/test_text_embedding.py
View File

@@ -12,21 +12,21 @@ def test_validate_credentials():
with pytest.raises(CredentialsValidateFailedError):
model.validate_credentials(
model='mistral:text',
model="mistral:text",
credentials={
'base_url': 'http://localhost:21434',
'mode': 'chat',
'context_size': 4096,
}
"base_url": "http://localhost:21434",
"mode": "chat",
"context_size": 4096,
},
)
model.validate_credentials(
model='mistral:text',
model="mistral:text",
credentials={
'base_url': os.environ.get('OLLAMA_BASE_URL'),
'mode': 'chat',
'context_size': 4096,
}
"base_url": os.environ.get("OLLAMA_BASE_URL"),
"mode": "chat",
"context_size": 4096,
},
)
@@ -34,17 +34,14 @@ def test_invoke_model():
model = OllamaEmbeddingModel()
result = model.invoke(
model='mistral:text',
model="mistral:text",
credentials={
'base_url': os.environ.get('OLLAMA_BASE_URL'),
'mode': 'chat',
'context_size': 4096,
"base_url": os.environ.get("OLLAMA_BASE_URL"),
"mode": "chat",
"context_size": 4096,
},
texts=[
"hello",
"world"
],
user="abc-123"
texts=["hello", "world"],
user="abc-123",
)
assert isinstance(result, TextEmbeddingResult)
@@ -56,16 +53,13 @@ def test_get_num_tokens():
model = OllamaEmbeddingModel()
num_tokens = model.get_num_tokens(
model='mistral:text',
model="mistral:text",
credentials={
'base_url': os.environ.get('OLLAMA_BASE_URL'),
'mode': 'chat',
'context_size': 4096,
"base_url": os.environ.get("OLLAMA_BASE_URL"),
"mode": "chat",
"context_size": 4096,
},
texts=[
"hello",
"world"
]
texts=["hello", "world"],
)
assert num_tokens == 2