PSOpenAI.psm1
|
$PrivateDirectory = Join-Path $PSScriptRoot 'Private' $PublicDirectory = Join-Path $PSScriptRoot 'Public' $PrivateFunctions = Get-ChildItem -LiteralPath $PrivateDirectory -Recurse -Filter '*.ps1' -File $PublicFunctions = Get-ChildItem -LiteralPath $PublicDirectory -Recurse -Filter '*.ps1' -File # Exclude on specific platfotm. $Exclude = @() if (-not $IsWindows -or $PSVersionTable.PSVersion -lt 7.4) { $Exclude = @( 'Start-RealtimeSessionAudioOutput.ps1' 'Stop-RealtimeSessionAudioOutput.ps1' 'Stop-RealtimeSessionAudioInput.ps1' 'Stop-RealtimeSessionAudioInput.ps1' ) } # Include Private functions $PrivateFunctions | Where-Object { $_.Name -notin $Exclude } | ForEach-Object { . $_.FullName } # Include Public functions $PublicFunctions | Where-Object { $_.Name -notin $Exclude } | ForEach-Object { . $_.FullName } # Export public functions $ExportFunctions = [string[]]@() $PublicFunctions | ForEach-Object { if (Test-Path -LiteralPath "Function:/$($_.BaseName)") { $ExportFunctions += $_.BaseName } } if ($ExportFunctions.Count -ge 1) { Export-ModuleMember -Function $ExportFunctions -Alias * } # Initialize context dictionary Clear-OpenAIContext # Export classes $ExportableTypes = @( [APIRequestException] [BadRequestException] [ContentFilteredException] [UnauthorizedException] [NotFoundException] [RateLimitExceededException] [QuotaLimitExceededException] ) # Get the internal TypeAccelerators class to use its static methods. $TypeAcceleratorsClass = [psobject].Assembly.GetType('System.Management.Automation.TypeAccelerators') # Add type accelerators for every exportable type. foreach ($Type in $ExportableTypes) { if ($Type.FullName -notin $ExistingTypeAccelerators.Keys) { $null = $TypeAcceleratorsClass::Add($Type.FullName, $Type) } } # Remove type accelerators when the module is removed. $MyInvocation.MyCommand.ScriptBlock.Module.OnRemove = { foreach ($Type in $ExportableTypes) { $null = $TypeAcceleratorsClass::Remove($Type.FullName) } }.GetNewClosure() |