import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";

const OPENAI_APIS = new Set(["openai-completions", "openai-responses"]);
const TEMPERATURE = 0.2;
const TOP_P = 0.9;

export default function (pi: ExtensionAPI) {
  pi.on("before_provider_request", (event, ctx) => {
    if (!ctx.model || !OPENAI_APIS.has(ctx.model.api)) {
      return;
    }
    if (!event.payload || typeof event.payload !== "object" || Array.isArray(event.payload)) {
      return;
    }

    return {
      ...(event.payload as Record<string, unknown>),
      temperature: TEMPERATURE,
      top_p: TOP_P,
    };
  });
}