format: Update python format

aws-samples · Oct 21, 2024 · f227999 · f227999
1 parent 75204eb
commit f227999
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/lib/model-interfaces/langchain/functions/request-handler/adapters/base/base.py b/lib/model-interfaces/langchain/functions/request-handler/adapters/base/base.py
@@ -37,6 +37,7 @@
 class Mode(Enum):
     CHAIN = "chain"
 
+
 class LLMStartHandler(BaseCallbackHandler):
     prompts = []
     usage = None

diff --git a/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/base.py b/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/base.py
@@ -245,7 +245,9 @@ def __init__(self, *args, **kwargs):
 # Register the adapters
 registry.register(r"^bedrock.ai21.jamba*", BedrockChatAdapter)
 registry.register(r"^bedrock.ai21.j2*", BedrockChatNoStreamingNoSystemPromptAdapter)
-registry.register(r"^bedrock\.cohere\.command-(text|light-text).*", BedrockChatNoSystemPromptAdapter)
+registry.register(
+    r"^bedrock\.cohere\.command-(text|light-text).*", BedrockChatNoSystemPromptAdapter
+)
 registry.register(r"^bedrock\.cohere\.command-r.*", BedrockChatAdapter)
 registry.register(r"^bedrock.anthropic.claude*", BedrockChatAdapter)
 registry.register(r"^bedrock.meta.llama*", BedrockChatAdapter)