AzureOpenAIChat.ps1

function Get-AzureOpenAIToken{
    <#  .SYNOPSIS
       Get an azure token for user or managed identity thats required to authenticate to Azure OpenAI with Rest API.
       Also construct the header if you are using an Azure OpenAI API key instead of Azure AD authentication.
    .PARAMETER ManagedIdentity
        Use this parameter if you want to use a managed identity to authenticate to Azure OpenAI.
    .PARAMETER User
        Use this parameter if you want to use a user to authenticate to Azure OpenAI.
    .PARAMETER APIKey
        Use this parameter if you want to use an API key to authenticate to Azure OpenAI.

    .EXAMPLE
        # Manually specify username and password to acquire an authentication token:
        Get-AzureOpenAIToken -APIKey "ghgkfhgfgfgkhgh"
        Get-AzureOpenAIToken -ManagedIdentity $true
        Get-AzureOpenAIToken -User $true
    .NOTES
        Author: Alexander Holmeset
        Twitter: @AlexHolmeset
        Website: https://www.alexholmeset.blog
        Created: 09-02-2023
        Updated: 
        Version history:
        1.0.0 - (09-02-2023) Function created  
    #>

[CmdletBinding()]
param (
    [Parameter(Mandatory=$false)]
    [string]$APIKey,
    [Parameter(Mandatory=$false)]
    [string]$ManagedIdentity,
    [Parameter(Mandatory=$false)]
    [string]$User
)

Process {
    $ErrorActionPreference = "Stop"

    if (Get-Module -ListAvailable -Name Az.Accounts) {
       # Write-Host "You have the Az.Accounts module installed"
    } 
    else {
        Write-Host "You need to install the Az.Accounts module";
        break
    }
If(!$MyHeader){


If($ManagedIdentity -eq $true){
    "managed"
    try {
        Connect-AzAccount -Identity

        $MyTokenRequest = Get-AzAccessToken -ResourceUrl "https://cognitiveservices.azure.com"
        $MyToken = $MyTokenRequest.token
            If(!$MyToken){
                Write-Warning "Failed to get API access token!"
                Exit 1
            }
        $Global:MyHeader = @{"Authorization" = "Bearer $MyToken" }
       }
    catch [System.Exception] {
        Write-Warning "Failed to get Access Token, Error message: $($_.Exception.Message)"; break
    }
    
    }
If($User -eq $true){
        "USER"
        try {
            Connect-AzAccount
    
            $MyTokenRequest = Get-AzAccessToken -ResourceUrl "https://cognitiveservices.azure.com"
            $MyToken = $MyTokenRequest.token
                If(!$MyToken){
                    Write-Warning "Failed to get API access token!"
                    Exit 1
                }
            $Global:MyHeader = @{"Authorization" = "Bearer $MyToken" }
           }
        catch [System.Exception] {
            Write-Warning "Failed to get Access Token, Error message: $($_.Exception.Message)"; break
        }
   
    }
If($APIkey){
    "APIKEY"

    $Global:MyHeader = @{"api-key"  = $apikey }

        

}
}




}
}

function Get-Chat {
    <#  .SYNOPSIS
        Get a resposne from the chat endpoint in Azure OpenAI.
    .PARAMETER DeploymentName
        A deployment name should be provided.
    .PARAMETER ResourceName
        A Resource  name should be provided.
    .PARAMETER Prompt
        A prompt name should be provided.
    .PARAMETER Token
        A token name should be provided.                
    .EXAMPLE
        Get-Chat -DeploymentName $DeploymentName -ResourceName $ResourceName -maxtokens 1000 -prompt "What is the meaning of life?" -AssitantInstruction $AssitantInstruction
    .NOTES
        Author: Alexander Holmeset
        Twitter: @AlexHolmeset
        Website: https://www.alexholmeset.blog
        Created: 09-02-2023
        Updated: 
        Version history:
        1.0.0 - (09-02-2023) Function created      
    #>[CmdletBinding()]
    param (
        [parameter(Mandatory = $true, HelpMessage = "Your azure openai deployment name")]
        [ValidateNotNullOrEmpty()]
        [string]$DeploymentName,
        [parameter(Mandatory = $true, HelpMessage = "your azure openai resource name")]
        [ValidateNotNullOrEmpty()]
        [string]$ResourceName,
        [parameter(Mandatory = $true, HelpMessage = "Your Azure OpenAI prompt")]
        [ValidateNotNullOrEmpty()]
        [string]$Prompt,
        [parameter(Mandatory = $true, HelpMessage = "Your Azure OpenAI service instructions")]
        [ValidateNotNullOrEmpty()]
        [string]$AssitantInstruction
        )

Process {
    $ErrorActionPreference = "Stop"
    $APIVersion = "2023-03-15-preview"
    # Construct URI
    $uri = "https://$ResourceName.openai.azure.com/openai/deployments/$DeploymentName/chat/completions?api-version=$ApiVersion"

    $script:conversation += @"
    {"role": "user", "content": "$prompt"},
"@


    # Construct Body
    $Body = @"
    {

"messages":
[

$script:conversation
{"role": "system", "content": "$AssitantInstruction"}

]

    }
"@


    try {
        $body
        $Global:Request = invoke-restmethod -Method POST -Uri $uri -ContentType "application/json" -Body $body  -Headers $Global:MyHeader
        $script:conversation += @"
        {"role": "assistant", "content": "$($Request.choices.message.content)"},
"@

       }
    catch [System.Exception] {
      Write-Warning "Failed to to POST request: $($_.Exception.Message)"; break
    }   

    
    "User: $Prompt"

    "Assitant: $($Request.choices.message.content)"
    }
}


Get-AzureOpenAIToken -APIKey "Enter API Token"
$DeploymentName = "Enter deploymentname"
$ResourceName = "Enter resource name"
$AssitantInstruction = "You are a helpfull assistant."
$Prompt = "What is Azure OpenAI?"

Get-Chat -DeploymentName $DeploymentName -ResourceName $ResourceName -Prompt $Prompt -AssitantInstruction $AssitantInstruction