diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2023-02-11 20:06:00 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-02-11 20:06:00 +0100 |
commit | 7dcada4f7521ec2fc5224e693ffa46744246b60c (patch) | |
tree | 846b3224b987c1874563be74c1906a2b6dc5f38a /model-evaluation | |
parent | 25dae690c989d6b0651cd619b8d3d2f7955d6b96 (diff) | |
parent | 5141973a055324c411aa8f745fe5b9300d02956b (diff) |
Merge pull request #25993 from vespa-engine/arnej/get-global-evaluator-from-registry-2
handle that the component graph contains multiple ModelsEvaluator instances
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 |