[Store] Add Pinecone ManagedStore

src/store/src/Bridge/Cache/Store.php

@@ -91,7 +91,7 @@ public function query(Vector $vector, array $options = []): iterable
         yield from $this->distanceCalculator->calculate($vectorDocuments, $vector, $options['maxItems'] ?? null);
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->cache->clear();
     }

src/store/src/Bridge/ClickHouse/Store.php

@@ -48,7 +48,7 @@ public function setup(array $options = []): void
         $this->execute('POST', $sql);
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->execute('POST', 'DROP TABLE IF EXISTS {{ table }}');
     }

src/store/src/Bridge/Cloudflare/Store.php

@@ -52,7 +52,7 @@ public function setup(array $options = []): void
         ]);
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('DELETE', \sprintf('vectorize/v2/indexes/%s', $this->index));
     }

src/store/src/Bridge/ManticoreSearch/Store.php

@@ -50,7 +50,7 @@ public function setup(array $options = []): void
         ));
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('cli', \sprintf('DROP TABLE %s', $this->table));
     }

src/store/src/Bridge/MariaDb/Store.php

@@ -76,7 +76,7 @@ public function setup(array $options = []): void
         );
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->connection->exec(\sprintf('DROP TABLE IF EXISTS %s', $this->tableName));
     }

src/store/src/Bridge/Meilisearch/Store.php

@@ -101,7 +101,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('DELETE', \sprintf('indexes/%s', $this->indexName), []);
     }

src/store/src/Bridge/Milvus/Store.php

@@ -122,7 +122,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('POST', 'v2/vectordb/databases/drop', [
             'dbName' => $this->database,

src/store/src/Bridge/MongoDb/Store.php

@@ -99,7 +99,7 @@ public function setup(array $options = []): void
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->getCollection()->drop();
     }

src/store/src/Bridge/Neo4j/Store.php

@@ -79,7 +79,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('POST', \sprintf('db/%s/query/v2', $this->databaseName), [
             'statement' => 'MATCH (n) DETACH DELETE n',

src/store/src/Bridge/OpenSearch/Store.php

@@ -60,7 +60,7 @@ public function setup(array $options = []): void
         ]);
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $indexExistResponse = $this->httpClient->request('HEAD', \sprintf('%s/%s', $this->endpoint, $this->indexName));
 

src/store/src/Bridge/Pinecone/Store.php

@@ -16,13 +16,15 @@
 use Symfony\AI\Platform\Vector\Vector;
 use Symfony\AI\Store\Document\Metadata;
 use Symfony\AI\Store\Document\VectorDocument;
+use Symfony\AI\Store\Exception\InvalidArgumentException;
+use Symfony\AI\Store\ManagedStoreInterface;
 use Symfony\AI\Store\StoreInterface;
 use Symfony\Component\Uid\Uuid;
 
 /**
  * @author Christopher Hertel <[email protected]>
  */
-final class Store implements StoreInterface
+final class Store implements StoreInterface, ManagedStoreInterface
 {
     /**
      * @param array<string, mixed> $filter
@@ -35,6 +37,23 @@ public function __construct(
     ) {
     }
 
+    public function setup(array $options = []): void
+    {
+        if (false === isset($options['indexName']) || false === isset($options['dimension'])) {
+            throw new InvalidArgumentException('No supported options.');
+        }
+
+        $this->pinecone
+            ->control()
+            ->index($options['indexName'])
+            ->createServerless(
+                $options['dimension'],
+                $options['metric'] ?? null,
+                $options['cloud'] ?? null,
+                $options['region'] ?? null,
+            );
+    }
+
     public function add(VectorDocument ...$documents): void
     {
         $vectors = [];
@@ -73,6 +92,18 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
+    public function drop(array $options = []): void
+    {
+        if (false === isset($options['indexName'])) {
+            throw new InvalidArgumentException('No supported options.');
+        }
+
+        $this->pinecone
+            ->control()
+            ->index($options['indexName'])
+            ->delete();
+    }
+
     private function getVectors(): VectorResource
     {
         return $this->pinecone->data()->vectors();

src/store/src/Bridge/Pinecone/Tests/StoreTest.php

@@ -13,6 +13,8 @@
 
 use PHPUnit\Framework\TestCase;
 use Probots\Pinecone\Client;
+use Probots\Pinecone\Resources\Control\IndexResource;
+use Probots\Pinecone\Resources\ControlResource;
 use Probots\Pinecone\Resources\Data\VectorResource;
 use Probots\Pinecone\Resources\DataResource;
 use Saloon\Http\Response;
@@ -24,6 +26,76 @@
 
 final class StoreTest extends TestCase
 {
+    public function testStoreCantSetupWithInvalidOptions()
+    {
+        $pinecone = $this->createMock(Client::class);
+        $store = new Store($pinecone);
+
+        $this->expectException(\InvalidArgumentException::class);
+
+        $store->setup();
+    }
+
+    public function testStoreCanSetup()
+    {
+        $pinecone = $this->createMock(Client::class);
+        $indexResource = $this->createMock(IndexResource::class);
+        $controlResource = $this->createMock(ControlResource::class);
+
+        $pinecone->expects($this->once())
+            ->method('control')
+            ->willReturn($controlResource);
+
+        $controlResource->expects($this->once())
+            ->method('index')
+            ->with('test-index')
+            ->willReturn($indexResource);
+
+        $indexResource->expects($this->once())
+            ->method('createServerless')
+            ->with(1536, null, null, null);
+
+        $store = new Store($pinecone);
+        $store->setup([
+            'indexName' => 'test-index',
+            'dimension' => 1536,
+        ]);
+    }
+
+    public function testStoreCantDropWithInvalidOptions()
+    {
+        $pinecone = $this->createMock(Client::class);
+        $store = new Store($pinecone);
+
+        $this->expectException(\InvalidArgumentException::class);
+
+        $store->drop();
+    }
+
+    public function testStoreCanDrop()
+    {
+        $pinecone = $this->createMock(Client::class);
+        $indexResource = $this->createMock(IndexResource::class);
+        $controlResource = $this->createMock(ControlResource::class);
+
+        $pinecone->expects($this->once())
+            ->method('control')
+            ->willReturn($controlResource);
+
+        $controlResource->expects($this->once())
+            ->method('index')
+            ->with('test-index')
+            ->willReturn($indexResource);
+
+        $indexResource->expects($this->once())
+            ->method('delete');
+
+        $store = new Store($pinecone);
+        $store->drop([
+            'indexName' => 'test-index',
+        ]);
+    }
+
     public function testAddSingleDocument()
     {
         $vectorResource = $this->createMock(VectorResource::class);

src/store/src/Bridge/Postgres/Store.php

@@ -75,7 +75,7 @@ public function setup(array $options = []): void
         );
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->connection->exec(\sprintf('DROP TABLE IF EXISTS %s', $this->tableName));
     }

src/store/src/Bridge/Qdrant/Store.php

@@ -103,7 +103,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('DELETE', \sprintf('collections/%s', $this->collectionName));
     }

src/store/src/Bridge/Redis/Store.php

@@ -67,7 +67,7 @@ public function setup(array $options = []): void
         $this->redis->clearLastError();
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         try {
             $this->redis->rawCommand('FT.DROPINDEX', $this->indexName);

src/store/src/Bridge/SurrealDb/Store.php

@@ -75,7 +75,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->authenticate();
 

src/store/src/Bridge/Typesense/Store.php

@@ -86,7 +86,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('DELETE', \sprintf('collections/%s', $this->collection), []);
     }

src/store/src/Bridge/Weaviate/Store.php

@@ -78,7 +78,7 @@ public function query(Vector $vector, array $options = []): iterable
         }
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->request('DELETE', \sprintf('v1/schema/%s', $this->collection), []);
     }

src/store/src/InMemory/Store.php

@@ -65,7 +65,7 @@ public function query(Vector $vector, array $options = []): iterable
         yield from $this->distanceCalculator->calculate($documents, $vector, $options['maxItems'] ?? null);
     }
 
-    public function drop(): void
+    public function drop(array $options = []): void
     {
         $this->documents = [];
     }

src/store/src/ManagedStoreInterface.php

@@ -21,5 +21,8 @@ interface ManagedStoreInterface
      */
     public function setup(array $options = []): void;
 
-    public function drop(): void;
+    /**
+     * @param array<mixed> $options
+     */
+    public function drop(array $options = []): void;
 }