@rasputin-ai/elysia 0.10.2 → 0.11.0

package/dist/index.js

@@ -8,6 +8,7 @@ import {
   scheduleAnnounceDeploy
 } from "@rasputin-ai/core";
 import {
+  bindAdaptiveRecorder,
   createExecutionRecorder,
   installAutomaticExecutionRuntime,
   installGlobalHandlers,
@@ -17,7 +18,7 @@ import { Elysia } from "elysia";
 
 // src/sdk-meta.ts
 var SDK_NAME = "@rasputin-ai/elysia";
-var SDK_VERSION = "0.10.2";
+var SDK_VERSION = "0.11.0";
 
 // src/rasputin-init.ts
 var publicExecution = (runtime) => ({
@@ -33,7 +34,7 @@ var publicExecution = (runtime) => ({
   getStats: runtime.getStats
 });
 var withExecution = (client, runtime, installRuntime, manifest, uninstallHandlers = () => {
-}) => {
+}, policy) => {
   const uninstallRuntime = installRuntime ? installAutomaticExecutionRuntime(runtime) : () => {
   };
   const manifestUpload = scheduleInstrumentationManifestUpload({
@@ -56,9 +57,11 @@ var withExecution = (client, runtime, installRuntime, manifest, uninstallHandler
         const startedAt = Date.now();
         await client.flush(timeoutMs);
         if (timeoutMs === void 0) {
+          await policy?.flush();
           await manifestUpload.wait();
           return;
         }
+        await policy?.flush(Math.max(0, timeoutMs - (Date.now() - startedAt)));
         await manifestUpload.wait(Math.max(0, timeoutMs - (Date.now() - startedAt)));
       },
       close: async () => {
@@ -66,6 +69,7 @@ var withExecution = (client, runtime, installRuntime, manifest, uninstallHandler
         uninstallRuntime();
         manifestUpload.disconnect();
         const startedAt = Date.now();
+        await policy?.close();
         await client.close();
         await manifestUpload.wait(Math.max(0, 2e3 - (Date.now() - startedAt)));
       }
@@ -102,10 +106,17 @@ var observeExecutionError = (scope, error) => {
     return void 0;
   }
 };
-var buildPlugin = (client, execution, recorderEnabled) => {
+var mergeRequestMetadata = (scope, request) => {
+  if (!scope) return;
+  try {
+    scope.mergeMetadata({ request });
+  } catch {
+  }
+};
+var buildPlugin = (client, runtime, recorderEnabled) => {
   const executionBoundary = ((handler, initialRequest) => {
     return (request) => {
-      const scope = execution.createScope({
+      const scope = runtime.createScope({
         kind: "http",
         request: { method: request.method || initialRequest.method }
       });
@@ -115,34 +126,31 @@ var buildPlugin = (client, execution, recorderEnabled) => {
       } catch {
       }
       return scope.run(() => {
-        try {
-          const response = handler(request);
-          if (isNativeResponsePromise(response)) {
-            Promise.prototype.then.call(
-              response,
-              (value) => {
-                finishExecution(scope, { request: { status: value.status } });
-              },
-              (error) => {
-                observeExecutionError(scope, error);
-                finishExecution(scope);
-              }
-            );
-            return response;
-          }
-          finishExecution(scope, { request: { status: response.status } });
+        const response = handler(request);
+        if (isNativeResponsePromise(response)) {
+          Promise.prototype.then.call(
+            response,
+            (value) => {
+              finishExecution(scope, { request: { status: value.status } });
+            },
+            (_error) => {
+            }
+          );
           return response;
-        } catch (error) {
-          observeExecutionError(scope, error);
-          finishExecution(scope);
-          throw error;
         }
+        finishExecution(scope, { request: { status: response.status } });
+        return response;
       });
     };
   });
   const plugin = new Elysia({ name: "rasputin" }).decorate("rasputin", client);
   if (recorderEnabled) plugin.wrap(executionBoundary);
-  return plugin.onError({ as: "global" }, ({ error, request, set, route }) => {
+  return plugin.onBeforeHandle({ as: "global" }, ({ request, route }) => {
+    mergeRequestMetadata(requestScopes.get(request), {
+      method: request.method,
+      route: route || void 0
+    });
+  }).onError({ as: "global" }, ({ error, request, set, route }) => {
     const requestContext = {
       method: request.method,
       route: route || void 0,
@@ -150,7 +158,9 @@ var buildPlugin = (client, execution, recorderEnabled) => {
     };
     try {
       const scope = requestScopes.get(request);
+      mergeRequestMetadata(scope, requestContext);
       const observation = scope ? observeExecutionError(scope, error) : void 0;
+      if (scope) finishExecution(scope, { request: requestContext });
       client.captureException(error, {
         request: requestContext,
         ...observation ? { observation } : {}
@@ -170,9 +180,19 @@ function RasputinInit(options, hooks = {}) {
     } = hooks;
     const enabled = isClientEnabled(options);
     const recorderEnabled = enabled && options.executionRecorder?.enabled !== false;
-    const runtime = createExecutionRecorder({
-      ...options.executionRecorder,
-      enabled: recorderEnabled
+    const { runtime, policy } = bindAdaptiveRecorder({
+      executionRecorder: options.executionRecorder,
+      recorderEnabled,
+      dsn: options.dsn,
+      projectApiKey: options.projectApiKey,
+      apiUrl: options.apiUrl,
+      release: options.release,
+      environment: options.environment,
+      fetch,
+      enabled: options.enabled,
+      sdkName: SDK_NAME,
+      sdkVersion: SDK_VERSION,
+      scrubPII: options.scrubPII
     });
     const client = createClient(options, {
       fetch,
@@ -199,7 +219,8 @@ function RasputinInit(options, hooks = {}) {
         sourceRoots: options.sourceRoots,
         logSuccess: options.logSuccess
       },
-      () => uninstallHandlers()
+      () => uninstallHandlers(),
+      policy
     );
     if (enabled && installProcessHandlers) {
       uninstallHandlers = installGlobalHandlers(wrappedClient);

package/dist/rasputin-init.d.ts

@@ -5,7 +5,8 @@
  * the framework-neutral Node recorder.
  */
 import { type RasputinOptions } from '@rasputin-ai/core';
-import type { ExecutionRecorderOptions, RasputinExecution, RasputinNodeClient } from '@rasputin-ai/node';
+import type { ExecutionRecorderOptions, RasputinNodeClient } from '@rasputin-ai/node';
+import { type ExecutionRuntime } from '@rasputin-ai/node/internal';
 import { Elysia } from 'elysia';
 /** Elysia plugin returned by `RasputinInit`. Register it with `.use(plugin)`. */
 export type RasputinElysiaPlugin = ReturnType<typeof buildPlugin>;
@@ -34,7 +35,7 @@ export type RasputinElysiaOptions = RasputinOptions & {
      */
     installTransportSignalHandlers?: boolean;
 };
-declare const buildPlugin: (client: RasputinNodeClient, execution: Pick<RasputinExecution, "createScope">, recorderEnabled: boolean) => Elysia<"", {
+declare const buildPlugin: (client: RasputinNodeClient, runtime: ExecutionRuntime, recorderEnabled: boolean) => Elysia<"", {
     decorator: {
         rasputin: RasputinNodeClient;
     };

package/dist/rasputin-init.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"rasputin-init.d.ts","sourceRoot":"","sources":["../src/rasputin-init.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAON,KAAK,eAAe,EAGpB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,KAAK,EACX,wBAAwB,EAExB,iBAAiB,EACjB,kBAAkB,EAClB,MAAM,mBAAmB,CAAC;AAQ3B,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAIhC,iFAAiF;AACjF,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAElE,MAAM,MAAM,cAAc,GAAG;IAC5B,8EAA8E;IAC9E,MAAM,EAAE,kBAAkB,CAAC;IAC3B,2EAA2E;IAC3E,MAAM,EAAE,oBAAoB,CAAC;CAC7B,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,eAAe,GAAG;IACrD;;;OAGG;IACH,iBAAiB,CAAC,EAAE,wBAAwB,CAAC;IAC7C;;;;;OAKG;IACH,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC;;;OAGG;IACH,8BAA8B,CAAC,EAAE,OAAO,CAAC;CACzC,CAAC;AAuHF,QAAA,MAAM,WAAW,GAChB,QAAQ,kBAAkB,EAC1B,WAAW,IAAI,CAAC,iBAAiB,EAAE,aAAa,CAAC,EACjD,iBAAiB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAiExB,CAAC;AAEF;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,qBAAqB,GAAG,cAAc,CAAC"}
+{"version":3,"file":"rasputin-init.d.ts","sourceRoot":"","sources":["../src/rasputin-init.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAON,KAAK,eAAe,EAGpB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,KAAK,EACX,wBAAwB,EAExB,kBAAkB,EAClB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAIN,KAAK,gBAAgB,EAKrB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAIhC,iFAAiF;AACjF,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAElE,MAAM,MAAM,cAAc,GAAG;IAC5B,8EAA8E;IAC9E,MAAM,EAAE,kBAAkB,CAAC;IAC3B,2EAA2E;IAC3E,MAAM,EAAE,oBAAoB,CAAC;CAC7B,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,eAAe,GAAG;IACrD;;;OAGG;IACH,iBAAiB,CAAC,EAAE,wBAAwB,CAAC;IAC7C;;;;;OAKG;IACH,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC;;;OAGG;IACH,8BAA8B,CAAC,EAAE,OAAO,CAAC;CACzC,CAAC;AAuIF,QAAA,MAAM,WAAW,GAChB,QAAQ,kBAAkB,EAC1B,SAAS,gBAAgB,EACzB,iBAAiB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAmExB,CAAC;AAEF;;;;;;;;;;;;;;;;GAgBG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,qBAAqB,GAAG,cAAc,CAAC"}

package/dist/sdk-meta.d.ts

@@ -1,4 +1,4 @@
 /** Generated by scripts/generate-sdk-meta.ts — do not edit. */
 export declare const SDK_NAME = "@rasputin-ai/elysia";
-export declare const SDK_VERSION = "0.10.2";
+export declare const SDK_VERSION = "0.11.0";
 //# sourceMappingURL=sdk-meta.d.ts.map

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rasputin-ai/elysia",
-	"version": "0.10.2",
+	"version": "0.11.0",
 	"description": "Official Rasputin AI SDK for Elysia.",
 	"type": "module",
 	"exports": {
@@ -35,8 +35,8 @@
 		"access": "public"
 	},
 	"dependencies": {
-		"@rasputin-ai/core": "0.10.2",
-		"@rasputin-ai/node": "0.10.2"
+		"@rasputin-ai/core": "0.11.0",
+		"@rasputin-ai/node": "0.11.0"
 	},
 	"peerDependencies": {
 		"elysia": ">=1.4.0 <2"