Public/VectorStores/New-VectorStore.ps1
|
function New-VectorStore { [CmdletBinding(DefaultParameterSetName = 'VectorStoreId')] [OutputType([pscustomobject])] param ( # Hidden param, for Set-Thread cmdlet [Parameter(DontShow, ParameterSetName = 'VectorStore', ValueFromPipeline, ValueFromPipelineByPropertyName)] [PSTypeName('PSOpenAI.VectorStore')]$VectorStore, [Parameter(DontShow, ParameterSetName = 'VectorStoreId', ValueFromPipeline, ValueFromPipelineByPropertyName)] [ValidateNotNullOrEmpty()] [string][UrlEncodeTransformation()]$VectorStoreId, [Parameter()] [Alias('file_ids')] [ValidateCount(0, 500)] [object[]]$FileId, [Parameter()] [string]$Name, [Parameter()] [ValidateRange(1, 365)] [uint16]$ExpiresAfterDays, [Parameter()] [Completions('last_active_at')] [string][LowerCaseTransformation()]$ExpiresAfterAnchor = 'last_active_at', [Parameter()] [System.Collections.IDictionary]$MetaData, [Parameter()] [int]$TimeoutSec = 0, [Parameter()] [ValidateRange(0, 100)] [int]$MaxRetryCount = 0, [Parameter()] [OpenAIApiType]$ApiType = [OpenAIApiType]::OpenAI, [Parameter()] [System.Uri]$ApiBase, [Parameter(DontShow)] [string]$ApiVersion, [Parameter()] [ValidateSet('openai', 'azure', 'azure_ad')] [string]$AuthType = 'openai', [Parameter()] [securestring][SecureStringTransformation()]$ApiKey, [Parameter()] [Alias('OrgId')] [string]$Organization, [Parameter()] [System.Collections.IDictionary]$AdditionalQuery, [Parameter()] [System.Collections.IDictionary]$AdditionalHeaders, [Parameter()] [object]$AdditionalBody ) begin { # Initialize API Key [securestring]$SecureToken = Initialize-APIKey -ApiKey $ApiKey -ErrorAction Stop # Initialize API Base $ApiBase = Initialize-APIBase -ApiBase $ApiBase -ApiType $ApiType -ErrorAction Stop # Initialize Organization ID $Organization = Initialize-OrganizationID -OrgId $Organization # Get API context $OpenAIParameter = Get-OpenAIContext -EndpointName 'VectorStores' -ApiType $ApiType -AuthType $AuthType -ApiBase $ApiBase -ApiVersion $ApiVersion -ErrorAction Stop } process { #region Construct parameters for API request if ($VectorStore) { $VectorStoreId = $VectorStore.id } if (-not [string]::IsNullOrEmpty($VectorStoreId)) { $QueryUri = $OpenAIParameter.Uri.ToString() + "/$VectorStoreId" } else { $QueryUri = $OpenAIParameter.Uri } #region Construct parameters for API request $PostBody = [System.Collections.Specialized.OrderedDictionary]::new() if ($FileId.Count -gt 0) { $list = [System.Collections.Generic.List[string]]::new($FileId.Count) foreach ($item in $FileId) { if ($item -is [string]) { $PostBody.file_ids += $item } elseif ($item.psobject.TypeNames -contains 'PSOpenAI.File') { $PostBody.file_ids += $item.id } } if ($list.Count -gt 0) { $PostBody.file_ids = $list.ToArray() } } if ($PSBoundParameters.ContainsKey('Name')) { $PostBody.name = $Name } if ($PSBoundParameters.ContainsKey('ExpiresAfterDays')) { $PostBody.expires_after = @{ 'anchor' = $ExpiresAfterAnchor 'days' = $ExpiresAfterDays } } if ($PSBoundParameters.ContainsKey('Metadata')) { $PostBody.metadata = $Metadata } #endregion #region Send API Request $params = @{ Method = $OpenAIParameter.Method Uri = $QueryUri ContentType = $OpenAIParameter.ContentType TimeoutSec = $TimeoutSec MaxRetryCount = $MaxRetryCount ApiKey = $SecureToken AuthType = $OpenAIParameter.AuthType Organization = $Organization Headers = @{'OpenAI-Beta' = 'assistants=v2' } Body = $PostBody AdditionalQuery = $AdditionalQuery AdditionalHeaders = $AdditionalHeaders AdditionalBody = $AdditionalBody } $Response = Invoke-OpenAIAPIRequest @params # error check if ($null -eq $Response) { return } #endregion #region Parse response object try { $Response = $Response | ConvertFrom-Json -ErrorAction Stop } catch { Write-Error -Exception $_.Exception } #endregion #region Output ParseVectorStoreObject -InputObject $Response #endregion } end { } } |