diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-09-28 10:49:57 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-09-28 10:49:57 +0200 |
commit | 53fcc095410411fa4f9e85072717e02f3c1ecaa4 (patch) | |
tree | f895775a4c878bbcec31910ab38cfa0efa1cd243 /config-model | |
parent | d706455e3b682b6a713d94e22a7a9b622946d45f (diff) |
Register model using resolved URL if present
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java b/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java index 084258ce1a6..76d93c38aee 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java @@ -55,13 +55,15 @@ class Model { } void registerOnnxModelCost(ApplicationContainerCluster c) { + var resolvedUrl = resolvedUrl().orElse(null); if (file != null) c.onnxModelCost().registerModel(file); - else c.onnxModelCost().registerModel(url); + else if (resolvedUrl != null) c.onnxModelCost().registerModel(resolvedUrl); } String name() { return paramName; } Optional<String> modelId() { return Optional.ofNullable(modelId); } Optional<URI> url() { return Optional.ofNullable(url); } + Optional<URI> resolvedUrl() { return ref.url().map(u -> URI.create(u.value())); } Optional<ApplicationFile> file() { return Optional.ofNullable(file); } ModelReference modelReference() { return ref; } } |