Resolve embedding provider by model identifier for taxonomy embeddings

Author: mikadamczyk

src/contracts/Search/Embedding/EmbeddingProviderResolverInterface.php

@@ -14,4 +14,9 @@ interface EmbeddingProviderResolverInterface
      * @throws \Ibexa\Contracts\Core\Search\Embedding\EmbeddingResolverNotFoundException
      */
     public function resolve(): EmbeddingProviderInterface;
+
+    /**
+     * @throws \Ibexa\Contracts\Core\Search\Embedding\EmbeddingResolverNotFoundException
+     */
+    public function resolveByModelIdentifier(string $modelIdentifier): EmbeddingProviderInterface;
 }

src/lib/Search/Embedding/EmbeddingProviderResolver.php

@@ -40,4 +40,18 @@ public function resolve(): EmbeddingProviderInterface
 
         return $this->registry->getEmbeddingProvider($defaultEmbeddingProvider);
     }
+
+    public function resolveByModelIdentifier(string $modelIdentifier): EmbeddingProviderInterface
+    {
+        $model = $this->embeddingConfiguration->getModel($modelIdentifier);
+        $providerIdentifier = $model['embedding_provider'];
+
+        if (!$this->registry->hasEmbeddingProvider($providerIdentifier)) {
+            throw new EmbeddingResolverNotFoundException(
+                $providerIdentifier
+            );
+        }
+
+        return $this->registry->getEmbeddingProvider($providerIdentifier);
+    }
 }

tests/lib/Search/Embedding/EmbeddingProviderResolverTest.php

@@ -79,4 +79,36 @@ public function testResolveThrowsWhenProviderMissing(): void
 
         $this->resolver->resolve();
     }
+
+    public function testResolveByModelIdentifierReturnsProviderConfiguredForModel(): void
+    {
+        $modelIdentifier = 'gemini_embedding_001_1536';
+        $embeddingProviderIdentifier = 'ibexa_gemini';
+        $mockProvider = $this->createMock(EmbeddingProviderInterface::class);
+
+        $this->configuration
+            ->expects(self::once())
+            ->method('getModel')
+            ->with($modelIdentifier)
+            ->willReturn([
+                'name' => 'models/gemini-embedding-001',
+                'dimensions' => 1536,
+                'field_suffix' => 'gemini_1536',
+                'embedding_provider' => $embeddingProviderIdentifier,
+            ]);
+
+        $this->registry
+            ->expects(self::once())
+            ->method('hasEmbeddingProvider')
+            ->with($embeddingProviderIdentifier)
+            ->willReturn(true);
+
+        $this->registry
+            ->expects(self::once())
+            ->method('getEmbeddingProvider')
+            ->with($embeddingProviderIdentifier)
+            ->willReturn($mockProvider);
+
+        self::assertSame($mockProvider, $this->resolver->resolveByModelIdentifier($modelIdentifier));
+    }
 }