fix: pgvector ingestion broken due to missing wait

iwilltry42 · iwilltry42 · commit 6559819f7cd6 · 2024-10-23T12:53:14.000+02:00
diff --git a/pkg/vectorstore/pgvector/pgvector.go b/pkg/vectorstore/pgvector/pgvector.go
@@ -318,6 +318,11 @@ func (v VectorStore) AddDocuments(ctx context.Context, docs []vs.Document, colle
 
 		docs[docIdx] = doc
 	}
+	wg.Wait()
+
+	if sharedErr != nil {
+		return nil, sharedErr
+	}
 
 	return ids, v.conn.SendBatch(ctx, b).Close()
 }

Original file line number	Diff line number	Diff line change
`@@ -318,6 +318,11 @@ func (v VectorStore) AddDocuments(ctx context.Context, docs []vs.Document, colle`
`318`	`318`
`319`	`319`	`docs[docIdx] = doc`
`320`	`320`	`}`
	`321`	`+ wg.Wait()`
	`322`	`+`
	`323`	`+ if sharedErr != nil {`
	`324`	`+ return nil, sharedErr`
	`325`	`+ }`
`321`	`326`
`322`	`327`	`return ids, v.conn.SendBatch(ctx, b).Close()`
`323`	`328`	`}`