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

chore: apply ruff rules on tests and app.py (#3605)

Browse source
This commit is contained in:
Bowen Liang 2024-04-18 20:24:05 +08:00 • committed by GitHub
commit d9b821cecc
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
92 changed files with 383 additions and 181 deletions
Download patch file Download diff file Expand all files Collapse all files

3
api/tests/integration_tests/model_runtime/huggingface_hub/test_llm.py
View file

@ -1,7 +1,8 @@
import os
from typing import Generator
from collections.abc import Generator
import pytest
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
from core.model_runtime.entities.message_entities import AssistantPromptMessage, UserPromptMessage
from core.model_runtime.errors.validate import CredentialsValidateFailedError

6
api/tests/integration_tests/model_runtime/huggingface_hub/test_text_embedding.py
View file

@ -1,10 +1,12 @@
import os
import pytest
from core.model_runtime.entities.text_embedding_entities import TextEmbeddingResult
from core.model_runtime.errors.validate import CredentialsValidateFailedError
from core.model_runtime.model_providers.huggingface_hub.text_embedding.text_embedding import \
HuggingfaceHubTextEmbeddingModel
from core.model_runtime.model_providers.huggingface_hub.text_embedding.text_embedding import (
HuggingfaceHubTextEmbeddingModel,
)
def test_hosted_inference_api_validate_credentials():