import type { Inputs, Tool } from "../types";
import examples from "./examples";

function toolDescription(tool: Tool) {
	let prompt = `name: ${tool.name} \ndescription: ${tool.description}`;

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

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

	return prompt;
}

export function generatePrompt(prompt: string, tools: Tool[], inputs: Inputs): string {
	if (tools.length === 0) {
		throw new Error("no tools selected");
	}

	let params = "";

	if (inputs.image) {
		params += `image`;
	}
	if (inputs.audio) {
		params += params ? "," : "";
		params += `audio`;
	}

	const exampleSnippet = examples
		.map(
			(example) => `
user: ${example.prompt}
function: 
\`\`\`js
${example.code}
\`\`\``
		)
		.join("\n-------\n");

	const toolSnippet = tools.map((tool) => toolDescription(tool)).join("\n-------\n");

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

In order to help in answering the above prompt, the function has access to the following methods to generate outputs.

${toolSnippet}

Examples:
${exampleSnippet}

If you need to send information to the user use \`message("message", data)\` and NOT \`console.log\`.

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

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

If you are not sure how to do it, try anyway and give it your best shot, as the user can always correct you later.`;

	return fullPrompt;
}