diff --git a/src/store/src/Bridge/ChromaDb/Store.php b/src/store/src/Bridge/ChromaDb/Store.php index f3893ad1..acf796c5 100644 --- a/src/store/src/Bridge/ChromaDb/Store.php +++ b/src/store/src/Bridge/ChromaDb/Store.php @@ -42,10 +42,11 @@ public function add(VectorDocument ...$documents): void $ids[] = (string) $document->id; $vectors[] = $document->vector->getData(); $metadata[] = $document->metadata->getArrayCopy(); + $originalDocuments[] = $document->metadata['content']; } $collection = $this->client->getOrCreateCollection($this->collectionName); - $collection->add($ids, $vectors, $metadata); + $collection->add($ids, $vectors, $metadata, $originalDocuments); } public function query(Vector $vector, array $options = []): array