chore: lint

aws-samples · Jan 14, 2025 · 1bb4d64 · 1bb4d64
1 parent 61023f4
commit 1bb4d64
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/backend/app/usecases/bot.py b/backend/app/usecases/bot.py
@@ -219,7 +219,9 @@ def create_new_bot(user_id: str, bot_input: BotInput) -> BotOutput:
                 if bot_input.bedrock_guardrails
                 else None
             ),
-            active_models=ActiveModelsModel.model_validate(dict(bot_input.active_models)),
+            active_models=ActiveModelsModel.model_validate(
+                dict(bot_input.active_models)
+            ),
         ),
     )
     return BotOutput(
@@ -261,7 +263,9 @@ def create_new_bot(user_id: str, bot_input: BotInput) -> BotOutput:
             ]
         ),
         bedrock_knowledge_base=(
-            BedrockKnowledgeBaseOutput(**(bot_input.bedrock_knowledge_base.model_dump()))
+            BedrockKnowledgeBaseOutput(
+                **(bot_input.bedrock_knowledge_base.model_dump())
+            )
             if bot_input.bedrock_knowledge_base
             else None
         ),
@@ -323,7 +327,8 @@ def modify_owned_bot(
             tools=[
                 AgentToolModel(name=t.name, description=t.description)
                 for t in [
-                    get_tool_by_name(tool_name) for tool_name in modify_input.agent.tools
+                    get_tool_by_name(tool_name)
+                    for tool_name in modify_input.agent.tools
                 ]
             ]
         )
@@ -371,7 +376,9 @@ def modify_owned_bot(
             ]
         ),
         bedrock_knowledge_base=(
-            BedrockKnowledgeBaseModel(**modify_input.bedrock_knowledge_base.model_dump())
+            BedrockKnowledgeBaseModel(
+                **modify_input.bedrock_knowledge_base.model_dump()
+            )
             if modify_input.bedrock_knowledge_base
             else None
         ),
@@ -380,7 +387,9 @@ def modify_owned_bot(
             if modify_input.bedrock_guardrails
             else None
         ),
-        active_models=ActiveModelsOutput.model_validate(dict(modify_input.active_models)),
+        active_models=ActiveModelsOutput.model_validate(
+            dict(modify_input.active_models)
+        ),
     )
 
     return BotModifyOutput(
@@ -424,7 +433,9 @@ def modify_owned_bot(
             if modify_input.bedrock_guardrails
             else None
         ),
-        active_models=ActiveModelsOutput.model_validate(dict(modify_input.active_models)),
+        active_models=ActiveModelsOutput.model_validate(
+            dict(modify_input.active_models)
+        ),
     )