bitfab 0.27.0 → 0.27.2

package/dist/{chunk-PFV4MPNS.js → chunk-DW246LVL.js}

@@ -10,10 +10,10 @@ import {
   toJsonSafe,
   toJsonSafeReport,
   warnOnce
-} from "./chunk-RNTDM6WM.js";
+} from "./chunk-MD4XQGAF.js";
 
 // src/version.generated.ts
-var __version__ = "0.27.0";
+var __version__ = "0.27.2";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -317,7 +317,7 @@ var HttpClient = class {
    * Start a replay session by fetching historical traces.
    * Blocking call — creates a test run and returns lightweight item references.
    */
-  async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId) {
+  async startReplay(traceFunctionKey, limit, traceIds, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId) {
     const payload = { traceFunctionKey };
     if (limit !== void 0) {
       payload.limit = limit;
@@ -325,6 +325,9 @@ var HttpClient = class {
     if (traceIds) {
       payload.traceIds = traceIds;
     }
+    if (name !== void 0) {
+      payload.name = name;
+    }
     if (codeChangeDescription !== void 0) {
       payload.codeChangeDescription = codeChangeDescription;
     }
@@ -1356,6 +1359,7 @@ function buildSnapshotRef(config, sdkWallClockBeforeFn) {
 
 // src/langgraph.ts
 var LANGSMITH_HIDDEN_TAG = "langsmith:hidden";
+var CHAIN_RUN_TYPES = /* @__PURE__ */ new Set(["chain", "parser", "prompt"]);
 var LANGGRAPH_METADATA_KEYS = [
   "langgraph_step",
   "langgraph_node",
@@ -1366,6 +1370,18 @@ var LANGGRAPH_METADATA_KEYS = [
 function nowIso2() {
   return (/* @__PURE__ */ new Date()).toISOString();
 }
+function normalizeChainStartArgs(parentRunIdOrRunType, runTypeOrRunName, runNameOrParentRunId) {
+  if (parentRunIdOrRunType && CHAIN_RUN_TYPES.has(parentRunIdOrRunType)) {
+    return {
+      parentRunId: runNameOrParentRunId,
+      runName: runTypeOrRunName
+    };
+  }
+  return {
+    parentRunId: parentRunIdOrRunType,
+    runName: runNameOrParentRunId
+  };
+}
 function convertMessage(message) {
   if (typeof message !== "object" || message === null) {
     return { role: "unknown", content: String(message) };
@@ -1577,6 +1593,7 @@ var BitfabLangGraphCallbackHandler = class {
     const willHide = tags?.includes(LANGSMITH_HIDDEN_TAG) === true;
     let invocation;
     let effectiveParentId;
+    let isRootInvocation = false;
     if (parentSpan) {
       const existing = this.invocations.get(parentSpan.rootRunId);
       if (existing) {
@@ -1607,6 +1624,7 @@ var BitfabLangGraphCallbackHandler = class {
       };
       this.invocations.set(runId, invocation);
       effectiveParentId = activeContext?.spanId ?? null;
+      isRootInvocation = true;
     }
     const lgMetadata = extractLangGraphMetadata(metadata);
     const contexts = Object.keys(lgMetadata).length > 0 ? [lgMetadata] : [];
@@ -1629,6 +1647,9 @@ var BitfabLangGraphCallbackHandler = class {
       spanInfo.hidden = true;
     }
     this.runToSpan.set(runId, spanInfo);
+    if (isRootInvocation) {
+      this.sendTraceStart(spanInfo);
+    }
     return spanInfo;
   }
   completeSpan(runId, output, error, extraContexts) {
@@ -1719,11 +1740,35 @@ var BitfabLangGraphCallbackHandler = class {
     } catch {
     }
   }
+  sendTraceStart(rootSpan) {
+    const traceData = {
+      type: "sdk-function",
+      source: "typescript-sdk-langgraph",
+      traceFunctionKey: this.traceFunctionKey,
+      externalTrace: {
+        id: rootSpan.traceId,
+        started_at: rootSpan.startedAt,
+        workflow_name: this.traceFunctionKey
+      },
+      completed: false
+    };
+    const finalized = finalizeTracePayload(traceData);
+    try {
+      this.httpClient.sendExternalTrace(finalized);
+    } catch {
+    }
+  }
   // ── chain callbacks (graph nodes) ─────────────────────────────
-  async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata) {
+  async handleChainStart(chain, inputs, runId, parentRunIdOrRunType, tags, metadata, runTypeOrRunName, runNameOrParentRunId) {
     try {
-      const idArr = chain.id;
-      const name = chain.name ?? idArr?.[idArr.length - 1] ?? "chain";
+      const { parentRunId, runName } = normalizeChainStartArgs(
+        parentRunIdOrRunType,
+        runTypeOrRunName,
+        runNameOrParentRunId
+      );
+      const serialized = chain ?? {};
+      const idArr = serialized.id;
+      const name = runName ?? serialized.name ?? idArr?.[idArr.length - 1] ?? "chain";
       this.startSpan(
         runId,
         parentRunId,
@@ -1758,11 +1803,12 @@ var BitfabLangGraphCallbackHandler = class {
     }
   }
   // ── LLM callbacks ─────────────────────────────────────────────
-  async handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, tags, metadata) {
+  async handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, tags, metadata, runName) {
     try {
-      const model = extractModelName(llm, metadata);
-      const idArr = llm.id;
-      const name = model ?? idArr?.[idArr.length - 1] ?? "llm";
+      const serialized = llm ?? {};
+      const model = extractModelName(serialized, metadata);
+      const idArr = serialized.id;
+      const name = runName ?? model ?? idArr?.[idArr.length - 1] ?? "llm";
       const converted = messages.map((batch) => batch.map(convertMessage));
       const spanInfo = this.startSpan(
         runId,
@@ -1777,11 +1823,12 @@ var BitfabLangGraphCallbackHandler = class {
     } catch {
     }
   }
-  async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata) {
+  async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata, runName) {
     try {
-      const model = extractModelName(llm, metadata);
-      const idArr = llm.id;
-      const name = model ?? idArr?.[idArr.length - 1] ?? "llm";
+      const serialized = llm ?? {};
+      const model = extractModelName(serialized, metadata);
+      const idArr = serialized.id;
+      const name = runName ?? model ?? idArr?.[idArr.length - 1] ?? "llm";
       const spanInfo = this.startSpan(
         runId,
         parentRunId,
@@ -1834,9 +1881,10 @@ var BitfabLangGraphCallbackHandler = class {
   async handleLLMNewToken() {
   }
   // ── tool callbacks ────────────────────────────────────────────
-  async handleToolStart(tool, input, runId, parentRunId, tags, metadata) {
+  async handleToolStart(tool, input, runId, parentRunId, tags, metadata, runName) {
     try {
-      const name = tool.name ?? "tool";
+      const serialized = tool ?? {};
+      const name = runName ?? serialized.name ?? "tool";
       this.startSpan(
         runId,
         parentRunId,
@@ -1866,9 +1914,10 @@ var BitfabLangGraphCallbackHandler = class {
     }
   }
   // ── retriever callbacks ───────────────────────────────────────
-  async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata) {
+  async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, runName) {
     try {
-      const name = retriever.name ?? "retriever";
+      const serialized = retriever ?? {};
+      const name = runName ?? serialized.name ?? "retriever";
       this.startSpan(
         runId,
         parentRunId,
@@ -3524,7 +3573,7 @@ var Bitfab = class {
         `Function is wrapped with trace function key '${wrappedKey}' but replay was called with '${traceFunctionKey}'. Pass matching keys, or pass the unwrapped function to replay it under the explicit key.`
       );
     }
-    const { replay: doReplay } = await import("./replay-66KNEAMO.js");
+    const { replay: doReplay } = await import("./replay-J3YTYE7V.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3749,4 +3798,4 @@ export {
   BitfabFunction,
   finalizers
 };
-//# sourceMappingURL=chunk-PFV4MPNS.js.map
+//# sourceMappingURL=chunk-DW246LVL.js.map