PsChat

0.0.37

Classes/PsChatUi.psm1

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

using module ".\Dialog.psm1"

using module ".\Options.psm1"

using module ".\OpenAiChat.psm1"

using module ".\Extensions\ExtensionContainer.psm1"

using module ".\Extensions\WordCountWarning.psm1"

using module ".\Extensions\AutoSave.psm1"

using module ".\Extensions\PreLoad.psm1"

using module ".\Extensions\Commands.psm1"

using module ".\Extensions\ShortTerm.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()

            [Commands]::new()

            [ShortTerm]::new()

        ))

    }

    Start() {

        [OutHelper]::Info("Starting PsChat v$((Get-Module -Name PsChat).Version).")

        $dlg = $this.Dialog

        $dlg.Question = $this.Options.InitialQuestion

        $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($this.Options.SingleQuestion) { 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)

        } 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.GetAnswer($dlg.Messages, $true)

        } else {

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

            [OutHelper]::Gpt($answer)

        }

        if($null -ne $answer) {

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

        }

        return $dlg

    }

}