Fix proto_utils, bug wasn't exposed until the base class was edited

holtskinner · holtskinner · commit 0b610f954bd5 · 2025-07-30T17:06:30.000+01:00
diff --git a/src/a2a/utils/proto_utils.py b/src/a2a/utils/proto_utils.py
@@ -363,6 +363,12 @@ def security_scheme(
                     flows=cls.oauth2_flows(scheme.root.flows),
                 )
             )
+        if isinstance(scheme.root, types.MutualTLSSecurityScheme):
+            return a2a_pb2.SecurityScheme(
+                mtls_security_scheme=a2a_pb2.MutualTlsSecurityScheme(
+                    description=scheme.root.description,
+                )
+            )
         return a2a_pb2.SecurityScheme(
             open_id_connect_security_scheme=a2a_pb2.OpenIdConnectSecurityScheme(
                 description=scheme.root.description,

Original file line number	Diff line number	Diff line change
`@@ -363,6 +363,12 @@ def security_scheme(`
`363`	`363`	`flows=cls.oauth2_flows(scheme.root.flows),`
`364`	`364`	`)`
`365`	`365`	`)`
	`366`	`+ if isinstance(scheme.root, types.MutualTLSSecurityScheme):`
	`367`	`+ return a2a_pb2.SecurityScheme(`
	`368`	`+ mtls_security_scheme=a2a_pb2.MutualTlsSecurityScheme(`
	`369`	`+ description=scheme.root.description,`
	`370`	`+ )`
	`371`	`+ )`
`366`	`372`	`return a2a_pb2.SecurityScheme(`
`367`	`373`	`open_id_connect_security_scheme=a2a_pb2.OpenIdConnectSecurityScheme(`
`368`	`374`	`description=scheme.root.description,`