PsChat

0.0.39

Classes/Extensions/PreLoad.psm1

using module "..\Dialog.psm1"
using module "..\..\Private\OutHelper.psm1"

class PreLoad {
    [string]$Prompt
    [string]$Path
    [string]$Role = "user"
    [bool]$Lock # ensures that the initial messages are always the first in the dialog
    [DialogMessage[]]$InitialMessages

    [Dialog] BeforeAnswer([Dialog]$dialog) {
        if(!$this.Lock) { return $dialog }

        # remove any "locked" messages
        $nlm = @()
        foreach($m in $dialog.Messages) {
            if($m.Locked) {
                continue
            }
            $nlm += $m
        }

        # re-insert locked messages
        $dialog.Messages = $this.InitialMessages += $nlm

        return $dialog
    }

    [Dialog] BeforeChatLoop([Dialog]$dialog) {
        if($this.Path -and $this.Prompt) {
            [OutHelper]::NonCriticalError("PreLoad: Cannot use both -Preload_Prompt and -Preload_Path")
            return $dialog
        }

        # load from path
        $p = $this.Path
        if($p -and (Test-Path $p)) {
            [OutHelper]::Info("- Preloading messages from: $p")
            # $this.InitialMessages = Get-Content $p | ConvertFrom-Json -NoEnumerate -AsHashtable
            $this.InitialMessages = [DialogMessage]::ImportMessages((Get-Content $p))
        }

        # load from prompt
        if($this.Prompt) {
            $this.InitialMessages = @( [DialogMessage]::new($this.Role, $this.Prompt) )
            [OutHelper]::Info("- Preloaded $($this.Role)-prompt: $($this.Prompt)")
        }

        if(!$this.InitialMessages) {
            return $dialog
        }

        [OutHelper]::Info("- $($this.InitialMessages.Count) messages loaded, approx. $([Dialog]::CalculateWords($this.InitialMessages)) words.")

        if($this.Lock) {
            # mark messages as locked
            foreach($m in $this.InitialMessages) {
                $m.Locked = $true
            }
        }

        $dialog.Messages = $this.InitialMessages

        return $dialog
    }
}