ilikelollies/agent-catalog/utils/buildPrompt.ts

import { Message } from "@/features/chat/messages";

export function buildPrompt(config: {name: string, system_prompt: string}, messages: Message[]) {
  let prompt = "";
  for (let m of messages) {
    switch(m.role) {
      case 'system':
        prompt += config.system_prompt+"\n\n";
        break;
      case 'user':
        prompt += `User: ${m.content}\n`;
        break;
      case 'assistant':
        prompt += `${config.name}: ${m.content}\n`;
        break;
    }
  }
  prompt += `${config.name}:`;
  return prompt;
}

export function buildVisionPrompt(config: {name: string, vision_system_prompt: string},messages: Message[]) {
  let prompt = "";
  for (let m of messages) {
    switch(m.role) {
      case 'system':
        prompt += config.vision_system_prompt+"\n\n";
        break;
      case 'user':
        prompt += `User: ${m.content}\n`;
        break;
      case 'assistant':
        prompt += `${config.name}: ${m.content}\n`;
        break;
    }
  }
  prompt += `${config.name}:`;
  return prompt;
}