update library and tests to use snake_case

Author: holtskinner

src/a2a/client/helpers.py

@@ -18,5 +18,7 @@ def create_text_message_object(
         A `Message` object with a new UUID messageId.
     """
     return Message(
-        role=role, parts=[Part(TextPart(text=content))], messageId=str(uuid4())
+        role=role,
+        parts=[Part(root=TextPart(text=content))],
+        message_id=str(uuid4()),
     )

src/a2a/server/tasks/task_manager.py

@@ -96,7 +96,7 @@ async def save_task_event(
                          when the TaskManager's ID is already set.
         """
         task_id_from_event = (
-            event.id if isinstance(event, Task) else event.taskId
+            event.id if isinstance(event, Task) else event.task_id
         )
         # If task id is known, make sure it is matched
         if self.task_id and self.task_id != task_id_from_event:
@@ -107,8 +107,8 @@ async def save_task_event(
             )
         if not self.task_id:
             self.task_id = task_id_from_event
-        if not self.context_id and self.context_id != event.contextId:
-            self.context_id = event.contextId
+        if not self.context_id and self.context_id != event.context_id:
+            self.context_id = event.context_id
 
         logger.debug(
             'Processing save of task event of type %s for task_id: %s',
@@ -160,12 +160,12 @@ async def ensure_task(
         if not task:
             logger.info(
                 'Task not found or task_id not set. Creating new task for event (task_id: %s, context_id: %s).',
-                event.taskId,
-                event.contextId,
+                event.task_id,
+                event.context_id,
             )
             # streaming agent did not previously stream task object.
             # Create a task object with the available information and persist the event
-            task = self._init_task_obj(event.taskId, event.contextId)
+            task = self._init_task_obj(event.task_id, event.context_id)
             await self._save_task(task)
 
         return task
@@ -207,7 +207,7 @@ def _init_task_obj(self, task_id: str, context_id: str) -> Task:
         history = [self._initial_message] if self._initial_message else []
         return Task(
             id=task_id,
-            contextId=context_id,
+            context_id=context_id,
             status=TaskStatus(state=TaskState.submitted),
             history=history,
         )
@@ -224,7 +224,7 @@ async def _save_task(self, task: Task) -> None:
         if not self.task_id:
             logger.info('New task created with id: %s', task.id)
             self.task_id = task.id
-            self.context_id = task.contextId
+            self.context_id = task.context_id
 
     def update_with_message(self, message: Message, task: Task) -> Task:
         """Updates a task object in memory by adding a new message to its history.

src/a2a/utils/artifact.py

@@ -21,7 +21,7 @@ def new_artifact(
         A new `Artifact` object with a generated artifactId.
     """
     return Artifact(
-        artifactId=str(uuid.uuid4()),
+        artifact_id=str(uuid.uuid4()),
         parts=parts,
         name=name,
         description=description,

src/a2a/utils/helpers.py

@@ -36,12 +36,12 @@ def create_task_obj(message_send_params: MessageSendParams) -> Task:
     Returns:
         A new `Task` object initialized with 'submitted' status and the input message in history.
     """
-    if not message_send_params.message.contextId:
-        message_send_params.message.contextId = str(uuid4())
+    if not message_send_params.message.context_id:
+        message_send_params.message.context_id = str(uuid4())
 
     return Task(
         id=str(uuid4()),
-        contextId=message_send_params.message.contextId,
+        context_id=message_send_params.message.context_id,
         status=TaskStatus(state=TaskState.submitted),
         history=[message_send_params.message],
     )
@@ -62,15 +62,15 @@ def append_artifact_to_task(task: Task, event: TaskArtifactUpdateEvent) -> None:
         task.artifacts = []
 
     new_artifact_data: Artifact = event.artifact
-    artifact_id: str = new_artifact_data.artifactId
+    artifact_id: str = new_artifact_data.artifact_id
     append_parts: bool = event.append or False
 
     existing_artifact: Artifact | None = None
     existing_artifact_list_index: int | None = None
 
     # Find existing artifact by its id
     for i, art in enumerate(task.artifacts):
-        if art.artifactId == artifact_id:
+        if art.artifact_id == artifact_id:
             existing_artifact = art
             existing_artifact_list_index = i
             break
@@ -115,7 +115,7 @@ def build_text_artifact(text: str, artifact_id: str) -> Artifact:
     """
     text_part = TextPart(text=text)
     part = Part(root=text_part)
-    return Artifact(parts=[part], artifactId=artifact_id)
+    return Artifact(parts=[part], artifact_id=artifact_id)
 
 
 def validate(

src/a2a/utils/message.py

@@ -28,9 +28,9 @@ def new_agent_text_message(
     return Message(
         role=Role.agent,
         parts=[Part(root=TextPart(text=text))],
-        messageId=str(uuid.uuid4()),
-        taskId=task_id,
-        contextId=context_id,
+        message_id=str(uuid.uuid4()),
+        task_id=task_id,
+        context_id=context_id,
     )
 
 
@@ -52,9 +52,9 @@ def new_agent_parts_message(
     return Message(
         role=Role.agent,
         parts=parts,
-        messageId=str(uuid.uuid4()),
-        taskId=task_id,
-        contextId=context_id,
+        message_id=str(uuid.uuid4()),
+        task_id=task_id,
+        context_id=context_id,
     )
 
 

src/a2a/utils/task.py

@@ -18,9 +18,9 @@ def new_task(request: Message) -> Task:
     """
     return Task(
         status=TaskStatus(state=TaskState.submitted),
-        id=(request.taskId if request.taskId else str(uuid.uuid4())),
-        contextId=(
-            request.contextId if request.contextId else str(uuid.uuid4())
+        id=(request.task_id if request.task_id else str(uuid.uuid4())),
+        context_id=(
+            request.context_id if request.context_id else str(uuid.uuid4())
         ),
         history=[request],
     )
@@ -51,7 +51,7 @@ def completed_task(
     return Task(
         status=TaskStatus(state=TaskState.completed),
         id=task_id,
-        contextId=context_id,
+        context_id=context_id,
         artifacts=artifacts,
         history=history,
     )