Update src/a2a/utils/task.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: pankaj-bind

src/a2a/utils/task.py

@@ -29,18 +29,18 @@ def new_task(request: Message) -> Task:
             raise ValueError('TextPart content cannot be empty')
 
     context_id_str = request.context_id
-    if context_id_str is not None:
+    context_id = request.context_id
+    if context_id is not None:
         try:
-            # Validate that the provided context_id is a valid UUID
-            uuid.UUID(context_id_str)
-            context_id = context_id_str
-        except (ValueError, AttributeError, TypeError):
-            # Catch a variety of potential issues with the UUID validation
+            # Validate that the provided context_id is a valid UUID.
+            uuid.UUID(context_id)
+        except ValueError:
+            # Re-raise as ValueError with a more specific message.
             raise ValueError(
-                f"Invalid context_id: '{context_id_str}' is not a valid UUID."
-            )
+                f"Invalid context_id: '{context_id}' is not a valid UUID."
+            ) from None
     else:
-        # Generate a new UUID if no context_id is provided
+        # Generate a new UUID if no context_id is provided.
         context_id = str(uuid.uuid4())
 
     return Task(