Fix language detecting.

yiwen-ai · Aug 13, 2023 · 6b6149d · 6b6149d
1 parent bd1bb49
commit 6b6149d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "jarvis"
-version = "0.7.6"
+version = "0.7.7"
 edition = "2021"
 rust-version = "1.64"
 description = ""

diff --git a/src/lang.rs b/src/lang.rs
@@ -1,8 +1,7 @@
+pub use isolang::Language;
 use lingua::LanguageDetectorBuilder;
+use std::str::FromStr;
 
-// pub use lingua::IsoCode639_1;
-// pub use lingua::Language;
-pub use isolang::Language;
 pub struct LanguageDetector {
     detector: lingua::LanguageDetector,
 }
@@ -22,9 +21,9 @@ impl LanguageDetector {
 
     pub fn detect_lang(&self, text: &str) -> Language {
         match self.detect(text) {
-            Some(lang) => match Language::from_name(lang.iso_code_639_3().to_string().as_str()) {
-                Some(lang) => lang,
-                None => Language::default(),
+            Some(lang) => match Language::from_str(lang.iso_code_639_3().to_string().as_str()) {
+                Ok(lang) => lang,
+                Err(_) => Language::default(),
             },
             None => Language::default(),
         }