src/lib/agents/promptGeneration.ts

import type { Tool } from './tools/tool';

function toolDescription(tool: Tool<any, any>) {
	let prompt = `	name: ${tool.name} \n	description: ${tool.description}`;

	const examples = tool.examples.slice(0, 1).map((example) => {
		return `	prompt: ${example.prompt} \n	command generated: \`${example.command}\``;
	});

	prompt += `\n` + examples.join('\n');

	return prompt;
}

export function generatePrompt(
	prompt: string,
	tools: Tool<any, any>[],
	image?: boolean,
	audio?: boolean
) {
	if (tools.length === 0) {
		throw new Error('no tools selected');
	}

	let params = '';

	if (image) {
		params += `image`;
	}
	if (audio) {
		params += params ? ',' : '';
		params += `audio`;
	}

	// describe all the tools
	const fullPrompt = `
Create a function that does the following: ${prompt}. 

Examples:
For the prompt: "Caption the image and give me the caption read out loud."
\`\`\`js
async function generate(image) {
	const caption = await imageToText(image);
	message("First we caption the image", caption);
	const output = await textToSpeech(caption);
	message("Then we read the caption out loud", output);
	return output;
};
\`\`\`

For the prompt "Display an image of a yellow dog wearing a top hat"
\`\`\`js
async function generate() {
	const output = await textToImage("yellow dog wearing a top hat");
	message("We generate the dog picture", output);
	return output;
}
\`\`\`

For the prompt "transcribe the attached audio"

\`\`\`js
async function generate(audio) {
	const output = await speechToText(audio)
	message("We read the text", output);
	return output;
}
\`\`\`

In order to help in answering the above prompt, the function has access to the following methods to generate outputs.
${tools.map((tool) => toolDescription(tool)).join('\n\n ')}

Use the above methods and only the above methods to answer the prompt: ${prompt}.

It must match the following signature:
\`\`\`js
async function generate(${params}}) {
// your code here
return output;
};
\`\`\``;

	return fullPrompt;
}