@builder.io/ai-utils 0.3.10 → 0.3.12

package/package.json

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

package/src/codegen.ts

@@ -70,11 +70,13 @@ export type CompletionStopReason =
   | "content_filter"
   | "error"
   | "aborted"
+  | "pause_turn"
+  | "refusal"
   | null;
 
 export interface ViewPathToolInput {
-  filePath: string;
-  viewRange?: [number, number];
+  file_path: string;
+  view_range?: [number, number];
 }
 
 export interface GlobSearchToolInput {
@@ -83,8 +85,8 @@ export interface GlobSearchToolInput {
 
 export interface GrepSearchToolInput {
   query: string;
-  includeGlob?: string;
-  excludeGlob?: string;
+  include_glob?: string;
+  exclude_glob?: string;
 }
 
 export interface GetRuleToolInput {
@@ -98,7 +100,7 @@ export interface GetStyleInspirationToolInput {
 export interface GetBuildOutputToolInput {}
 export interface DevServerControlInput {
   restart?: boolean;
-  getLogs?: boolean;
+  get_logs?: boolean;
 }
 
 export interface BashToolInput {
@@ -110,6 +112,19 @@ export interface WebSearchToolInput {
   query: string;
 }
 
+export interface WriteFileInput {
+  title: string;
+  file_path: string;
+  content: string;
+}
+
+export interface SearchReplaceInput {
+  title: string;
+  file_path: string;
+  old_str: string;
+  new_str: string;
+}
+
 export interface CodeGenToolMap {
   view_path: ViewPathToolInput;
   glob_search: GlobSearchToolInput;
@@ -120,10 +135,17 @@ export interface CodeGenToolMap {
   dev_server_control: DevServerControlInput;
   bash: BashToolInput;
   web_search: WebSearchToolInput;
+  write_file: WriteFileInput;
+  search_replace_file: SearchReplaceInput;
 }
 
 export type CodeGenTools = keyof CodeGenToolMap;
 
+export type AllCodeGenTools =
+  | CodeGenTools
+  | "str_replace_editor"
+  | "web_search";
+
 export type CodeGenMode =
   | "exact" // @deprecated
   | "precise" // tries to match the design as close
@@ -166,7 +188,6 @@ export interface CodeGenInputOptions {
   maxPages?: number;
   autoContinue?: number;
   promptCaching?: boolean;
-  isAutoContinue?: boolean;
   llmSuggestions?: boolean;
   conclusionText?: boolean;
 
@@ -175,6 +196,10 @@ export interface CodeGenInputOptions {
   // Prettier options
   prettierConfig?: PrettierOptions;
 
+  // Role
+  role?: "user" | "agent";
+  user?: UserSource;
+
   /** @deprecated */
   history?: (UserMessageParam | AssistantMessageParam)[];
   /** @deprecated */
@@ -223,7 +248,8 @@ export interface ActionItem {
     | "thinking"
     | "tool"
     | "suggestion"
-    | "tool_result";
+    | "tool_result"
+    | "server_tool_result";
   id?: string;
   content: string;
   filePath?: string;
@@ -349,6 +375,7 @@ export interface GenerateCompletionStepStart {
   id: string | undefined;
   title: string;
   content: string;
+  filePath?: string;
 }
 
 export interface GenerateCompletionStepDelta {
@@ -455,6 +482,13 @@ export interface GenerateCompletionStepSession {
   hasChanges: boolean;
 }
 
+export interface GenerateCompletionStepServerToolResult {
+  type: "server_tool_result";
+  content: any;
+  title: string;
+  id: string;
+}
+
 export type GenerateCompletionStep = { timestamp?: number } & (
   | GenerateCompletionStepPlanning
   | GenerateCompletionStepStart
@@ -474,6 +508,7 @@ export type GenerateCompletionStep = { timestamp?: number } & (
   | GenerateCompletionStepState
   | GenerateCompletionStepStdio
   | GenerateCompletionStepSession
+  | GenerateCompletionStepServerToolResult
   | GenerateCompletionStepToolResult
 );
 
@@ -486,7 +521,30 @@ export interface ApplyActionsResult {
   oldContent?: string;
 }
 
+export interface UserSourceBuilder {
+  source: "builder.io";
+  userId?: string;
+  userName?: string;
+  userEmail?: string;
+  role: "user";
+}
+
+export interface UserSourceGithub {
+  source: "github";
+  userName: string;
+  link?: string;
+  role: "user" | "agent";
+}
+
+export interface UserSourceAgent {
+  source: "agent";
+  role: "agent";
+}
+
+export type UserSource = UserSourceBuilder | UserSourceGithub | UserSourceAgent;
+
 export interface GenerateUserMessage {
+  user?: UserSource;
   userPrompt: string;
   ephemeralUserPrompt?: string;
   displayPrompt?: string;
@@ -594,6 +652,8 @@ export interface GenerateCodeEventUser {
   type: "user";
   id: string;
   displayPrompt: string | undefined;
+  user: UserSource;
+  /** @deprecated */
   role: "user" | "agent";
 }
 

package/src/events.ts

@@ -245,6 +245,15 @@ interface ContentPatchBase {
    * If there was an error applying the patch, this will contain the error message.
    */
   error?: string;
+
+  /**
+   * If `true`, Editor AI attempted to process this patch. It could be that the patch
+   * had invalid syntax or that the LLM made its changes incorrectly. This can be used
+   * to differentiate between a patch that is actively being streamed in, as indicated by
+   * the "incomplete" property, and a patch that is has finished streaming and is being
+   * processed.
+   */
+  attemptedProcess?: boolean;
 }
 
 export interface ContentTsUpdateComponentPatch extends ContentPatchBase {

package/src/messages.ts

@@ -13,6 +13,7 @@ export interface ContentMessageItemText {
   type: "text";
   text: string;
   cache?: boolean;
+  citations?: TextCitationParam[] | null;
   ephemeral?: boolean;
 }
 
@@ -48,7 +49,7 @@ export interface ContentMessageItemToolResult {
   tool_input?: string;
   title?: string;
   content: string;
-  is_error: boolean;
+  is_error?: boolean;
   cache?: boolean;
   ephemeral?: boolean;
 }
@@ -83,6 +84,104 @@ export interface ContentMessageItemToolUse {
   name: string;
 }
 
+export interface CitationCharLocationParam {
+  cited_text: string;
+
+  document_index: number;
+
+  document_title: string | null;
+
+  end_char_index: number;
+
+  start_char_index: number;
+
+  type: "char_location";
+}
+
+export interface CitationContentBlockLocationParam {
+  cited_text: string;
+
+  document_index: number;
+
+  document_title: string | null;
+
+  end_block_index: number;
+
+  start_block_index: number;
+
+  type: "content_block_location";
+}
+
+export interface CitationPageLocationParam {
+  cited_text: string;
+
+  document_index: number;
+
+  document_title: string | null;
+
+  end_page_number: number;
+
+  start_page_number: number;
+
+  type: "page_location";
+}
+
+export interface CitationWebSearchResultLocationParam {
+  cited_text: string;
+
+  encrypted_index: string;
+
+  title: string | null;
+
+  type: "web_search_result_location";
+
+  url: string;
+}
+
+export type TextCitationParam =
+  | CitationCharLocationParam
+  | CitationPageLocationParam
+  | CitationContentBlockLocationParam
+  | CitationWebSearchResultLocationParam;
+
+export interface ServerToolUseContent {
+  type: "web_search_result";
+  title: string;
+  url: string;
+  page_age: string | null;
+  encrypted_content: string;
+}
+
+export interface ServerToolUseContentError {
+  error_code:
+    | "invalid_tool_input"
+    | "unavailable"
+    | "max_uses_exceeded"
+    | "too_many_requests"
+    | "query_too_long";
+
+  type: "web_search_tool_result_error";
+}
+
+export type ServerToolUseContentUnion =
+  | ServerToolUseContent[]
+  | ServerToolUseContentError;
+
+export interface ContentMessageItemServerToolUse {
+  type: "server_tool_use";
+  id: string;
+  name: "web_search";
+  input: unknown;
+  completion?: string;
+  content?: ServerToolUseContentUnion;
+}
+
+export interface ContentMessageItemWebSearchToolResult {
+  content: ServerToolUseContent[] | ServerToolUseContentError;
+  tool_use_id: string;
+  type: "web_search_tool_result";
+}
+
 export type ContentMessageItem =
   | ContentMessageItemText
   | ContentMessageItemImage
@@ -90,7 +189,9 @@ export type ContentMessageItem =
   | ContentMessageItemThinking
   | ContentMessageItemRedactedThinking
   | ContentMessageItemToolUse
-  | ContentMessageItemToolResult;
+  | ContentMessageItemToolResult
+  | ContentMessageItemWebSearchToolResult
+  | ContentMessageItemServerToolUse;
 
 export type ContentMessage = ContentMessageItem[];