using XFEExtension.NetCore.XFEChatGPT.ChatGPTInnerClass.DefaultClass;

namespace XFEExtension.NetCore.XFEChatGPT;

/// <summary>
/// XFEChatGPT的扩展方法
/// </summary>
public static class XFEChatGPTExtension
{
    /// <summary>
    /// 检查是否合法（system在最前，接着是user，然后是Assistant，以此类推...，不能连着2个user或者Assistant）
    /// </summary>
    /// <returns>是否合法</returns>
    public static bool CheckLegal(this GPTMessage[] xFEGPTMessages)
    {
        if (xFEGPTMessages.Length == 0)
        {
            return true;
        }

        if (xFEGPTMessages[0].Role != "system")
        {
            return false;
        }

        var lastRole = xFEGPTMessages[0].Role;
        for (var i = 1; i < xFEGPTMessages.Length; i++)
        {
            if (xFEGPTMessages[i].Role == lastRole)
            {
                return false;
            }

            lastRole = xFEGPTMessages[i].Role;
        }
        return true;
    }
}