neatlogs 1.1.2 → 1.1.4

package/dist/index.d.ts

@@ -590,6 +590,6 @@ declare function registerCrewaiTask(task: {
  * SDK version. Kept in sync with package.json by the `version:sync` script
  * (runs automatically on `prebuild`). Do not edit by hand — bump package.json.
  */
-declare const __version__ = "1.1.2";
+declare const __version__ = "1.1.4";
 
 export { type CachedPrompt, type InitOptions, type MaskFunction, PromptApiError, PromptClient, PromptClientError, PromptHandle, type PromptMessage, PromptNotFoundError, PromptTemplate, Span, type SpanKind, type SpanOptions, type TraceOptions, UserPromptTemplate, __version__, bindTemplates, createPrompt, deletePrompt, fetchPrompt, flush, getMastraObservability, getPrompt, getSessionConfig, identify, init, isDebugEnabled, listPrompts, log, registerCrewaiTask, removeTag, saveAsVersion, shutdown, span, trace, updatePrompt };

package/dist/index.mjs

@@ -2136,9 +2136,84 @@ function applyMask(spanData, globalMask) {
   }
 }
 
-// src/prompt/template.ts
+// src/core/identity.ts
 import { AsyncLocalStorage } from "async_hooks";
-var _promptStorage = new AsyncLocalStorage();
+var _GLOBAL_KEY = /* @__PURE__ */ Symbol.for("neatlogs.identity.storage");
+var _g = globalThis;
+var _storage = _g[_GLOBAL_KEY] ?? (_g[_GLOBAL_KEY] = new AsyncLocalStorage());
+function currentSessionId() {
+  return _storage.getStore()?.sessionId;
+}
+function currentEndUserId() {
+  return _storage.getStore()?.endUserId;
+}
+function currentEndUserMetadata() {
+  return _storage.getStore()?.endUserMetadata;
+}
+function identify(opts, fn) {
+  const prev = _storage.getStore();
+  const next = { ...prev ?? {} };
+  if (opts.sessionId !== void 0) next.sessionId = opts.sessionId;
+  if (opts.endUserId !== void 0) next.endUserId = opts.endUserId;
+  if (opts.endUserMetadata !== void 0) next.endUserMetadata = opts.endUserMetadata;
+  return _storage.run(next, fn);
+}
+
+// src/core/session.ts
+var logger4 = getLogger();
+var SESSION_ID_KEY = "neatlogs.session.id";
+function applySessionAttributes(span2, sessionId, isRoot = true) {
+  const resolved = isRoot ? sessionId ?? currentSessionId() : sessionId;
+  if (!resolved) return;
+  if (!isRoot) {
+    logger4.debug(
+      "[session] Ignoring sessionId on a non-root span \u2014 declare it on the trace root (top-level trace()/span()) or identify()."
+    );
+    return;
+  }
+  span2.setAttribute(SESSION_ID_KEY, String(resolved));
+}
+
+// src/core/end-user.ts
+import { trace as otelTrace, context as otelContext } from "@opentelemetry/api";
+var logger5 = getLogger();
+var END_USER_ID_KEY = "neatlogs.end_user.id";
+var END_USER_METADATA_KEY = "neatlogs.end_user.metadata";
+function normalizeEndUserMetadata(metadata) {
+  if (metadata === void 0 || metadata === null) return void 0;
+  if (typeof metadata === "string") return metadata || void 0;
+  try {
+    return JSON.stringify(metadata);
+  } catch {
+    return JSON.stringify(String(metadata));
+  }
+}
+function isRootSpan() {
+  const current = otelTrace.getSpan(otelContext.active());
+  return !(current && current.isRecording());
+}
+function applyEndUserAttributes(span2, endUserId, endUserMetadata, isRoot = true) {
+  const resolvedId = isRoot ? endUserId ?? currentEndUserId() : endUserId;
+  const resolvedMeta = isRoot ? endUserMetadata ?? currentEndUserMetadata() : endUserMetadata;
+  if (!resolvedId && !resolvedMeta) return;
+  if (!isRoot) {
+    logger5.debug(
+      "[end_user] Ignoring endUserId/endUserMetadata on a non-root span \u2014 declare it on the trace root (top-level trace()/span()) or identify()."
+    );
+    return;
+  }
+  if (resolvedId) {
+    span2.setAttribute(END_USER_ID_KEY, String(resolvedId));
+  }
+  const metaJson = normalizeEndUserMetadata(resolvedMeta);
+  if (metaJson) {
+    span2.setAttribute(END_USER_METADATA_KEY, metaJson);
+  }
+}
+
+// src/prompt/template.ts
+import { AsyncLocalStorage as AsyncLocalStorage2 } from "async_hooks";
+var _promptStorage = new AsyncLocalStorage2();
 var PromptContext = class {
   /**
    * Store system prompt template and variables in context.
@@ -2165,7 +2240,7 @@ var PromptContext = class {
     _promptStorage.enterWith(void 0);
   }
 };
-var _userPromptStorage = new AsyncLocalStorage();
+var _userPromptStorage = new AsyncLocalStorage2();
 var UserPromptContext = class {
   /**
    * Store user prompt template and variables in context.
@@ -2297,83 +2372,6 @@ import {
   SpanStatusCode as SpanStatusCode2
 } from "@opentelemetry/api";
 
-// src/core/end-user.ts
-import { trace as otelTrace, context as otelContext } from "@opentelemetry/api";
-
-// src/core/identity.ts
-import { AsyncLocalStorage as AsyncLocalStorage2 } from "async_hooks";
-var _GLOBAL_KEY = /* @__PURE__ */ Symbol.for("neatlogs.identity.storage");
-var _g = globalThis;
-var _storage = _g[_GLOBAL_KEY] ?? (_g[_GLOBAL_KEY] = new AsyncLocalStorage2());
-function currentSessionId() {
-  return _storage.getStore()?.sessionId;
-}
-function currentEndUserId() {
-  return _storage.getStore()?.endUserId;
-}
-function currentEndUserMetadata() {
-  return _storage.getStore()?.endUserMetadata;
-}
-function identify(opts, fn) {
-  const prev = _storage.getStore();
-  const next = { ...prev ?? {} };
-  if (opts.sessionId !== void 0) next.sessionId = opts.sessionId;
-  if (opts.endUserId !== void 0) next.endUserId = opts.endUserId;
-  if (opts.endUserMetadata !== void 0) next.endUserMetadata = opts.endUserMetadata;
-  return _storage.run(next, fn);
-}
-
-// src/core/end-user.ts
-var logger4 = getLogger();
-var END_USER_ID_KEY = "neatlogs.end_user.id";
-var END_USER_METADATA_KEY = "neatlogs.end_user.metadata";
-function normalizeEndUserMetadata(metadata) {
-  if (metadata === void 0 || metadata === null) return void 0;
-  if (typeof metadata === "string") return metadata || void 0;
-  try {
-    return JSON.stringify(metadata);
-  } catch {
-    return JSON.stringify(String(metadata));
-  }
-}
-function isRootSpan() {
-  const current = otelTrace.getSpan(otelContext.active());
-  return !(current && current.isRecording());
-}
-function applyEndUserAttributes(span2, endUserId, endUserMetadata, isRoot = true) {
-  const resolvedId = isRoot ? endUserId ?? currentEndUserId() : endUserId;
-  const resolvedMeta = isRoot ? endUserMetadata ?? currentEndUserMetadata() : endUserMetadata;
-  if (!resolvedId && !resolvedMeta) return;
-  if (!isRoot) {
-    logger4.debug(
-      "[end_user] Ignoring endUserId/endUserMetadata on a non-root span \u2014 declare it on the trace root (top-level trace()/span()) or identify()."
-    );
-    return;
-  }
-  if (resolvedId) {
-    span2.setAttribute(END_USER_ID_KEY, String(resolvedId));
-  }
-  const metaJson = normalizeEndUserMetadata(resolvedMeta);
-  if (metaJson) {
-    span2.setAttribute(END_USER_METADATA_KEY, metaJson);
-  }
-}
-
-// src/core/session.ts
-var logger5 = getLogger();
-var SESSION_ID_KEY = "neatlogs.session.id";
-function applySessionAttributes(span2, sessionId, isRoot = true) {
-  const resolved = isRoot ? sessionId ?? currentSessionId() : sessionId;
-  if (!resolved) return;
-  if (!isRoot) {
-    logger5.debug(
-      "[session] Ignoring sessionId on a non-root span \u2014 declare it on the trace root (top-level trace()/span()) or identify()."
-    );
-    return;
-  }
-  span2.setAttribute(SESSION_ID_KEY, String(resolved));
-}
-
 // src/decorators/base.ts
 import { trace, SpanStatusCode } from "@opentelemetry/api";
 var logger6 = getLogger();
@@ -4351,6 +4349,10 @@ var NeatlogsSpanProcessor = class {
   onStart(span2, parentContext) {
     const startTime = performance.now();
     try {
+      if (!span2.parentSpanId) {
+        applySessionAttributes(span2, void 0, true);
+        applyEndUserAttributes(span2, void 0, void 0, true);
+      }
       const attrs = span2.attributes ?? {};
       const spanKind = attrs["openinference.span.kind"];
       const spanName = typeof span2.name === "string" ? span2.name : "";
@@ -5743,7 +5745,7 @@ async function removeTag(name, tag) {
 }
 
 // src/version.ts
-var __version__ = "1.1.2";
+var __version__ = "1.1.4";
 
 // src/init.ts
 var logger13 = getLogger();