Implement Pinecone managed store and update ManagedStoreInterface::drop()

MolloKhan · MolloKhan · commit 766ad1aa9d2a · 2025-12-12T11:53:09.000-07:00
diff --git a/src/store/src/Bridge/Cache/Store.php b/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();
     }
diff --git a/src/store/src/Bridge/ClickHouse/Store.php b/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 }}');
     }
diff --git a/src/store/src/Bridge/Cloudflare/Store.php b/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));
     }
diff --git a/src/store/src/Bridge/ManticoreSearch/Store.php b/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));
     }
diff --git a/src/store/src/Bridge/MariaDb/Store.php b/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));
     }
diff --git a/src/store/src/Bridge/Meilisearch/Store.php b/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), []);
     }
diff --git a/src/store/src/Bridge/Milvus/Store.php b/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,
diff --git a/src/store/src/Bridge/MongoDb/Store.php b/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();
     }
diff --git a/src/store/src/Bridge/Neo4j/Store.php b/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',
diff --git a/src/store/src/Bridge/OpenSearch/Store.php b/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));
 
diff --git a/src/store/src/Bridge/Pinecone/ManagedStore.php b/src/store/src/Bridge/Pinecone/ManagedStore.php
diff --git a/src/store/src/Bridge/Pinecone/Store.php b/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 <mail@christopher-hertel.de>
  */
-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();
diff --git a/src/store/src/Bridge/Pinecone/Tests/StoreTest.php b/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);
diff --git a/src/store/src/Bridge/Postgres/Store.php b/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));
     }
diff --git a/src/store/src/Bridge/Qdrant/Store.php b/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));
     }
diff --git a/src/store/src/Bridge/Redis/Store.php b/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);
diff --git a/src/store/src/Bridge/SurrealDb/Store.php b/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();
 
diff --git a/src/store/src/Bridge/Typesense/Store.php b/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), []);
     }
diff --git a/src/store/src/Bridge/Weaviate/Store.php b/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), []);
     }
diff --git a/src/store/src/ManagedStoreInterface.php b/src/store/src/ManagedStoreInterface.php
@@ -21,5 +21,5 @@ interface ManagedStoreInterface
      */
     public function setup(array $options = []): void;
 
-    public function drop(): void;
+    public function drop(array $options = []): void;
 }

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ public function query(Vector $vector, array $options = []): iterable`
`91`	`91`	`yield from $this->distanceCalculator->calculate($vectorDocuments, $vector, $options['maxItems'] ?? null);`
`92`	`92`	`}`
`93`	`93`
`94`		`- public function drop(): void`
	`94`	`+ public function drop(array $options = []): void`
`95`	`95`	`{`
`96`	`96`	`$this->cache->clear();`
`97`	`97`	`}`
Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ public function setup(array $options = []): void`
`48`	`48`	`$this->execute('POST', $sql);`
`49`	`49`	`}`
`50`	`50`
`51`		`- public function drop(): void`
	`51`	`+ public function drop(array $options = []): void`
`52`	`52`	`{`
`53`	`53`	`$this->execute('POST', 'DROP TABLE IF EXISTS {{ table }}');`
`54`	`54`	`}`
Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ public function setup(array $options = []): void`
`52`	`52`	`]);`
`53`	`53`	`}`
`54`	`54`
`55`		`- public function drop(): void`
	`55`	`+ public function drop(array $options = []): void`
`56`	`56`	`{`
`57`	`57`	`$this->request('DELETE', \sprintf('vectorize/v2/indexes/%s', $this->index));`
`58`	`58`	`}`
Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ public function setup(array $options = []): void`
`50`	`50`	`));`
`51`	`51`	`}`
`52`	`52`
`53`		`- public function drop(): void`
	`53`	`+ public function drop(array $options = []): void`
`54`	`54`	`{`
`55`	`55`	`$this->request('cli', \sprintf('DROP TABLE %s', $this->table));`
`56`	`56`	`}`
Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,7 @@ public function setup(array $options = []): void`
`76`	`76`	`);`
`77`	`77`	`}`
`78`	`78`
`79`		`- public function drop(): void`
	`79`	`+ public function drop(array $options = []): void`
`80`	`80`	`{`
`81`	`81`	`$this->connection->exec(\sprintf('DROP TABLE IF EXISTS %s', $this->tableName));`
`82`	`82`	`}`
Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,7 @@ public function query(Vector $vector, array $options = []): iterable`
`101`	`101`	`}`
`102`	`102`	`}`
`103`	`103`
`104`		`- public function drop(): void`
	`104`	`+ public function drop(array $options = []): void`
`105`	`105`	`{`
`106`	`106`	`$this->request('DELETE', \sprintf('indexes/%s', $this->indexName), []);`
`107`	`107`	`}`
Original file line number	Diff line number	Diff line change
`@@ -122,7 +122,7 @@ public function query(Vector $vector, array $options = []): iterable`
`122`	`122`	`}`
`123`	`123`	`}`
`124`	`124`
`125`		`- public function drop(): void`
	`125`	`+ public function drop(array $options = []): void`
`126`	`126`	`{`
`127`	`127`	`$this->request('POST', 'v2/vectordb/databases/drop', [`
`128`	`128`	`'dbName' => $this->database,`
Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,7 @@ public function setup(array $options = []): void`
`99`	`99`	`}`
`100`	`100`	`}`
`101`	`101`
`102`		`- public function drop(): void`
	`102`	`+ public function drop(array $options = []): void`
`103`	`103`	`{`
`104`	`104`	`$this->getCollection()->drop();`
`105`	`105`	`}`
Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,7 @@ public function query(Vector $vector, array $options = []): iterable`
`79`	`79`	`}`
`80`	`80`	`}`
`81`	`81`
`82`		`- public function drop(): void`
	`82`	`+ public function drop(array $options = []): void`
`83`	`83`	`{`
`84`	`84`	`$this->request('POST', \sprintf('db/%s/query/v2', $this->databaseName), [`
`85`	`85`	`'statement' => 'MATCH (n) DETACH DELETE n',`
Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ public function setup(array $options = []): void`
`60`	`60`	`]);`
`61`	`61`	`}`
`62`	`62`
`63`		`- public function drop(): void`
	`63`	`+ public function drop(array $options = []): void`
`64`	`64`	`{`
`65`	`65`	`$indexExistResponse = $this->httpClient->request('HEAD', \sprintf('%s/%s', $this->endpoint, $this->indexName));`
`66`	`66`