Format types

Author: holtskinner

src/a2a/types.py

@@ -133,7 +133,9 @@ class AgentSkill(A2ABaseModel):
     """
     Supported media types for output.
     """
-    tags: list[str] = Field(..., examples=[['cooking', 'customer support', 'billing']])
+    tags: list[str] = Field(
+        ..., examples=[['cooking', 'customer support', 'billing']]
+    )
     """
     Set of tagwords describing classes of capabilities for this specific skill.
     """
@@ -1271,7 +1273,9 @@ class Artifact(A2ABaseModel):
 
 
 class DeleteTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse
     """
@@ -1280,7 +1284,9 @@ class DeleteTaskPushNotificationConfigResponse(
 
 
 class GetTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
     """
@@ -1289,7 +1295,9 @@ class GetTaskPushNotificationConfigResponse(
 
 
 class ListTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse
     """
@@ -1444,7 +1452,9 @@ class SendStreamingMessageRequest(A2ABaseModel):
 
 
 class SetTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
     """
@@ -1497,7 +1507,9 @@ class TaskStatus(A2ABaseModel):
     Additional status updates for client
     """
     state: TaskState
-    timestamp: str | None = Field(default=None, examples=['2023-10-27T10:00:00Z'])
+    timestamp: str | None = Field(
+        default=None, examples=['2023-10-27T10:00:00Z']
+    )
     """
     ISO 8601 datetime string when the status was recorded.
     """
@@ -1757,7 +1769,9 @@ class SendStreamingMessageSuccessResponse(A2ABaseModel):
     """
 
 
-class CancelTaskResponse(RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]):
+class CancelTaskResponse(
+    RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]
+):
     root: JSONRPCErrorResponse | CancelTaskSuccessResponse
     """
     JSON-RPC response for the 'tasks/cancel' method.
@@ -1800,7 +1814,9 @@ class JSONRPCResponse(
     """
 
 
-class SendMessageResponse(RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]):
+class SendMessageResponse(
+    RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]
+):
     root: JSONRPCErrorResponse | SendMessageSuccessResponse
     """
     JSON-RPC response model for the 'message/send' method.