Skip to content

Commit 5ef9546

Browse files
jtara1missinglink
authored andcommitted
feat: included probability in language_classifier
1 parent 559587f commit 5ef9546

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/language_classifier.cc

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,14 @@ void LanguageClassifier(const Nan::FunctionCallbackInfo<v8::Value>& info) {
2727

2828
if (response != NULL) {
2929
v8::Local<v8::Array> lang_array = Nan::New<v8::Array>(response->num_languages);
30-
for (size_t i = 0; i < response->num_languages; i++) {
31-
const char* language = response->languages[i]; // Directly access the array
30+
31+
for (size_t i = 0; i < response->num_languages; ++i) {
32+
const char *language = response->languages[i];
33+
const double probability = response->probs[i];
3234

3335
v8::Local<v8::Object> lang_obj = Nan::New<v8::Object>();
3436
Nan::Set(lang_obj, Nan::New("language").ToLocalChecked(), Nan::New(language).ToLocalChecked());
37+
Nan::Set(lang_obj, Nan::New("probability").ToLocalChecked(), Nan::New(probability));
3538

3639
Nan::Set(lang_array, i, lang_obj);
3740
}

0 commit comments

Comments
 (0)