fix: linting

dynamiq-ai · Feb 27, 2025 · 2f5fb47 · 2f5fb47
1 parent 36836f9
commit 2f5fb47
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dynamiq/nodes/agents/base.py b/dynamiq/nodes/agents/base.py
@@ -118,8 +118,10 @@ class AgentInputSchema(BaseModel):
 
     @model_validator(mode="after")
     def validate_input_fields(self, context):
-        messages = [context.context.get("input_message"), 
-                    Message(role = MessageRole.USER, content = context.context.get("role"))]
+        messages = [
+            context.context.get("input_message"),
+            Message(role=MessageRole.USER, content=context.context.get("role")),
+        ]
         required_parameters = Prompt(messages=messages).get_required_parameters()
 
         provided_parameters = set(self.model_dump().keys())