veryfront 0.1.180 → 0.1.181

package/src/src/internal-agents/ag-ui-sse.ts

@@ -1,40 +1,27 @@
-import * as dntShim from "../../_dnt.shims.js";
-import type { AgentResponse } from "../agent/index.js";
+import type {
+  AgUiBrowserEncodedEvent,
+  AgUiBrowserEncoderState,
+  AgUiBrowserRunFinishedMetadata,
+  AgUiRuntimeStreamEvent,
+} from "../agent/ag-ui-browser-encoder.js";
+import {
+  createAgUiBrowserEncoderState,
+  finalizeAgUiBrowserEvents,
+  mapRuntimeStreamEventToAgUiBrowserEvents,
+} from "../agent/ag-ui-browser-encoder.js";
 import { serverLogger } from "../utils/index.js";
 import { z } from "zod";
 
 const encoder = new TextEncoder();
 const logger = serverLogger.component("internal-agents-ag-ui-sse");
 
-type RuntimeDataEvent = Record<string, unknown> & { type: string };
-
-export interface RunFinishedMetadata {
-  provider?: string;
-  model?: string;
-  inputTokens?: number;
-  outputTokens?: number;
-  totalTokens?: number;
-  finishReason?: string;
-}
-
-export interface StreamTransformState {
-  messageId: string | null;
-  textOpen: boolean;
-  reasoningMessageId: string | null;
-  sawVisibleOutput: boolean;
-  sawTerminalError: boolean;
-  metadata: RunFinishedMetadata;
-}
+type RuntimeDataEvent = AgUiRuntimeStreamEvent;
+export type RunFinishedMetadata = AgUiBrowserRunFinishedMetadata;
+export type StreamTransformState = AgUiBrowserEncoderState;
+export type MappedAgUiEvent = AgUiBrowserEncodedEvent;
 
 export function createStreamTransformState(): StreamTransformState {
-  return {
-    messageId: null,
-    textOpen: false,
-    reasoningMessageId: null,
-    sawVisibleOutput: false,
-    sawTerminalError: false,
-    metadata: {},
-  };
+  return createAgUiBrowserEncoderState();
 }
 
 const agUiEventPayloadSchemas = {
@@ -77,6 +64,12 @@ const agUiEventPayloadSchemas = {
   ReasoningMessageEnd: z.object({
     messageId: z.string().min(1),
   }),
+  StepStarted: z.object({
+    stepName: z.string().min(1),
+  }),
+  StepFinished: z.object({
+    stepName: z.string().min(1),
+  }),
   ToolCallStart: z.object({
     toolCallId: z.string().min(1),
     toolCallName: z.string().min(1),
@@ -93,6 +86,10 @@ const agUiEventPayloadSchemas = {
     result: z.unknown(),
     isError: z.boolean().optional(),
   }),
+  Custom: z.object({
+    name: z.string().min(1),
+    value: z.unknown(),
+  }),
   RunError: z.object({
     code: z.string().min(1).optional(),
     message: z.string().min(1),
@@ -147,254 +144,16 @@ export function parseSseJsonEvents(chunk: string): {
   return { events, remainder };
 }
 
-function getMessageId(state: StreamTransformState, event: RuntimeDataEvent): string {
-  if (typeof event.messageId === "string") {
-    state.messageId = event.messageId;
-    return event.messageId;
-  }
-
-  if (!state.messageId && typeof event.id === "string") {
-    state.messageId = event.id;
-  }
-
-  if (!state.messageId) {
-    state.messageId = dntShim.crypto.randomUUID();
-  }
-
-  return state.messageId;
-}
-
-function getReasoningMessageId(state: StreamTransformState, event: RuntimeDataEvent): string {
-  if (typeof event.id === "string" && event.id.length > 0) {
-    state.reasoningMessageId = state.messageId
-      ? `${state.messageId}:reasoning:${event.id}`
-      : event.id;
-    return state.reasoningMessageId;
-  }
-
-  if (!state.reasoningMessageId) {
-    state.reasoningMessageId = state.messageId
-      ? `${state.messageId}:reasoning:${dntShim.crypto.randomUUID()}`
-      : dntShim.crypto.randomUUID();
-  }
-
-  return state.reasoningMessageId;
-}
-
-function applyDataMetadata(state: StreamTransformState, event: RuntimeDataEvent): void {
-  const data = event.data && typeof event.data === "object" && !Array.isArray(event.data)
-    ? event.data as Record<string, unknown>
-    : event;
-
-  if (typeof data.model === "string") {
-    state.metadata.model = data.model;
-    const provider = data.model.split("/")[0];
-    if (provider) {
-      state.metadata.provider = provider;
-    }
-  }
-}
-
-function applyResponseMetadata(state: StreamTransformState, response: AgentResponse | null): void {
-  if (!response) return;
-
-  if (response.usage) {
-    state.metadata.inputTokens = response.usage.promptTokens;
-    state.metadata.outputTokens = response.usage.completionTokens;
-    state.metadata.totalTokens = response.usage.totalTokens;
-  }
-
-  const finishReason = response.metadata && typeof response.metadata === "object"
-    ? response.metadata.finishReason
-    : undefined;
-  if (typeof finishReason === "string") {
-    state.metadata.finishReason = finishReason;
-  }
-}
-
 export function mapRuntimeEventToAgUi(
   state: StreamTransformState,
   event: RuntimeDataEvent,
-): Array<{ event: string; payload: Record<string, unknown> }> {
-  switch (event.type) {
-    case "message-start":
-      getMessageId(state, event);
-      return [];
-
-    case "text-start": {
-      if (state.textOpen) return [];
-      const messageId = getMessageId(state, event);
-      state.textOpen = true;
-      state.sawVisibleOutput = true;
-      return [{
-        event: "TextMessageStart",
-        payload: { messageId, role: "assistant" },
-      }];
-    }
-
-    case "text-delta": {
-      const messageId = getMessageId(state, event);
-      state.sawVisibleOutput = true;
-      if (!state.textOpen) {
-        state.textOpen = true;
-        return [
-          { event: "TextMessageStart", payload: { messageId, role: "assistant" } },
-          {
-            event: "TextMessageContent",
-            payload: { messageId, delta: typeof event.delta === "string" ? event.delta : "" },
-          },
-        ];
-      }
-
-      return [{
-        event: "TextMessageContent",
-        payload: { messageId, delta: typeof event.delta === "string" ? event.delta : "" },
-      }];
-    }
-
-    case "text-end": {
-      if (!state.textOpen) return [];
-      state.textOpen = false;
-      return [{
-        event: "TextMessageEnd",
-        payload: { messageId: getMessageId(state, event) },
-      }];
-    }
-
-    case "reasoning-start":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ReasoningMessageStart",
-        payload: { messageId: getReasoningMessageId(state, event), role: "reasoning" },
-      }];
-
-    case "reasoning-delta":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ReasoningMessageContent",
-        payload: {
-          messageId: getReasoningMessageId(state, event),
-          delta: typeof event.delta === "string" ? event.delta : "",
-        },
-      }];
-
-    case "reasoning-end": {
-      const messageId = getReasoningMessageId(state, event);
-      state.reasoningMessageId = null;
-      return [{
-        event: "ReasoningMessageEnd",
-        payload: { messageId },
-      }];
-    }
-
-    case "tool-input-start":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ToolCallStart",
-        payload: {
-          toolCallId: event.toolCallId,
-          toolCallName: event.toolName,
-        },
-      }];
-
-    case "tool-input-delta":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ToolCallArgs",
-        payload: {
-          toolCallId: event.toolCallId,
-          delta: typeof event.inputTextDelta === "string" ? event.inputTextDelta : "",
-        },
-      }];
-
-    case "tool-input-available":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ToolCallEnd",
-        payload: { toolCallId: event.toolCallId },
-      }];
-
-    case "tool-output-available":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ToolCallResult",
-        payload: {
-          toolCallId: event.toolCallId,
-          result: event.output,
-        },
-      }];
-
-    case "tool-output-error":
-      state.sawVisibleOutput = true;
-      return [{
-        event: "ToolCallResult",
-        payload: {
-          toolCallId: event.toolCallId,
-          result: { error: event.errorText },
-          isError: true,
-        },
-      }];
-
-    case "step-start":
-      return [];
-
-    case "step-end":
-      return [];
-
-    case "data":
-      applyDataMetadata(state, event);
-      return [];
-
-    case "error":
-      state.sawTerminalError = true;
-      return [{
-        event: "RunError",
-        payload: {
-          message: typeof event.error === "string" ? event.error : "Agent run failed",
-        },
-      }];
-
-    default:
-      return [];
-  }
+): MappedAgUiEvent[] {
+  return mapRuntimeStreamEventToAgUiBrowserEvents(state, event);
 }
 
 export function finalizeRunEvents(
   state: StreamTransformState,
-  response: AgentResponse | null,
-): Array<{ event: string; payload: Record<string, unknown> }> {
-  applyResponseMetadata(state, response);
-
-  if (state.sawTerminalError) {
-    return [];
-  }
-
-  if (!state.sawVisibleOutput) {
-    state.sawTerminalError = true;
-    return [{
-      event: "RunError",
-      payload: {
-        code: "EMPTY_ASSISTANT_OUTPUT",
-        message: "Agent run produced no assistant-visible output",
-      },
-    }];
-  }
-
-  const events: Array<{ event: string; payload: Record<string, unknown> }> = [];
-  if (state.textOpen) {
-    state.textOpen = false;
-    events.push({
-      event: "TextMessageEnd",
-      payload: { messageId: getMessageId(state, { type: "text-end" }) },
-    });
-  }
-
-  events.push({
-    event: "RunFinished",
-    payload: {
-      metadata: state.metadata,
-    },
-  });
-
-  return events;
+  response: Parameters<typeof finalizeAgUiBrowserEvents>[1],
+): MappedAgUiEvent[] {
+  return finalizeAgUiBrowserEvents(state, response);
 }

package/src/src/rendering/orchestrator/pipeline.ts

@@ -687,14 +687,7 @@ export class RenderPipeline {
 
     const providers: string[] = [];
 
-    let projectUpdatedAt: string | undefined;
-    const fs = this.config.adapter?.fs;
-    if (fs && isExtendedFSAdapter(fs) && fs.isVeryfrontAdapter()) {
-      const wrappedAdapter = fs.getUnderlyingAdapter() as {
-        getProjectData?: () => { updated_at?: string } | undefined;
-      };
-      projectUpdatedAt = wrappedAdapter.getProjectData?.()?.updated_at;
-    }
+    const projectUpdatedAt = this.resolveProjectUpdatedAt();
 
     const appPath = await this.resolveAppPath();
 
@@ -783,6 +776,18 @@ export class RenderPipeline {
     return undefined;
   }
 
+  private resolveProjectUpdatedAt(): string | undefined {
+    const fs = this.config.adapter?.fs;
+    if (!fs || !isExtendedFSAdapter(fs) || !fs.isVeryfrontAdapter()) {
+      return undefined;
+    }
+
+    const wrappedAdapter = fs.getUnderlyingAdapter() as {
+      getProjectData?: () => { updated_at?: string } | undefined;
+    };
+    return wrappedAdapter.getProjectData?.()?.updated_at;
+  }
+
   private async resolvePageDataCss(
     slug: string,
     options: RenderOptions | undefined,

package/src/src/utils/version-constant.ts

@@ -1,3 +1,3 @@
 // Keep in sync with deno.json version.
 // scripts/release.ts updates this constant during releases.
-export const VERSION = "0.1.180";
+export const VERSION = "0.1.181";