@assistant-ui/react/dist/runtimes/edge/streams/runResultStream.js

React components for AI chat.

"use strict";
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);

// src/runtimes/edge/streams/runResultStream.ts
var runResultStream_exports = {};
__export(runResultStream_exports, {
  runResultStream: () => runResultStream
});
module.exports = __toCommonJS(runResultStream_exports);
var import_parse_partial_json = require("../partial-json/parse-partial-json.cjs");
function runResultStream() {
  let message = {
    content: [],
    status: { type: "running" }
  };
  return new TransformStream({
    transform(chunk, controller) {
      const chunkType = chunk.type;
      switch (chunkType) {
        case "text-delta": {
          message = appendOrUpdateText(message, chunk.textDelta);
          controller.enqueue(message);
          break;
        }
        case "tool-call-delta": {
          const { toolCallId, toolName, argsTextDelta } = chunk;
          message = appendOrUpdateToolCall(
            message,
            toolCallId,
            toolName,
            argsTextDelta
          );
          controller.enqueue(message);
          break;
        }
        case "tool-call":
        case "response-metadata":
          break;
        case "tool-result": {
          message = appendOrUpdateToolResult(
            message,
            chunk.toolCallId,
            chunk.toolName,
            chunk.result
          );
          controller.enqueue(message);
          break;
        }
        case "step-finish": {
          message = appendOrUpdateStepFinish(message, chunk);
          controller.enqueue(message);
          break;
        }
        case "finish": {
          message = appendOrUpdateFinish(message, chunk);
          controller.enqueue(message);
          break;
        }
        case "error": {
          if (chunk.error instanceof Error && chunk.error.name === "AbortError") {
            message = appendOrUpdateCancel(message);
            controller.enqueue(message);
            break;
          } else {
            throw chunk.error;
          }
        }
        default: {
          const unhandledType = chunkType;
          throw new Error(`Unhandled chunk type: ${unhandledType}`);
        }
      }
    }
  });
}
var appendOrUpdateText = (message, textDelta) => {
  let contentParts = message.content ?? [];
  let contentPart = message.content?.at(-1);
  if (contentPart?.type !== "text") {
    contentPart = { type: "text", text: textDelta };
  } else {
    contentParts = contentParts.slice(0, -1);
    contentPart = { type: "text", text: contentPart.text + textDelta };
  }
  return {
    ...message,
    content: contentParts.concat([contentPart])
  };
};
var appendOrUpdateToolCall = (message, toolCallId, toolName, argsTextDelta) => {
  let contentParts = message.content ?? [];
  const contentPartIdx = contentParts.findIndex(
    (c) => c.type === "tool-call" && c.toolCallId === toolCallId
  );
  let contentPart = contentPartIdx === -1 ? null : contentParts[contentPartIdx];
  if (contentPart == null) {
    contentPart = {
      type: "tool-call",
      toolCallId,
      toolName,
      argsText: argsTextDelta,
      args: (0, import_parse_partial_json.parsePartialJson)(argsTextDelta)
    };
    contentParts = [...contentParts, contentPart];
  } else {
    const argsText = contentPart.argsText + argsTextDelta;
    contentPart = {
      ...contentPart,
      argsText,
      args: (0, import_parse_partial_json.parsePartialJson)(argsText)
    };
    contentParts = [
      ...contentParts.slice(0, contentPartIdx),
      contentPart,
      ...contentParts.slice(contentPartIdx + 1)
    ];
  }
  return {
    ...message,
    content: contentParts
  };
};
var appendOrUpdateToolResult = (message, toolCallId, toolName, result) => {
  let found = false;
  const newContentParts = message.content?.map((part) => {
    if (part.type !== "tool-call" || part.toolCallId !== toolCallId)
      return part;
    found = true;
    if (part.toolName !== toolName)
      throw new Error(
        `Tool call ${toolCallId} found with tool name ${part.toolName}, but expected ${toolName}`
      );
    return {
      ...part,
      result
    };
  });
  if (!found)
    throw new Error(
      `Received tool result for unknown tool call "${toolName}" / "${toolCallId}". This is likely an internal bug in assistant-ui.`
    );
  return {
    ...message,
    content: newContentParts
  };
};
var appendOrUpdateStepFinish = (message, chunk) => {
  const { type, ...rest } = chunk;
  const steps = [
    ...message.metadata?.steps ?? [],
    {
      usage: rest.usage
    }
  ];
  return {
    ...message,
    metadata: {
      ...message.metadata,
      roundtrips: steps,
      steps
    }
  };
};
var appendOrUpdateFinish = (message, chunk) => {
  const { type, ...rest } = chunk;
  const steps = [
    ...message.metadata?.steps ?? [],
    {
      logprobs: rest.logprobs,
      usage: rest.usage
    }
  ];
  return {
    ...message,
    status: getStatus(chunk),
    metadata: {
      ...message.metadata,
      roundtrips: steps,
      steps
    }
  };
};
var getStatus = (chunk) => {
  if (chunk.finishReason === "tool-calls") {
    return {
      type: "requires-action",
      reason: "tool-calls"
    };
  } else if (chunk.finishReason === "stop" || chunk.finishReason === "unknown") {
    return {
      type: "complete",
      reason: chunk.finishReason
    };
  } else {
    return {
      type: "incomplete",
      reason: chunk.finishReason
    };
  }
};
var appendOrUpdateCancel = (message) => {
  return {
    ...message,
    status: {
      type: "incomplete",
      reason: "cancelled"
    }
  };
};
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
  runResultStream
});
//# sourceMappingURL=runResultStream.js.map