@cydm/pie 1.0.13 → 1.0.15

package/dist/chunks/{src-LZC56DRG.js → src-6WPNVGT2.js}

@@ -3,7 +3,8 @@ import {
   Agent,
   agentLoop,
   agentLoopContinue
-} from "./chunk-D7NAXU7F.js";
+} from "./chunk-VE2HDCNB.js";
+import "./chunk-5DA2D3K2.js";
 import "./chunk-TG2EQLX2.js";
 export {
   Agent,

package/dist/chunks/test-stream-ZSKNLUEJ.js

@@ -0,0 +1,94 @@
+import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
+import {
+  EventStream
+} from "./chunk-5DA2D3K2.js";
+import "./chunk-TG2EQLX2.js";
+
+// src/app/test-stream.ts
+function createAssistantMessage(model, text, stopReason = "stop", errorMessage) {
+  return {
+    role: "assistant",
+    content: [{ type: "text", text }],
+    api: model.api,
+    provider: model.provider,
+    model: model.id,
+    usage: {
+      input: 1,
+      output: text ? 1 : 0,
+      cacheRead: 0,
+      cacheWrite: 0,
+      totalTokens: text ? 2 : 1,
+      cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+    },
+    stopReason,
+    errorMessage,
+    timestamp: Date.now()
+  };
+}
+function parseMockStreamSequence(value) {
+  const parsed = JSON.parse(value);
+  if (!Array.isArray(parsed)) {
+    throw new Error("PIE_TEST_MOCK_STREAM_SEQUENCE must be a JSON array");
+  }
+  return parsed.map((item) => {
+    if (!item || typeof item !== "object") {
+      throw new Error("PIE_TEST_MOCK_STREAM_SEQUENCE entries must be objects");
+    }
+    const record = item;
+    if (record.type === "error" && typeof record.error === "string") {
+      return {
+        type: "error",
+        error: record.error,
+        partialText: typeof record.partialText === "string" ? record.partialText : void 0
+      };
+    }
+    if (record.type === "text" && typeof record.text === "string") {
+      return { type: "text", text: record.text };
+    }
+    throw new Error("PIE_TEST_MOCK_STREAM_SEQUENCE entries must be text or error steps");
+  });
+}
+function createTestStreamFnFromSequenceEnv(value = process.env.PIE_TEST_MOCK_STREAM_SEQUENCE) {
+  if (!value) {
+    return void 0;
+  }
+  const steps = parseMockStreamSequence(value);
+  let callIndex = 0;
+  return async (model) => {
+    const step = steps[Math.min(callIndex, steps.length - 1)];
+    callIndex += 1;
+    const stream = new EventStream(
+      (event) => event.type === "done" || event.type === "error",
+      (event) => event.type === "done" ? event.message : event.error
+    );
+    queueMicrotask(() => {
+      if (!step) {
+        const message2 = createAssistantMessage(model, "");
+        stream.push({ type: "done", reason: "stop", message: message2 });
+        return;
+      }
+      if (step.type === "error") {
+        if (step.partialText) {
+          const partial = createAssistantMessage(model, step.partialText);
+          stream.push({ type: "start", partial });
+          stream.push({ type: "text_start", contentIndex: 0, partial });
+          stream.push({ type: "text_delta", contentIndex: 0, delta: step.partialText, partial });
+          stream.push({ type: "text_end", contentIndex: 0, content: step.partialText, partial });
+        }
+        const error = createAssistantMessage(model, "", "error", step.error);
+        stream.push({ type: "error", reason: "error", error });
+        return;
+      }
+      const message = createAssistantMessage(model, step.text);
+      stream.push({ type: "start", partial: message });
+      stream.push({ type: "text_start", contentIndex: 0, partial: message });
+      stream.push({ type: "text_delta", contentIndex: 0, delta: step.text, partial: message });
+      stream.push({ type: "text_end", contentIndex: 0, content: step.text, partial: message });
+      stream.push({ type: "done", reason: "stop", message });
+    });
+    return stream;
+  };
+}
+export {
+  createTestStreamFnFromSequenceEnv
+};