@anvia/studio 0.5.1 → 0.5.2

package/dist/index.d.ts

@@ -161,6 +161,7 @@ type StudioTranscriptChatEntry = {
     kind: "message";
     role: "user" | "assistant";
     text: string;
+    tone?: "error";
     traceId?: string;
 };
 type StudioTranscriptReasoningEntry = {

package/dist/index.js

@@ -3440,8 +3440,62 @@ function sessionUserId(session) {
   return typeof userId === "string" && userId.trim().length > 0 ? userId : DEFAULT_USER_ID;
 }
 
+// src/runtime/streams.ts
+import { createEventStream } from "@anvia/server";
+var studioJsonlHeaders = {
+  "cache-control": "no-cache, no-transform",
+  connection: "keep-alive",
+  "content-type": "application/x-ndjson; charset=utf-8",
+  "transfer-encoding": "chunked",
+  "x-accel-buffering": "no"
+};
+function streamStudioJsonl(events) {
+  return createEventStream(withStudioStreamErrors(events), {
+    format: "jsonl",
+    headers: studioJsonlHeaders
+  });
+}
+function withStudioStreamErrors(events) {
+  const iterator = events[Symbol.asyncIterator]();
+  let done = false;
+  return {
+    [Symbol.asyncIterator]() {
+      return {
+        async next() {
+          if (done) {
+            return { done: true, value: void 0 };
+          }
+          try {
+            const next = await iterator.next();
+            if (next.done === true) {
+              done = true;
+            }
+            return next;
+          } catch (error) {
+            done = true;
+            return {
+              done: false,
+              value: studioStreamError(error)
+            };
+          }
+        },
+        async return() {
+          done = true;
+          await iterator.return?.();
+          return { done: true, value: void 0 };
+        }
+      };
+    }
+  };
+}
+function studioStreamError(error) {
+  return {
+    type: "error",
+    error: serializeError2(error)
+  };
+}
+
 // src/runtime/observability.ts
-import { stream as streamResponse } from "hono/streaming";
 var defaultBufferSize = 1e3;
 var StudioObservabilityHub = class {
   subscriptions = /* @__PURE__ */ new Set();
@@ -3485,28 +3539,23 @@ function registerObservabilityRoutes(app, hub) {
         400
       );
     }
-    c.header("content-type", "application/x-ndjson; charset=utf-8");
-    c.header("cache-control", "no-cache, no-transform");
-    c.header("connection", "keep-alive");
-    c.header("transfer-encoding", "chunked");
-    c.header("x-accel-buffering", "no");
-    return streamResponse(c, async (stream) => {
-      const subscription = hub.subscribe(types === void 0 ? {} : { types });
-      try {
-        while (true) {
-          const next = await subscription.next();
-          if (next.done === true) {
-            break;
-          }
-          await stream.write(`${JSON.stringify(next.value)}
-`);
-        }
-      } finally {
-        subscription.close();
-      }
-    });
+    return streamStudioJsonl(observabilityEvents(hub, types));
   });
 }
+function observabilityEvents(hub, types) {
+  const subscription = hub.subscribe(types === void 0 ? {} : { types });
+  return {
+    [Symbol.asyncIterator]() {
+      return {
+        next: () => subscription.next(),
+        async return() {
+          subscription.close();
+          return { done: true, value: void 0 };
+        }
+      };
+    }
+  };
+}
 function createSubscription(types) {
   const values = [];
   const resolvers = [];
@@ -3630,9 +3679,6 @@ function traceSummary2(trace) {
   };
 }
 
-// src/runtime/pipelines.ts
-import { stream as streamResponse3 } from "hono/streaming";
-
 // src/runtime/pipeline-logs.ts
 async function appendPipelineLog(store, input) {
   return store?.appendPipelineLog(input);
@@ -3781,7 +3827,6 @@ function formatUnknown(value) {
 }
 
 // src/runtime/runs.ts
-import { stream as streamResponse2 } from "hono/streaming";
 var AsyncEventQueue = class {
   values = [];
   resolvers = [];
@@ -3854,25 +3899,8 @@ async function* mergeRunAndApprovalEvents(runEvents, approvalEvents) {
     approvalEvents.close();
   }
 }
-function streamAgentRunEvents(c, events) {
-  c.header("content-type", "application/x-ndjson; charset=utf-8");
-  c.header("cache-control", "no-cache, no-transform");
-  c.header("connection", "keep-alive");
-  c.header("transfer-encoding", "chunked");
-  c.header("x-accel-buffering", "no");
-  return streamResponse2(
-    c,
-    async (stream) => {
-      for await (const event of events) {
-        await stream.write(`${JSON.stringify(event)}
-`);
-      }
-    },
-    async (error, stream) => {
-      await stream.write(`${JSON.stringify({ type: "error", error: serializeError2(error) })}
-`);
-    }
-  );
+function streamAgentRunEvents(_c, events) {
+  return streamStudioJsonl(events);
 }
 function traceForRun(trace, agentId, session) {
   const metadata = {
@@ -3912,6 +3940,7 @@ async function* persistStreamingSessionTranscript(props) {
       yield event;
     }
   } catch (error) {
+    appendTranscriptAssistantError(transcript, errorText(error));
     await props.store.saveSessionRunTranscript({
       id: props.session.id,
       runId: props.runId,
@@ -4040,6 +4069,9 @@ function acceptTranscriptStreamEvent(transcript, event) {
   if (event.type === "final" && event.trace?.traceId !== void 0) {
     assignTranscriptTraceId(transcript, event.trace.traceId);
   }
+  if (event.type === "error") {
+    appendTranscriptAssistantError(transcript, errorText(event.error));
+  }
 }
 function approvalCallId(approval) {
   return approval.callId ?? approval.toolCallId;
@@ -4134,9 +4166,6 @@ function childAgentTranscriptEvent(event) {
   return void 0;
 }
 function errorText(error) {
-  if (error instanceof Error) {
-    return error.message;
-  }
   if (typeof error === "string") {
     return error;
   }
@@ -4217,7 +4246,7 @@ function messageToTranscriptEntry(message, entryId) {
 }
 function appendTranscriptAssistantText(transcript, delta) {
   const last = transcript.at(-1);
-  if (last?.kind === "message" && last.role === "assistant") {
+  if (last?.kind === "message" && last.role === "assistant" && last.tone !== "error") {
     last.text = `${last.text}${delta}`;
     return;
   }
@@ -4228,6 +4257,19 @@ function appendTranscriptAssistantText(transcript, delta) {
     text: delta
   });
 }
+function appendTranscriptAssistantError(transcript, text) {
+  const last = transcript.at(-1);
+  if (last?.kind === "message" && last.role === "assistant" && last.tone === "error" && last.text === text) {
+    return;
+  }
+  transcript.push({
+    entryId: transcript.length,
+    kind: "message",
+    role: "assistant",
+    text,
+    tone: "error"
+  });
+}
 function assignTranscriptTraceId(transcript, traceId) {
   for (let index = transcript.length - 1; index >= 0; index -= 1) {
     const entry = transcript[index];
@@ -4592,25 +4634,8 @@ function pipelineDetail(pipeline) {
     graph
   };
 }
-function streamPipelineRun(c, props) {
-  c.header("content-type", "application/x-ndjson; charset=utf-8");
-  c.header("cache-control", "no-cache, no-transform");
-  c.header("connection", "keep-alive");
-  c.header("transfer-encoding", "chunked");
-  c.header("x-accel-buffering", "no");
-  return streamResponse3(
-    c,
-    async (stream) => {
-      for await (const event of pipelineRunEvents(props)) {
-        await stream.write(`${JSON.stringify(event)}
-`);
-      }
-    },
-    async (error, stream) => {
-      await stream.write(`${JSON.stringify({ type: "error", error: serializeError2(error) })}
-`);
-    }
-  );
+function streamPipelineRun(_c, props) {
+  return streamStudioJsonl(pipelineRunEvents(props));
 }
 async function* pipelineRunEvents(props) {
   yield* emitPipelineLog(props.logStore, pipelineRunStartedLog(props.pipeline, props.runId));