@builder.io/ai-utils 0.2.0 → 0.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@builder.io/ai-utils",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Builder.io AI utils",
   "type": "module",
   "main": "src/index.js",

package/src/codegen.ts

@@ -345,6 +345,7 @@ export interface GenerateCompletionStepContinue {
 
 export interface GenerateCompletionStepWaitForInput {
   type: "wait-for-input";
+  state: GenerateCompletionState;
 }
 
 export interface GenerateCompletionStepAbort {
@@ -359,6 +360,26 @@ export interface GenerateCompletionStepRestore {
   commitHash: string | undefined;
 }
 
+export type GenerateCompletionState =
+  | "unknown"
+  | "initial-with-url"
+  | "initial-without-url"
+  | "generating"
+  | "success"
+  | "abort"
+  | "error"
+  | "close";
+
+export interface GenerateCompletionStepState {
+  type: "state";
+  previousState: GenerateCompletionState;
+  newState: GenerateCompletionState;
+}
+
+export interface GenerateCompletionStepClose {
+  type: "close";
+}
+
 export type GenerateCompletionStep = { timestamp?: number } & (
   | GenerateCompletionStepPlanning
   | GenerateCompletionStepStart
@@ -375,6 +396,7 @@ export type GenerateCompletionStep = { timestamp?: number } & (
   | GenerateCompletionStepUserInput
   | GenerateCompletionStepText
   | GenerateCompletionStepRestore
+  | GenerateCompletionStepState
   | GenerateCompletionStepToolResult
 );
 

package/src/events.ts

@@ -44,6 +44,7 @@ export type BuilderAssistantEvent =
   | AppWebImportEvent
   | AppAiTemplatesEvent
   | AssistantContentInitialEvent
+  | ThreadMessageSummaryEvent
   | AssistantHeartbeatEvent;
 
 export interface AssistantCompletionResultEvent {
@@ -267,6 +268,11 @@ export interface AssistantStats {
    * The assistant's response message.
    */
   assistantMessage: string;
+  /**
+   * The assistant's summary of what it did. This appears after assistantMessage
+   * as well as any changes it made.
+   */
+  assistantSummary: string;
   /**
    * The user's prompt message.
    */
@@ -425,6 +431,11 @@ export interface ThreadMessageRetryEvent {
   type: "assistant.thread.message.retry";
 }
 
+export interface ThreadMessageSummaryEvent {
+  type: "assistant.thread.message.summary.delta";
+  data: ThreadMessageDelta;
+}
+
 export interface ThreadMessageCompleted extends AssistantMessage {
   platformId: string;
   threadId: string;

package/src/messages.ts

@@ -175,6 +175,8 @@ export interface UserMessage extends UserMessageParam {
 export interface AssistantMessage extends AssistantMessageParam {
   id: string;
   status?: "accepted" | "rejected" | "aborted";
+  // Optional summary from the LLM that explains why it did
+  summary?: string;
 }
 
 export interface AssistantActionMessage {