Providers/OpenAI.ps1
|
<#
.SYNOPSIS Invokes the OpenAI API to generate responses using specified models. .DESCRIPTION The Invoke-OpenAIProvider function sends requests to the OpenAI API and returns the generated content. It requires an API key to be set in the environment variable 'OpenAIKey'. .PARAMETER ModelName The name of the OpenAI model to use (e.g., 'gpt-4', 'gpt-3.5-turbo'). .PARAMETER Messages An array of hashtables containing the messages to send to the model. .EXAMPLE $Message = New-ChatMessage -Prompt 'Write a PowerShell function to calculate factorial' $response = Invoke-OpenAIProvider -ModelName 'gpt-4' -Message $Message .NOTES Requires the OpenAIKey environment variable to be set with a valid API key. Includes 'assistants=v2' beta header for compatibility with newer API features. API Reference: https://platform.openai.com/docs/api-reference #> function Invoke-OpenAIProvider { param( [Parameter(Mandatory)] [string]$ModelName, [Parameter(Mandatory)] [hashtable[]]$Messages ) $headers = @{ 'OpenAI-Beta' = 'assistants=v2' 'Authorization' = "Bearer $env:OpenAIKey" 'content-type' = 'application/json' } $body = @{ 'model' = $ModelName 'messages' = $Messages } $Uri = "https://api.openai.com/v1/chat/completions" $params = @{ Uri = $Uri Method = 'POST' Headers = $headers Body = $body | ConvertTo-Json -Depth 10 } try { $response = Invoke-RestMethod @params return $response.choices[0].message.content } catch { $statusCode = $_.Exception.Response.StatusCode.value__ $errorMessage = $_.ErrorDetails.Message Write-Error "OpenAI API Error (HTTP $statusCode): $errorMessage" return "Error calling OpenAI API: $($_.Exception.Message)" } } |