fix: ensure vectorStore is ready

Signed-off-by: Manuel Ruck <[email protected]>
demokratie-live · Aug 12, 2024 · d4a2ab9 · d4a2ab9
1 parent 11e0424
commit d4a2ab9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/services/non-named-votes-ai/src/utils.ts b/services/non-named-votes-ai/src/utils.ts
@@ -63,10 +63,14 @@ export const ensureFile = async ({ pdfUrl, file_id }: { pdfUrl: string; file_id?
 
 export const retriveVectorStore = async (vectorStoreId: string) =>
   await openai.beta.vectorStores.retrieve(vectorStoreId);
-export const createVectorStore = async ({ file_ids }: { file_ids: string[] }) =>
-  await openai.beta.vectorStores.create({
+export const createVectorStore = async ({ file_ids }: { file_ids: string[] }) => {
+  const vectorStore = await openai.beta.vectorStores.create({});
+
+  await openai.beta.vectorStores.fileBatches.createAndPoll(vectorStore.id, {
     file_ids,
   });
+  return vectorStore;
+};
 export const ensureVectorStore = async ({
   file_id,
   vector_store_id,