@builder.io/ai-utils 0.0.74 → 0.0.75

package/src/fetch.js

@@ -0,0 +1,83 @@
+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/index.js

@@ -0,0 +1,10 @@
+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

@@ -0,0 +1,77 @@
+import type { ESMImport } from "./codegen";
+export interface UserContext {
+    client: string;
+    clientVersion: string;
+    nodeVersion: string;
+    systemPlatform: string;
+    frameworks: string[];
+    systemEOL: string;
+    systemArch: string;
+    systemShell?: string;
+    inGitRepo?: boolean;
+    [key: string]: string | string[] | boolean | undefined;
+}
+export type ExportType = "default" | "named";
+/**
+ * Gets the latest component mappings for a space
+ */
+export interface FigmaMappingsData {
+    id: string;
+    figmaBuilderLinks: FigmaBuilderLink[];
+    version?: number;
+    createdDate?: string;
+    local: boolean;
+    userEmail?: string;
+    remoteUrl?: string;
+}
+export interface FigmaBuilderLink {
+    builderName: string;
+    figmaName: string;
+    figmaKey: string;
+    figmaUrl?: string;
+    inputMapper?: string;
+    originalInputMapper?: string;
+    exportType?: ExportType;
+    importName?: string;
+    importPath?: string;
+    source: string;
+    loc?: string;
+    imports?: ESMImport[];
+}
+export interface FigmaMapperFile {
+    filePath: string;
+    content: string;
+}
+export interface PublishedMapping {
+    figmaBuilderLinks: FigmaBuilderLink[];
+    mapperFiles: FigmaMapperFile[];
+    remoteUrl?: string;
+    defaultBranch?: string;
+    currentBranch?: string;
+    commit?: string;
+    spaceKind?: string;
+    userContext?: UserContext;
+}
+export interface FigmaComponentInfo {
+    documentName: string;
+    key: string;
+    tree?: string;
+    jsx?: string;
+    type: string;
+    name: string;
+    exportJson?: any;
+    inputs: FigmaComponentInput[];
+    description: string;
+    documentationLinks: string[];
+    instanceId: string;
+}
+export interface FigmaComponentInput {
+    id: string;
+    name: string;
+    value?: any;
+    type: string;
+    baseType: "text" | "variant" | "boolean" | "slot";
+    variantOptions?: string[];
+    isDefault: boolean;
+    ref?: string;
+}

package/src/mapping.js

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

package/src/messages.d.ts

@@ -0,0 +1,132 @@
+import type { ContentUpdatePatch } from "./events.js";
+/**
+ * Message param does not know the id of the message.
+ * This is an input message.
+ */
+export type MessageParam = SystemMessageParam | UserMessageParam | AssistantMessageParam;
+export interface ContentMessageItemText {
+    type: "text";
+    text: string;
+    cache?: boolean;
+}
+export interface ContentMessageItemImage {
+    type: "image";
+    source: ImageSource;
+    cache?: boolean;
+}
+export interface ImageSource {
+    type: "base64";
+    media_type: "image/webp" | "image/png" | "image/jpeg";
+    data: string;
+}
+export interface ContentMessageItemThinking {
+    type: "thinking";
+    thinking: string;
+    signature: string;
+}
+export interface ContentMessageItemRedactedThinking {
+    type: "redacted_thinking";
+    data: string;
+}
+export interface ContentMessageItemToolUse {
+    type: "tool_use";
+    id: string;
+    input: unknown;
+    name: string;
+}
+export type ContentMessageItem = ContentMessageItemText | ContentMessageItemImage | ContentMessageItemThinking | ContentMessageItemRedactedThinking | ContentMessageItemToolUse;
+export type ContentMessage = ContentMessageItem[];
+export interface SystemMessageParam {
+    /**
+     * The contents of the system message.
+     */
+    content: string | ContentMessageItemText[];
+    responseId?: string;
+    /**
+     * The role of the messages author, in this case `system`.
+     */
+    role: "system";
+    id?: string;
+}
+export interface UserMessageParam {
+    /**
+     * The contents of the user message.
+     */
+    content: string | (ContentMessageItemText | ContentMessageItemImage)[];
+    responseId?: string;
+    /**
+     * The role of the messages author, in this case `user`.
+     */
+    role: "user";
+    id?: string;
+}
+export interface AssistantMessageParam {
+    /**
+     * The contents of the assistant message.
+     */
+    content: string | ContentMessageItem[];
+    /**
+     * The role of the messages author, in this case `assistant`.
+     */
+    role: "assistant";
+    responseId?: string;
+    /**
+     * The function call name and arguments
+     */
+    action?: {
+        /**
+         * The specific function name
+         */
+        name: string;
+        /** This is arbitrary JSON */
+        arguments: any;
+    };
+    id?: string;
+    skipDelta?: boolean;
+    /**
+     * A summary of the patches which the assistant has made.
+     * Useful for genai.
+     */
+    patches?: ContentUpdatePatch[];
+    state?: "error";
+}
+export interface SystemMessage extends SystemMessageParam {
+    id: string;
+}
+export interface UserMessage extends UserMessageParam {
+    id: string;
+}
+export interface AssistantMessage extends AssistantMessageParam {
+    id: string;
+    status?: "accepted" | "rejected" | "aborted";
+}
+export interface AssistantActionMessage {
+    /**
+     * The role of the messages author, in this case `assistant`.
+     */
+    role: "assistant";
+    id: string;
+}
+/**
+ * Message DOES know the id of the message.
+ * This message is after an id has been assigned
+ * and is the output message.
+ */
+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 interface FileUpload {
+    type: "upload";
+    contentType: string;
+    name: string;
+    dataUrl: string;
+    size: number;
+    id: string;
+}
+export interface Template {
+    type: "template";
+    name: string;
+    id: number;
+}

package/src/messages.js

@@ -0,0 +1,16 @@
+export function getContentText(message) {
+    if (typeof message === "string") {
+        return message;
+    }
+    return message
+        .map((item) => (item.type === "text" ? item.text : ""))
+        .join("");
+}
+export function getContentAttachments(message) {
+    if (typeof message === "string") {
+        return [];
+    }
+    return message
+        .filter((item) => item.type === "image")
+        .map((item) => item.source);
+}

package/src/settings.d.ts

@@ -0,0 +1,23 @@
+export interface AssistantSettings {
+    assistantType?: string;
+    viewId?: string;
+    cssOverrides?: string;
+    messagePlaceholder: string;
+    showPrompt?: boolean;
+    theme?: "dark" | "light";
+}
+interface IframeSettings extends Partial<AssistantSettings> {
+    local?: boolean;
+    /**
+     * The URL of the assistant.
+     */
+    baseUrl?: string;
+}
+export declare function getAssistantUrl(opts?: IframeSettings): string;
+export declare function parseAssistantUrlSettings(url: string): Partial<AssistantSettings>;
+export interface BuilderEditorAuth {
+    spaceId: string;
+    userId: string;
+    authToken: string;
+}
+export {};

package/src/settings.js

@@ -0,0 +1,31 @@
+const urlParamSettings = [
+    "assistantType",
+    "showPrompt",
+    "theme",
+    "viewId",
+];
+export function getAssistantUrl(opts = {}) {
+    var _a;
+    const url = new URL((_a = opts.baseUrl) !== null && _a !== void 0 ? _a : (opts.local ? "http://localhost:7242" : "https://assistant.builder.io"));
+    urlParamSettings.forEach((key) => {
+        const value = opts[key];
+        if (typeof value === "string" || typeof value === "boolean") {
+            url.searchParams.set(key, String(value));
+        }
+    });
+    return url.href;
+}
+export function parseAssistantUrlSettings(url) {
+    const parsed = new URL(url);
+    const settings = {};
+    urlParamSettings.forEach((key) => {
+        const value = parsed.searchParams.get(key);
+        if (value === "true" || value === "false") {
+            settings[key] = value === "true";
+        }
+        else if (value) {
+            settings[key] = value;
+        }
+    });
+    return settings;
+}

package/src/thread.d.ts

@@ -0,0 +1,27 @@
+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

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

package/src/vcp.d.ts

@@ -0,0 +1,32 @@
+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

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

package/README.md

@@ -1,3 +0,0 @@
-# Builder.io AI types and utilities
-
-Shared types between AI systems, threads, servers, etc.

package/src/codegen.ts

@@ -1,111 +0,0 @@
-import type { UserMessageParam } from "./messages";
-
-import type { BuilderContent } from "./completion";
-
-import type { Options as PrettierOptions } from "prettier";
-import type { UserContext } from "./mapping";
-import type { AssistantMessageParam } from "./messages";
-
-// Define the import type as a union of literal types.
-export type ImportType = "named" | "default";
-
-// Define an interface for the import descriptor.
-export interface ESMImport {
-  importName: string; // e.g. "Button"
-  importPath: string; // e.g. "path"
-  importType: ImportType; // Either 'named' or 'default'
-}
-
-export interface ProjectFile {
-  filePath: string;
-  content?: string;
-  importance?: number;
-  dropReason?: string;
-  wasIncluded?: boolean;
-}
-
-export interface CustomInstruction {
-  id: string;
-  name: string;
-  content: string;
-}
-
-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 CodeGenMode =
-  | "exact" // @deprecated
-  | "precise" // tries to match the design as close
-  | "precise_vision" // tries to match the design as close, also uses vision to match
-  | "creative" // adapts the design to some generic design language
-  | "creative_vision" // adapts the design to some generic design language, also uses vision to match
-  | "creative_only_vision"; // adapts the design to some generic design language, but only uses vision to match
-
-export interface CodeGenInputOptions {
-  position: string;
-  eventName?: string;
-  sessionId: string;
-
-  codeGenMode?: "fast" | "quality-v2";
-  url?: string;
-  diffActions?: boolean;
-  planningPrompt?: boolean;
-  customInstructions?: CustomInstruction[];
-  userPrompt?: string;
-  files?: ProjectFile[];
-  rerankFiles?: number;
-
-  // Code options
-  builderContent?: BuilderContent;
-  framework?: CodeGenFramework;
-  styleLibrary?: CodeGenStyleLibrary;
-  typescript?: boolean;
-  userContext?: UserContext;
-
-  // Options
-  maxTokens?: number;
-  maxPages?: number;
-  autoContinue?: number;
-  promptCaching?: boolean;
-  isAutoContinue?: boolean;
-  llmSuggestions?: boolean;
-  requestActions?: boolean;
-  conclusionText?: boolean;
-
-  searchResponse?: any | null;
-
-  // Prettier options
-  prettierConfig?: PrettierOptions;
-
-  /** @deprecated */
-  history?: (UserMessageParam | AssistantMessageParam)[];
-  /** @deprecated */
-  prevId?: string;
-  /** @deprecated */
-  nextPage?: boolean;
-  /** @deprecated */
-  vcpId?: string;
-}

package/src/complete-json.ts

@@ -1,55 +0,0 @@
-/**
- * Complete a partial JSON string with the necessary closing brackets and braces
- */
-export function completeJson(input: string): string {
-  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;
-}