Fix Status for Wait

potatoqualitee · Jun 19, 2024 · 58d3c4c · 58d3c4c
1 parent e24c0ff
commit 58d3c4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/public/Initialize-VectorStore.ps1 b/public/Initialize-VectorStore.ps1
@@ -121,7 +121,7 @@ function Initialize-VectorStore {
 
                 $uploads = Get-ChildItem $commandfiles | Register-OpenAIFile -Purpose "assistants"
                 $filebatch = PSOpenAI\Start-VectorStoreFileBatch -VectorStore $vectorStore -FileId $uploads.id
-                $null = PSOpenAI\Wait-VectorStoreFileBatch -VectorStore $vectorStore -BatchId $filebatch.id -StatusForWait 'completed'
+                $null = PSOpenAI\Wait-VectorStoreFileBatch -VectorStore $vectorStore -BatchId $filebatch.id -StatusForWait ('queued', 'in_progress', 'cancelling')
                 Write-Verbose "Uploaded batch $($i + 1) of $batches to vector store: $storename"
                 Start-Sleep 10