package cn.iocoder.yudao.module.qcreport.service.aiimport.llm;

/**
 * 一次待执行的大模型调用。
 * <p>
 * 由 {@link QcReportLlmCallPlanner} 纯函数算出，交给薄执行器真正发出去。
 * 拆成「先算计划、再执行」是为了让多页 / 多文件 / 超预算这些控制流可以被单测完整覆盖——
 * 否则只能靠 mock 掉 {@code AiChatApi} 才能测，那测的是 mock 不是逻辑。
 * <p>
 * 每个调用自带全部入参（含 TEXT 通道的正文），执行器不需要回看抽取结果。
 *
 * @param kind        调用方式，决定走 {@code chat} 还是 {@code chatWithImage}
 * @param sourceLabel 来源标签，形如「扫描件.pdf 第 2 页」，用于软提示与错误文案
 * @param text        TEXT 通道的正文；IMAGE 通道恒为空串
 * @param imageBase64 IMAGE 通道的**裸 base64**（不含 {@code data:image/png;base64,} 前缀，
 *                    因为 {@code AiChatApiImpl} 直接对它做 Base64 解码）；TEXT 通道恒为空串
 * @param mimeType    IMAGE 通道的裸 MIME 类型，如 {@code image/png}
 */
public record QcReportLlmCall(
        Kind kind,
        String sourceLabel,
        String text,
        String imageBase64,
        String mimeType) {

    /**
     * 调用方式。
     */
    public enum Kind {
        /** 纯文本，一次调用搞定整份文件 */
        TEXT,
        /** 多模态，一次调用只认一张图，所以逐页一调 */
        IMAGE
    }

    public static QcReportLlmCall ofText(String sourceLabel, String text) {
        return new QcReportLlmCall(Kind.TEXT, sourceLabel, text, "", "");
    }

    public static QcReportLlmCall ofImage(String sourceLabel, String imageBase64, String mimeType) {
        return new QcReportLlmCall(Kind.IMAGE, sourceLabel, "", imageBase64, mimeType);
    }

}