fix gemini-code-assist issues

Author: pankaj-bind

src/a2a/utils/task.py

@@ -15,7 +15,7 @@ def new_task(request: Message) -> Task:
 
     Returns:
         A new `Task` object initialized with 'submitted' status and the input message in history.
-    
+
     Raises:
         TypeError: If the message role is None.
         ValueError: If the message parts are empty.
@@ -24,7 +24,7 @@ def new_task(request: Message) -> Task:
         raise TypeError('Message role cannot be None')
     if not request.parts:
         raise ValueError('Message parts cannot be empty')
-        
+
     return Task(
         status=TaskStatus(state=TaskState.submitted),
         id=(request.taskId if request.taskId else str(uuid.uuid4())),

tests/utils/test_task.py

@@ -126,12 +126,13 @@ def test_new_task_invalid_message_empty_parts(self):
             )
 
     def test_new_task_invalid_message_none_role(self):
-        with self.assertRaises(ValidationError):
-            Message(
+        with self.assertRaises(TypeError):
+            msg = Message.model_construct(
                 role=None,
                 parts=[Part(root=TextPart(text='test message'))],
                 messageId=str(uuid.uuid4()),
             )
+            new_task(msg)
 
 
 if __name__ == '__main__':