Formatting

Author: holtskinner

tests/utils/test_artifact.py

@@ -93,7 +93,7 @@ class TestGetArtifactText(unittest.TestCase):
     def test_get_artifact_text_single_part(self):
         # Setup
         artifact = Artifact(
-            name="test-artifact",
+            name='test-artifact',
             parts=[Part(root=TextPart(text='Hello world'))],
             artifact_id='test-artifact-id',
         )
@@ -107,7 +107,7 @@ def test_get_artifact_text_single_part(self):
     def test_get_artifact_text_multiple_parts(self):
         # Setup
         artifact = Artifact(
-            name="test-artifact",
+            name='test-artifact',
             parts=[
                 Part(root=TextPart(text='First line')),
                 Part(root=TextPart(text='Second line')),
@@ -125,7 +125,7 @@ def test_get_artifact_text_multiple_parts(self):
     def test_get_artifact_text_custom_delimiter(self):
         # Setup
         artifact = Artifact(
-            name="test-artifact",
+            name='test-artifact',
             parts=[
                 Part(root=TextPart(text='First part')),
                 Part(root=TextPart(text='Second part')),
@@ -143,7 +143,7 @@ def test_get_artifact_text_custom_delimiter(self):
     def test_get_artifact_text_empty_parts(self):
         # Setup
         artifact = Artifact(
-            name="test-artifact",
+            name='test-artifact',
             parts=[],
             artifact_id='test-artifact-id',
         )
@@ -154,5 +154,6 @@ def test_get_artifact_text_empty_parts(self):
         # Verify
         assert result == ''
 
+
 if __name__ == '__main__':
     unittest.main()

tests/utils/test_parts.py

@@ -1,4 +1,3 @@
-
 from a2a.types import (
     DataPart,
     FilePart,
@@ -8,9 +7,9 @@
     TextPart,
 )
 from a2a.utils.parts import (
-    get_text_parts,
     get_data_parts,
     get_file_parts,
+    get_text_parts,
 )