fix python style

langgenius · Dec 11, 2024 · d8edd32 · d8edd32
1 parent 368bd67
commit d8edd32
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/api/core/model_runtime/model_providers/bedrock/llm/llm.py b/api/core/model_runtime/model_providers/bedrock/llm/llm.py
@@ -7,6 +7,7 @@
 
 # 3rd import
 import boto3
+from api.core.model_runtime.model_providers.bedrock.get_bedrock_client import get_bedrock_client
 from botocore.config import Config
 from botocore.exceptions import (
     ClientError,
@@ -15,6 +16,7 @@
     ServiceNotInRegionError,
     UnknownServiceError,
 )
+
 # local import
 from core.model_runtime.callbacks.base_callback import Callback
 from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
@@ -40,8 +42,6 @@
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
 
-from api.core.model_runtime.model_providers.bedrock.get_bedrock_client import get_bedrock_client
-
 logger = logging.getLogger(__name__)
 ANTHROPIC_BLOCK_MODE_PROMPT = """You should always follow the instructions and output a valid {{block}} object.
 The structure of the {{block}} object you can found in the instructions, use {"answer": "$your_answer"} as the default structure

diff --git a/api/core/model_runtime/model_providers/bedrock/rerank/rerank.py b/api/core/model_runtime/model_providers/bedrock/rerank/rerank.py
@@ -1,5 +1,7 @@
 from typing import Optional
 
+from api.core.model_runtime.model_providers.bedrock.get_bedrock_client import get_bedrock_client
+
 from core.model_runtime.entities.rerank_entities import RerankDocument, RerankResult
 from core.model_runtime.errors.invoke import (
     InvokeAuthorizationError,
@@ -12,8 +14,6 @@
 from core.model_runtime.errors.validate import CredentialsValidateFailedError
 from core.model_runtime.model_providers.__base.rerank_model import RerankModel
 
-from api.core.model_runtime.model_providers.bedrock.get_bedrock_client import get_bedrock_client
-
 
 class BedrockRerankModel(RerankModel):
     """

diff --git a/api/core/model_runtime/model_providers/bedrock/text_embedding/text_embedding.py b/api/core/model_runtime/model_providers/bedrock/text_embedding/text_embedding.py
@@ -3,13 +3,15 @@
 import time
 from typing import Optional
 
+from api.core.model_runtime.model_providers.bedrock.get_bedrock_client import get_bedrock_client
 from botocore.exceptions import (
     ClientError,
     EndpointConnectionError,
     NoRegionError,
     ServiceNotInRegionError,
     UnknownServiceError,
 )
+
 from core.entities.embedding_type import EmbeddingInputType
 from core.model_runtime.entities.model_entities import PriceType
 from core.model_runtime.entities.text_embedding_entities import EmbeddingUsage, TextEmbeddingResult
@@ -23,8 +25,6 @@
 )
 from core.model_runtime.model_providers.__base.text_embedding_model import TextEmbeddingModel
 
-from api.core.model_runtime.model_providers.bedrock.get_bedrock_client import get_bedrock_client
-
 logger = logging.getLogger(__name__)