ai 3.0.15 → 3.0.16

package/dist/index.mjs

@@ -94,6 +94,26 @@ var InvalidDataContentError = class extends Error {
   }
 };
 
+// spec/errors/invalid-prompt-error.ts
+var InvalidPromptError = class extends Error {
+  constructor({ prompt: prompt2, message }) {
+    super(`Invalid prompt: ${message}`);
+    this.name = "AI_InvalidPromptError";
+    this.prompt = prompt2;
+  }
+  static isInvalidPromptError(error) {
+    return error instanceof Error && error.name === "AI_InvalidPromptError" && prompt != null;
+  }
+  toJSON() {
+    return {
+      name: this.name,
+      message: this.message,
+      stack: this.stack,
+      prompt: this.prompt
+    };
+  }
+};
+
 // spec/util/get-error-message.ts
 function getErrorMessage(error) {
   if (error == null) {
@@ -355,89 +375,101 @@ function convertDataContentToUint8Array(content) {
 }
 
 // core/prompt/convert-to-language-model-prompt.ts
-function convertToLanguageModelPrompt({
-  system,
-  prompt,
-  messages
-}) {
-  if (prompt == null && messages == null) {
-    throw new Error("prompt or messages must be defined");
-  }
-  if (prompt != null && messages != null) {
-    throw new Error("prompt and messages cannot be defined at the same time");
-  }
+function convertToLanguageModelPrompt(prompt2) {
   const languageModelMessages = [];
-  if (system != null) {
-    languageModelMessages.push({ role: "system", content: system });
-  }
-  if (typeof prompt === "string") {
-    languageModelMessages.push({
-      role: "user",
-      content: [{ type: "text", text: prompt }]
-    });
-  } else {
-    messages = messages;
-    languageModelMessages.push(
-      ...messages.map((message) => {
-        switch (message.role) {
-          case "user": {
-            if (typeof message.content === "string") {
+  if (prompt2.system != null) {
+    languageModelMessages.push({ role: "system", content: prompt2.system });
+  }
+  switch (prompt2.type) {
+    case "prompt": {
+      languageModelMessages.push({
+        role: "user",
+        content: [{ type: "text", text: prompt2.prompt }]
+      });
+      break;
+    }
+    case "messages": {
+      languageModelMessages.push(
+        ...prompt2.messages.map((message) => {
+          switch (message.role) {
+            case "user": {
+              if (typeof message.content === "string") {
+                return {
+                  role: "user",
+                  content: [{ type: "text", text: message.content }]
+                };
+              }
               return {
                 role: "user",
-                content: [{ type: "text", text: message.content }]
-              };
-            }
-            return {
-              role: "user",
-              content: message.content.map(
-                (part) => {
-                  switch (part.type) {
-                    case "text": {
-                      return part;
-                    }
-                    case "image": {
-                      return {
-                        type: "image",
-                        image: part.image instanceof URL ? part.image : convertDataContentToUint8Array(part.image),
-                        mimeType: part.mimeType
-                      };
+                content: message.content.map(
+                  (part) => {
+                    switch (part.type) {
+                      case "text": {
+                        return part;
+                      }
+                      case "image": {
+                        return {
+                          type: "image",
+                          image: part.image instanceof URL ? part.image : convertDataContentToUint8Array(part.image),
+                          mimeType: part.mimeType
+                        };
+                      }
                     }
                   }
-                }
-              )
-            };
-          }
-          case "assistant": {
-            if (typeof message.content === "string") {
-              return {
-                role: "assistant",
-                content: [{ type: "text", text: message.content }]
+                )
               };
             }
-            return { role: "assistant", content: message.content };
-          }
-          case "tool": {
-            return message;
+            case "assistant": {
+              if (typeof message.content === "string") {
+                return {
+                  role: "assistant",
+                  content: [{ type: "text", text: message.content }]
+                };
+              }
+              return { role: "assistant", content: message.content };
+            }
+            case "tool": {
+              return message;
+            }
           }
-        }
-      })
-    );
+        })
+      );
+      break;
+    }
+    default: {
+      const _exhaustiveCheck = prompt2;
+      throw new Error(`Unsupported prompt type: ${_exhaustiveCheck}`);
+    }
   }
   return languageModelMessages;
 }
 
-// core/prompt/get-input-format.ts
-function getInputFormat({
-  prompt,
-  messages
-}) {
-  if (prompt == null && messages == null) {
-    throw new Error("prompt or messages must be defined");
+// core/prompt/get-validated-prompt.ts
+function getValidatedPrompt(prompt2) {
+  if (prompt2.prompt == null && prompt2.messages == null) {
+    throw new InvalidPromptError({
+      prompt: prompt2,
+      message: "prompt or messages must be defined"
+    });
   }
-  if (prompt != null && messages != null) {
-    throw new Error("prompt and messages cannot be defined at the same time");
+  if (prompt2.prompt != null && prompt2.messages != null) {
+    throw new InvalidPromptError({
+      prompt: prompt2,
+      message: "prompt and messages cannot be defined at the same time"
+    });
   }
-  return prompt != null ? "prompt" : "messages";
+  return prompt2.prompt != null ? {
+    type: "prompt",
+    prompt: prompt2.prompt,
+    messages: void 0,
+    system: prompt2.system
+  } : {
+    type: "messages",
+    prompt: void 0,
+    messages: prompt2.messages,
+    // only possible case bc of checks above
+    system: prompt2.system
+  };
 }
 
 // core/prompt/prepare-call-settings.ts
@@ -649,7 +681,7 @@ async function experimental_generateObject({
   schema,
   mode,
   system,
-  prompt,
+  prompt: prompt2,
   messages,
   maxRetries,
   abortSignal,
@@ -667,19 +699,20 @@ async function experimental_generateObject({
   let warnings;
   switch (mode) {
     case "json": {
-      const generateResult = await retry(
-        () => model.doGenerate({
+      const validatedPrompt = getValidatedPrompt({
+        system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+        prompt: prompt2,
+        messages
+      });
+      const generateResult = await retry(() => {
+        return model.doGenerate({
           mode: { type: "object-json" },
           ...prepareCallSettings(settings),
-          inputFormat: getInputFormat({ prompt, messages }),
-          prompt: convertToLanguageModelPrompt({
-            system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
-            prompt,
-            messages
-          }),
+          inputFormat: validatedPrompt.type,
+          prompt: convertToLanguageModelPrompt(validatedPrompt),
           abortSignal
-        })
-      );
+        });
+      });
       if (generateResult.text === void 0) {
         throw new NoTextGeneratedError();
       }
@@ -690,16 +723,17 @@ async function experimental_generateObject({
       break;
     }
     case "grammar": {
+      const validatedPrompt = getValidatedPrompt({
+        system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+        prompt: prompt2,
+        messages
+      });
       const generateResult = await retry(
         () => model.doGenerate({
           mode: { type: "object-grammar", schema: jsonSchema },
           ...settings,
-          inputFormat: getInputFormat({ prompt, messages }),
-          prompt: convertToLanguageModelPrompt({
-            system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
-            prompt,
-            messages
-          }),
+          inputFormat: validatedPrompt.type,
+          prompt: convertToLanguageModelPrompt(validatedPrompt),
           abortSignal
         })
       );
@@ -713,6 +747,11 @@ async function experimental_generateObject({
       break;
     }
     case "tool": {
+      const validatedPrompt = getValidatedPrompt({
+        system,
+        prompt: prompt2,
+        messages
+      });
       const generateResult = await retry(
         () => model.doGenerate({
           mode: {
@@ -725,8 +764,8 @@ async function experimental_generateObject({
             }
           },
           ...settings,
-          inputFormat: getInputFormat({ prompt, messages }),
-          prompt: convertToLanguageModelPrompt({ system, prompt, messages }),
+          inputFormat: validatedPrompt.type,
+          prompt: convertToLanguageModelPrompt(validatedPrompt),
           abortSignal
         })
       );
@@ -1166,7 +1205,7 @@ async function experimental_streamObject({
   schema,
   mode,
   system,
-  prompt,
+  prompt: prompt2,
   messages,
   maxRetries,
   abortSignal,
@@ -1181,15 +1220,16 @@ async function experimental_streamObject({
   let transformer;
   switch (mode) {
     case "json": {
+      const validatedPrompt = getValidatedPrompt({
+        system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+        prompt: prompt2,
+        messages
+      });
       callOptions = {
         mode: { type: "object-json" },
         ...prepareCallSettings(settings),
-        inputFormat: getInputFormat({ prompt, messages }),
-        prompt: convertToLanguageModelPrompt({
-          system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
-          prompt,
-          messages
-        }),
+        inputFormat: validatedPrompt.type,
+        prompt: convertToLanguageModelPrompt(validatedPrompt),
         abortSignal
       };
       transformer = {
@@ -1207,15 +1247,16 @@ async function experimental_streamObject({
       break;
     }
     case "grammar": {
+      const validatedPrompt = getValidatedPrompt({
+        system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
+        prompt: prompt2,
+        messages
+      });
       callOptions = {
         mode: { type: "object-grammar", schema: jsonSchema },
         ...settings,
-        inputFormat: getInputFormat({ prompt, messages }),
-        prompt: convertToLanguageModelPrompt({
-          system: injectJsonSchemaIntoSystem({ system, schema: jsonSchema }),
-          prompt,
-          messages
-        }),
+        inputFormat: validatedPrompt.type,
+        prompt: convertToLanguageModelPrompt(validatedPrompt),
         abortSignal
       };
       transformer = {
@@ -1233,6 +1274,11 @@ async function experimental_streamObject({
       break;
     }
     case "tool": {
+      const validatedPrompt = getValidatedPrompt({
+        system,
+        prompt: prompt2,
+        messages
+      });
       callOptions = {
         mode: {
           type: "object-tool",
@@ -1244,8 +1290,8 @@ async function experimental_streamObject({
           }
         },
         ...settings,
-        inputFormat: getInputFormat({ prompt, messages }),
-        prompt: convertToLanguageModelPrompt({ system, prompt, messages }),
+        inputFormat: validatedPrompt.type,
+        prompt: convertToLanguageModelPrompt(validatedPrompt),
         abortSignal
       };
       transformer = {
@@ -1352,7 +1398,7 @@ async function experimental_generateText({
   model,
   tools,
   system,
-  prompt,
+  prompt: prompt2,
   messages,
   maxRetries,
   abortSignal,
@@ -1360,8 +1406,9 @@ async function experimental_generateText({
 }) {
   var _a, _b;
   const retry = retryWithExponentialBackoff({ maxRetries });
-  const modelResponse = await retry(
-    () => model.doGenerate({
+  const validatedPrompt = getValidatedPrompt({ system, prompt: prompt2, messages });
+  const modelResponse = await retry(() => {
+    return model.doGenerate({
       mode: {
         type: "regular",
         tools: tools == null ? void 0 : Object.entries(tools).map(([name, tool2]) => ({
@@ -1372,15 +1419,11 @@ async function experimental_generateText({
         }))
       },
       ...prepareCallSettings(settings),
-      inputFormat: getInputFormat({ prompt, messages }),
-      prompt: convertToLanguageModelPrompt({
-        system,
-        prompt,
-        messages
-      }),
+      inputFormat: validatedPrompt.type,
+      prompt: convertToLanguageModelPrompt(validatedPrompt),
       abortSignal
-    })
-  );
+    });
+  });
   const toolCalls = [];
   for (const modelToolCall of (_a = modelResponse.toolCalls) != null ? _a : []) {
     toolCalls.push(parseToolCall({ toolCall: modelToolCall, tools }));
@@ -1589,13 +1632,14 @@ async function experimental_streamText({
   model,
   tools,
   system,
-  prompt,
+  prompt: prompt2,
   messages,
   maxRetries,
   abortSignal,
   ...settings
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
+  const validatedPrompt = getValidatedPrompt({ system, prompt: prompt2, messages });
   const { stream, warnings } = await retry(
     () => model.doStream({
       mode: {
@@ -1608,12 +1652,8 @@ async function experimental_streamText({
         }))
       },
       ...prepareCallSettings(settings),
-      inputFormat: getInputFormat({ prompt, messages }),
-      prompt: convertToLanguageModelPrompt({
-        system,
-        prompt,
-        messages
-      }),
+      inputFormat: validatedPrompt.type,
+      prompt: convertToLanguageModelPrompt(validatedPrompt),
       abortSignal
     })
   );