Undo changes and silence lint

Author: holtskinner

.github/linters/.mypy.ini

@@ -1,6 +1,6 @@
 [mypy]
 exclude = examples/
-disable_error_code = import-not-found
+disable_error_code = import-not-found,annotation-unchecked
 
 [mypy-examples.*]
 follow_imports = skip

src/a2a/utils/telemetry.py

@@ -215,7 +215,7 @@ def sync_wrapper(*args, **kwargs):
 def trace_class(
     include_list: list[str] | None = None,
     exclude_list: list[str] | None = None,
-    kind: SpanKind = SpanKind.INTERNAL,
+    kind=SpanKind.INTERNAL,
 ):
     """A class decorator to automatically trace specified methods of a class.
 
@@ -276,37 +276,21 @@ def decorator(cls):
             if name.startswith('__') and name.endswith('__'):
                 continue
 
-            # Apply inclusion/exclusion rules
-            if include_list is not None:
-                # If include_list is specified, only trace methods in it
-                if name not in include_list:
-                    logger.debug(
-                        f'Skipping method not in include_list: {cls.__name__}.{name}'
-                    )
-                    continue
-            elif exclude_list is not None:
-                # If include_list is not specified, exclude methods in exclude_list
-                if name in exclude_list:
-                    logger.debug(
-                        f'Skipping method in exclude_list: {cls.__name__}.{name}'
-                    )
-                    continue
-            # If neither list is specified, all non-dunder methods are traced by default
-
-            # Construct default span name including class name
-            default_span_name = f'{cls.__module__}.{cls.__name__}.{name}'
-
-            # Apply the trace_function decorator to the method
-            logger.debug(
-                f'Applying trace_function to method: {default_span_name}'
-            )
-            traced_method = trace_function(
-                span_name=default_span_name, kind=kind
-            )(method)
-
-            # Replace the original method with the traced version
-            setattr(cls, name, traced_method)
+            # Skip if include list is defined but the method not included.
+            if include_list and name not in include_list:
+                continue
+            # Skip if include list is not defined but the method is in excludes.
+            if not include_list and name in exclude_list:
+                continue
 
+            all_methods[name] = method
+            span_name = f'{cls.__module__}.{cls.__name__}.{name}'
+            # Set the decorator on the method.
+            setattr(
+                cls,
+                name,
+                trace_function(span_name=span_name, kind=kind)(method),
+            )
         return cls
 
     return decorator