AWS S3 Vectors Bridge

1
2
3
4
5
6
7
8
9
10
11
12
13
14

use AsyncAws\S3Vectors\S3VectorsClient;
use Symfony\AI\Store\Bridge\S3Vectors\Store;

$client = new S3VectorsClient([
    'region' => 'us-east-1',
]);

$store = new Store(
    client: $client,
    vectorBucketName: 'my-vector-bucket',
    indexName: 'my-index',
    filter: [], // Optional: default filter for queries
    topK: 3, // Optional: default number of results
);

1
2
3
4
5
6
7
8
9
10
11

use Symfony\AI\Platform\Vector\Vector;
use Symfony\AI\Store\Document\Metadata;
use Symfony\AI\Store\Document\VectorDocument;
use Symfony\Component\Uid\Uuid;

$document = new VectorDocument(
    id: Uuid::v4(),
    vector: new Vector([0.1, 0.2, 0.3, ...]),
    metadata: new Metadata(['title' => 'My Document'])
);
$store->add($document);

1
2
3
4
5
6
7
8
9
10
11
12
13
14

use Symfony\AI\Platform\Vector\Vector;
use Symfony\AI\Store\Query\VectorQuery;

$results = $store->query(
    query: new VectorQuery(new Vector([0.1, 0.2, 0.3, ...])),
    options: [
        'topK' => 5,
        'filter' => ['category' => 'documentation'],
    ]
);

foreach ($results as $result) {
    echo $result->getMetadata()['title'] . ' (score: ' . $result->getScore() . ')' . PHP_EOL;
}