refactor: enhance error handling in tokenization and detokenization processes with clearer messages
This commit is contained in:
@@ -1,6 +1,5 @@
|
|||||||
package de.extio.lmlib.token;
|
package de.extio.lmlib.token;
|
||||||
|
|
||||||
import java.io.InputStream;
|
|
||||||
import java.nio.charset.StandardCharsets;
|
import java.nio.charset.StandardCharsets;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
|
||||||
@@ -41,11 +40,14 @@ final class LlamaServerTokenizer implements Tokenizer {
|
|||||||
.onStatus(status -> {
|
.onStatus(status -> {
|
||||||
if (status.getStatusCode().isError()) {
|
if (status.getStatusCode().isError()) {
|
||||||
LOGGER.error("Tokenization request failed with status code {} {}: {}", status.getStatusCode(), status.getStatusText(), new String(status.getBody().readAllBytes(), StandardCharsets.UTF_8));
|
LOGGER.error("Tokenization request failed with status code {} {}: {}", status.getStatusCode(), status.getStatusText(), new String(status.getBody().readAllBytes(), StandardCharsets.UTF_8));
|
||||||
throw new RuntimeException("Tokenization request failed with status code " + status.getStatusCode() + " " + status.getStatusText());
|
throw new RuntimeException("Tokenization request failed with status code " + status.getStatusCode() + " " + status.getStatusText() + ". Your LLM provider may not support the /tokenize endpoint");
|
||||||
}
|
}
|
||||||
return false;
|
return false;
|
||||||
})
|
})
|
||||||
.body(TokenizeResponse.class);
|
.body(TokenizeResponse.class);
|
||||||
|
if (response == null || response.tokens() == null) {
|
||||||
|
throw new RuntimeException("Tokenization request failed: response is null. Your LLM provider may not support the /tokenize endpoint");
|
||||||
|
}
|
||||||
return response.tokens();
|
return response.tokens();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -67,11 +69,14 @@ final class LlamaServerTokenizer implements Tokenizer {
|
|||||||
.onStatus(status -> {
|
.onStatus(status -> {
|
||||||
if (status.getStatusCode().isError()) {
|
if (status.getStatusCode().isError()) {
|
||||||
LOGGER.error("Detokenization request failed with status code {} {}: {}", status.getStatusCode(), status.getStatusText(), new String(status.getBody().readAllBytes(), StandardCharsets.UTF_8));
|
LOGGER.error("Detokenization request failed with status code {} {}: {}", status.getStatusCode(), status.getStatusText(), new String(status.getBody().readAllBytes(), StandardCharsets.UTF_8));
|
||||||
throw new RuntimeException("Detokenization request failed with status code " + status.getStatusCode() + " " + status.getStatusText());
|
throw new RuntimeException("Detokenization request failed with status code " + status.getStatusCode() + " " + status.getStatusText() + ". Your LLM provider may not support the /detokenize endpoint");
|
||||||
}
|
}
|
||||||
return false;
|
return false;
|
||||||
})
|
})
|
||||||
.body(DetokenizeResponse.class);
|
.body(DetokenizeResponse.class);
|
||||||
|
if (response == null || response.content() == null) {
|
||||||
|
throw new RuntimeException("Detokenization request failed: response is null. Your LLM provider may not support the /detokenize endpoint");
|
||||||
|
}
|
||||||
return response.content();
|
return response.content();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user