Merge branch 'main' into add-a2a-routes-to-existing-app

Author: MadaraUchiha-314

scripts/grpc_gen_post_processor.py

@@ -1,5 +1,5 @@
-"""
-Fix absolute imports in *_pb2_grpc.py files.
+"""Fix absolute imports in *_pb2_grpc.py files.
+
 Example:
 import a2a_pb2 as a2a__pb2
 from . import a2a_pb2 as a2a__pb2
@@ -16,40 +16,40 @@ def process_generated_code(src_folder: str = 'src/a2a/grpc'):
     dir_path = Path(src_folder)
     print(dir_path)
     if not dir_path.is_dir():
-      print('Source folder not found')
-      sys.exit(1)
+        print('Source folder not found')
+        sys.exit(1)
 
     grpc_pattern = '**/*_pb2_grpc.py'
     files = dir_path.glob(grpc_pattern)
 
     for file in files:
-      print(f'Processing {file}')
-      try:
-        with file.open('r', encoding='utf-8') as f:
-          src_content = f.read()
-
-          # Change import a2a_pb2 as a2a__pb2
-          import_pattern = r'^import (\w+_pb2) as (\w+__pb2)$'
-          # to from . import a2a_pb2 as a2a__pb2
-          replacement_pattern = r'from . import \1 as \2'
-
-          fixed_src_content = re.sub(
-            import_pattern,
-            replacement_pattern,
-            src_content,
-            flags=re.MULTILINE,
-          )
-
-          if fixed_src_content != src_content:
-            with file.open('w', encoding='utf-8') as f:
-              f.write(fixed_src_content)
-              print('Imports fixed')
-          else:
-              print('No changes needed')
-
-      except Exception as e:
-        print(f'Error processing file {file}: {e}')
-        sys.exit(1)
+        print(f'Processing {file}')
+        try:
+            with file.open('r', encoding='utf-8') as f:
+                src_content = f.read()
+
+                # Change import a2a_pb2 as a2a__pb2
+                import_pattern = r'^import (\w+_pb2) as (\w+__pb2)$'
+                # to from . import a2a_pb2 as a2a__pb2
+                replacement_pattern = r'from . import \1 as \2'
+
+                fixed_src_content = re.sub(
+                    import_pattern,
+                    replacement_pattern,
+                    src_content,
+                    flags=re.MULTILINE,
+                )
+
+                if fixed_src_content != src_content:
+                    with file.open('w', encoding='utf-8') as f:
+                        f.write(fixed_src_content)
+                        print('Imports fixed')
+                else:
+                    print('No changes needed')
+
+        except Exception as e:
+            print(f'Error processing file {file}: {e}')
+            sys.exit(1)
 
 
 if __name__ == '__main__':

src/a2a/server/events/event_consumer.py

@@ -112,7 +112,7 @@ async def consume_all(self) -> AsyncGenerator[Event]:
                             TaskState.failed,
                             TaskState.rejected,
                             TaskState.unknown,
-                            TaskState.input_required
+                            TaskState.input_required,
                         )
                     )
                 )

src/a2a/server/request_handlers/grpc_handler.py

@@ -55,7 +55,7 @@ def __init__(
         self,
         agent_card: AgentCard,
         request_handler: RequestHandler,
-        context_builder: CallContextBuilder | None = None
+        context_builder: CallContextBuilder | None = None,
     ):
         """Initializes the GrpcHandler.