@puckeditor/cloud-client 0.3.0-canary.41e65b36 → 0.3.0-canary.78ff547c

package/dist/index.js

@@ -32,6 +32,7 @@ var index_exports = {};
 __export(index_exports, {
   chat: () => chat,
   endpoints: () => endpoints,
+  generate: () => generate,
   puckHandler: () => puckHandler,
   tool: () => tool
 });
@@ -40,6 +41,186 @@ module.exports = __toCommonJS(index_exports);
 // ../tsup-config/react-import.js
 var import_react = __toESM(require("react"));
 
+// src/api/tool.ts
+var tool = (t) => t;
+
+// ../../node_modules/.pnpm/@ai-sdk+provider@2.0.0/node_modules/@ai-sdk/provider/dist/index.mjs
+var marker = "vercel.ai.error";
+var symbol = Symbol.for(marker);
+var _a;
+var _AISDKError = class _AISDKError2 extends Error {
+  /**
+   * Creates an AI SDK Error.
+   *
+   * @param {Object} params - The parameters for creating the error.
+   * @param {string} params.name - The name of the error.
+   * @param {string} params.message - The error message.
+   * @param {unknown} [params.cause] - The underlying cause of the error.
+   */
+  constructor({
+    name: name143,
+    message,
+    cause
+  }) {
+    super(message);
+    this[_a] = true;
+    this.name = name143;
+    this.cause = cause;
+  }
+  /**
+   * Checks if the given error is an AI SDK Error.
+   * @param {unknown} error - The error to check.
+   * @returns {boolean} True if the error is an AI SDK Error, false otherwise.
+   */
+  static isInstance(error45) {
+    return _AISDKError2.hasMarker(error45, marker);
+  }
+  static hasMarker(error45, marker153) {
+    const markerSymbol = Symbol.for(marker153);
+    return error45 != null && typeof error45 === "object" && markerSymbol in error45 && typeof error45[markerSymbol] === "boolean" && error45[markerSymbol] === true;
+  }
+};
+_a = symbol;
+var AISDKError = _AISDKError;
+var name = "AI_APICallError";
+var marker2 = `vercel.ai.error.${name}`;
+var symbol2 = Symbol.for(marker2);
+var _a2;
+_a2 = symbol2;
+var name2 = "AI_EmptyResponseBodyError";
+var marker3 = `vercel.ai.error.${name2}`;
+var symbol3 = Symbol.for(marker3);
+var _a3;
+_a3 = symbol3;
+function getErrorMessage(error45) {
+  if (error45 == null) {
+    return "unknown error";
+  }
+  if (typeof error45 === "string") {
+    return error45;
+  }
+  if (error45 instanceof Error) {
+    return error45.message;
+  }
+  return JSON.stringify(error45);
+}
+var name3 = "AI_InvalidArgumentError";
+var marker4 = `vercel.ai.error.${name3}`;
+var symbol4 = Symbol.for(marker4);
+var _a4;
+var InvalidArgumentError = class extends AISDKError {
+  constructor({
+    message,
+    cause,
+    argument
+  }) {
+    super({ name: name3, message, cause });
+    this[_a4] = true;
+    this.argument = argument;
+  }
+  static isInstance(error45) {
+    return AISDKError.hasMarker(error45, marker4);
+  }
+};
+_a4 = symbol4;
+var name4 = "AI_InvalidPromptError";
+var marker5 = `vercel.ai.error.${name4}`;
+var symbol5 = Symbol.for(marker5);
+var _a5;
+_a5 = symbol5;
+var name5 = "AI_InvalidResponseDataError";
+var marker6 = `vercel.ai.error.${name5}`;
+var symbol6 = Symbol.for(marker6);
+var _a6;
+_a6 = symbol6;
+var name6 = "AI_JSONParseError";
+var marker7 = `vercel.ai.error.${name6}`;
+var symbol7 = Symbol.for(marker7);
+var _a7;
+var JSONParseError = class extends AISDKError {
+  constructor({ text: text2, cause }) {
+    super({
+      name: name6,
+      message: `JSON parsing failed: Text: ${text2}.
+Error message: ${getErrorMessage(cause)}`,
+      cause
+    });
+    this[_a7] = true;
+    this.text = text2;
+  }
+  static isInstance(error45) {
+    return AISDKError.hasMarker(error45, marker7);
+  }
+};
+_a7 = symbol7;
+var name7 = "AI_LoadAPIKeyError";
+var marker8 = `vercel.ai.error.${name7}`;
+var symbol8 = Symbol.for(marker8);
+var _a8;
+_a8 = symbol8;
+var name8 = "AI_LoadSettingError";
+var marker9 = `vercel.ai.error.${name8}`;
+var symbol9 = Symbol.for(marker9);
+var _a9;
+_a9 = symbol9;
+var name9 = "AI_NoContentGeneratedError";
+var marker10 = `vercel.ai.error.${name9}`;
+var symbol10 = Symbol.for(marker10);
+var _a10;
+_a10 = symbol10;
+var name10 = "AI_NoSuchModelError";
+var marker11 = `vercel.ai.error.${name10}`;
+var symbol11 = Symbol.for(marker11);
+var _a11;
+_a11 = symbol11;
+var name11 = "AI_TooManyEmbeddingValuesForCallError";
+var marker12 = `vercel.ai.error.${name11}`;
+var symbol12 = Symbol.for(marker12);
+var _a12;
+_a12 = symbol12;
+var name12 = "AI_TypeValidationError";
+var marker13 = `vercel.ai.error.${name12}`;
+var symbol13 = Symbol.for(marker13);
+var _a13;
+var _TypeValidationError = class _TypeValidationError2 extends AISDKError {
+  constructor({ value, cause }) {
+    super({
+      name: name12,
+      message: `Type validation failed: Value: ${JSON.stringify(value)}.
+Error message: ${getErrorMessage(cause)}`,
+      cause
+    });
+    this[_a13] = true;
+    this.value = value;
+  }
+  static isInstance(error45) {
+    return AISDKError.hasMarker(error45, marker13);
+  }
+  /**
+   * Wraps an error into a TypeValidationError.
+   * If the cause is already a TypeValidationError with the same value, it returns the cause.
+   * Otherwise, it creates a new TypeValidationError.
+   *
+   * @param {Object} params - The parameters for wrapping the error.
+   * @param {unknown} params.value - The value that failed validation.
+   * @param {unknown} params.cause - The original error or cause of the validation failure.
+   * @returns {TypeValidationError} A TypeValidationError instance.
+   */
+  static wrap({
+    value,
+    cause
+  }) {
+    return _TypeValidationError2.isInstance(cause) && cause.value === value ? cause : new _TypeValidationError2({ value, cause });
+  }
+};
+_a13 = symbol13;
+var TypeValidationError = _TypeValidationError;
+var name13 = "AI_UnsupportedFunctionalityError";
+var marker14 = `vercel.ai.error.${name13}`;
+var symbol14 = Symbol.for(marker14);
+var _a14;
+_a14 = symbol14;
+
 // ../../node_modules/.pnpm/zod@4.1.9/node_modules/zod/v4/classic/external.js
 var external_exports = {};
 __export(external_exports, {
@@ -244,7 +425,7 @@ __export(external_exports, {
   stringbool: () => stringbool,
   success: () => success,
   superRefine: () => superRefine,
-  symbol: () => symbol,
+  symbol: () => symbol15,
   templateLiteral: () => templateLiteral,
   toJSONSchema: () => toJSONSchema,
   toLowerCase: () => _toLowerCase,
@@ -11819,7 +12000,7 @@ var ZodSymbol = /* @__PURE__ */ $constructor("ZodSymbol", (inst, def) => {
   $ZodSymbol.init(inst, def);
   ZodType.init(inst, def);
 });
-function symbol(params) {
+function symbol15(params) {
   return _symbol(ZodSymbol, params);
 }
 var ZodUndefined = /* @__PURE__ */ $constructor("ZodUndefined", (inst, def) => {
@@ -12462,198 +12643,6 @@ var classic_default = external_exports;
 // ../../node_modules/.pnpm/zod@4.1.9/node_modules/zod/v4/index.js
 var v4_default = classic_default;
 
-// src/lib/tool.ts
-var tool = (t) => t;
-var prepareUserTools = (toolRegistry) => Object.keys(toolRegistry).reduce(
-  (acc, key) => ({
-    ...acc,
-    [key]: {
-      name: toolRegistry[key].name ?? key,
-      description: toolRegistry[key].description,
-      inputSchema: v4_default.toJSONSchema(toolRegistry[key].inputSchema),
-      outputSchema: toolRegistry[key].outputSchema ? v4_default.toJSONSchema(toolRegistry[key].outputSchema) : null
-    }
-  }),
-  {}
-);
-
-// ../../node_modules/.pnpm/@ai-sdk+provider@2.0.0/node_modules/@ai-sdk/provider/dist/index.mjs
-var marker = "vercel.ai.error";
-var symbol2 = Symbol.for(marker);
-var _a;
-var _AISDKError = class _AISDKError2 extends Error {
-  /**
-   * Creates an AI SDK Error.
-   *
-   * @param {Object} params - The parameters for creating the error.
-   * @param {string} params.name - The name of the error.
-   * @param {string} params.message - The error message.
-   * @param {unknown} [params.cause] - The underlying cause of the error.
-   */
-  constructor({
-    name: name143,
-    message,
-    cause
-  }) {
-    super(message);
-    this[_a] = true;
-    this.name = name143;
-    this.cause = cause;
-  }
-  /**
-   * Checks if the given error is an AI SDK Error.
-   * @param {unknown} error - The error to check.
-   * @returns {boolean} True if the error is an AI SDK Error, false otherwise.
-   */
-  static isInstance(error45) {
-    return _AISDKError2.hasMarker(error45, marker);
-  }
-  static hasMarker(error45, marker153) {
-    const markerSymbol = Symbol.for(marker153);
-    return error45 != null && typeof error45 === "object" && markerSymbol in error45 && typeof error45[markerSymbol] === "boolean" && error45[markerSymbol] === true;
-  }
-};
-_a = symbol2;
-var AISDKError = _AISDKError;
-var name = "AI_APICallError";
-var marker2 = `vercel.ai.error.${name}`;
-var symbol22 = Symbol.for(marker2);
-var _a2;
-_a2 = symbol22;
-var name2 = "AI_EmptyResponseBodyError";
-var marker3 = `vercel.ai.error.${name2}`;
-var symbol3 = Symbol.for(marker3);
-var _a3;
-_a3 = symbol3;
-function getErrorMessage(error45) {
-  if (error45 == null) {
-    return "unknown error";
-  }
-  if (typeof error45 === "string") {
-    return error45;
-  }
-  if (error45 instanceof Error) {
-    return error45.message;
-  }
-  return JSON.stringify(error45);
-}
-var name3 = "AI_InvalidArgumentError";
-var marker4 = `vercel.ai.error.${name3}`;
-var symbol4 = Symbol.for(marker4);
-var _a4;
-var InvalidArgumentError = class extends AISDKError {
-  constructor({
-    message,
-    cause,
-    argument
-  }) {
-    super({ name: name3, message, cause });
-    this[_a4] = true;
-    this.argument = argument;
-  }
-  static isInstance(error45) {
-    return AISDKError.hasMarker(error45, marker4);
-  }
-};
-_a4 = symbol4;
-var name4 = "AI_InvalidPromptError";
-var marker5 = `vercel.ai.error.${name4}`;
-var symbol5 = Symbol.for(marker5);
-var _a5;
-_a5 = symbol5;
-var name5 = "AI_InvalidResponseDataError";
-var marker6 = `vercel.ai.error.${name5}`;
-var symbol6 = Symbol.for(marker6);
-var _a6;
-_a6 = symbol6;
-var name6 = "AI_JSONParseError";
-var marker7 = `vercel.ai.error.${name6}`;
-var symbol7 = Symbol.for(marker7);
-var _a7;
-var JSONParseError = class extends AISDKError {
-  constructor({ text: text2, cause }) {
-    super({
-      name: name6,
-      message: `JSON parsing failed: Text: ${text2}.
-Error message: ${getErrorMessage(cause)}`,
-      cause
-    });
-    this[_a7] = true;
-    this.text = text2;
-  }
-  static isInstance(error45) {
-    return AISDKError.hasMarker(error45, marker7);
-  }
-};
-_a7 = symbol7;
-var name7 = "AI_LoadAPIKeyError";
-var marker8 = `vercel.ai.error.${name7}`;
-var symbol8 = Symbol.for(marker8);
-var _a8;
-_a8 = symbol8;
-var name8 = "AI_LoadSettingError";
-var marker9 = `vercel.ai.error.${name8}`;
-var symbol9 = Symbol.for(marker9);
-var _a9;
-_a9 = symbol9;
-var name9 = "AI_NoContentGeneratedError";
-var marker10 = `vercel.ai.error.${name9}`;
-var symbol10 = Symbol.for(marker10);
-var _a10;
-_a10 = symbol10;
-var name10 = "AI_NoSuchModelError";
-var marker11 = `vercel.ai.error.${name10}`;
-var symbol11 = Symbol.for(marker11);
-var _a11;
-_a11 = symbol11;
-var name11 = "AI_TooManyEmbeddingValuesForCallError";
-var marker12 = `vercel.ai.error.${name11}`;
-var symbol12 = Symbol.for(marker12);
-var _a12;
-_a12 = symbol12;
-var name12 = "AI_TypeValidationError";
-var marker13 = `vercel.ai.error.${name12}`;
-var symbol13 = Symbol.for(marker13);
-var _a13;
-var _TypeValidationError = class _TypeValidationError2 extends AISDKError {
-  constructor({ value, cause }) {
-    super({
-      name: name12,
-      message: `Type validation failed: Value: ${JSON.stringify(value)}.
-Error message: ${getErrorMessage(cause)}`,
-      cause
-    });
-    this[_a13] = true;
-    this.value = value;
-  }
-  static isInstance(error45) {
-    return AISDKError.hasMarker(error45, marker13);
-  }
-  /**
-   * Wraps an error into a TypeValidationError.
-   * If the cause is already a TypeValidationError with the same value, it returns the cause.
-   * Otherwise, it creates a new TypeValidationError.
-   *
-   * @param {Object} params - The parameters for wrapping the error.
-   * @param {unknown} params.value - The value that failed validation.
-   * @param {unknown} params.cause - The original error or cause of the validation failure.
-   * @returns {TypeValidationError} A TypeValidationError instance.
-   */
-  static wrap({
-    value,
-    cause
-  }) {
-    return _TypeValidationError2.isInstance(cause) && cause.value === value ? cause : new _TypeValidationError2({ value, cause });
-  }
-};
-_a13 = symbol13;
-var TypeValidationError = _TypeValidationError;
-var name13 = "AI_UnsupportedFunctionalityError";
-var marker14 = `vercel.ai.error.${name13}`;
-var symbol14 = Symbol.for(marker14);
-var _a14;
-_a14 = symbol14;
-
 // ../../node_modules/.pnpm/zod@4.1.9/node_modules/zod/v3/helpers/util.js
 var util;
 (function(util2) {
@@ -17866,14 +17855,14 @@ var __export2 = (target, all) => {
 };
 var name14 = "AI_NoOutputSpecifiedError";
 var marker15 = `vercel.ai.error.${name14}`;
-var symbol15 = Symbol.for(marker15);
+var symbol16 = Symbol.for(marker15);
 var _a15;
-_a15 = symbol15;
+_a15 = symbol16;
 var name22 = "AI_InvalidArgumentError";
 var marker22 = `vercel.ai.error.${name22}`;
-var symbol23 = Symbol.for(marker22);
+var symbol22 = Symbol.for(marker22);
 var _a22;
-_a22 = symbol23;
+_a22 = symbol22;
 var name32 = "AI_InvalidStreamPartError";
 var marker32 = `vercel.ai.error.${name32}`;
 var symbol32 = Symbol.for(marker32);
@@ -19270,9 +19259,9 @@ var object2 = ({
 };
 var name16 = "AI_NoSuchProviderError";
 var marker16 = `vercel.ai.error.${name16}`;
-var symbol16 = Symbol.for(marker16);
+var symbol162 = Symbol.for(marker16);
 var _a16;
-_a16 = symbol16;
+_a16 = symbol162;
 var ClientOrServerImplementationSchema = external_exports.looseObject({
   name: external_exports.string(),
   version: external_exports.string()
@@ -19625,8 +19614,64 @@ function createUIMessageStream({
   });
 }
 
-// src/lib/chat.ts
-async function* iterSSE(body) {
+// src/lib/prepare-user-tools.ts
+var prepareUserTools = (toolRegistry) => Object.keys(toolRegistry).reduce(
+  (acc, key) => ({
+    ...acc,
+    [key]: {
+      name: toolRegistry[key].name ?? key,
+      description: toolRegistry[key].description,
+      inputSchema: v4_default.toJSONSchema(toolRegistry[key].inputSchema),
+      outputSchema: toolRegistry[key].outputSchema ? v4_default.toJSONSchema(toolRegistry[key].outputSchema) : null
+    }
+  }),
+  {}
+);
+
+// src/lib/get-api-key.ts
+var getApiKey = () => process.env.PUCK_API_KEY;
+
+// src/lib/process-user-tool.ts
+var processUserTool = async ({
+  data,
+  tools,
+  options
+}) => {
+  const { toolCallId, tools: requestTools } = data;
+  const { apiKey = getApiKey(), host = "https://cloud.puckeditor.com/api" } = options;
+  if (!apiKey) {
+    throw new Error(
+      "No Puck API key specified. Set the PUCK_API_KEY environment variable, or provide one"
+    );
+  }
+  const responses = [];
+  for (let i = 0; i < requestTools.length; i++) {
+    const { name: name17, input } = requestTools[i];
+    const tool2 = tools?.[name17];
+    if (!tool2) {
+      throw new Error(`An unknown tool was called: ${name17}`);
+    }
+    if (!input) {
+      throw new Error(`Input was not present in tool call: ${toolCallId}`);
+    }
+    const output = await tool2.execute(input);
+    responses.push({ name: name17, input, output });
+  }
+  const res = await fetch(`${host}/tool`, {
+    headers: { "x-api-key": apiKey },
+    method: "post",
+    body: JSON.stringify({
+      id: toolCallId,
+      responses
+    })
+  });
+  if (res.status !== 200) {
+    throw new Error("Tool result could not be sent to server");
+  }
+};
+
+// src/lib/iterate-sse.ts
+async function* iterateSSE(body) {
   const reader = body.getReader();
   const dec = new TextDecoder();
   let buf = "";
@@ -19647,81 +19692,84 @@ async function* iterSSE(body) {
     }
   }
 }
-var getApiKey = () => process.env.PUCK_API_KEY;
-function chat({ chatId, messages: initialMessages, config: config2, pageData }, {
-  ai = {},
-  apiKey = getApiKey(),
-  host = "https://cloud.puckeditor.com/api"
-}) {
+
+// src/lib/cloud-api.ts
+var cloudApi = async (path, body, options = {}, onChunk) => {
+  const {
+    ai = {},
+    apiKey = getApiKey(),
+    host = "https://cloud.puckeditor.com/api"
+  } = options;
   if (!apiKey) {
     throw new Error(
-      `No Puck API key specified. Set the PUCK_API_KEY environment variable, or provide one to the function.
-
-Visit https://cloud.puckeditor.com/sign-up to register for an account and generate a key.`
+      "No Puck API key specified. Set the PUCK_API_KEY environment variable, or provide one to the function"
     );
   }
   const { context, tools = {} } = ai;
-  let messages = initialMessages;
+  const res = await fetch(`${host}/${path}`, {
+    headers: { "x-api-key": apiKey },
+    method: "post",
+    body: JSON.stringify({
+      ...body,
+      context,
+      tools: prepareUserTools(tools)
+    })
+  });
+  if (!res.body) {
+    throw new Error(`Puck ${res.status} (${res.statusText})`);
+  } else if (!res.ok) {
+    const body2 = await res.json();
+    throw new Error(
+      `Puck ${res.status} (${res.statusText}): ${body2.error ?? "Unknown reason"}`
+    );
+  }
+  for await (const chunk of iterateSSE(res.body)) {
+    if (chunk.type === "data-user-tool") {
+      await processUserTool({ data: chunk.data, tools, options });
+    } else {
+      onChunk(chunk);
+    }
+  }
+  return res;
+};
+
+// src/api/chat.ts
+function chat({ chatId, messages, config: config2, pageData }, options = {}) {
   const stream = createUIMessageStream({
     execute: async ({ writer }) => {
-      const res = await fetch(`${host}/chat`, {
-        headers: { "x-api-key": apiKey },
-        method: "post",
-        body: JSON.stringify({
-          chatId,
-          context,
-          messages,
-          config: config2,
-          pageData,
-          tools: prepareUserTools(tools)
-        })
-      });
-      if (!res.body) {
-        throw new Error(`Puck ${res.status} (${res.statusText})`);
-      } else if (!res.ok) {
-        const body = await res.json();
-        throw new Error(
-          `Puck ${res.status} (${res.statusText}): ${body.error ?? "Unknown reason"}`
-        );
-      }
-      for await (const chunk of iterSSE(res.body)) {
-        if (chunk.type === "data-user-tool") {
-          const { toolCallId, tools: requestTools } = chunk.data;
-          const responses = [];
-          for (let i = 0; i < requestTools.length; i++) {
-            const { name: name17, input } = requestTools[i];
-            const tool2 = tools?.[name17];
-            if (!tool2) {
-              throw new Error(`An unknown tool was called: ${name17}`);
-            }
-            if (!input) {
-              throw new Error(
-                `Input was not present in tool call: ${toolCallId}`
-              );
-            }
-            const output = await tool2.execute(input);
-            responses.push({ name: name17, input, output });
-          }
-          const res2 = await fetch(`${host}/chat/tool`, {
-            headers: { "x-api-key": apiKey },
-            method: "post",
-            body: JSON.stringify({
-              id: toolCallId,
-              responses
-            })
-          });
-          if (res2.status !== 200) {
-            throw new Error("Tool result could not be sent to server");
-          }
-        } else {
+      await cloudApi(
+        "chat",
+        { chatId, config: config2, messages, pageData },
+        options,
+        (chunk) => {
           writer.write(chunk);
         }
-      }
+      );
     }
   });
   return createUIMessageStreamResponse({ stream });
 }
 
+// src/api/generate.ts
+async function generate({
+  prompt,
+  config: config2,
+  pageData,
+  context,
+  tools,
+  apiKey,
+  host
+}) {
+  let result = null;
+  const options = { ai: { context, tools }, apiKey, host };
+  await cloudApi("generate", { prompt, config: config2, pageData }, options, (chunk) => {
+    if (chunk.type === "data-page") {
+      result = chunk.data;
+    }
+  });
+  return result;
+}
+
 // src/routes.ts
 var routeRegistry = {
   "/api/puck/chat": {
@@ -19732,7 +19780,7 @@ var findRoute = (pathname, method) => {
   return routeRegistry[pathname][method];
 };
 
-// src/lib/puck-handler.ts
+// src/api/puck-handler.ts
 async function puckHandler(request, options) {
   const cloned = request.clone();
   const url2 = new URL(cloned.url);
@@ -19759,6 +19807,7 @@ var endpoints = ["chat"];
 0 && (module.exports = {
   chat,
   endpoints,
+  generate,
   puckHandler,
   tool
 });