import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
import z from 'zod';

import { workflowNamingPromptTemplate } from '@/prompts/chains/workflow-name.prompt';

export async function workflowNameChain(llm: BaseChatModel, initialPrompt: string) {
	// Use structured output for the workflow name to ensure it meets the required format and length
	const nameSchema = z.object({
		name: z.string().min(10).max(128).describe('Name of the workflow based on the prompt'),
	});

	const modelWithStructure = llm.withStructuredOutput(nameSchema);

	const prompt = await workflowNamingPromptTemplate.invoke({
		initialPrompt,
	});

	const structuredOutput = (await modelWithStructure.invoke(prompt)) as z.infer<typeof nameSchema>;

	return {
		name: structuredOutput.name,
	};
}