Added Get-CompletionFromMessages

Author: dfinke

PowerShellAI.psd1

@@ -22,9 +22,10 @@ The PowerShell AI module integrates with the OpenAI API and let's you easily acc
         'Disable-ChatPersistence'
         'Enable-AIShortCutKey'
         'Enable-ChatPersistence'
-        'Get-DalleImage'
-        'Get-ChatCompletion'
+        'Get-ChatCompletion'        
         'Get-ChatPersistence'
+        'Get-CompletionFromMessages'
+        'Get-DalleImage'
         'Get-GPT3Completion'
         'Get-GPT4Completion'
         'Get-GPT4Response'

Public/Get-CompletionFromMessages.ps1

@@ -0,0 +1,23 @@
+function Get-CompletionFromMessages {
+    param(
+        [Parameter(Mandatory)]
+        $Messages
+    )
+
+    $payload = (Get-ChatSessionOptions).Clone()
+
+    $payload.messages = $messages
+    $payload = $payload | ConvertTo-Json -Depth 10
+    
+    $body = [System.Text.Encoding]::UTF8.GetBytes($payload)
+
+    if ((Get-ChatAPIProvider) -eq 'OpenAI') {
+        $uri = Get-OpenAIChatCompletionUri
+    }
+    elseif ((Get-ChatAPIProvider) -eq 'AzureOpenAI') {
+        $uri = Get-ChatAzureOpenAIURI
+    }
+
+    $result = Invoke-OpenAIAPI -Uri $uri -Method 'Post' -Body $body 
+    $result.choices.message
+}

__tests__/Get-CompletionFromMessages.tests.ps1

@@ -0,0 +1,39 @@
+Import-Module $PSScriptRoot\..\PowerShellAI.psd1 -Force
+
+Describe "Get-CompletionFromMessages" -Tag "Get-CompletionFromMessages" {
+
+    It "tests the function Get-CompletionFromMessages exists" {
+        $actual = Get-Command Get-CompletionFromMessages -ErrorAction SilentlyContinue
+        $actual | Should -Not -BeNullOrEmpty
+    }
+
+    It "tests Get-CompletionFromMessages has a parameter named Messages" {
+        $actual = Get-Command Get-CompletionFromMessages -ErrorAction SilentlyContinue
+        $actual.Parameters.Keys | Should -Contain Messages
+    }
+
+    It "tests Get-CompletionFromMessages returns a response" {
+        Mock Invoke-RestMethodWithProgress -ModuleName PowerShellAI -ParameterFilter { 
+            $Params.Method -eq 'Post' -and $Params.Uri -eq (Get-OpenAIChatCompletionUri) 
+        } -MockWith {
+            [PSCustomObject]@{
+                choices = @(
+                    [PSCustomObject]@{
+                        message = [PSCustomObject]@{
+                            content = 'Mocked Get-GPT4Completion call'
+                        }
+                    }
+                )
+            }
+        } 
+        
+        $messages = $(
+            New-ChatMessageTemplate -Role system "I am a bot"
+            New-ChatMessageTemplate -Role user "Hello"
+        )
+
+        $actual = Get-CompletionFromMessages -Messages $messages
+
+        $actual.content | Should -BeExactly "Mocked Get-GPT4Completion call"
+    }
+}