@jaypie/express 1.2.23 → 1.2.25

package/dist/esm/index.js

@@ -1,3 +1,4 @@
+import { flushLlmObs, hasDatadogEnv, submitMetric, DATADOG, loadDatadogApiKey } from '@jaypie/datadog';
 import { Readable, Writable } from 'node:stream';
 import { ServerResponse } from 'node:http';
 import { CorsError, BadRequestError, UnhandledError, GatewayTimeoutError, UnavailableError, BadGatewayError, InternalError, TeapotError, GoneError, MethodNotAllowedError, NotFoundError, ForbiddenError, UnauthorizedError, NotImplementedError } from '@jaypie/errors';
@@ -5,7 +6,6 @@ import { force, envBoolean, JAYPIE, HTTP, getHeaderFrom, jaypieHandler } from '@
 import expressCors from 'cors';
 import { loadEnvSecrets, getContentTypeForFormat, formatStreamError } from '@jaypie/aws';
 import { log, redactAuth } from '@jaypie/logger';
-import { hasDatadogEnv, submitMetric, DATADOG } from '@jaypie/datadog';
 
 //
 //
@@ -1099,6 +1099,9 @@ function createLambdaHandler(app, _options) {
         finally {
             // Clear current invoke context
             clearCurrentInvoke();
+            // Flush buffered LLM Observability spans before the Lambda freezes.
+            // No-op unless DD_LLMOBS_ENABLED; never throws.
+            flushLlmObs();
         }
     };
 }
@@ -1142,6 +1145,9 @@ function createLambdaStreamHandler(app, _options) {
         finally {
             // Clear current invoke context
             clearCurrentInvoke();
+            // Flush buffered LLM Observability spans before the Lambda freezes.
+            // No-op unless DD_LLMOBS_ENABLED; never throws.
+            flushLlmObs();
         }
     });
 }
@@ -1863,6 +1869,10 @@ function expressHandler(handlerOrOptions, optionsOrHandler) {
         //
         // Build a request-local setup list to avoid mutating the shared array
         const requestSetup = [];
+        // Load the Datadog LLM Observability API key into the environment when enabled
+        requestSetup.push(async () => {
+            await loadDatadogApiKey();
+        });
         // Load secrets into process.env if configured
         if (secrets && secrets.length > 0) {
             const secretsToLoad = secrets;