From 1afc25ff380a7860369904e5be09bdae4f2dc4f1 Mon Sep 17 00:00:00 2001 From: Oskar Stark Date: Fri, 6 Dec 2024 13:00:41 +0100 Subject: [PATCH] feat: Add `score` to `VectorDocument` (#152) * Add `score` to `VectorDocument` * - --- src/Bridge/MongoDB/Store.php | 1 + src/Bridge/Pinecone/Store.php | 1 + src/Document/VectorDocument.php | 1 + 3 files changed, 3 insertions(+) diff --git a/src/Bridge/MongoDB/Store.php b/src/Bridge/MongoDB/Store.php index 9332549..6a62243 100644 --- a/src/Bridge/MongoDB/Store.php +++ b/src/Bridge/MongoDB/Store.php @@ -134,6 +134,7 @@ public function query(Vector $vector, array $options = [], ?float $minScore = nu id: $this->toUuid($result['_id']), vector: new Vector($result[$this->vectorFieldName]), metadata: new Metadata($result['metadata'] ?? []), + score: $result['score'], ); } diff --git a/src/Bridge/Pinecone/Store.php b/src/Bridge/Pinecone/Store.php index 1b2164c..b940a0c 100644 --- a/src/Bridge/Pinecone/Store.php +++ b/src/Bridge/Pinecone/Store.php @@ -59,6 +59,7 @@ public function query(Vector $vector, array $options = [], ?float $minScore = nu id: Uuid::fromString($match['id']), vector: new Vector($match['values']), metadata: new Metadata($match['metadata']), + score: $match['score'], ); } diff --git a/src/Document/VectorDocument.php b/src/Document/VectorDocument.php index 27f36b1..b0bc241 100644 --- a/src/Document/VectorDocument.php +++ b/src/Document/VectorDocument.php @@ -12,6 +12,7 @@ public function __construct( public Uuid $id, public VectorInterface $vector, public Metadata $metadata = new Metadata(), + public ?float $score = null, ) { } }