[Store] Add a way to store document content when using Chroma DB

dorrogeray · dorrogeray · commit 0f076860a647 · 2025-08-18T08:43:35.000+02:00
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -42,10 +42,11 @@ public function add(VectorDocument ...$documents): void`
`42`	`42`	`$ids[] = (string) $document->id;`
`43`	`43`	`$vectors[] = $document->vector->getData();`
`44`	`44`	`$metadata[] = $document->metadata->getArrayCopy();`
	`45`	`+ $originalDocuments[] = $document->metadata['content'];`
`45`	`46`	`}`
`46`	`47`
`47`	`48`	`$collection = $this->client->getOrCreateCollection($this->collectionName);`
`48`		`- $collection->add($ids, $vectors, $metadata);`
	`49`	`+ $collection->add($ids, $vectors, $metadata, $originalDocuments);`
`49`	`50`	`}`
`50`	`51`
`51`	`52`	`public function query(Vector $vector, array $options = []): array`