fix: Remove the judgment of the Agent type

Author: BhAem

veadk/tracing/telemetry/telemetry.py

@@ -57,8 +57,6 @@ def trace_send_data(): ...
 def set_common_attributes(
     invocation_context: InvocationContext, current_span: _Span, **kwargs
 ) -> None:
-    from veadk.agent import Agent
-
     if current_span.context:
         current_span_id = current_span.context.trace_id
     else:
@@ -67,31 +65,26 @@ def set_common_attributes(
         )
         return
 
-    if isinstance(invocation_context.agent, Agent):
-        try:
-            spans = _INMEMORY_EXPORTER_INSTANCE.processor.spans  #  # type: ignore
-
-            spans_in_current_trace = [
-                span
-                for span in spans
-                if span.context and span.context.trace_id == current_span_id
-            ]
-
-            common_attributes = ATTRIBUTES.get("common", {})
-            for span in spans_in_current_trace:
-                if span.name.startswith("invocation"):
-                    span.set_attribute("gen_ai.operation.name", "chain")
-                elif span.name.startswith("agent_run"):
-                    span.set_attribute("gen_ai.operation.name", "agent")
-                for attr_name, attr_extractor in common_attributes.items():
-                    value = attr_extractor(**kwargs)
-                    span.set_attribute(attr_name, value)
-        except Exception as e:
-            logger.error(f"Failed to set common attributes for spans: {e}")
-    else:
-        logger.warning(
-            "Failed to set common attributes for spans as your agent is not VeADK Agent. Skip this."
-        )
+    try:
+        spans = _INMEMORY_EXPORTER_INSTANCE.processor.spans  #  # type: ignore
+
+        spans_in_current_trace = [
+            span
+            for span in spans
+            if span.context and span.context.trace_id == current_span_id
+        ]
+
+        common_attributes = ATTRIBUTES.get("common", {})
+        for span in spans_in_current_trace:
+            if span.name.startswith("invocation"):
+                span.set_attribute("gen_ai.operation.name", "chain")
+            elif span.name.startswith("agent_run"):
+                span.set_attribute("gen_ai.operation.name", "agent")
+            for attr_name, attr_extractor in common_attributes.items():
+                value = attr_extractor(**kwargs)
+                span.set_attribute(attr_name, value)
+    except Exception as e:
+        logger.error(f"Failed to set common attributes for spans: {e}")
 
 
 def trace_tool_call(