@builder.io/ai-utils 0.1.1 → 0.1.2

package/package.json

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

package/src/codegen.ts

@@ -1,4 +1,5 @@
 import type {
+  Attachment,
   ContentMessageItemToolResult,
   UserMessageParam,
 } from "./messages";
@@ -117,6 +118,7 @@ export interface CodeGenInputOptions {
   files?: ProjectFile[];
   rerankFiles?: number;
   toolResults?: ContentMessageItemToolResult[];
+  attachments?: Attachment[];
 
   // Code options
   builderContent?: BuilderContent;
@@ -248,7 +250,6 @@ export interface RankedResult {
 
 export interface GenerateCompletionStepThinking {
   type: "thinking";
-  id: string;
 }
 
 export interface FileInfo {
@@ -265,7 +266,6 @@ export interface GenerateCompletionStepUserInput {
 export interface GenerateCompletionStepPlanning {
   type: "planning";
   content: string;
-  id: string;
 }
 
 export interface GenerateCompletionStepUser {
@@ -300,7 +300,6 @@ export interface GenerateCompletionStepTool {
 export interface GenerateCompletionStepText {
   type: "text";
   content: string;
-  id: string;
 }
 
 export interface GenerateCompletionStepDone {
@@ -311,7 +310,8 @@ export interface GenerateCompletionStepDone {
   actions: ActionItem[];
   usage: CodegenUsage | undefined;
   url?: string;
-  stopReason?: "limit" | "done";
+  stopReason?: "error" | "limit" | CompletionStopReason;
+  stopReasonMessage?: string;
 }
 
 export interface GenerateCompletionStepStart {
@@ -333,7 +333,6 @@ export interface GenerateCompletionStepDelta {
 
 export interface GenerateCompletionStepError {
   type: "error";
-  id: string;
   error: string;
   metadata: any;
 }
@@ -346,7 +345,6 @@ export interface GenerateCompletionStepContinue {
 
 export interface GenerateCompletionStepWaitForInput {
   type: "wait-for-input";
-  id: string;
 }
 
 export interface GenerateCompletionStepAbort {
@@ -378,3 +376,11 @@ export interface ApplyActionsResult {
   content?: string;
   oldContent?: string;
 }
+
+export interface GenerateUserMessage {
+  userPrompt: string;
+  files?: string[];
+  includeBaseFiles?: boolean;
+  skipSearch?: boolean;
+  attachments?: Attachment[];
+}

package/src/events.ts

@@ -42,7 +42,8 @@ export type BuilderAssistantEvent =
   | ThreadMessageRetryEvent
   | AppFigmaImportEvent
   | AppWebImportEvent
-  | AppAiTemplatesEvent;
+  | AppAiTemplatesEvent
+  | AssistantContentInitialEvent;
 
 export interface AssistantCompletionResultEvent {
   type: "assistant.result";
@@ -442,3 +443,10 @@ export type DeepPartial<T> = T extends object
       [P in keyof T]?: DeepPartial<T[P]>;
     }
   : T;
+
+export interface AssistantContentInitialEvent {
+  type: "assistant.content.initial";
+  data: {
+    code: string;
+  };
+}