@bitfab/sdk 0.27.1 → 0.27.2

package/dist/index.cjs

@@ -636,7 +636,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/version.generated.ts
-var __version__ = "0.27.1";
+var __version__ = "0.27.2";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -1996,6 +1996,7 @@ function buildSnapshotRef(config, sdkWallClockBeforeFn) {
 init_randomUuid();
 init_serialize();
 var LANGSMITH_HIDDEN_TAG = "langsmith:hidden";
+var CHAIN_RUN_TYPES = /* @__PURE__ */ new Set(["chain", "parser", "prompt"]);
 var LANGGRAPH_METADATA_KEYS = [
   "langgraph_step",
   "langgraph_node",
@@ -2006,6 +2007,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) };
@@ -2217,6 +2230,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) {
@@ -2247,6 +2261,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] : [];
@@ -2269,6 +2284,9 @@ var BitfabLangGraphCallbackHandler = class {
       spanInfo.hidden = true;
     }
     this.runToSpan.set(runId, spanInfo);
+    if (isRootInvocation) {
+      this.sendTraceStart(spanInfo);
+    }
     return spanInfo;
   }
   completeSpan(runId, output, error, extraContexts) {
@@ -2359,11 +2377,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,
@@ -2398,11 +2440,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,
@@ -2417,11 +2460,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,
@@ -2474,9 +2518,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,
@@ -2506,9 +2551,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,