Patch #420

llmkira/extra/voice_hook.py

@@ -11,18 +11,32 @@
 from llmkira.task.schema import EventMessage, Location
 
 
-def check_string(text):
+def detect_text(text: str) -> list:
     """
-    检查字符串是否 TTS 可以处理
-    :param text: 字符串
-    :return: 是否符合要求
+    检测文本的语言
+    :param text: 文本
+    :return: 语言
     """
+    text = text.replace("\n", "")
+    text = text[:200]
     parsed_text = detect_multilingual(text)
     if not parsed_text:
-        return False
+        return []
     lang_kinds = []
     for lang in parsed_text:
-        lang_kinds.append(lang.get("lang", "ru"))
+        lang_ = lang.get("lang", None)
+        if lang_:
+            lang_kinds.append(lang_)
+    return lang_kinds
+
+
+def check_string(text):
+    """
+    检查字符串是否 TTS 可以处理
+    :param text: 字符串
+    :return: 是否符合要求
+    """
+    lang_kinds = detect_text(text)
     limit = 200
     if len(set(lang_kinds)) == 1:
         if lang_kinds[0] in ["en"]:
@@ -57,12 +71,7 @@ async def hook_run(self, *args, **kwargs):
         for message in messages:
             if not check_string(message.text):
                 return args, kwargs
-            parsed_text = detect_multilingual(message.text)
-            if not parsed_text:
-                return args, kwargs
-            lang_kinds = []
-            for lang in parsed_text:
-                lang_kinds.append(lang.get("lang"))
+            lang_kinds = detect_text(message.text)
             reecho_api_key = await EnvManager(locate.uid).get_env(
                 "REECHO_VOICE_KEY", None
             )

pyproject.toml

@@ -19,7 +19,6 @@ dependencies = [
     "shortuuid<2.0.0,>=1.0.11",
     "contextvars<3.0,>=2.4",
     "pytz<2024.0.0,>=2023.3.post1",
-    "fasttext-wheel<1.0.0,>=0.9.2",
     "tenacity<9.0.0,>=8.2.3",
     "pysocks<2.0.0,>=1.7.1",
     "flask-sqlalchemy<4.0.0,>=3.1.1",
@@ -53,7 +52,7 @@ dependencies = [
     "apscheduler>=3.10.4",
     "montydb[lmdb]>=2.5.2",
     "pymongo>=4.6.3",
-    "fast-langdetect>=0.1.1",
+    "fast-langdetect>=0.2.1",
     "lmdb>=1.4.1",
     "e2b>=0.14.14",
     "e2b-code-interpreter>=0.0.3",