diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-02-10 14:39:35 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-02-10 14:44:38 +0000 |
commit | e292a99c6a1a4e3cd7b63361784e3a16c359ed4e (patch) | |
tree | 1c2c15ea91b4208d116821fe66f79054ce951314 /model-evaluation | |
parent | 54ebec2f273e408f2a7be8a610b6b1bc34305a57 (diff) |
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 | 10 |
1 files changed, 10 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..18d81e45819 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,6 +41,14 @@ public class ModelsEvaluationHandler extends ThreadedHttpRequestHandler { private final ModelsEvaluator modelsEvaluator; + @Inject + public ModelsEvaluationHandler(ComponentRegistry<ModelsEvaluator> registry, + Executor executor) + { + this(registry.getComponent("default/component/" + ModelsEvaluator.class.getName()), + executor); + } + public ModelsEvaluationHandler(ModelsEvaluator modelsEvaluator, Executor executor) { super(executor); this.modelsEvaluator = modelsEvaluator; |