ai 3.2.39 → 3.2.41

package/dist/index.js

@@ -258,6 +258,33 @@ function recordSpan({
   });
 }
 
+// core/telemetry/select-telemetry-attributes.ts
+function selectTelemetryAttributes({
+  telemetry,
+  attributes
+}) {
+  return Object.entries(attributes).reduce((attributes2, [key, value]) => {
+    if (value === void 0) {
+      return attributes2;
+    }
+    if (typeof value === "object" && "input" in value && typeof value.input === "function") {
+      if ((telemetry == null ? void 0 : telemetry.recordInputs) === false) {
+        return attributes2;
+      }
+      const result = value.input();
+      return result === void 0 ? attributes2 : { ...attributes2, [key]: result };
+    }
+    if (typeof value === "object" && "output" in value && typeof value.output === "function") {
+      if ((telemetry == null ? void 0 : telemetry.recordOutputs) === false) {
+        return attributes2;
+      }
+      const result = value.output();
+      return result === void 0 ? attributes2 : { ...attributes2, [key]: result };
+    }
+    return { ...attributes2, [key]: value };
+  }, {});
+}
+
 // core/util/retry-with-exponential-backoff.ts
 var import_provider = require("@ai-sdk/provider");
 var import_provider_utils = require("@ai-sdk/provider-utils");
@@ -340,11 +367,13 @@ async function embed({
   const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   return recordSpan({
     name: "ai.embed",
-    attributes: {
-      ...baseTelemetryAttributes,
-      // specific settings that only make sense on the outer level:
-      "ai.value": JSON.stringify(value)
-    },
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...baseTelemetryAttributes,
+        "ai.value": { input: () => JSON.stringify(value) }
+      }
+    }),
     tracer,
     fn: async (span) => {
       const retry = retryWithExponentialBackoff({ maxRetries });
@@ -353,11 +382,14 @@ async function embed({
           // nested spans to align with the embedMany telemetry data:
           recordSpan({
             name: "ai.embed.doEmbed",
-            attributes: {
-              ...baseTelemetryAttributes,
-              // specific settings that only make sense on the outer level:
-              "ai.values": [JSON.stringify(value)]
-            },
+            attributes: selectTelemetryAttributes({
+              telemetry,
+              attributes: {
+                ...baseTelemetryAttributes,
+                // specific settings that only make sense on the outer level:
+                "ai.values": { input: () => [JSON.stringify(value)] }
+              }
+            }),
             tracer,
             fn: async (doEmbedSpan) => {
               var _a2;
@@ -368,12 +400,19 @@ async function embed({
               });
               const embedding2 = modelResponse.embeddings[0];
               const usage2 = (_a2 = modelResponse.usage) != null ? _a2 : { tokens: NaN };
-              doEmbedSpan.setAttributes({
-                "ai.embeddings": modelResponse.embeddings.map(
-                  (embedding3) => JSON.stringify(embedding3)
-                ),
-                "ai.usage.tokens": usage2.tokens
-              });
+              doEmbedSpan.setAttributes(
+                selectTelemetryAttributes({
+                  telemetry,
+                  attributes: {
+                    "ai.embeddings": {
+                      output: () => modelResponse.embeddings.map(
+                        (embedding3) => JSON.stringify(embedding3)
+                      )
+                    },
+                    "ai.usage.tokens": usage2.tokens
+                  }
+                })
+              );
               return {
                 embedding: embedding2,
                 usage: usage2,
@@ -383,10 +422,15 @@ async function embed({
           })
         )
       );
-      span.setAttributes({
-        "ai.embedding": JSON.stringify(embedding),
-        "ai.usage.tokens": usage.tokens
-      });
+      span.setAttributes(
+        selectTelemetryAttributes({
+          telemetry,
+          attributes: {
+            "ai.embedding": { output: () => JSON.stringify(embedding) },
+            "ai.usage.tokens": usage.tokens
+          }
+        })
+      );
       return new DefaultEmbedResult({ value, embedding, usage, rawResponse });
     }
   });
@@ -432,11 +476,16 @@ async function embedMany({
   const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   return recordSpan({
     name: "ai.embedMany",
-    attributes: {
-      ...baseTelemetryAttributes,
-      // specific settings that only make sense on the outer level:
-      "ai.values": values.map((value) => JSON.stringify(value))
-    },
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...baseTelemetryAttributes,
+        // specific settings that only make sense on the outer level:
+        "ai.values": {
+          input: () => values.map((value) => JSON.stringify(value))
+        }
+      }
+    }),
     tracer,
     fn: async (span) => {
       const retry = retryWithExponentialBackoff({ maxRetries });
@@ -445,11 +494,16 @@ async function embedMany({
         const { embeddings: embeddings2, usage } = await retry(() => {
           return recordSpan({
             name: "ai.embedMany.doEmbed",
-            attributes: {
-              ...baseTelemetryAttributes,
-              // specific settings that only make sense on the outer level:
-              "ai.values": values.map((value) => JSON.stringify(value))
-            },
+            attributes: selectTelemetryAttributes({
+              telemetry,
+              attributes: {
+                ...baseTelemetryAttributes,
+                // specific settings that only make sense on the outer level:
+                "ai.values": {
+                  input: () => values.map((value) => JSON.stringify(value))
+                }
+              }
+            }),
             tracer,
             fn: async (doEmbedSpan) => {
               var _a2;
@@ -460,22 +514,32 @@ async function embedMany({
               });
               const embeddings3 = modelResponse.embeddings;
               const usage2 = (_a2 = modelResponse.usage) != null ? _a2 : { tokens: NaN };
-              doEmbedSpan.setAttributes({
-                "ai.embeddings": embeddings3.map(
-                  (embedding) => JSON.stringify(embedding)
-                ),
-                "ai.usage.tokens": usage2.tokens
-              });
+              doEmbedSpan.setAttributes(
+                selectTelemetryAttributes({
+                  telemetry,
+                  attributes: {
+                    "ai.embeddings": {
+                      output: () => embeddings3.map((embedding) => JSON.stringify(embedding))
+                    },
+                    "ai.usage.tokens": usage2.tokens
+                  }
+                })
+              );
               return { embeddings: embeddings3, usage: usage2 };
             }
           });
         });
-        span.setAttributes({
-          "ai.embeddings": embeddings2.map(
-            (embedding) => JSON.stringify(embedding)
-          ),
-          "ai.usage.tokens": usage.tokens
-        });
+        span.setAttributes(
+          selectTelemetryAttributes({
+            telemetry,
+            attributes: {
+              "ai.embeddings": {
+                output: () => embeddings2.map((embedding) => JSON.stringify(embedding))
+              },
+              "ai.usage.tokens": usage.tokens
+            }
+          })
+        );
         return new DefaultEmbedManyResult({ values, embeddings: embeddings2, usage });
       }
       const valueChunks = splitArray(values, maxEmbeddingsPerCall);
@@ -485,11 +549,16 @@ async function embedMany({
         const { embeddings: responseEmbeddings, usage } = await retry(() => {
           return recordSpan({
             name: "ai.embedMany.doEmbed",
-            attributes: {
-              ...baseTelemetryAttributes,
-              // specific settings that only make sense on the outer level:
-              "ai.values": chunk.map((value) => JSON.stringify(value))
-            },
+            attributes: selectTelemetryAttributes({
+              telemetry,
+              attributes: {
+                ...baseTelemetryAttributes,
+                // specific settings that only make sense on the outer level:
+                "ai.values": {
+                  input: () => chunk.map((value) => JSON.stringify(value))
+                }
+              }
+            }),
             tracer,
             fn: async (doEmbedSpan) => {
               var _a2;
@@ -500,12 +569,17 @@ async function embedMany({
               });
               const embeddings2 = modelResponse.embeddings;
               const usage2 = (_a2 = modelResponse.usage) != null ? _a2 : { tokens: NaN };
-              doEmbedSpan.setAttributes({
-                "ai.embeddings": embeddings2.map(
-                  (embedding) => JSON.stringify(embedding)
-                ),
-                "ai.usage.tokens": usage2.tokens
-              });
+              doEmbedSpan.setAttributes(
+                selectTelemetryAttributes({
+                  telemetry,
+                  attributes: {
+                    "ai.embeddings": {
+                      output: () => embeddings2.map((embedding) => JSON.stringify(embedding))
+                    },
+                    "ai.usage.tokens": usage2.tokens
+                  }
+                })
+              );
               return { embeddings: embeddings2, usage: usage2 };
             }
           });
@@ -513,10 +587,17 @@ async function embedMany({
         embeddings.push(...responseEmbeddings);
         tokens += usage.tokens;
       }
-      span.setAttributes({
-        "ai.embeddings": embeddings.map((embedding) => JSON.stringify(embedding)),
-        "ai.usage.tokens": tokens
-      });
+      span.setAttributes(
+        selectTelemetryAttributes({
+          telemetry,
+          attributes: {
+            "ai.embeddings": {
+              output: () => embeddings.map((embedding) => JSON.stringify(embedding))
+            },
+            "ai.usage.tokens": tokens
+          }
+        })
+      );
       return new DefaultEmbedManyResult({
         values,
         embeddings,
@@ -1073,13 +1154,20 @@ async function generateObject({
   const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   return recordSpan({
     name: "ai.generateObject",
-    attributes: {
-      ...baseTelemetryAttributes,
-      // specific settings that only make sense on the outer level:
-      "ai.prompt": JSON.stringify({ system, prompt, messages }),
-      "ai.schema": JSON.stringify(schema.jsonSchema),
-      "ai.settings.mode": mode
-    },
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...baseTelemetryAttributes,
+        // specific settings that only make sense on the outer level:
+        "ai.prompt": {
+          input: () => JSON.stringify({ system, prompt, messages })
+        },
+        "ai.schema": {
+          input: () => JSON.stringify(schema.jsonSchema)
+        },
+        "ai.settings.mode": mode
+      }
+    }),
     tracer,
     fn: async (span) => {
       const retry = retryWithExponentialBackoff({ maxRetries });
@@ -1110,12 +1198,19 @@ async function generateObject({
           const generateResult = await retry(
             () => recordSpan({
               name: "ai.generateObject.doGenerate",
-              attributes: {
-                ...baseTelemetryAttributes,
-                "ai.prompt.format": inputFormat,
-                "ai.prompt.messages": JSON.stringify(promptMessages),
-                "ai.settings.mode": mode
-              },
+              attributes: selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  ...baseTelemetryAttributes,
+                  "ai.prompt.format": {
+                    input: () => inputFormat
+                  },
+                  "ai.prompt.messages": {
+                    input: () => JSON.stringify(promptMessages)
+                  },
+                  "ai.settings.mode": mode
+                }
+              }),
               tracer,
               fn: async (span2) => {
                 const result2 = await model.doGenerate({
@@ -1129,12 +1224,17 @@ async function generateObject({
                 if (result2.text === void 0) {
                   throw new import_provider6.NoObjectGeneratedError();
                 }
-                span2.setAttributes({
-                  "ai.finishReason": result2.finishReason,
-                  "ai.usage.promptTokens": result2.usage.promptTokens,
-                  "ai.usage.completionTokens": result2.usage.completionTokens,
-                  "ai.result.object": result2.text
-                });
+                span2.setAttributes(
+                  selectTelemetryAttributes({
+                    telemetry,
+                    attributes: {
+                      "ai.finishReason": result2.finishReason,
+                      "ai.usage.promptTokens": result2.usage.promptTokens,
+                      "ai.usage.completionTokens": result2.usage.completionTokens,
+                      "ai.result.object": { output: () => result2.text }
+                    }
+                  })
+                );
                 return { ...result2, objectText: result2.text };
               }
             })
@@ -1161,12 +1261,19 @@ async function generateObject({
           const generateResult = await retry(
             () => recordSpan({
               name: "ai.generateObject.doGenerate",
-              attributes: {
-                ...baseTelemetryAttributes,
-                "ai.prompt.format": inputFormat,
-                "ai.prompt.messages": JSON.stringify(promptMessages),
-                "ai.settings.mode": mode
-              },
+              attributes: selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  ...baseTelemetryAttributes,
+                  "ai.prompt.format": {
+                    input: () => inputFormat
+                  },
+                  "ai.prompt.messages": {
+                    input: () => JSON.stringify(promptMessages)
+                  },
+                  "ai.settings.mode": mode
+                }
+              }),
               tracer,
               fn: async (span2) => {
                 var _a2, _b;
@@ -1190,12 +1297,17 @@ async function generateObject({
                 if (objectText === void 0) {
                   throw new import_provider6.NoObjectGeneratedError();
                 }
-                span2.setAttributes({
-                  "ai.finishReason": result2.finishReason,
-                  "ai.usage.promptTokens": result2.usage.promptTokens,
-                  "ai.usage.completionTokens": result2.usage.completionTokens,
-                  "ai.result.object": objectText
-                });
+                span2.setAttributes(
+                  selectTelemetryAttributes({
+                    telemetry,
+                    attributes: {
+                      "ai.finishReason": result2.finishReason,
+                      "ai.usage.promptTokens": result2.usage.promptTokens,
+                      "ai.usage.completionTokens": result2.usage.completionTokens,
+                      "ai.result.object": { output: () => objectText }
+                    }
+                  })
+                );
                 return { ...result2, objectText };
               }
             })
@@ -1222,12 +1334,19 @@ async function generateObject({
       if (!parseResult.success) {
         throw parseResult.error;
       }
-      span.setAttributes({
-        "ai.finishReason": finishReason,
-        "ai.usage.promptTokens": usage.promptTokens,
-        "ai.usage.completionTokens": usage.completionTokens,
-        "ai.result.object": JSON.stringify(parseResult.value)
-      });
+      span.setAttributes(
+        selectTelemetryAttributes({
+          telemetry,
+          attributes: {
+            "ai.finishReason": finishReason,
+            "ai.usage.promptTokens": usage.promptTokens,
+            "ai.usage.completionTokens": usage.completionTokens,
+            "ai.result.object": {
+              output: () => JSON.stringify(parseResult.value)
+            }
+          }
+        })
+      );
       return new DefaultGenerateObjectResult({
         object: parseResult.value,
         finishReason,
@@ -1347,13 +1466,18 @@ async function streamObject({
   const schema = asSchema(inputSchema);
   return recordSpan({
     name: "ai.streamObject",
-    attributes: {
-      ...baseTelemetryAttributes,
-      // specific settings that only make sense on the outer level:
-      "ai.prompt": JSON.stringify({ system, prompt, messages }),
-      "ai.schema": JSON.stringify(schema.jsonSchema),
-      "ai.settings.mode": mode
-    },
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...baseTelemetryAttributes,
+        // specific settings that only make sense on the outer level:
+        "ai.prompt": {
+          input: () => JSON.stringify({ system, prompt, messages })
+        },
+        "ai.schema": { input: () => JSON.stringify(schema.jsonSchema) },
+        "ai.settings.mode": mode
+      }
+    }),
     tracer,
     endWhenDone: false,
     fn: async (rootSpan) => {
@@ -1454,20 +1578,25 @@ async function streamObject({
       } = await retry(
         () => recordSpan({
           name: "ai.streamObject.doStream",
-          attributes: {
-            ...baseTelemetryAttributes,
-            "ai.prompt.format": callOptions.inputFormat,
-            "ai.prompt.messages": JSON.stringify(callOptions.prompt),
-            "ai.settings.mode": mode
-          },
+          attributes: selectTelemetryAttributes({
+            telemetry,
+            attributes: {
+              ...baseTelemetryAttributes,
+              "ai.prompt.format": {
+                input: () => callOptions.inputFormat
+              },
+              "ai.prompt.messages": {
+                input: () => JSON.stringify(callOptions.prompt)
+              },
+              "ai.settings.mode": mode
+            }
+          }),
           tracer,
           endWhenDone: false,
-          fn: async (doStreamSpan2) => {
-            return {
-              result: await model.doStream(callOptions),
-              doStreamSpan: doStreamSpan2
-            };
-          }
+          fn: async (doStreamSpan2) => ({
+            result: await model.doStream(callOptions),
+            doStreamSpan: doStreamSpan2
+          })
         })
       );
       return new DefaultStreamObjectResult({
@@ -1477,7 +1606,8 @@ async function streamObject({
         schema,
         onFinish,
         rootSpan,
-        doStreamSpan
+        doStreamSpan,
+        telemetry
       });
     }
   });
@@ -1490,7 +1620,8 @@ var DefaultStreamObjectResult = class {
     schema,
     onFinish,
     rootSpan,
-    doStreamSpan
+    doStreamSpan,
+    telemetry
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -1572,17 +1703,31 @@ var DefaultStreamObjectResult = class {
               completionTokens: NaN,
               totalTokens: NaN
             };
-            doStreamSpan.setAttributes({
-              "ai.usage.promptTokens": finalUsage.promptTokens,
-              "ai.usage.completionTokens": finalUsage.completionTokens,
-              "ai.result.object": JSON.stringify(object)
-            });
+            doStreamSpan.setAttributes(
+              selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  "ai.usage.promptTokens": finalUsage.promptTokens,
+                  "ai.usage.completionTokens": finalUsage.completionTokens,
+                  "ai.result.object": {
+                    output: () => JSON.stringify(object)
+                  }
+                }
+              })
+            );
             doStreamSpan.end();
-            rootSpan.setAttributes({
-              "ai.usage.promptTokens": finalUsage.promptTokens,
-              "ai.usage.completionTokens": finalUsage.completionTokens,
-              "ai.result.object": JSON.stringify(object)
-            });
+            rootSpan.setAttributes(
+              selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  "ai.usage.promptTokens": finalUsage.promptTokens,
+                  "ai.usage.completionTokens": finalUsage.completionTokens,
+                  "ai.result.object": {
+                    output: () => JSON.stringify(object)
+                  }
+                }
+              })
+            );
             await (onFinish == null ? void 0 : onFinish({
               usage: finalUsage,
               object,
@@ -1777,12 +1922,17 @@ async function generateText({
   const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   return recordSpan({
     name: "ai.generateText",
-    attributes: {
-      ...baseTelemetryAttributes,
-      // specific settings that only make sense on the outer level:
-      "ai.prompt": JSON.stringify({ system, prompt, messages }),
-      "ai.settings.maxToolRoundtrips": maxToolRoundtrips
-    },
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...baseTelemetryAttributes,
+        // specific settings that only make sense on the outer level:
+        "ai.prompt": {
+          input: () => JSON.stringify({ system, prompt, messages })
+        },
+        "ai.settings.maxToolRoundtrips": maxToolRoundtrips
+      }
+    }),
     tracer,
     fn: async (span) => {
       var _a2, _b, _c, _d;
@@ -1817,11 +1967,16 @@ async function generateText({
         currentModelResponse = await retry(
           () => recordSpan({
             name: "ai.generateText.doGenerate",
-            attributes: {
-              ...baseTelemetryAttributes,
-              "ai.prompt.format": currentInputFormat,
-              "ai.prompt.messages": JSON.stringify(promptMessages)
-            },
+            attributes: selectTelemetryAttributes({
+              telemetry,
+              attributes: {
+                ...baseTelemetryAttributes,
+                "ai.prompt.format": { input: () => currentInputFormat },
+                "ai.prompt.messages": {
+                  input: () => JSON.stringify(promptMessages)
+                }
+              }
+            }),
             tracer,
             fn: async (span2) => {
               const result = await model.doGenerate({
@@ -1832,13 +1987,22 @@ async function generateText({
                 abortSignal,
                 headers
               });
-              span2.setAttributes({
-                "ai.finishReason": result.finishReason,
-                "ai.usage.promptTokens": result.usage.promptTokens,
-                "ai.usage.completionTokens": result.usage.completionTokens,
-                "ai.result.text": result.text,
-                "ai.result.toolCalls": JSON.stringify(result.toolCalls)
-              });
+              span2.setAttributes(
+                selectTelemetryAttributes({
+                  telemetry,
+                  attributes: {
+                    "ai.finishReason": result.finishReason,
+                    "ai.usage.promptTokens": result.usage.promptTokens,
+                    "ai.usage.completionTokens": result.usage.completionTokens,
+                    "ai.result.text": {
+                      output: () => result.text
+                    },
+                    "ai.result.toolCalls": {
+                      output: () => JSON.stringify(result.toolCalls)
+                    }
+                  }
+                })
+              );
               return result;
             }
           })
@@ -1849,7 +2013,8 @@ async function generateText({
         currentToolResults = tools == null ? [] : await executeTools({
           toolCalls: currentToolCalls,
           tools,
-          tracer
+          tracer,
+          telemetry
         });
         const currentUsage = calculateCompletionTokenUsage(
           currentModelResponse.usage
@@ -1883,13 +2048,22 @@ async function generateText({
         currentToolResults.length === currentToolCalls.length && // the number of roundtrips is less than the maximum:
         roundtripCount++ < maxToolRoundtrips
       );
-      span.setAttributes({
-        "ai.finishReason": currentModelResponse.finishReason,
-        "ai.usage.promptTokens": currentModelResponse.usage.promptTokens,
-        "ai.usage.completionTokens": currentModelResponse.usage.completionTokens,
-        "ai.result.text": currentModelResponse.text,
-        "ai.result.toolCalls": JSON.stringify(currentModelResponse.toolCalls)
-      });
+      span.setAttributes(
+        selectTelemetryAttributes({
+          telemetry,
+          attributes: {
+            "ai.finishReason": currentModelResponse.finishReason,
+            "ai.usage.promptTokens": currentModelResponse.usage.promptTokens,
+            "ai.usage.completionTokens": currentModelResponse.usage.completionTokens,
+            "ai.result.text": {
+              output: () => currentModelResponse.text
+            },
+            "ai.result.toolCalls": {
+              output: () => JSON.stringify(currentModelResponse.toolCalls)
+            }
+          }
+        })
+      );
       return new DefaultGenerateTextResult({
         // Always return a string so that the caller doesn't have to check for undefined.
         // If they need to check if the model did not return any text,
@@ -1911,7 +2085,8 @@ async function generateText({
 async function executeTools({
   toolCalls,
   tools,
-  tracer
+  tracer,
+  telemetry
 }) {
   const toolResults = await Promise.all(
     toolCalls.map(async (toolCall) => {
@@ -1921,18 +2096,30 @@ async function executeTools({
       }
       const result = await recordSpan({
         name: "ai.toolCall",
-        attributes: {
-          "ai.toolCall.name": toolCall.toolName,
-          "ai.toolCall.id": toolCall.toolCallId,
-          "ai.toolCall.args": JSON.stringify(toolCall.args)
-        },
+        attributes: selectTelemetryAttributes({
+          telemetry,
+          attributes: {
+            "ai.toolCall.name": toolCall.toolName,
+            "ai.toolCall.id": toolCall.toolCallId,
+            "ai.toolCall.args": {
+              output: () => JSON.stringify(toolCall.args)
+            }
+          }
+        }),
         tracer,
         fn: async (span) => {
           const result2 = await tool2.execute(toolCall.args);
           try {
-            span.setAttributes({
-              "ai.toolCall.result": JSON.stringify(result2)
-            });
+            span.setAttributes(
+              selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  "ai.toolCall.result": {
+                    output: () => JSON.stringify(result2)
+                  }
+                }
+              })
+            );
           } catch (ignored) {
           }
           return result2;
@@ -2084,7 +2271,8 @@ function runToolsTransformation({
   tools,
   generatorStream,
   toolCallStreaming,
-  tracer
+  tracer,
+  telemetry
 }) {
   let canClose = false;
   const outstandingToolCalls = /* @__PURE__ */ new Set();
@@ -2154,11 +2342,16 @@ function runToolsTransformation({
               outstandingToolCalls.add(toolExecutionId);
               recordSpan({
                 name: "ai.toolCall",
-                attributes: {
-                  "ai.toolCall.name": toolCall.toolName,
-                  "ai.toolCall.id": toolCall.toolCallId,
-                  "ai.toolCall.args": JSON.stringify(toolCall.args)
-                },
+                attributes: selectTelemetryAttributes({
+                  telemetry,
+                  attributes: {
+                    "ai.toolCall.name": toolCall.toolName,
+                    "ai.toolCall.id": toolCall.toolCallId,
+                    "ai.toolCall.args": {
+                      output: () => JSON.stringify(toolCall.args)
+                    }
+                  }
+                }),
                 tracer,
                 fn: async (span) => tool2.execute(toolCall.args).then(
                   (result) => {
@@ -2172,9 +2365,16 @@ function runToolsTransformation({
                       toolResultsStreamController.close();
                     }
                     try {
-                      span.setAttributes({
-                        "ai.toolCall.result": JSON.stringify(result)
-                      });
+                      span.setAttributes(
+                        selectTelemetryAttributes({
+                          telemetry,
+                          attributes: {
+                            "ai.toolCall.result": {
+                              output: () => JSON.stringify(result)
+                            }
+                          }
+                        })
+                      );
                     } catch (ignored) {
                     }
                   },
@@ -2275,11 +2475,16 @@ async function streamText({
   const tracer = getTracer({ isEnabled: (_a = telemetry == null ? void 0 : telemetry.isEnabled) != null ? _a : false });
   return recordSpan({
     name: "ai.streamText",
-    attributes: {
-      ...baseTelemetryAttributes,
-      // specific settings that only make sense on the outer level:
-      "ai.prompt": JSON.stringify({ system, prompt, messages })
-    },
+    attributes: selectTelemetryAttributes({
+      telemetry,
+      attributes: {
+        ...baseTelemetryAttributes,
+        // specific settings that only make sense on the outer level:
+        "ai.prompt": {
+          input: () => JSON.stringify({ system, prompt, messages })
+        }
+      }
+    }),
     tracer,
     endWhenDone: false,
     fn: async (rootSpan) => {
@@ -2295,11 +2500,18 @@ async function streamText({
       } = await retry(
         () => recordSpan({
           name: "ai.streamText.doStream",
-          attributes: {
-            ...baseTelemetryAttributes,
-            "ai.prompt.format": validatedPrompt.type,
-            "ai.prompt.messages": JSON.stringify(promptMessages)
-          },
+          attributes: selectTelemetryAttributes({
+            telemetry,
+            attributes: {
+              ...baseTelemetryAttributes,
+              "ai.prompt.format": {
+                input: () => validatedPrompt.type
+              },
+              "ai.prompt.messages": {
+                input: () => JSON.stringify(promptMessages)
+              }
+            }
+          }),
           tracer,
           endWhenDone: false,
           fn: async (doStreamSpan2) => {
@@ -2325,13 +2537,15 @@ async function streamText({
           tools,
           generatorStream: stream,
           toolCallStreaming,
-          tracer
+          tracer,
+          telemetry
         }),
         warnings,
         rawResponse,
         onFinish,
         rootSpan,
-        doStreamSpan
+        doStreamSpan,
+        telemetry
       });
     }
   });
@@ -2343,7 +2557,8 @@ var DefaultStreamTextResult = class {
     rawResponse,
     onFinish,
     rootSpan,
-    doStreamSpan
+    doStreamSpan,
+    telemetry
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
@@ -2423,21 +2638,31 @@ var DefaultStreamTextResult = class {
             };
             const finalFinishReason = finishReason != null ? finishReason : "unknown";
             const telemetryToolCalls = toolCalls.length > 0 ? JSON.stringify(toolCalls) : void 0;
-            doStreamSpan.setAttributes({
-              "ai.finishReason": finalFinishReason,
-              "ai.usage.promptTokens": finalUsage.promptTokens,
-              "ai.usage.completionTokens": finalUsage.completionTokens,
-              "ai.result.text": text,
-              "ai.result.toolCalls": telemetryToolCalls
-            });
+            doStreamSpan.setAttributes(
+              selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  "ai.finishReason": finalFinishReason,
+                  "ai.usage.promptTokens": finalUsage.promptTokens,
+                  "ai.usage.completionTokens": finalUsage.completionTokens,
+                  "ai.result.text": { output: () => text },
+                  "ai.result.toolCalls": { output: () => telemetryToolCalls }
+                }
+              })
+            );
             doStreamSpan.end();
-            rootSpan.setAttributes({
-              "ai.finishReason": finalFinishReason,
-              "ai.usage.promptTokens": finalUsage.promptTokens,
-              "ai.usage.completionTokens": finalUsage.completionTokens,
-              "ai.result.text": text,
-              "ai.result.toolCalls": telemetryToolCalls
-            });
+            rootSpan.setAttributes(
+              selectTelemetryAttributes({
+                telemetry,
+                attributes: {
+                  "ai.finishReason": finalFinishReason,
+                  "ai.usage.promptTokens": finalUsage.promptTokens,
+                  "ai.usage.completionTokens": finalUsage.completionTokens,
+                  "ai.result.text": { output: () => text },
+                  "ai.result.toolCalls": { output: () => telemetryToolCalls }
+                }
+              })
+            );
             resolveToolResults(toolResults);
             await ((_a = self.onFinish) == null ? void 0 : _a.call(self, {
               finishReason: finalFinishReason,
@@ -3476,7 +3701,8 @@ function InkeepStream(res, callbacks) {
 var langchain_adapter_exports = {};
 __export(langchain_adapter_exports, {
   toAIStream: () => toAIStream,
-  toDataStream: () => toDataStream
+  toDataStream: () => toDataStream,
+  toDataStreamResponse: () => toDataStreamResponse
 });
 function toAIStream(stream, callbacks) {
   return toDataStream(stream, callbacks);
@@ -3504,6 +3730,21 @@ function toDataStream(stream, callbacks) {
     })
   ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
+function toDataStreamResponse(stream, options) {
+  var _a;
+  const dataStream = toDataStream(stream, options == null ? void 0 : options.callbacks);
+  const data = options == null ? void 0 : options.data;
+  const init = options == null ? void 0 : options.init;
+  const responseStream = data ? mergeStreams(data.stream, dataStream) : dataStream;
+  return new Response(responseStream, {
+    status: (_a = init == null ? void 0 : init.status) != null ? _a : 200,
+    statusText: init == null ? void 0 : init.statusText,
+    headers: prepareResponseHeaders(init, {
+      contentType: "text/plain; charset=utf-8",
+      dataStreamVersion: "v1"
+    })
+  });
+}
 function forwardAIMessageChunk(chunk, controller) {
   if (typeof chunk.content === "string") {
     controller.enqueue(chunk.content);