fix (ai/core): Add null check for OTEL attributes

vercel · Feb 8, 2025 · be01951 · be01951
1 parent eb99d48
commit be01951
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/...lemetry/select-temetry-attributes.test.ts → ...metry/select-telemetry-attributes.test.ts b/...lemetry/select-temetry-attributes.test.ts → ...metry/select-telemetry-attributes.test.ts
@@ -99,6 +99,7 @@ it('should handle mixed attribute types correctly', () => {
       input: { input: () => 'input value' },
       output: { output: () => 'output value' },
       undefined: undefined,
+      someNull: null,
     },
   });
   expect(result).toEqual({

diff --git a/packages/ai/core/telemetry/select-telemetry-attributes.ts b/packages/ai/core/telemetry/select-telemetry-attributes.ts
@@ -11,7 +11,8 @@ export function selectTelemetryAttributes({
       | AttributeValue
       | { input: () => AttributeValue | undefined }
       | { output: () => AttributeValue | undefined }
-      | undefined;
+      | undefined
+      | null;
   };
 }): Attributes {
   // when telemetry is disabled, return an empty object to avoid serialization overhead:
@@ -20,7 +21,7 @@ export function selectTelemetryAttributes({
   }
 
   return Object.entries(attributes).reduce((attributes, [key, value]) => {
-    if (value === undefined) {
+    if (value === undefined || value === null) {
       return attributes;
     }