ai/prompts/dist/index.mjs

Vercel AI SDK - The AI Toolkit for TypeScript and JavaScript

// prompts/anthropic.ts
function experimental_buildAnthropicPrompt(messages) {
  return messages.map(({ content, role }) => {
    if (role === "user") {
      return `

Human: ${content}`;
    } else {
      return `

Assistant: ${content}`;
    }
  }) + "\n\nAssistant:";
}
function experimental_buildAnthropicMessages(messages) {
  return messages.map(({ content, role }) => {
    if (!["assistant", "user"].includes(role)) {
      throw new Error(`Cannot use ${role} on Anthropic V3 Messages API`);
    }
    return {
      role,
      content: [{ type: "text", text: content }]
    };
  });
}

// prompts/huggingface.ts
function experimental_buildStarChatBetaPrompt(messages) {
  return messages.map(({ content, role }) => {
    if (role === "user") {
      return `<|user|>
${content}<|end|>
`;
    } else if (role === "assistant") {
      return `<|assistant|>
${content}<|end|>
`;
    } else if (role === "system") {
      return `<|system|>
${content}<|end|>
`;
    } else if (role === "function") {
      throw new Error("StarChat Beta does not support function calls.");
    }
  }).join("") + "<|assistant|>";
}
function experimental_buildOpenAssistantPrompt(messages) {
  return messages.map(({ content, role }) => {
    if (role === "user") {
      return `<|prompter|>${content}<|endoftext|>`;
    } else if (role === "function") {
      throw new Error("OpenAssistant does not support function calls.");
    } else if (role === "system") {
      throw new Error("OpenAssistant does not support system messages.");
    } else {
      return `<|assistant|>${content}<|endoftext|>`;
    }
  }).join("") + "<|assistant|>";
}
function experimental_buildLlama2Prompt(messages) {
  const startPrompt = `<s>[INST] `;
  const endPrompt = ` [/INST]`;
  const conversation = messages.map(({ content, role }, index) => {
    if (role === "user") {
      return content.trim();
    } else if (role === "assistant") {
      return ` [/INST] ${content}</s><s>[INST] `;
    } else if (role === "function") {
      throw new Error("Llama 2 does not support function calls.");
    } else if (role === "system" && index === 0) {
      return `<<SYS>>
${content}
<</SYS>>

`;
    } else {
      throw new Error(`Invalid message role: ${role}`);
    }
  });
  return startPrompt + conversation.join("") + endPrompt;
}

// prompts/openai.tsx
function experimental_buildOpenAIMessages(messages) {
  return messages.map((message) => {
    switch (message.role) {
      case "system":
      case "user":
        return {
          role: message.role,
          content: message.content
        };
      case "assistant": {
        const function_call = message.function_call;
        if (function_call !== void 0 && (typeof function_call === "string" || function_call.arguments === void 0 || function_call.name === void 0)) {
          throw new Error(
            "Invalid function call in message. Expected a function call object"
          );
        }
        return {
          role: message.role,
          content: message.content,
          function_call: function_call === void 0 ? void 0 : {
            name: function_call.name,
            arguments: function_call.arguments
          }
        };
      }
      case "function": {
        if (message.name === void 0) {
          throw new Error("Invalid function call in message. Expected a name");
        }
        return {
          role: message.role,
          content: message.content,
          name: message.name
        };
      }
      case "data": {
        throw "unsupported message role 'data'";
      }
      case "tool": {
        if (message.name === void 0) {
          throw new Error("Invalid tool message. Expected a name");
        }
        if (message.tool_call_id === void 0) {
          throw new Error("Invalid tool message. Expected a tool_call_id");
        }
        return {
          role: message.role,
          content: message.content,
          tool_call_id: message.tool_call_id
        };
      }
    }
  });
}
export {
  experimental_buildAnthropicMessages,
  experimental_buildAnthropicPrompt,
  experimental_buildLlama2Prompt,
  experimental_buildOpenAIMessages,
  experimental_buildOpenAssistantPrompt,
  experimental_buildStarChatBetaPrompt
};
//# sourceMappingURL=index.mjs.map