improve: introduce isort for linting Python imports (#1983)

api/tests/integration_tests/model_runtime/__mock/anthropic.py

@@ -1,16 +1,14 @@
-import anthropic
-from anthropic import Anthropic
-from anthropic.resources.completions import Completions
-from anthropic.types import completion_create_params, Completion
-from anthropic._types import NOT_GIVEN, NotGiven, Headers, Query, Body
-
-from _pytest.monkeypatch import MonkeyPatch
-
-from typing import List, Union, Literal, Any, Generator
-from time import sleep
-
-import pytest
 import os
+from time import sleep
+from typing import Any, Generator, List, Literal, Union
+
+import anthropic
+import pytest
+from _pytest.monkeypatch import MonkeyPatch
+from anthropic import Anthropic
+from anthropic._types import NOT_GIVEN, Body, Headers, NotGiven, Query
+from anthropic.resources.completions import Completions
+from anthropic.types import Completion, completion_create_params
 
 MOCK = os.getenv('MOCK_SWITCH', 'false') == 'true'
 

api/tests/integration_tests/model_runtime/__mock/google.py

@@ -1,17 +1,15 @@
+from typing import Generator, List
+
+import google.generativeai.types.content_types as content_types
+import google.generativeai.types.generation_types as generation_config_types
+import google.generativeai.types.safety_types as safety_types
+import pytest
+from _pytest.monkeypatch import MonkeyPatch
+from google.ai import generativelanguage as glm
 from google.generativeai import GenerativeModel
+from google.generativeai.client import _ClientManager, configure
 from google.generativeai.types import GenerateContentResponse
 from google.generativeai.types.generation_types import BaseGenerateContentResponse
-import google.generativeai.types.generation_types as generation_config_types
-import google.generativeai.types.content_types as content_types
-import google.generativeai.types.safety_types as safety_types
-from google.generativeai.client import _ClientManager, configure
-
-from google.ai import generativelanguage as glm
-
-from typing import Generator, List
-from _pytest.monkeypatch import MonkeyPatch
-
-import pytest
 
 current_api_key = ''
 

api/tests/integration_tests/model_runtime/__mock/huggingface.py

@@ -1,12 +1,10 @@
-from tests.integration_tests.model_runtime.__mock.huggingface_chat import MockHuggingfaceChatClass
-
-from huggingface_hub import InferenceClient
-
-from _pytest.monkeypatch import MonkeyPatch
-from typing import List, Dict, Any
+import os
+from typing import Any, Dict, List
 
 import pytest
-import os
+from _pytest.monkeypatch import MonkeyPatch
+from huggingface_hub import InferenceClient
+from tests.integration_tests.model_runtime.__mock.huggingface_chat import MockHuggingfaceChatClass
 
 MOCK = os.getenv('MOCK_SWITCH', 'false').lower() == 'true'
 

api/tests/integration_tests/model_runtime/__mock/huggingface_chat.py

@@ -1,11 +1,12 @@
+import re
+from typing import Any, Generator, List, Literal, Optional, Union
+
+from _pytest.monkeypatch import MonkeyPatch
 from huggingface_hub import InferenceClient
-from huggingface_hub.inference._text_generation import TextGenerationResponse, TextGenerationStreamResponse, Details, StreamDetails, Token
+from huggingface_hub.inference._text_generation import (Details, StreamDetails, TextGenerationResponse,
+                                                        TextGenerationStreamResponse, Token)
 from huggingface_hub.utils import BadRequestError
 
-from typing import Literal, Optional, List, Generator, Union, Any
-from _pytest.monkeypatch import MonkeyPatch
-
-import re
 
 class MockHuggingfaceChatClass(object):
     @staticmethod

api/tests/integration_tests/model_runtime/__mock/openai.py

@@ -1,22 +1,22 @@
-from tests.integration_tests.model_runtime.__mock.openai_completion import MockCompletionsClass
-from tests.integration_tests.model_runtime.__mock.openai_chat import MockChatClass
-from tests.integration_tests.model_runtime.__mock.openai_remote import MockModelClass
-from tests.integration_tests.model_runtime.__mock.openai_moderation import MockModerationClass
-from tests.integration_tests.model_runtime.__mock.openai_speech2text import MockSpeech2TextClass
-from tests.integration_tests.model_runtime.__mock.openai_embeddings import MockEmbeddingsClass
-from openai.resources.completions import Completions
-from openai.resources.chat import Completions as ChatCompletions
-from openai.resources.models import Models
-from openai.resources.moderations import Moderations
-from openai.resources.audio.transcriptions import Transcriptions
-from openai.resources.embeddings import Embeddings
+import os
+from typing import Callable, List, Literal
 
+import pytest
 # import monkeypatch
 from _pytest.monkeypatch import MonkeyPatch
-from typing import Literal, Callable, List
+from openai.resources.audio.transcriptions import Transcriptions
+from openai.resources.chat import Completions as ChatCompletions
+from openai.resources.completions import Completions
+from openai.resources.embeddings import Embeddings
+from openai.resources.models import Models
+from openai.resources.moderations import Moderations
+from tests.integration_tests.model_runtime.__mock.openai_chat import MockChatClass
+from tests.integration_tests.model_runtime.__mock.openai_completion import MockCompletionsClass
+from tests.integration_tests.model_runtime.__mock.openai_embeddings import MockEmbeddingsClass
+from tests.integration_tests.model_runtime.__mock.openai_moderation import MockModerationClass
+from tests.integration_tests.model_runtime.__mock.openai_remote import MockModelClass
+from tests.integration_tests.model_runtime.__mock.openai_speech2text import MockSpeech2TextClass
 
-import os
-import pytest
 
 def mock_openai(monkeypatch: MonkeyPatch, methods: List[Literal["completion", "chat", "remote", "moderation", "speech2text", "text_embedding"]]) -> Callable[[], None]:
     """

api/tests/integration_tests/model_runtime/__mock/openai_chat.py

@@ -1,27 +1,26 @@
-from openai import OpenAI
-from openai.types import Completion as CompletionMessage
-from openai._types import NotGiven, NOT_GIVEN
-from openai.types.chat import ChatCompletion, ChatCompletionChunk, ChatCompletionMessageParam, \
-    ChatCompletionToolChoiceOptionParam, ChatCompletionToolParam, ChatCompletionMessageToolCall
-from openai.types.chat.chat_completion_chunk import ChoiceDeltaToolCall, ChoiceDeltaFunctionCall,\
-    Choice, ChoiceDelta, ChoiceDeltaToolCallFunction
-from openai.types.chat.chat_completion import Choice as _ChatCompletionChoice, ChatCompletion as _ChatCompletion
-from openai.types.chat.chat_completion_message import FunctionCall, ChatCompletionMessage
-from openai.types.chat.chat_completion_message_tool_call import Function
-from openai.types.completion_usage import CompletionUsage
-from openai.resources.chat.completions import Completions
-from openai import AzureOpenAI
+import re
+from json import dumps, loads
+from time import sleep, time
+# import monkeypatch
+from typing import Any, Generator, List, Literal, Optional, Union
 
 import openai.types.chat.completion_create_params as completion_create_params
-
-# import monkeypatch
-from typing import List, Any, Generator, Union, Optional, Literal
-from time import time, sleep
-from json import dumps, loads
-
 from core.model_runtime.errors.invoke import InvokeAuthorizationError
+from openai import AzureOpenAI, OpenAI
+from openai._types import NOT_GIVEN, NotGiven
+from openai.resources.chat.completions import Completions
+from openai.types import Completion as CompletionMessage
+from openai.types.chat import (ChatCompletion, ChatCompletionChunk, ChatCompletionMessageParam,
+                               ChatCompletionMessageToolCall, ChatCompletionToolChoiceOptionParam,
+                               ChatCompletionToolParam)
+from openai.types.chat.chat_completion import ChatCompletion as _ChatCompletion
+from openai.types.chat.chat_completion import Choice as _ChatCompletionChoice
+from openai.types.chat.chat_completion_chunk import (Choice, ChoiceDelta, ChoiceDeltaFunctionCall, ChoiceDeltaToolCall,
+                                                     ChoiceDeltaToolCallFunction)
+from openai.types.chat.chat_completion_message import ChatCompletionMessage, FunctionCall
+from openai.types.chat.chat_completion_message_tool_call import Function
+from openai.types.completion_usage import CompletionUsage
 
-import re
 
 class MockChatClass(object):
     @staticmethod

api/tests/integration_tests/model_runtime/__mock/openai_completion.py

@@ -1,17 +1,16 @@
-from openai import BadRequestError, OpenAI, AzureOpenAI
-from openai.types import Completion as CompletionMessage
-from openai._types import NotGiven, NOT_GIVEN
-from openai.types.completion import CompletionChoice
-from openai.types.completion_usage import CompletionUsage
-from openai.resources.completions import Completions
-
+import re
+from time import sleep, time
 # import monkeypatch
-from typing import List, Any, Generator, Union, Optional, Literal
-from time import time, sleep
+from typing import Any, Generator, List, Literal, Optional, Union
 
 from core.model_runtime.errors.invoke import InvokeAuthorizationError
+from openai import AzureOpenAI, BadRequestError, OpenAI
+from openai._types import NOT_GIVEN, NotGiven
+from openai.resources.completions import Completions
+from openai.types import Completion as CompletionMessage
+from openai.types.completion import CompletionChoice
+from openai.types.completion_usage import CompletionUsage
 
-import re
 
 class MockCompletionsClass(object):
     @staticmethod

api/tests/integration_tests/model_runtime/__mock/openai_embeddings.py

@@ -1,14 +1,13 @@
-from openai.resources.embeddings import Embeddings
-from openai._types import NotGiven, NOT_GIVEN
-from openai.types.create_embedding_response import CreateEmbeddingResponse, Usage
-from openai.types.embedding import Embedding
-from openai import OpenAI
-
-from typing import Union, List, Literal, Any
+import re
+from typing import Any, List, Literal, Union
 
 from core.model_runtime.errors.invoke import InvokeAuthorizationError
+from openai import OpenAI
+from openai._types import NOT_GIVEN, NotGiven
+from openai.resources.embeddings import Embeddings
+from openai.types.create_embedding_response import CreateEmbeddingResponse, Usage
+from openai.types.embedding import Embedding
 
-import re
 
 class MockEmbeddingsClass(object):
     def create_embeddings(

api/tests/integration_tests/model_runtime/__mock/openai_moderation.py

@@ -1,13 +1,12 @@
-from openai.resources.moderations import Moderations
-from openai.types import ModerationCreateResponse
-from openai.types.moderation import Moderation, Categories, CategoryScores
-from openai._types import NotGiven, NOT_GIVEN
-
-from typing import Union, List, Literal, Any
+import re
+from typing import Any, List, Literal, Union
 
 from core.model_runtime.errors.invoke import InvokeAuthorizationError
+from openai._types import NOT_GIVEN, NotGiven
+from openai.resources.moderations import Moderations
+from openai.types import ModerationCreateResponse
+from openai.types.moderation import Categories, CategoryScores, Moderation
 
-import re
 
 class MockModerationClass(object):
     def moderation_create(self: Moderations,*,

api/tests/integration_tests/model_runtime/__mock/openai_remote.py

@@ -1,8 +1,9 @@
+from time import time
+from typing import List
+
 from openai.resources.models import Models
 from openai.types.model import Model
 
-from typing import List
-from time import time
 
 class MockModelClass(object):
     """

api/tests/integration_tests/model_runtime/__mock/openai_speech2text.py

@@ -1,12 +1,11 @@
-from openai.resources.audio.transcriptions import Transcriptions
-from openai._types import NotGiven, NOT_GIVEN, FileTypes
-from openai.types.audio.transcription import Transcription
-
-from typing import Union, List, Literal, Any
+import re
+from typing import Any, List, Literal, Union
 
 from core.model_runtime.errors.invoke import InvokeAuthorizationError
+from openai._types import NOT_GIVEN, FileTypes, NotGiven
+from openai.resources.audio.transcriptions import Transcriptions
+from openai.types.audio.transcription import Transcription
 
-import re
 
 class MockSpeech2TextClass(object):
     def speech2text_create(self: Transcriptions,

api/tests/integration_tests/model_runtime/__mock/xinference.py

@@ -1,17 +1,17 @@
-from xinference_client.client.restful.restful_client import Client, \
-    RESTfulChatModelHandle, RESTfulGenerateModelHandle, RESTfulChatglmCppChatModelHandle, \
-    RESTfulEmbeddingModelHandle, RESTfulRerankModelHandle
-from xinference_client.types import Embedding, EmbeddingData, EmbeddingUsage
-
-from requests.sessions import Session
-from requests import Response
-from requests.exceptions import ConnectionError
-from typing import Union, List
-
-from _pytest.monkeypatch import MonkeyPatch
-import pytest
 import os
 import re
+from typing import List, Union
+
+import pytest
+from _pytest.monkeypatch import MonkeyPatch
+from requests import Response
+from requests.exceptions import ConnectionError
+from requests.sessions import Session
+from xinference_client.client.restful.restful_client import (Client, RESTfulChatglmCppChatModelHandle,
+                                                             RESTfulChatModelHandle, RESTfulEmbeddingModelHandle,
+                                                             RESTfulGenerateModelHandle, RESTfulRerankModelHandle)
+from xinference_client.types import Embedding, EmbeddingData, EmbeddingUsage
+
 
 class MockXinferenceClass(object):
     def get_chat_model(self: Client, model_uid: str) -> Union[RESTfulChatglmCppChatModelHandle, RESTfulGenerateModelHandle, RESTfulChatModelHandle]:

api/tests/integration_tests/model_runtime/anthropic/test_llm.py

@@ -2,15 +2,13 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage, AssistantPromptMessage
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.anthropic.llm.llm import AnthropicLargeLanguageModel
-
 from tests.integration_tests.model_runtime.__mock.anthropic import setup_anthropic_mock
 
+
 @pytest.mark.parametrize('setup_anthropic_mock', [['none']], indirect=True)
 def test_validate_credentials(setup_anthropic_mock):
     model = AnthropicLargeLanguageModel()

api/tests/integration_tests/model_runtime/anthropic/test_provider.py

@@ -1,12 +1,11 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.anthropic.anthropic import AnthropicProvider
-
 from tests.integration_tests.model_runtime.__mock.anthropic import setup_anthropic_mock
 
+
 @pytest.mark.parametrize('setup_anthropic_mock', [['none']], indirect=True)
 def test_validate_provider_credentials(setup_anthropic_mock):
     provider = AnthropicProvider()

api/tests/integration_tests/model_runtime/azure_openai/test_llm.py

@@ -2,16 +2,15 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, TextPromptMessageContent, \
-    SystemPromptMessage, ImagePromptMessageContent, PromptMessageTool, UserPromptMessage
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import (AssistantPromptMessage, ImagePromptMessageContent,
+                                                          PromptMessageTool, SystemPromptMessage,
+                                                          TextPromptMessageContent, UserPromptMessage)
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.azure_openai.llm.llm import AzureOpenAILargeLanguageModel
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['chat']], indirect=True)
 def test_validate_credentials_for_chat_model(setup_openai_mock):
     model = AzureOpenAILargeLanguageModel()

api/tests/integration_tests/model_runtime/azure_openai/test_text_embedding.py

@@ -1,13 +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.azure_openai.text_embedding.text_embedding import AzureOpenAITextEmbeddingModel
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['text_embedding']], indirect=True)
 def test_validate_credentials(setup_openai_mock):
     model = AzureOpenAITextEmbeddingModel()

api/tests/integration_tests/model_runtime/baichuan/test_llm.py

@@ -1,16 +1,15 @@
 import os
-import pytest
-
-from typing import Generator
 from time import sleep
+from typing import Generator
 
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, UserPromptMessage, SystemPromptMessage
+import pytest
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.entities.model_entities import AIModelEntity
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.baichuan.llm.llm import BaichuanLarguageModel
 
+
 def test_predefined_models():
     model = BaichuanLarguageModel()
     model_schemas = model.predefined_models()

api/tests/integration_tests/model_runtime/baichuan/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.baichuan.baichuan import BaichuanProvider
 

api/tests/integration_tests/model_runtime/baichuan/test_text_embedding.py

@@ -1,11 +1,11 @@
 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.baichuan.text_embedding.text_embedding import BaichuanTextEmbeddingModel
 
+
 def test_validate_credentials():
     model = BaichuanTextEmbeddingModel()
 

api/tests/integration_tests/model_runtime/chatglm/test_llm.py

@@ -1,18 +1,17 @@
 import os
-import pytest
-
 from typing import Generator
 
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, TextPromptMessageContent, UserPromptMessage, \
-    SystemPromptMessage, PromptMessageTool
+import pytest
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import (AssistantPromptMessage, PromptMessageTool,
+                                                          SystemPromptMessage, TextPromptMessageContent,
+                                                          UserPromptMessage)
 from core.model_runtime.entities.model_entities import AIModelEntity
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.chatglm.llm.llm import ChatGLMLargeLanguageModel
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 def test_predefined_models():
     model = ChatGLMLargeLanguageModel()
     model_schemas = model.predefined_models()

api/tests/integration_tests/model_runtime/chatglm/test_provider.py

@@ -1,12 +1,11 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.chatglm.chatglm import ChatGLMProvider
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['chat']], indirect=True)
 def test_validate_provider_credentials(setup_openai_mock):
     provider = ChatGLMProvider()

api/tests/integration_tests/model_runtime/cohere/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.cohere.cohere import CohereProvider
 

api/tests/integration_tests/model_runtime/cohere/test_rerank.py

@@ -1,6 +1,6 @@
 import os
-import pytest
 
+import pytest
 from core.model_runtime.entities.rerank_entities import RerankResult
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.cohere.rerank.rerank import CohereRerankModel

api/tests/integration_tests/model_runtime/google/test_llm.py

@@ -2,15 +2,15 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage, AssistantPromptMessage,  TextPromptMessageContent, ImagePromptMessageContent
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import (AssistantPromptMessage, ImagePromptMessageContent,
+                                                          SystemPromptMessage, TextPromptMessageContent,
+                                                          UserPromptMessage)
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.google.llm.llm import GoogleLargeLanguageModel
-
 from tests.integration_tests.model_runtime.__mock.google import setup_google_mock
 
+
 @pytest.mark.parametrize('setup_google_mock', [['none']], indirect=True)
 def test_validate_credentials(setup_google_mock):
     model = GoogleLargeLanguageModel()

api/tests/integration_tests/model_runtime/google/test_provider.py

@@ -1,12 +1,11 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.google.google import GoogleProvider
-
 from tests.integration_tests.model_runtime.__mock.google import setup_google_mock
 
+
 @pytest.mark.parametrize('setup_google_mock', [['none']], indirect=True)
 def test_validate_provider_credentials(setup_google_mock):
     provider = GoogleProvider()

api/tests/integration_tests/model_runtime/huggingface_hub/test_llm.py

@@ -2,15 +2,13 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
-from core.model_runtime.entities.message_entities import UserPromptMessage, AssistantPromptMessage
+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
 from core.model_runtime.model_providers.huggingface_hub.llm.llm import HuggingfaceHubLargeLanguageModel
-
 from tests.integration_tests.model_runtime.__mock.huggingface import setup_huggingface_mock
 
+
 @pytest.mark.parametrize('setup_huggingface_mock', [['none']], indirect=True)
 def test_hosted_inference_api_validate_credentials(setup_huggingface_mock):
     model = HuggingfaceHubLargeLanguageModel()

api/tests/integration_tests/model_runtime/huggingface_hub/test_text_embedding.py

@@ -1,7 +1,6 @@
 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 \

api/tests/integration_tests/model_runtime/jina/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.jina.jina import JinaProvider
 

api/tests/integration_tests/model_runtime/jina/test_text_embedding.py

@@ -1,7 +1,6 @@
 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.jina.text_embedding.text_embedding import JinaTextEmbeddingModel

api/tests/integration_tests/model_runtime/localai/test_llm.py

@@ -1,16 +1,16 @@
 import os
-import pytest
-
 from typing import Generator
 
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, TextPromptMessageContent, UserPromptMessage, \
-    SystemPromptMessage, PromptMessageTool
+import pytest
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import (AssistantPromptMessage, PromptMessageTool,
+                                                          SystemPromptMessage, TextPromptMessageContent,
+                                                          UserPromptMessage)
 from core.model_runtime.entities.model_entities import ParameterRule
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.localai.llm.llm import LocalAILarguageModel
 
+
 def test_validate_credentials_for_chat_model():
     model = LocalAILarguageModel()
 

api/tests/integration_tests/model_runtime/minimax/test_embedding.py

@@ -1,11 +1,11 @@
 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.minimax.text_embedding.text_embedding import MinimaxTextEmbeddingModel
 
+
 def test_validate_credentials():
     model = MinimaxTextEmbeddingModel()
 

api/tests/integration_tests/model_runtime/minimax/test_llm.py

@@ -1,16 +1,15 @@
 import os
-import pytest
-
-from typing import Generator
 from time import sleep
+from typing 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.entities.model_entities import AIModelEntity
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.minimax.llm.llm import MinimaxLargeLanguageModel
 
+
 def test_predefined_models():
     model = MinimaxLargeLanguageModel()
     model_schemas = model.predefined_models()

api/tests/integration_tests/model_runtime/minimax/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.minimax.minimax import MinimaxProvider
 

api/tests/integration_tests/model_runtime/openai/test_llm.py

@@ -2,19 +2,19 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, TextPromptMessageContent, UserPromptMessage, \
-    SystemPromptMessage, ImagePromptMessageContent, PromptMessageTool
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import (AssistantPromptMessage, ImagePromptMessageContent,
+                                                          PromptMessageTool, SystemPromptMessage,
+                                                          TextPromptMessageContent, UserPromptMessage)
 from core.model_runtime.entities.model_entities import AIModelEntity, ModelType
-from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
+from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
 from core.model_runtime.model_providers.openai.llm.llm import OpenAILargeLanguageModel
 
 """FOR MOCK FIXTURES, DO NOT REMOVE"""
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 def test_predefined_models():
     model = OpenAILargeLanguageModel()
     model_schemas = model.predefined_models()

api/tests/integration_tests/model_runtime/openai/test_moderation.py

@@ -1,12 +1,11 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.openai.moderation.moderation import OpenAIModerationModel
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['moderation']], indirect=True)
 def test_validate_credentials(setup_openai_mock):
     model = OpenAIModerationModel()

api/tests/integration_tests/model_runtime/openai/test_provider.py

@@ -1,12 +1,11 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.openai.openai import OpenAIProvider
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['chat']], indirect=True)
 def test_validate_provider_credentials(setup_openai_mock):
     provider = OpenAIProvider()

api/tests/integration_tests/model_runtime/openai/test_speech2text.py

@@ -1,12 +1,11 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.openai.speech2text.speech2text import OpenAISpeech2TextModel
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['speech2text']], indirect=True)
 def test_validate_credentials(setup_openai_mock):
     model = OpenAISpeech2TextModel()

api/tests/integration_tests/model_runtime/openai/test_text_embedding.py

@@ -1,13 +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.openai.text_embedding.text_embedding import OpenAITextEmbeddingModel
-
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock', [['text_embedding']], indirect=True)
 def test_validate_credentials(setup_openai_mock):
     model = OpenAITextEmbeddingModel()

api/tests/integration_tests/model_runtime/openai_api_compatible/test_llm.py

@@ -2,11 +2,9 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, UserPromptMessage, \
-    SystemPromptMessage, PromptMessageTool
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
+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.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.openai_api_compatible.llm.llm import OAIAPICompatLargeLanguageModel
 

api/tests/integration_tests/model_runtime/openai_api_compatible/test_text_embedding.py

@@ -1,10 +1,10 @@
 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.openai_api_compatible.text_embedding.text_embedding import OAICompatEmbeddingModel
+from core.model_runtime.model_providers.openai_api_compatible.text_embedding.text_embedding import \
+    OAICompatEmbeddingModel
 
 """
 Using OpenAI's API as testing endpoint

api/tests/integration_tests/model_runtime/openllm/test_embedding.py

@@ -1,11 +1,11 @@
 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.openllm.text_embedding.text_embedding import OpenLLMTextEmbeddingModel
 
+
 def test_validate_credentials():
     model = OpenLLMTextEmbeddingModel()
 

api/tests/integration_tests/model_runtime/openllm/test_llm.py

@@ -1,14 +1,13 @@
 import os
-import pytest
-
 from typing 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.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.openllm.llm.llm import OpenLLMLargeLanguageModel
 
+
 def test_validate_credentials_for_chat_model():
     model = OpenLLMLargeLanguageModel()
 

api/tests/integration_tests/model_runtime/replicate/test_llm.py

@@ -2,10 +2,8 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage, AssistantPromptMessage
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.replicate.llm.llm import ReplicateLargeLanguageModel
 

api/tests/integration_tests/model_runtime/replicate/test_text_embedding.py

@@ -1,7 +1,6 @@
 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.replicate.text_embedding.text_embedding import ReplicateEmbeddingModel

api/tests/integration_tests/model_runtime/spark/test_llm.py

@@ -2,10 +2,8 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage, AssistantPromptMessage
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.spark.llm.llm import SparkLargeLanguageModel
 

api/tests/integration_tests/model_runtime/spark/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.spark.spark import SparkProvider
 

api/tests/integration_tests/model_runtime/test_model_provider_factory.py

@@ -2,8 +2,8 @@ import logging
 import os
 
 from core.model_runtime.entities.model_entities import ModelType
-from core.model_runtime.entities.provider_entities import SimpleProviderEntity, ProviderConfig, ProviderEntity
-from core.model_runtime.model_providers.model_provider_factory import ModelProviderFactory, ModelProviderExtension
+from core.model_runtime.entities.provider_entities import ProviderConfig, ProviderEntity, SimpleProviderEntity
+from core.model_runtime.model_providers.model_provider_factory import ModelProviderExtension, ModelProviderFactory
 
 logger = logging.getLogger(__name__)
 

api/tests/integration_tests/model_runtime/togetherai/test_llm.py

@@ -2,11 +2,9 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, UserPromptMessage, \
-    SystemPromptMessage, PromptMessageTool
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
+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.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.togetherai.llm.llm import TogetherAILargeLanguageModel
 

api/tests/integration_tests/model_runtime/tongyi/test_llm.py

@@ -2,10 +2,8 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage, AssistantPromptMessage
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.tongyi.llm.llm import TongyiLargeLanguageModel
 

api/tests/integration_tests/model_runtime/tongyi/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.tongyi.tongyi import TongyiProvider
 

api/tests/integration_tests/model_runtime/wenxin/test_llm.py

@@ -1,16 +1,15 @@
 import os
-import pytest
-
-from typing import Generator
 from time import sleep
+from typing import Generator
 
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, UserPromptMessage, SystemPromptMessage
+import pytest
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.entities.model_entities import AIModelEntity
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.wenxin.llm.llm import ErnieBotLarguageModel
 
+
 def test_predefined_models():
     model = ErnieBotLarguageModel()
     model_schemas = model.predefined_models()

api/tests/integration_tests/model_runtime/wenxin/test_provider.py

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.wenxin.wenxin import WenxinProvider
 

api/tests/integration_tests/model_runtime/xinference/test_embeddings.py

@@ -1,12 +1,11 @@
 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.xinference.text_embedding.text_embedding import XinferenceTextEmbeddingModel
+from tests.integration_tests.model_runtime.__mock.xinference import MOCK, setup_xinference_mock
 
-from tests.integration_tests.model_runtime.__mock.xinference import setup_xinference_mock, MOCK
 
 @pytest.mark.parametrize('setup_xinference_mock', [['none']], indirect=True)
 def test_validate_credentials(setup_xinference_mock):

api/tests/integration_tests/model_runtime/xinference/test_llm.py

@@ -1,13 +1,12 @@
 import os
-import pytest
-
 from typing import Generator
 
-from core.model_runtime.entities.message_entities import AssistantPromptMessage, TextPromptMessageContent, UserPromptMessage, \
-    SystemPromptMessage, PromptMessageTool
+import pytest
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import (AssistantPromptMessage, PromptMessageTool,
+                                                          SystemPromptMessage, TextPromptMessageContent,
+                                                          UserPromptMessage)
 from core.model_runtime.entities.model_entities import AIModelEntity
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunkDelta, \
-    LLMResultChunk
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.xinference.llm.llm import XinferenceAILargeLanguageModel
 
@@ -15,6 +14,7 @@ from core.model_runtime.model_providers.xinference.llm.llm import XinferenceAILa
 from tests.integration_tests.model_runtime.__mock.openai import setup_openai_mock
 from tests.integration_tests.model_runtime.__mock.xinference import setup_xinference_mock
 
+
 @pytest.mark.parametrize('setup_openai_mock, setup_xinference_mock', [['chat', 'none']], indirect=True)
 def test_validate_credentials_for_chat_model(setup_openai_mock, setup_xinference_mock):
     model = XinferenceAILargeLanguageModel()

api/tests/integration_tests/model_runtime/xinference/test_rerank.py

@@ -1,11 +1,11 @@
 import os
-import pytest
 
+import pytest
 from core.model_runtime.entities.rerank_entities import RerankResult
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.xinference.rerank.rerank import XinferenceRerankModel
+from tests.integration_tests.model_runtime.__mock.xinference import MOCK, setup_xinference_mock
 
-from tests.integration_tests.model_runtime.__mock.xinference import setup_xinference_mock, MOCK
 
 @pytest.mark.parametrize('setup_xinference_mock', [['none']], indirect=True)
 def test_validate_credentials(setup_xinference_mock):

api/tests/integration_tests/model_runtime/zhipuai/test_llm.py

@@ -2,10 +2,8 @@ import os
 from typing import Generator
 
 import pytest
-
-from core.model_runtime.entities.message_entities import SystemPromptMessage, UserPromptMessage, AssistantPromptMessage
-from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, \
-    LLMResultChunkDelta
+from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
+from core.model_runtime.entities.message_entities import AssistantPromptMessage, SystemPromptMessage, UserPromptMessage
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.zhipuai.llm.llm import ZhipuAILargeLanguageModel
 

api/tests/integration_tests/model_runtime/zhipuai/test_provider.py

@@ -1,10 +1,10 @@
 import os
 
 import pytest
-
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.zhipuai.zhipuai import ZhipuaiProvider
 
+
 def test_validate_provider_credentials():
     provider = ZhipuaiProvider()
 

api/tests/integration_tests/model_runtime/zhipuai/test_text_embedding.py

@@ -1,7 +1,6 @@
 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.zhipuai.text_embedding.text_embedding import ZhipuAITextEmbeddingModel