Add Get, Enable, Disable ChatPersistence

Author: dfinke

PowerShellAI.psd1

@@ -19,9 +19,12 @@ The PowerShell AI module integrates with the OpenAI API and let's you easily acc
         'ConvertFrom-GPTMarkdownTable'
         'copilot'
         'Disable-AIShortCutKey'
+        'Disable-ChatPersistence'
         'Enable-AIShortCutKey'
+        'Enable-ChatPersistence'
         'Get-DalleImage'
         'Get-ChatCompletion'
+        'Get-ChatPersistence'
         'Get-GPT3Completion'
         'Get-GPT4Completion'
         'Get-GPT4Response'

PowerShellAI.psm1

@@ -1,6 +1,37 @@
-$Script:OpenAIKey = $null
+# Set the OpenAI key to null
+$Script:OpenAIKey = $null
 
+# Set the chat API provider to OpenAI
+$Script:ChatAPIProvider = 'OpenAI'
 
+# Set the chat in progress flag to false
+$Script:ChatInProgress = $false
+
+# Create an array list to store chat messages
+[System.Collections.ArrayList]$Script:ChatMessages = @()
+
+# Enable chat persistence
+$Script:ChatPersistence = $true
+
+# Set the options for the chat session
+$Script:ChatSessionOptions = @{
+    'model'             = 'gpt-4'
+    'temperature'       = 0.0
+    'max_tokens'        = 256
+    'top_p'             = 1.0
+    'frequency_penalty' = 0
+    'presence_penalty'  = 0
+    'stop'              = $null
+}
+
+# Set the options for the Azure OpenAI API
+$Script:AzureOpenAIOptions = @{
+    Endpoint       = 'not set'
+    DeploymentName = 'not set'
+    ApiVersion     = 'not set'
+}
+
+# Load all PowerShell scripts in the Public and Private directories
 foreach ($directory in @('Public', 'Private')) {
     Get-ChildItem -Path "$PSScriptRoot\$directory\*.ps1" | ForEach-Object { . $_.FullName }
 }

Public/Disable-ChatPersistence.ps1

@@ -0,0 +1,3 @@
+function Disable-ChatPersistence {
+    $Script:ChatPersistence = $false
+}

Public/Enable-ChatPersistence.ps1

@@ -0,0 +1,3 @@
+function Enable-ChatPersistence {
+    $Script:ChatPersistence = $true
+}

Public/Get-ChatPersistence.ps1

@@ -0,0 +1,3 @@
+function Get-ChatPersistence {
+    $Script:ChatPersistence
+}

Public/Get-GPT4Completion.ps1

@@ -1,24 +1,3 @@
-$Script:ChatSessionOptions = @{
-    'model'             = 'gpt-4'
-    'temperature'       = 0.0
-    'max_tokens'        = 256
-    'top_p'             = 1.0
-    'frequency_penalty' = 0
-    'presence_penalty'  = 0
-    'stop'              = $null
-}
-
-$Script:AzureOpenAIOptions = @{
-    Endpoint       = 'not set'
-    DeploymentName = 'not set'
-    ApiVersion     = 'not set'
-}
-
-$Script:ChatAPIProvider = 'OpenAI'
-$Script:ChatInProgress = $false
-
-[System.Collections.ArrayList]$Script:ChatMessages = @()
-
 function Get-AzureOpenAIOptions {
     [CmdletBinding()]
     param()
@@ -197,6 +176,8 @@ function Reset-ChatSessionOptions {
         'presence_penalty'  = 0
         'stop'              = $null
     }
+
+    Enable-ChatPersistence
 }
 
 function Clear-ChatMessages {

__tests__/Disable-ChatPersistence.tests.ps1

@@ -0,0 +1,19 @@
+Describe 'Disable-ChatPersistence' -Tag ChatPersistence {
+
+    BeforeEach {
+        Reset-ChatSessionOptions
+    }
+    
+
+    It 'tests the function Disable-ChatPersistence exists' {
+        $actual = Get-Command Disable-ChatPersistence -ErrorAction SilentlyContinue
+        $actual | Should -Not -BeNullOrEmpty
+    }
+
+    It 'tests that it disables chat persistence' {
+        Disable-ChatPersistence
+        $actual = Get-ChatPersistence
+
+        $actual | Should -Be $false
+    }
+}

__tests__/Enable-ChatPersistence.tests.ps1

@@ -0,0 +1,18 @@
+Describe 'Enable-ChatPersistence' -Tag ChatPersistence {
+
+    AfterEach {
+        Reset-ChatSessionOptions
+    }
+
+    It 'tests the function Enable-ChatPersistence exists' {
+        $actual = Get-Command Enable-ChatPersistence -ErrorAction SilentlyContinue
+        $actual | Should -Not -BeNullOrEmpty
+    }
+
+    It 'tests that it enables chat persistence' {
+        Enable-ChatPersistence
+        $actual = Get-ChatPersistence
+
+        $actual | Should -Be $true
+    }
+}

__tests__/Get-ChatPersistence.tests.ps1

@@ -0,0 +1,12 @@
+
+Describe 'Get-ChatPersistence' -Tag ChatPersistence {
+    It 'tests the function Get-ChatPersistence exists' {
+        $actual = Get-Command Get-ChatPersistence -ErrorAction SilentlyContinue
+        $actual | Should -Not -BeNullOrEmpty
+    }
+
+    It 'tests the function Get-ChatPersistence returns a boolean' {
+        $actual = Get-ChatPersistence
+        $actual.GetType().Name | Should -Be 'Boolean'
+    }
+}