@builder.io/ai-utils 0.3.17 → 0.3.18

package/package.json

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

package/src/codegen.ts

@@ -141,10 +141,7 @@ export interface CodeGenToolMap {
 
 export type CodeGenTools = keyof CodeGenToolMap;
 
-export type AllCodeGenTools =
-  | CodeGenTools
-  | "str_replace_editor"
-  | "web_search";
+export type AllCodeGenTools = CodeGenTools | "web_search";
 
 export type CodeGenMode =
   | "exact" // @deprecated
@@ -190,6 +187,7 @@ export interface CodeGenInputOptions {
   promptCaching?: boolean;
   llmSuggestions?: boolean;
   conclusionText?: boolean;
+  mcpServers?: boolean;
 
   searchResponse?: any | null;
 
@@ -205,8 +203,6 @@ export interface CodeGenInputOptions {
   /** @deprecated */
   prevId?: string;
   /** @deprecated */
-  nextPage?: boolean;
-  /** @deprecated */
   vcpId?: string;
 }
 
@@ -667,6 +663,10 @@ export interface GenerateCodeEventDelta {
   content: string;
 }
 
+export interface GenerateCodeEventPing {
+  type: "ping";
+}
+
 export interface GenerateCodeEventUser {
   type: "user";
   id: string;
@@ -683,4 +683,5 @@ export type GenerateCodeEvent =
   | GenerateCodeEventPagination
   | GenerateCodeEventDelta
   | GenerateCodeEventError
+  | GenerateCodeEventPing
   | GenerateCodeEventUser;

package/src/messages.ts

@@ -48,7 +48,7 @@ export interface ContentMessageItemToolResult {
   tool_name?: string;
   tool_input?: string;
   title?: string;
-  content: string;
+  content: string | (ContentMessageItemText | ContentMessageItemImage)[];
   is_error?: boolean;
   cache?: boolean;
   ephemeral?: boolean;
@@ -56,7 +56,7 @@ export interface ContentMessageItemToolResult {
 
 export interface ImageBase64Source {
   type: "base64";
-  media_type: "image/webp" | "image/png" | "image/jpeg";
+  media_type: "image/webp" | "image/png" | "image/jpeg" | "image/gif";
   data: string;
 }
 
@@ -170,16 +170,65 @@ export type ServerToolUseContentUnion =
 export interface ContentMessageItemServerToolUse {
   type: "server_tool_use";
   id: string;
-  name: "web_search";
+  name: "web_search" | "code_execution";
   input: unknown;
   completion?: string;
-  content?: ServerToolUseContentUnion;
+  content?: any;
+  cache?: boolean;
+}
+
+export interface MCPServerURLDefinition {
+  name: string;
+
+  type: "url";
+
+  url: string;
+
+  authorization_token?: string | null;
+
+  tool_configuration?: MCPServerToolConfiguration | null;
+}
+
+export interface MCPServerToolConfiguration {
+  allowed_tools?: Array<string> | null;
+
+  enabled?: boolean | null;
 }
 
 export interface ContentMessageItemWebSearchToolResult {
   content: ServerToolUseContent[] | ServerToolUseContentError;
   tool_use_id: string;
   type: "web_search_tool_result";
+  cache?: boolean;
+}
+
+export interface ContentMessageItemMCPToolUse {
+  type: "mcp_tool_use";
+  id: string;
+  name: string;
+  input: unknown;
+  server_name: string;
+  cache?: boolean;
+}
+
+export interface ContentMessageItemContainerUpload {
+  type: "container_upload";
+  file_id: string;
+}
+
+export interface ContentMessageItemMCPToolResult {
+  type: "mcp_tool_result";
+  tool_use_id: string;
+  content: string | ContentMessageItemText[];
+  is_error: boolean;
+  cache?: boolean;
+}
+
+export interface ContentMessageItemCodeExecutionToolResult {
+  type: "code_execution_tool_result";
+  tool_use_id: string;
+  content: any;
+  cache?: boolean;
 }
 
 export type ContentMessageItem =
@@ -191,7 +240,11 @@ export type ContentMessageItem =
   | ContentMessageItemToolUse
   | ContentMessageItemToolResult
   | ContentMessageItemWebSearchToolResult
-  | ContentMessageItemServerToolUse;
+  | ContentMessageItemServerToolUse
+  | ContentMessageItemMCPToolUse
+  | ContentMessageItemMCPToolResult
+  | ContentMessageItemContainerUpload
+  | ContentMessageItemCodeExecutionToolResult;
 
 export type ContentMessage = ContentMessageItem[];