Skip to content

Commit 2bd1fa3

Browse files
author
Pazuzzu
committed
resolves #1881 NLP: fix case, where no intialized sagemaker client returns OK
1 parent f766790 commit 2bd1fa3

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

nlp/model/sagemaker/src/main/kotlin/SagemakerEngineProvider.kt

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,11 +59,22 @@ class SagemakerEngineProvider : NlpEngineProvider {
5959
}
6060

6161
override fun healthcheck(): () -> NlpHealthcheckResult = {
62-
val grouped = SagemakerClientProvider.getAllClient().groupBy { it.name }.withDefault { emptyList() }
63-
NlpHealthcheckResult(
64-
entityClassifier = grouped.getValue(SagemakerEntityClassifier.CLIENT_TYPE.clientName).all { it.healthcheck() },
65-
intentClassifier = grouped.getValue(SagemakerIntentClassifier.CLIENT_TYPE.clientName).all { it.healthcheck() },
66-
)
62+
val clients = SagemakerClientProvider.getAllClient()
63+
if (clients.isEmpty()) {
64+
NlpHealthcheckResult(
65+
entityClassifier = false,
66+
intentClassifier = false
67+
)
68+
} else {
69+
val grouped = clients.groupBy { it.name }.withDefault { emptyList() }
70+
val entityClients = grouped.getValue(SagemakerEntityClassifier.CLIENT_TYPE.clientName)
71+
val intentClients = grouped.getValue(SagemakerIntentClassifier.CLIENT_TYPE.clientName)
72+
73+
NlpHealthcheckResult(
74+
entityClassifier = entityClients.isNotEmpty() && entityClients.all { it.healthcheck() },
75+
intentClassifier = intentClients.isNotEmpty() && intentClients.all { it.healthcheck() }
76+
)
77+
}
6778
}
6879
}
6980

0 commit comments

Comments
 (0)