PsChat

0.0.20

Classes/PsChatUi.psm1

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

using module "..\Private\SpecialCommands.psm1"

using module ".\Options.psm1"

using module ".\OpenAiChat.psm1"

using module ".\Dialog.psm1"

using module ".\Extensions\ExtensionContainer.psm1"

using module ".\Extensions\WordCountWarning.psm1"

using module ".\Extensions\AutoSave.psm1"

using module ".\Extensions\PreLoad.psm1"

class PsChatUi {

    [string]$OpenAiAuthKey

    [Options]$Options

    [OpenAiChat]$ChatApi

    [Dialog]$Dialog

    [ExtensionContainer]$ExtensionContainer

    [bool]$Stream = $true

    PsChatUi([string]$openAiAuthKey, [Options]$options) {

        $this.OpenAiAuthKey = $openAiAuthKey

        $this.Options = $options

        $this.ChatApi = [OpenAiChat]::new($this.OpenAiAuthKey)

        $this.Dialog = [Dialog]::new()

        $this.ExtensionContainer = [ExtensionContainer]::new($this.ChatApi, $this.Options, @(

            [WordCountWarning]::new()

            [AutoSave]::new()

            [PreLoad]::new()

        ))

    }

    Start([string]$question, [bool]$single) {

        [OutHelper]::Info("Starting PsChat v$((Get-Module -Name PsChat).Version).$(if(!$single) { " Press 'h' for help." })")

        $dlg = $this.Dialog

        $dlg.Question = $Question

        $dlg = $this.ExtensionContainer.Invoke("BeforeChatLoop", $dlg)

        do {

            # call api with all previous messages

            if(![string]::IsNullOrEmpty($dlg.Question)) {

                $dlg = $this.ExtensionContainer.Invoke("BeforeAnswer", $dlg)

                $dlg = $this.Invoke($dlg)

                $dlg = $this.ExtensionContainer.Invoke("AfterAnswer", $dlg)

                if($Single) { break }

            }

            # execute extension logic before a question

            $dlg = $this.ExtensionContainer.Invoke("BeforeQuestion", $dlg)

            $dlg.PromptUser()

            # execute extension logic after a question

            $dlg = $this.ExtensionContainer.Invoke("AfterQuestion", $dlg)

            # handle special commands

            $dlg = Invoke-SpecialCommand $this.ChatApi $dlg

        } while($dlg.Question -ne "q" )

        $this.ExtensionContainer.Invoke("AfterChatLoop", $dlg) | Out-Null

    }

    [Dialog] Invoke([Dialog]$dlg) {

        $dlg.AddMessage("user", $dlg.Question)

        $answer = $null

        if($this.Stream) {

            $answer = $this.ChatApi.GetStreamedAnswer($dlg.Messages)

        } else {

            $answer = $this.ChatApi.GetAnswer($dlg.Messages)

            [OutHelper]::Gpt($answer)

        }

        if($null -ne $answer) {

            $dlg.AddMessage("assistant", $answer)

        }

        return $dlg

    }

}