diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-02-10 21:22:58 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-02-10 22:17:47 +0000 |
commit | 5141973a055324c411aa8f745fe5b9300d02956b (patch) | |
tree | 1e6101dfaae616bdd97862c61fcc8a3641e18bd9 /model-evaluation | |
parent | 6f6a7ee65709642ea32e250eca618ffcf6a08e20 (diff) |
handle that the component graph contains multiple ModelsEvaluator instances (take 2)
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java b/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java index 1bcd6363d2d..07388460422 100644 --- a/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java +++ b/model-evaluation/src/main/java/ai/vespa/models/handler/ModelsEvaluationHandler.java @@ -4,6 +4,8 @@ package ai.vespa.models.handler; import ai.vespa.models.evaluation.FunctionEvaluator; import ai.vespa.models.evaluation.Model; import ai.vespa.models.evaluation.ModelsEvaluator; +import com.yahoo.component.annotation.Inject; +import com.yahoo.component.provider.ComponentRegistry; import com.yahoo.container.jdisc.HttpRequest; import com.yahoo.container.jdisc.HttpResponse; import com.yahoo.container.jdisc.ThreadedHttpRequestHandler; @@ -39,9 +41,19 @@ public class ModelsEvaluationHandler extends ThreadedHttpRequestHandler { private final ModelsEvaluator modelsEvaluator; + @Inject + public ModelsEvaluationHandler(ComponentRegistry<ModelsEvaluator> registry, + Executor executor) + { + this(registry.getComponent(ModelsEvaluator.class.getName()), executor); + } + public ModelsEvaluationHandler(ModelsEvaluator modelsEvaluator, Executor executor) { super(executor); this.modelsEvaluator = modelsEvaluator; + if (modelsEvaluator == null) { + throw new IllegalArgumentException("missing ModelsEvaluator"); + } } @Override |