feat: collect input and output from entities (workflows, tasks, etc.)

traceloop · Jan 17, 2024 · 276340f · 276340f
1 parent 1ed3a6a
commit 276340f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/packages/ai-semantic-conventions/src/SemanticAttributes.ts b/packages/ai-semantic-conventions/src/SemanticAttributes.ts
@@ -40,6 +40,8 @@ export const SpanAttributes = {
   TRACELOOP_WORKFLOW_NAME: "traceloop.workflow.name",
   TRACELOOP_ENTITY_NAME: "traceloop.entity.name",
   TRACELOOP_ASSOCIATION_PROPERTIES: "traceloop.association.properties",
+  TRACELOOP_ENTITY_INPUT: "traceloop.entity.input",
+  TRACELOOP_ENTITY_OUTPUT: "traceloop.entity.output",
 };
 
 export enum LLMRequestTypeValues {

diff --git a/packages/traceloop-sdk/src/lib/tracing/decorators.ts b/packages/traceloop-sdk/src/lib/tracing/decorators.ts
@@ -38,8 +38,15 @@ function withEntity<
           }
           span.setAttribute(SpanAttributes.TRACELOOP_SPAN_KIND, type);
           span.setAttribute(SpanAttributes.TRACELOOP_ENTITY_NAME, name);
+
           const res = await fn.apply(thisArg, args);
+
+          span.setAttribute(
+            SpanAttributes.TRACELOOP_ENTITY_OUTPUT,
+            JSON.stringify(res),
+          );
           span.end();
+
           return res;
         },
       ),