fix(traceloop-sdk): ai-sdk fixes (#865)

Author: nina-kollman

packages/traceloop-sdk/src/lib/tracing/ai-sdk-transformations.ts

@@ -723,6 +723,14 @@ const shouldHandleSpan = (span: ReadableSpan): boolean => {
   return span.instrumentationScope?.name === "ai";
 };
 
+const getAiSdkVersion = (): string | undefined => {
+  try {
+    return require("ai/package.json").version;
+  } catch {
+    return undefined;
+  }
+};
+
 const TOP_LEVEL_AI_SPANS = [
   AI_GENERATE_TEXT,
   AI_STREAM_TEXT,
@@ -750,6 +758,12 @@ export const transformAiSdkSpanAttributes = (span: ReadableSpan): void => {
   if (!shouldHandleSpan(span)) {
     return;
   }
+
+  const aiSdkVersion = getAiSdkVersion();
+  if (aiSdkVersion) {
+    span.attributes["ai.sdk.version"] = aiSdkVersion;
+  }
+
   transformLLMSpans(span.attributes, span.name);
   transformToolCalls(span);
 };

packages/traceloop-sdk/src/lib/tracing/tracing.ts

@@ -1,6 +1,9 @@
 import { trace, createContextKey, Context } from "@opentelemetry/api";
+import { version } from "../../../package.json";
 
 const TRACER_NAME = "@traceloop/node-server-sdk";
+const TRACER_VERSION = version;
+
 export const WORKFLOW_NAME_KEY = createContextKey("workflow_name");
 export const ENTITY_NAME_KEY = createContextKey("entity_name");
 export const AGENT_NAME_KEY = createContextKey("agent_name");
@@ -9,7 +12,7 @@ export const ASSOCATION_PROPERTIES_KEY = createContextKey(
 );
 
 export const getTracer = () => {
-  return trace.getTracer(TRACER_NAME);
+  return trace.getTracer(TRACER_NAME, TRACER_VERSION);
 };
 
 export const getTraceloopTracer = getTracer;

…dk/test/ai-sdk-agent-integration.test.ts …/ai-sdk/ai-sdk-agent-integration.test.tspackages/traceloop-sdk/test/ai-sdk-agent-integration.test.ts renamed to packages/traceloop-sdk/test/ai-sdk/ai-sdk-agent-integration.test.ts packages/traceloop-sdk/test/ai-sdk-agent-integration.test.ts renamed to packages/traceloop-sdk/test/ai-sdk/ai-sdk-agent-integration.test.ts

@@ -22,13 +22,13 @@ import { z } from "zod";
 import { SpanAttributes } from "@traceloop/ai-semantic-conventions";
 import { ATTR_GEN_AI_AGENT_NAME } from "@opentelemetry/semantic-conventions/incubating";
 
-import * as traceloop from "../src";
+import * as traceloop from "../../src";
 
 import { Polly, setupMocha as setupPolly } from "@pollyjs/core";
 import NodeHttpAdapter from "@pollyjs/adapter-node-http";
 import FetchAdapter from "@pollyjs/adapter-fetch";
 import FSPersister from "@pollyjs/persister-fs";
-import { initializeSharedTraceloop, getSharedExporter } from "./test-setup";
+import { initializeSharedTraceloop, getSharedExporter } from "../test-setup";
 
 const memoryExporter = getSharedExporter();
 

…loop-sdk/test/ai-sdk-integration.test.ts …k/test/ai-sdk/ai-sdk-integration.test.tspackages/traceloop-sdk/test/ai-sdk-integration.test.ts renamed to packages/traceloop-sdk/test/ai-sdk/ai-sdk-integration.test.ts packages/traceloop-sdk/test/ai-sdk-integration.test.ts renamed to packages/traceloop-sdk/test/ai-sdk/ai-sdk-integration.test.ts

@@ -28,13 +28,13 @@ import {
   ATTR_GEN_AI_SYSTEM,
 } from "@opentelemetry/semantic-conventions/incubating";
 
-import * as traceloop from "../src";
+import * as traceloop from "../../src";
 
 import { Polly, setupMocha as setupPolly } from "@pollyjs/core";
 import NodeHttpAdapter from "@pollyjs/adapter-node-http";
 import FetchAdapter from "@pollyjs/adapter-fetch";
 import FSPersister from "@pollyjs/persister-fs";
-import { initializeSharedTraceloop, getSharedExporter } from "./test-setup";
+import { initializeSharedTraceloop, getSharedExporter } from "../test-setup";
 
 const memoryExporter = getSharedExporter();
 

…-sdk/test/ai-sdk-transformations.test.ts …st/ai-sdk/ai-sdk-transformations.test.tspackages/traceloop-sdk/test/ai-sdk-transformations.test.ts renamed to packages/traceloop-sdk/test/ai-sdk/ai-sdk-transformations.test.ts packages/traceloop-sdk/test/ai-sdk-transformations.test.ts renamed to packages/traceloop-sdk/test/ai-sdk/ai-sdk-transformations.test.ts

@@ -25,12 +25,12 @@ import {
   ATTR_GEN_AI_USAGE_PROMPT_TOKENS,
 } from "@opentelemetry/semantic-conventions/incubating";
 import { context } from "@opentelemetry/api";
-import { ASSOCATION_PROPERTIES_KEY } from "../src/lib/tracing/tracing";
+import { ASSOCATION_PROPERTIES_KEY } from "../../src/lib/tracing/tracing";
 import {
   transformLLMSpans,
   transformAiSdkSpanAttributes,
   transformAiSdkSpanNames,
-} from "../src/lib/tracing/ai-sdk-transformations";
+} from "../../src/lib/tracing/ai-sdk-transformations";
 
 describe("AI SDK Transformations", () => {
   describe("transformAiSdkAttributes - response text", () => {