Public/Invoke-Completion.ps1
|
<#
.SYNOPSIS Invokes a completion request to a specified AI model provider. .DESCRIPTION The Invoke-Completion function sends a prompt to a specified AI model provider and returns the completion response. The model provider and model name must be specified in the 'provider:model' format. The function dynamically constructs the provider-specific function name and invokes it to get the response. .PARAMETER Prompt The prompt string to be sent to the AI model for completion. This parameter is mandatory. .PARAMETER Model The model to be used for the completion request, specified in 'provider:model' format. Defaults to "openai:gpt-4o-mini". .PARAMETER TextOnly A switch parameter that, if specified, returns only the response text. If not specified, a custom object containing the response text, model information, and timestamp is returned. .PARAMETER IncludeElapsedTime A switch parameter that, if specified, measures and includes the elapsed time of the API request in the output. .EXAMPLE Invoke-Completion -Prompt "Hello, world!" -Model "openai:gpt-4o-mini" Sends the prompt "Hello, world!" to the OpenAI GPT-4o-mini model and returns the completion response. .EXAMPLE Invoke-Completion -Prompt "Hello, world!" -Model "openai:gpt-4o-mini" -TextOnly Sends the prompt "Hello, world!" to the OpenAI GPT-4o-mini model and returns only the completion response text. .EXAMPLE Invoke-Completion -Prompt "Hello, world!" -Model "openai:gpt-4o-mini" -IncludeElapsedTime Sends the prompt and returns the completion response with the elapsed time information. .NOTES The function dynamically constructs the provider-specific function name based on the provider specified in the Model parameter. If the provider function does not exist, an error is thrown. #> function Invoke-Completion { [CmdletBinding()] param( [Parameter(Mandatory)] [string]$Prompt, [string]$Model = "openai:gpt-4o-mini", [switch]$TextOnly, [switch]$IncludeElapsedTime ) # Parse provider and model from the Model parameter if ($Model -match "^([^:]+):(.+)$") { $provider = $Matches[1].ToLower() $modelName = $Matches[2] } else { throw "Model must be specified in 'provider:model' format." } # Dynamically construct provider function name $providerFunction = "Invoke-${provider}Provider" # Check if the provider function exists if (-not (Get-Command $providerFunction -ErrorAction SilentlyContinue)) { throw "Unsupported provider: $provider. No function named $providerFunction found." } # Start measuring execution time if requested if ($IncludeElapsedTime) { $stopwatch = [System.Diagnostics.Stopwatch]::StartNew() } # Invoke the provider-specific implementation $responseText = & $providerFunction $modelName $Prompt # Stop measuring execution time if requested if ($IncludeElapsedTime) { $stopwatch.Stop() $elapsedTime = $stopwatch.Elapsed } # Return the result based on the TextOnly switch if ($TextOnly) { # Format the text output to include the elapsed time if requested if ($IncludeElapsedTime) { return "$responseText`n`nElapsed Time: $($elapsedTime.ToString('hh\:mm\:ss\.fff'))" } else { return $responseText } } else { # Create the base response object $responseObject = [PSCustomObject]@{ Prompt = $Prompt Response = $responseText Model = $Model Provider = $provider ModelName = $modelName Timestamp = Get-Date } # Add elapsed time if requested if ($IncludeElapsedTime) { $responseObject | Add-Member -MemberType NoteProperty -Name 'ElapsedTime' -Value $elapsedTime } return $responseObject } } |