PsChat

0.0.63

Classes/Extensions/PreLoad.psm1

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

class PreLoad {
    [string]$Prompt
    [string]$Path    
    [object[]]$Objects 
    [string]$Role = "user"
    [bool]$Verbose = $false
    [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) {
        $this.InitialMessages = @()

        # load from objects
        $objs = $this.Objects
        if($objs) {
            [OutHelper]::Info("- Preload: From: $($objs.Count) objects")
            # $this.InitialMessages = [DialogMessage]::ImportMessages((Get-Content $p))
            foreach($obj in $objs) {
                $message = [DialogMessage]::new($obj.Role, $obj.Content)
                if($obj.Locked) {
                    $message.Locked = $true
                }
                $this.InitialMessages += $message
            }
        }

        # load from path
        $p = $this.Path
        if($p -and (Test-Path $p)) {
            [OutHelper]::Info("- Preload: From: $p")
            $this.InitialMessages = [DialogMessage]::ImportMessages((Get-Content $p))
        }
        
        # load from prompt
        if($this.Prompt) {
            $this.InitialMessages = @( [DialogMessage]::new($this.Role, $this.Prompt) )
            if($this.Verbose) {
                [OutHelper]::Info("- Preload: $($this.Role)-prompt: $($this.Prompt)")
            }
        }

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

        if($this.Verbose) {
            [OutHelper]::Info("- Preload: $($this.InitialMessages.Count) messages loaded, approx. $([Dialog]::ApproximateTokens($this.InitialMessages)) tokens.")
        }

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

        $dialog.Messages = $this.InitialMessages

        return $dialog
    }
}