fix: JSCPD errors

- Refactor to remove copy/paste

Author: holtskinner

src/a2a/utils/telemetry.py

@@ -136,10 +136,7 @@ def trace_function(
         f'Start tracing for {actual_span_name}, is_async_func {is_async_func}'
     )
 
-    @functools.wraps(func)
-    async def async_wrapper(*args, **kwargs) -> any:
-        """Async Wrapper for the decorator."""
-        logger.debug('Start async tracer')
+    async def _invoke_with_tracing(*args, **kwargs):
         tracer = trace.get_tracer(
             INSTRUMENTING_MODULE_NAME, INSTRUMENTING_MODULE_VERSION
         )
@@ -153,7 +150,11 @@ async def async_wrapper(*args, **kwargs) -> any:
 
             try:
                 # Async wrapper, await for the function call to complete.
-                result = await func(*args, **kwargs)
+                if is_async_func:
+                    result = await func(*args, **kwargs)
+                # Sync wrapper, execute the function call.
+                else:
+                    result = func(*args, **kwargs)
                 span.set_status(StatusCode.OK)
                 return result
 
@@ -173,39 +174,22 @@ async def async_wrapper(*args, **kwargs) -> any:
                             f'attribute_extractor error in span {actual_span_name}: {attr_e}'
                         )
 
+    @functools.wraps(func)
+    async def async_wrapper(*args, **kwargs):
+        """Async Wrapper for the decorator."""
+        logger.debug('Start async tracer')
+        return await _invoke_with_tracing(
+            *args,
+            **kwargs,
+        )
+
     @functools.wraps(func)
     def sync_wrapper(*args, **kwargs):
         """Sync Wrapper for the decorator."""
-        tracer = trace.get_tracer(INSTRUMENTING_MODULE_NAME)
-        with tracer.start_as_current_span(actual_span_name, kind=kind) as span:
-            if attributes:
-                for k, v in attributes.items():
-                    span.set_attribute(k, v)
-
-            result = None
-            exception = None
-
-            try:
-                # Sync wrapper, execute the function call.
-                result = func(*args, **kwargs)
-                span.set_status(StatusCode.OK)
-                return result
-
-            except Exception as e:
-                exception = e
-                span.record_exception(e)
-                span.set_status(StatusCode.ERROR, description=str(e))
-                raise
-            finally:
-                if attribute_extractor:
-                    try:
-                        attribute_extractor(
-                            span, args, kwargs, result, exception
-                        )
-                    except Exception as attr_e:
-                        logger.error(
-                            f'attribute_extractor error in span {actual_span_name}: {attr_e}'
-                        )
+        return _invoke_with_tracing(
+            *args,
+            **kwargs,
+        )
 
     return async_wrapper if is_async_func else sync_wrapper