import { streamText, type ModelMessage } from "ai";
import { providersRegistry } from "../providers";
import { logger } from "../utils/logger";

export const beautifyAgent = async (
  {
    model,
    messages,
  }: {
    model: string;
    messages: ModelMessage[];
  },
  options: { headers?: Record<string, string> } = {}
) => {
  const [providerName = "openai", modelName = "gpt-4o-mini"] =
    model.split(":::");
  const aiModel = await providersRegistry.getModel(providerName, modelName);

  return streamText({
    ...options,
    model: aiModel,
    messages: [
      {
        role: "system",
        content:
          "Use provided messages to generate a better implementation-focused prompt.  Limit the response to 300 symbols, make it actionable for development, and use plain text only. Prioritize clarity and practical instructions.",
      },
      messages[messages.length - 1] as any,
    ],
    onError: (err) => {
      logger.dump(err);
    },
  });
};