Merge pull request #221 from jmartisk/provider-selection

Fail if selected chat/embedding model isn't the one found on classpath
quarkiverse · Jan 14, 2024 · e6c425b · e6c425b
2 parents d98578e + 2d3a71c
commit e6c425b
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/core/deployment/src/main/java/io/quarkiverse/langchain4j/deployment/BeansProcessor.java b/core/deployment/src/main/java/io/quarkiverse/langchain4j/deployment/BeansProcessor.java
@@ -139,6 +139,12 @@ private <T extends ProviderHolder> String selectProvider(
                     requestedBeanName));
         }
         if (availableProviders.size() == 1) {
+            // user has selected a provider, but it's not the one that is available
+            if (userSelectedProvider.isPresent() && !availableProviders.get(0).equals(userSelectedProvider.get())) {
+                throw new ConfigurationException(String.format(
+                        "A %s bean with provider=%s was requested was requested via configuration, but the only provider found on the classpath is %s.",
+                        requestedBeanName, userSelectedProvider.get(), availableProviders.get(0)));
+            }
             return availableProviders.get(0);
         }
         // multiple providers exist, so we now need the configuration to select the proper one
@@ -170,10 +176,16 @@ private <T extends ProviderHolder> String selectEmbeddingModelProvider(
             throw new ConfigurationException(String.format(
                     "A %s bean was requested, but no langchain4j providers were configured and no in-process embedding model were found on the classpath. "
                             +
-                            "Consider adding an extension like 'quarkus-langchain4j-openai' or one of the in-process embedding model.",
+                            "Consider adding an extension like 'quarkus-langchain4j-openai' or one of the in-process embedding models.",
                     requestedBeanName));
         }
         if (availableProviders.size() == 1) {
+            // user has selected a provider, but it's not the one that is available
+            if (userSelectedProvider.isPresent() && !availableProviders.get(0).equals(userSelectedProvider.get())) {
+                throw new ConfigurationException(String.format(
+                        "Embedding model provider %s was requested via configuration, but the only provider found on the classpath is %s.",
+                        userSelectedProvider.get(), availableProviders.get(0)));
+            }
             return availableProviders.get(0);
         }
         // multiple providers exist, so we now need the configuration to select the proper one