prettier

Author: galkleinman

packages/ai-semantic-conventions/src/SemanticAttributes.ts

@@ -35,7 +35,8 @@ export const SpanAttributes = {
   LLM_USAGE_COMPLETION_TOKENS: "gen_ai.usage.completion_tokens",
   LLM_USAGE_INPUT_TOKENS: ATTR_GEN_AI_USAGE_INPUT_TOKENS,
   LLM_USAGE_OUTPUT_TOKENS: ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,
-  LLM_USAGE_CACHE_CREATION_INPUT_TOKENS: "gen_ai.usage.cache_creation_input_tokens",
+  LLM_USAGE_CACHE_CREATION_INPUT_TOKENS:
+    "gen_ai.usage.cache_creation_input_tokens",
   LLM_USAGE_CACHE_READ_INPUT_TOKENS: "gen_ai.usage.cache_read_input_tokens",
 
   GEN_AI_AGENT_NAME: "gen_ai.agent.name",

packages/traceloop-sdk/recordings/Test-AI-SDK-Integration-with-Recording_156038438/should-capture-and-transform-cache-tokens-from-OpenAI-with-prompt-caching_4027203422/recording.har

@@ -327,4 +327,4 @@
     "pages": [],
     "version": "1.2"
   }
-}
+}

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

@@ -33,7 +33,8 @@ const AI_PROMPT = "ai.prompt";
 const AI_USAGE_PROMPT_TOKENS = "ai.usage.promptTokens";
 const AI_USAGE_COMPLETION_TOKENS = "ai.usage.completionTokens";
 const AI_USAGE_CACHED_INPUT_TOKENS = "ai.usage.cachedInputTokens";
-const AI_USAGE_CACHE_CREATION_INPUT_TOKENS = "ai.usage.cacheCreationInputTokens";
+const AI_USAGE_CACHE_CREATION_INPUT_TOKENS =
+  "ai.usage.cacheCreationInputTokens";
 const AI_USAGE_CACHE_READ_INPUT_TOKENS = "ai.usage.cacheReadInputTokens";
 const AI_MODEL_PROVIDER = "ai.model.provider";
 const AI_PROMPT_TOOLS = "ai.prompt.tools";
@@ -371,7 +372,9 @@ const calculateTotalTokens = (attributes: Record<string, any>): void => {
   }
 };
 
-const transformCacheCreationInputTokens = (attributes: Record<string, any>): void => {
+const transformCacheCreationInputTokens = (
+  attributes: Record<string, any>,
+): void => {
   // Transform ai.usage.cacheCreationInputTokens to gen_ai.usage.cache_creation_input_tokens
   if (AI_USAGE_CACHE_CREATION_INPUT_TOKENS in attributes) {
     attributes[SpanAttributes.LLM_USAGE_CACHE_CREATION_INPUT_TOKENS] =
@@ -380,7 +383,9 @@ const transformCacheCreationInputTokens = (attributes: Record<string, any>): voi
   }
 };
 
-const transformCacheReadInputTokens = (attributes: Record<string, any>): void => {
+const transformCacheReadInputTokens = (
+  attributes: Record<string, any>,
+): void => {
   // Transform ai.usage.cacheReadInputTokens to gen_ai.usage.cache_read_input_tokens
   if (AI_USAGE_CACHE_READ_INPUT_TOKENS in attributes) {
     attributes[SpanAttributes.LLM_USAGE_CACHE_READ_INPUT_TOKENS] =

packages/traceloop-sdk/test/ai-sdk-transformations.test.ts

@@ -953,7 +953,10 @@ describe("AI SDK Transformations", () => {
         attributes[SpanAttributes.LLM_USAGE_CACHE_CREATION_INPUT_TOKENS],
         1024,
       );
-      assert.strictEqual(attributes["ai.usage.cacheCreationInputTokens"], undefined);
+      assert.strictEqual(
+        attributes["ai.usage.cacheCreationInputTokens"],
+        undefined,
+      );
       assert.strictEqual(attributes.someOtherAttr, "value");
     });
 
@@ -969,7 +972,10 @@ describe("AI SDK Transformations", () => {
         attributes[SpanAttributes.LLM_USAGE_CACHE_READ_INPUT_TOKENS],
         512,
       );
-      assert.strictEqual(attributes["ai.usage.cacheReadInputTokens"], undefined);
+      assert.strictEqual(
+        attributes["ai.usage.cacheReadInputTokens"],
+        undefined,
+      );
       assert.strictEqual(attributes.someOtherAttr, "value");
     });
 
@@ -1007,11 +1013,26 @@ describe("AI SDK Transformations", () => {
         attributes[SpanAttributes.LLM_USAGE_CACHE_READ_INPUT_TOKENS],
         512,
       );
-      assert.strictEqual(attributes[SpanAttributes.LLM_USAGE_INPUT_TOKENS], 2048);
-      assert.strictEqual(attributes[SpanAttributes.LLM_USAGE_OUTPUT_TOKENS], 100);
-      assert.strictEqual(attributes[SpanAttributes.LLM_USAGE_TOTAL_TOKENS], 2148);
-      assert.strictEqual(attributes["ai.usage.cacheCreationInputTokens"], undefined);
-      assert.strictEqual(attributes["ai.usage.cacheReadInputTokens"], undefined);
+      assert.strictEqual(
+        attributes[SpanAttributes.LLM_USAGE_INPUT_TOKENS],
+        2048,
+      );
+      assert.strictEqual(
+        attributes[SpanAttributes.LLM_USAGE_OUTPUT_TOKENS],
+        100,
+      );
+      assert.strictEqual(
+        attributes[SpanAttributes.LLM_USAGE_TOTAL_TOKENS],
+        2148,
+      );
+      assert.strictEqual(
+        attributes["ai.usage.cacheCreationInputTokens"],
+        undefined,
+      );
+      assert.strictEqual(
+        attributes["ai.usage.cacheReadInputTokens"],
+        undefined,
+      );
     });
 
     it("should prefer cacheReadInputTokens over cachedInputTokens when both present", () => {
@@ -1029,7 +1050,10 @@ describe("AI SDK Transformations", () => {
         attributes[SpanAttributes.LLM_USAGE_CACHE_READ_INPUT_TOKENS],
         256,
       );
-      assert.strictEqual(attributes["ai.usage.cacheReadInputTokens"], undefined);
+      assert.strictEqual(
+        attributes["ai.usage.cacheReadInputTokens"],
+        undefined,
+      );
       assert.strictEqual(attributes["ai.usage.cachedInputTokens"], undefined);
     });
 
@@ -1079,7 +1103,10 @@ describe("AI SDK Transformations", () => {
       transformLLMSpans(attributes);
 
       // Should preserve input tokens and add total tokens
-      assert.strictEqual(attributes[SpanAttributes.LLM_USAGE_INPUT_TOKENS], 100);
+      assert.strictEqual(
+        attributes[SpanAttributes.LLM_USAGE_INPUT_TOKENS],
+        100,
+      );
       assert.strictEqual(attributes.someOtherAttr, "value");
       assert.strictEqual(
         attributes[SpanAttributes.LLM_USAGE_CACHE_CREATION_INPUT_TOKENS],
@@ -1128,8 +1155,14 @@ describe("AI SDK Transformations", () => {
       assert.strictEqual(attributes[SpanAttributes.LLM_SYSTEM], "Anthropic");
 
       // Check original attributes are removed
-      assert.strictEqual(attributes["ai.usage.cacheCreationInputTokens"], undefined);
-      assert.strictEqual(attributes["ai.usage.cacheReadInputTokens"], undefined);
+      assert.strictEqual(
+        attributes["ai.usage.cacheCreationInputTokens"],
+        undefined,
+      );
+      assert.strictEqual(
+        attributes["ai.usage.cacheReadInputTokens"],
+        undefined,
+      );
       assert.strictEqual(attributes["ai.response.text"], undefined);
       assert.strictEqual(attributes["ai.model.provider"], undefined);
       assert.strictEqual(attributes.someOtherAttr, "value");