@builder.io/ai-utils 0.0.75 → 0.1.0

package/package.json

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

package/src/codegen.d.ts

@@ -1,4 +1,4 @@
-import type { UserMessageParam } from "./messages";
+import type { ContentMessageItemToolResult, UserMessageParam } from "./messages";
 import type { BuilderContent } from "./completion";
 import type { Options as PrettierOptions } from "prettier";
 import type { UserContext } from "./mapping";
@@ -23,13 +23,35 @@ export interface CustomInstruction {
 }
 export type CodeGenFramework = "react" | "html" | "mitosis" | "react-native" | "angular" | "vue" | "svelte" | "qwik" | "solid" | "marko" | "swiftui" | "jetpack-compose" | "flutter";
 export type CodeGenStyleLibrary = "tailwind" | "tailwind-precise" | "emotion" | "styled-components" | "styled-jsx" | "react-native" | undefined;
-export type CodeGenStopReason = "error" | "max_tokens" | "end" | "limit";
+export type CompletionStopReason = "max_tokens" | "stop_sequence" | "tool_use" | "end_turn" | "content_filter" | null;
+export type CodeGenStopReason = "error" | "limit" | CompletionStopReason;
+export interface ViewPathToolInput {
+    filePath: string;
+}
+export interface GlobSearchToolInput {
+    pattern: string;
+}
+export interface GrepSearchToolInput {
+    query: string;
+    includeGlob?: string;
+    excludeGlob?: string;
+}
+export interface AskUserToolInput {
+    question: string;
+}
+export interface CodeGenToolMap {
+    view_path: ViewPathToolInput;
+    glob_search: GlobSearchToolInput;
+    grep_search: GrepSearchToolInput;
+    ask_user: AskUserToolInput;
+}
+export type CodeGenTools = keyof CodeGenToolMap;
 export type CodeGenMode = "exact" | "precise" | "precise_vision" | "creative" | "creative_vision" | "creative_only_vision";
 export interface CodeGenInputOptions {
     position: string;
     eventName?: string;
     sessionId: string;
-    codeGenMode?: "fast" | "quality-v2";
+    codeGenMode?: "fast" | "quality" | "quality-v3";
     url?: string;
     diffActions?: boolean;
     planningPrompt?: boolean;
@@ -37,18 +59,19 @@ export interface CodeGenInputOptions {
     userPrompt?: string;
     files?: ProjectFile[];
     rerankFiles?: number;
+    toolResults?: ContentMessageItemToolResult[];
     builderContent?: BuilderContent;
     framework?: CodeGenFramework;
     styleLibrary?: CodeGenStyleLibrary;
     typescript?: boolean;
     userContext?: UserContext;
+    enabledTools?: CodeGenTools[];
     maxTokens?: number;
     maxPages?: number;
     autoContinue?: number;
     promptCaching?: boolean;
     isAutoContinue?: boolean;
     llmSuggestions?: boolean;
-    requestActions?: boolean;
     conclusionText?: boolean;
     searchResponse?: any | null;
     prettierConfig?: PrettierOptions;
@@ -61,3 +84,41 @@ export interface CodeGenInputOptions {
     /** @deprecated */
     vcpId?: string;
 }
+export type Feature = "component-mapping";
+export interface CodegenUsage {
+    total: number;
+    fast: number;
+    quality: number;
+    features: Feature[];
+    limits: {
+        aiGeneration: number;
+        aiGenerationContextWindow: number;
+    };
+}
+export interface PromptSuggestion {
+    type: "missing-imports" | "lazy-code" | "syntax-error" | "llm-suggested" | "diff-apply";
+    filePath?: string;
+    line?: number;
+    importance: "high" | "medium" | "low";
+    column?: number;
+    summary: string;
+    prompt: string;
+}
+export interface ActionItem {
+    type: "file" | "text" | "delta" | "done" | "diff" | "thinking" | "user" | "tool" | "suggestion" | "continue";
+    id?: string;
+    content: string;
+    filePath?: string;
+    artifactTitle?: string;
+    actionTitle?: string;
+    synthetic?: boolean;
+    incomplete?: boolean;
+    nextUrl?: string;
+    actions?: ActionItem[];
+    stopReason?: CodeGenStopReason;
+    suggestions?: PromptSuggestion[];
+    messageIndex?: number;
+    usage?: CodegenUsage;
+    errors?: string[];
+    sessionUsage?: number;
+}

package/src/completion.d.ts

@@ -1,7 +1,6 @@
 import type { BuilderContent, BuilderElement, Component } from "@builder.io/sdk";
 import type { Message, MessageParam, Attachment } from "./messages.js";
 import type { BuilderModel } from "./events.js";
-export type BuilderContentData = BuilderContent["data"];
 export type { BuilderContent, BuilderElement, Component };
 export interface CompletionOptions {
     /**
@@ -147,6 +146,7 @@ export interface BuilderEditorState {
      * Custom instructions that could be add into the prompt.
      */
     customInstructions?: Record<string, string>;
+    styleInspirationURL?: string;
     /**
      * All targeting attributes of the content.
      */

package/src/events.d.ts

@@ -2,7 +2,7 @@ import type { BuilderContent, BuilderElement } from "@builder.io/sdk";
 import type { AssistantMessage } from "./messages.js";
 import type { AssistantSettings } from "./settings.js";
 export type BuilderAssistantEventHandler = (ev: BuilderAssistantEvent) => void;
-export type BuilderAssistantEvent = AssistantCompletionResultEvent | AssistantErrorEvent | AssistantStreamErrorEvent | AppCloseEvent | AppMessagesClickEvent | AppMessagesGenerationEvent | AppMessageEditCustomInstructionsEvent | AppPromptAbortEvent | AppPromptFocusEvent | AppPromptTextEvent | AppReadyEvent | AppSettingsResetEvent | AppSettingsSetEvent | AppThreadNewEvent | AssistantStatsEvent | AssistantThemeEvent | BuilderEditorAuthEvent | BuilderEditorStateEvent | ContentUpdateEvent | ContentApplySnapshotEvent | ModelCreateEvent | ModelUpdateEvent | ModelUndoEvent | ModelRedoEvent | ResultEvent | ThreadCreatedEvent | ThreadMessageCompletedEvent | ThreadMessageCreatedEvent | ThreadMessageDeltaEvent | ThreadMessageFeedbackEvent | ThreadRunRequiresActionEvent | ThreadRunStepCreatedEvent | ThreadRunStepDeltaEvent | AppAcceptChangeEvent | AppAcceptRejectEvent | AssistantTrackEvent | AssistantEditorAuthMessage | AppAttachmentTemplateEvent | ThreadMessageRetryEvent;
+export type BuilderAssistantEvent = AssistantCompletionResultEvent | AssistantErrorEvent | AssistantStreamErrorEvent | AppCloseEvent | AppMessagesClickEvent | AppMessagesGenerationEvent | AppMessageEditCustomInstructionsEvent | AppPromptAbortEvent | AppPromptFocusEvent | AppReadyEvent | AppSettingsSetEvent | AppThreadNewEvent | AssistantStatsEvent | AssistantThemeEvent | BuilderEditorAuthEvent | BuilderEditorStateEvent | ContentUpdateEvent | ContentApplySnapshotEvent | ModelUndoEvent | ModelRedoEvent | ResultEvent | ThreadCreatedEvent | ThreadMessageCompletedEvent | ThreadMessageCreatedEvent | ThreadMessageDeltaEvent | ThreadMessageFeedbackEvent | ThreadRunStepCreatedEvent | ThreadRunStepDeltaEvent | AppAcceptChangeEvent | AppAcceptRejectEvent | AssistantTrackEvent | AssistantEditorAuthMessage | AppAttachmentTemplateEvent | ThreadMessageRetryEvent;
 export interface AssistantCompletionResultEvent {
     type: "assistant.result";
     data: {
@@ -78,18 +78,9 @@ export interface AppPromptAbortEvent {
 export interface AppPromptFocusEvent {
     type: "assistant.app.prompt.focus";
 }
-export interface AppPromptTextEvent {
-    type: "assistant.app.prompt.text";
-    data: {
-        text: string;
-    };
-}
 export interface AppReadyEvent {
     type: "assistant.app.ready";
 }
-export interface AppSettingsResetEvent {
-    type: "assistant.app.settings.reset";
-}
 export interface AppSettingsSetEvent {
     type: "assistant.app.settings.set";
     data: Partial<AssistantSettings>;
@@ -140,7 +131,7 @@ export interface ContentUpdateEvent {
     type: "assistant.content.update";
     data: ContentUpdatePatch[];
 }
-export type ContentUpdatePatch = ContentTsCodePatch | ContentInsertBeforePatch | ContentInsertAfterPatch | ContentPropsUpdatePatch | ContentSetChildrenPatch | ContentStyleUpdatePatch | ContentDeletePatch | ContentSetStorePatch | ContentSetMountPatch | ContentMovePatch | ContentMoveAfterPatch | ContentMoveBeforePatch | ContentSetTagNamePatch | ContentTsUpdateComponentPatch;
+export type ContentUpdatePatch = ContentTsUpdateComponentPatch;
 interface ContentPatchBase {
     id: string;
     nodeId?: string;
@@ -161,48 +152,6 @@ interface ContentPatchBase {
 export interface ContentTsUpdateComponentPatch extends ContentPatchBase {
     type: "update_component";
 }
-export interface ContentTsCodePatch extends ContentPatchBase {
-    type: "ts_code";
-}
-export interface ContentSetTagNamePatch extends ContentPatchBase {
-    type: "set_tag_name";
-}
-export interface ContentInsertBeforePatch extends ContentPatchBase {
-    type: "insert_before";
-    snapshotBuilderId?: string;
-}
-export interface ContentInsertAfterPatch extends ContentPatchBase {
-    type: "insert_after";
-    snapshotBuilderId?: string;
-}
-export interface ContentPropsUpdatePatch extends ContentPatchBase {
-    type: "props";
-}
-export interface ContentStyleUpdatePatch extends ContentPatchBase {
-    type: "style";
-}
-export interface ContentSetChildrenPatch extends ContentPatchBase {
-    type: "children";
-    snapshotBuilderId?: string;
-}
-export interface ContentDeletePatch extends ContentPatchBase {
-    type: "delete";
-}
-export interface ContentSetStorePatch extends ContentPatchBase {
-    type: "store";
-}
-export interface ContentSetMountPatch extends ContentPatchBase {
-    type: "mount";
-}
-export interface ContentMovePatch extends ContentPatchBase {
-    type: "move";
-}
-export interface ContentMoveAfterPatch extends ContentPatchBase {
-    type: "move_after";
-}
-export interface ContentMoveBeforePatch extends ContentPatchBase {
-    type: "move_before";
-}
 export interface AssistantStatsEvent {
     type: "assistant.stats";
     data: AssistantStats;
@@ -295,23 +244,12 @@ export interface AssistantStats {
      */
     cacheCreatedTokens?: number;
 }
-export interface ModelCreateEvent {
-    type: "assistant.model.create";
-    data: BuilderModel;
-}
-export interface ModelUpdateEvent {
-    type: "assistant.model.update";
-    data: ModelUpdate;
-}
 export interface ModelUndoEvent {
     type: "assistant.model.undo";
 }
 export interface ModelRedoEvent {
     type: "assistant.model.redo";
 }
-export interface ModelUpdate {
-    patches: ModelPatch[];
-}
 export interface BuilderModel {
     name?: string;
     friendlyName?: string;
@@ -324,11 +262,6 @@ export interface BuilderModelField {
     type?: string;
     description?: string;
 }
-export type ModelPatch = {
-    op: "add" | "remove" | "replace";
-    path: string;
-    value?: any;
-};
 export interface ThreadMessageFeedbackEvent {
     type: "assistant.thread.message.feedback";
     data: CompletionResponseFeedback;
@@ -383,18 +316,6 @@ export interface ThreadMessageCompleted extends AssistantMessage {
     threadId: string;
     commitId?: string;
 }
-export interface ThreadRunRequiresActionEvent {
-    type: "assistant.thread.run.requires_action";
-    data: ThreadRunRequiredAction;
-}
-export interface ThreadRunRequiredAction {
-    platformId: string;
-    threadId: string;
-    action: {
-        name: string;
-        arguments: string;
-    };
-}
 export interface ThreadRunStepDeltaEvent {
     type: "assistant.thread.run.step.delta";
     data: ThreadMessageStepDelta;

package/src/index.d.ts

@@ -1,10 +1,6 @@
-export * from "./complete-json.js";
 export * from "./completion.js";
 export * from "./events.js";
-export * from "./fetch.js";
 export * from "./messages.js";
 export * from "./settings.js";
-export * from "./thread.js";
 export * from "./mapping.js";
 export * from "./codegen.js";
-export * from "./vcp.js";

package/src/index.js

@@ -1,10 +1,6 @@
-export * from "./complete-json.js";
 export * from "./completion.js";
 export * from "./events.js";
-export * from "./fetch.js";
 export * from "./messages.js";
 export * from "./settings.js";
-export * from "./thread.js";
 export * from "./mapping.js";
 export * from "./codegen.js";
-export * from "./vcp.js";

package/src/mapping.d.ts

@@ -1,4 +1,12 @@
 import type { ESMImport } from "./codegen";
+export interface RawFigmaJson {
+    documents: any[];
+    document?: any;
+    components?: Record<string, any>;
+    styles?: Record<string, any>;
+    componentSets?: Record<string, any>;
+    schemaVersion?: number;
+}
 export interface UserContext {
     client: string;
     clientVersion: string;
@@ -18,6 +26,7 @@ export type ExportType = "default" | "named";
 export interface FigmaMappingsData {
     id: string;
     figmaBuilderLinks: FigmaBuilderLink[];
+    framework: string;
     version?: number;
     createdDate?: string;
     local: boolean;

package/src/messages.d.ts

@@ -11,14 +11,27 @@ export interface ContentMessageItemText {
 }
 export interface ContentMessageItemImage {
     type: "image";
-    source: ImageSource;
+    source: ImageBase64Source | ImageUrlSource;
     cache?: boolean;
 }
-export interface ImageSource {
+export interface ContentMessageItemToolResult {
+    type: "tool_result";
+    tool_use_id: string;
+    tool_name?: string;
+    tool_input?: string;
+    content: string;
+    is_error: boolean;
+    cache?: boolean;
+}
+export interface ImageBase64Source {
     type: "base64";
     media_type: "image/webp" | "image/png" | "image/jpeg";
     data: string;
 }
+export interface ImageUrlSource {
+    type: "url";
+    url: string;
+}
 export interface ContentMessageItemThinking {
     type: "thinking";
     thinking: string;
@@ -32,9 +45,10 @@ export interface ContentMessageItemToolUse {
     type: "tool_use";
     id: string;
     input: unknown;
+    completion?: string;
     name: string;
 }
-export type ContentMessageItem = ContentMessageItemText | ContentMessageItemImage | ContentMessageItemThinking | ContentMessageItemRedactedThinking | ContentMessageItemToolUse;
+export type ContentMessageItem = ContentMessageItemText | ContentMessageItemImage | ContentMessageItemThinking | ContentMessageItemRedactedThinking | ContentMessageItemToolUse | ContentMessageItemToolResult;
 export type ContentMessage = ContentMessageItem[];
 export interface SystemMessageParam {
     /**
@@ -52,7 +66,7 @@ export interface UserMessageParam {
     /**
      * The contents of the user message.
      */
-    content: string | (ContentMessageItemText | ContentMessageItemImage)[];
+    content: string | (ContentMessageItemText | ContentMessageItemImage | ContentMessageItemToolResult)[];
     responseId?: string;
     /**
      * The role of the messages author, in this case `user`.
@@ -115,8 +129,12 @@ export interface AssistantActionMessage {
 export type Message = SystemMessage | UserMessage | AssistantMessage;
 export type GeneratingMessage = null | Partial<AssistantActionMessage | AssistantMessage>;
 export declare function getContentText(message: string | ContentMessage): string;
-export declare function getContentAttachments(message: string | ContentMessage): ImageSource[];
-export type Attachment = FileUpload | Template;
+export declare function getContentAttachments(message: string | ContentMessage): (ImageBase64Source | ImageUrlSource)[];
+export type Attachment = FileUpload | Template | URL;
+export interface URL {
+    type: "url";
+    value: string;
+}
 export interface FileUpload {
     type: "upload";
     contentType: string;

package/src/settings.d.ts

@@ -1,9 +1,6 @@
 export interface AssistantSettings {
     assistantType?: string;
     viewId?: string;
-    cssOverrides?: string;
-    messagePlaceholder: string;
-    showPrompt?: boolean;
     theme?: "dark" | "light";
 }
 interface IframeSettings extends Partial<AssistantSettings> {

package/src/settings.js

@@ -1,6 +1,5 @@
 const urlParamSettings = [
     "assistantType",
-    "showPrompt",
     "theme",
     "viewId",
 ];

package/src/complete-json.d.ts

@@ -1,4 +0,0 @@
-/**
- * Complete a partial JSON string with the necessary closing brackets and braces
- */
-export declare function completeJson(input: string): string;

package/src/complete-json.js

@@ -1,52 +0,0 @@
-/**
- * Complete a partial JSON string with the necessary closing brackets and braces
- */
-export function completeJson(input) {
-    const stack = [];
-    let isInString = false;
-    let lastQuote = "";
-    // Scan the string to determine the necessary closures
-    for (const char of input) {
-        switch (char) {
-            case '"':
-            case "'":
-                if (isInString && char === lastQuote) {
-                    // Check for escaping
-                    if (input.charAt(input.indexOf(char) - 1) !== "\\") {
-                        isInString = false;
-                    }
-                }
-                else if (!isInString) {
-                    isInString = true;
-                    lastQuote = char;
-                }
-                break;
-            case "{":
-            case "[":
-                if (!isInString) {
-                    stack.push(char === "{" ? "}" : "]");
-                }
-                break;
-            case "}":
-            case "]":
-                if (!isInString) {
-                    if (stack.length > 0 && stack[stack.length - 1] === char) {
-                        stack.pop();
-                    }
-                }
-                break;
-            default:
-                break;
-        }
-    }
-    // Close any unclosed string
-    if (isInString) {
-        input += lastQuote;
-    }
-    // Close all remaining open brackets and braces
-    while (stack.length > 0) {
-        const charToClose = stack.pop();
-        input += charToClose;
-    }
-    return input;
-}

package/src/fetch.d.ts

@@ -1 +0,0 @@
-export declare function fetchJsonl<T>(input: string | URL, init: RequestInit, onData: (ev: T) => void): Promise<Response>;

package/src/fetch.js

@@ -1,83 +0,0 @@
-export async function fetchJsonl(input, init, onData) {
-    return new Promise(async (resolve, reject) => {
-        try {
-            init = {
-                ...init,
-                headers: {
-                    accept: "application/jsonl",
-                    "content-type": "application/json",
-                    ...init.headers,
-                },
-            };
-            const rsp = await (async () => {
-                try {
-                    return await fetch(input, init);
-                }
-                catch (err) {
-                    reject(new NetworkError(`${input}, status: network failure`));
-                }
-            })();
-            if (!rsp) {
-                return;
-            }
-            if (!rsp.ok) {
-                reject(new APIError(`${input}, status: ${rsp.status}`));
-                return;
-            }
-            if (!rsp.body) {
-                reject(new APIError(`${input}, missing body`));
-                return;
-            }
-            const reader = rsp.body.getReader();
-            const decoder = new TextDecoder();
-            let data = "";
-            const processStream = async () => {
-                const { done, value } = await reader.read();
-                data += decoder.decode(value, { stream: !done });
-                if (!done) {
-                    let lines = data.split("\n");
-                    if (!done) {
-                        data = lines.pop() || "";
-                    }
-                    for (let line of lines) {
-                        if (line) {
-                            try {
-                                line = line.trim();
-                                if (line.startsWith("{") && line.endsWith("}")) {
-                                    onData(JSON.parse(line));
-                                }
-                                else {
-                                    reject(`Invalid JSONL line: ${line}`);
-                                    return;
-                                }
-                            }
-                            catch (e) {
-                                console.error(`Failed to parse JSONL: ${line}`);
-                                reject(e);
-                                return;
-                            }
-                        }
-                    }
-                    await processStream();
-                }
-            };
-            await processStream();
-            resolve(rsp);
-        }
-        catch (e) {
-            reject(e);
-        }
-    });
-}
-class NetworkError extends Error {
-    constructor(message) {
-        super(message);
-        this.name = "NetworkError";
-    }
-}
-class APIError extends Error {
-    constructor(message) {
-        super(message);
-        this.name = "APIError";
-    }
-}

package/src/thread.d.ts

@@ -1,27 +0,0 @@
-import type { Message } from "./messages.js";
-export interface Thread {
-    /**
-     * The unique LOCAL identifier of the thread.
-     */
-    id: string;
-    /**
-     * The unique OPENAI identifier of the thread. This is only
-     * set after openai creates the thread.
-     */
-    threadId?: string;
-    platformId?: string;
-    title?: string;
-    /**
-     * The type of assistant that the thread is associated with.
-     * For example, "content-editor" or "docs".
-     */
-    assistantType?: string;
-    /**
-     * The unique identifier of the view that the thread is associated with.
-     * For example, the "content-editor" assistant would use the
-     * builder content id as the viewId.
-     */
-    viewId?: string;
-    created: number;
-    messages: Message[];
-}

package/src/thread.js

@@ -1 +0,0 @@
-export {};

package/src/vcp.d.ts

@@ -1,32 +0,0 @@
-import type { ESMImport } from "./codegen";
-export interface BuilderComponentInfo {
-    name: string;
-    inputs?: BuilderComponentInput[];
-    path: string;
-    imports: {
-        [key: string]: string | undefined;
-    };
-    esmImports?: ESMImport[];
-    options?: any;
-}
-export interface BuilderComponentInput {
-    name: string;
-    type: string;
-    value: any;
-}
-export interface BuilderComponentSlot {
-    figmaId: string;
-    builderName: string;
-}
-export interface JSXNode {
-    "@type": "jsx";
-    name: string;
-    props: Record<string, any>;
-    tagName?: string;
-    includeAirLayoutStyles?: boolean;
-    path?: string;
-    imports?: {
-        [key: string]: string | undefined;
-    };
-    esmImports?: ESMImport[];
-}

package/src/vcp.js

@@ -1 +0,0 @@
-export {};