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

2024-04-18 20:24:05 +08:00
parent d5448e07ab
commit d9b821cecc
92 changed files with 383 additions and 181 deletions
--- a/api/tests/integration_tests/model_runtime/openrouter/test_llm.py
+++ b/api/tests/integration_tests/model_runtime/openrouter/test_llm.py
@@ -1,10 +1,15 @@
 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, PromptMessageTool,
-                                                          SystemPromptMessage, UserPromptMessage)
+from core.model_runtime.entities.message_entities import (
+    AssistantPromptMessage,
+    PromptMessageTool,
+    SystemPromptMessage,
+    UserPromptMessage,
+)
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.openrouter.llm.llm import OpenRouterLargeLanguageModel