span_type -> span_kind

Author: abhishekg999

src/judgeval/tracer/__init__.py

@@ -274,7 +274,7 @@ def observe(
 
         name = func.__qualname__
         attributes = {
-            AttributeKeys.SPAN_TYPE: span_type,
+            AttributeKeys.JUDGMENT_SPAN_KIND: span_type,
         }
 
         if inspect.iscoroutinefunction(func):

src/judgeval/tracer/keys.py

@@ -8,9 +8,8 @@
 
 
 class AttributeKeys:
-    SPAN_TYPE = SpanAttributes.TYPE
-
     # General function tracing attributes (custom namespace)
+    JUDGMENT_SPAN_KIND = "judgment.span_kind"
     JUDGMENT_INPUT = "judgment.input"
     JUDGMENT_OUTPUT = "judgment.output"
 

src/judgeval/tracer/llm/__init__.py

@@ -53,7 +53,7 @@ def wrapped(function):
         @functools.wraps(function)
         def wrapper(*args, **kwargs):
             with sync_span_context(
-                tracer, span_name, {AttributeKeys.SPAN_TYPE: "llm"}
+                tracer, span_name, {AttributeKeys.JUDGMENT_SPAN_KIND: "llm"}
             ) as span:
                 span.set_attribute(AttributeKeys.GEN_AI_PROMPT, safe_serialize(kwargs))
                 try:
@@ -87,7 +87,7 @@ def wrapped_async(function):
         @functools.wraps(function)
         async def wrapper(*args, **kwargs):
             async with async_span_context(
-                tracer, span_name, {AttributeKeys.SPAN_TYPE: "llm"}
+                tracer, span_name, {AttributeKeys.JUDGMENT_SPAN_KIND: "llm"}
             ) as span:
                 span.set_attribute(AttributeKeys.GEN_AI_PROMPT, safe_serialize(kwargs))
                 try: