Merge branch 'feat/add-plugin-readme-buttom' of https://github.com/VanillaNahida/AstrBot into feat/add-plugin-readme-buttom

Author: VanillaNahida

Dockerfile

@@ -12,6 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     ca-certificates \
     bash \
     ffmpeg \
+    libavcodec-extra \
     curl \
     gnupg \
     git \

README.md

@@ -184,6 +184,7 @@ Connect AstrBot to your favorite chat platform.
 | Coze | LLMOps Platforms |
 | OpenAI Whisper | Speech-to-Text Services |
 | SenseVoice | Speech-to-Text Services |
+| Xiaomi MiMo Omni | Speech-to-Text Services |
 | OpenAI TTS | Text-to-Speech Services |
 | Gemini TTS | Text-to-Speech Services |
 | GPT-Sovits-Inference | Text-to-Speech Services |
@@ -193,6 +194,7 @@ Connect AstrBot to your favorite chat platform.
 | Alibaba Cloud Bailian TTS | Text-to-Speech Services |
 | Azure TTS | Text-to-Speech Services |
 | Minimax TTS | Text-to-Speech Services |
+| Xiaomi MiMo TTS | Text-to-Speech Services |
 | Volcano Engine TTS | Text-to-Speech Services |
 
 ## ❤️ Sponsors

README_fr.md

@@ -184,6 +184,7 @@ Connectez AstrBot à vos plateformes de chat préférées.
 | Coze | Plateformes LLMOps |
 | OpenAI Whisper | Services de reconnaissance vocale |
 | SenseVoice | Services de reconnaissance vocale |
+| Xiaomi MiMo Omni | Services de reconnaissance vocale |
 | OpenAI TTS | Services de synthèse vocale |
 | Gemini TTS | Services de synthèse vocale |
 | GPT-Sovits-Inference | Services de synthèse vocale |
@@ -193,6 +194,7 @@ Connectez AstrBot à vos plateformes de chat préférées.
 | Alibaba Cloud Bailian TTS | Services de synthèse vocale |
 | Azure TTS | Services de synthèse vocale |
 | Minimax TTS | Services de synthèse vocale |
+| Xiaomi MiMo TTS | Services de synthèse vocale |
 | Volcano Engine TTS | Services de synthèse vocale |
 
 ## ❤️ Contribuer

README_ja.md

@@ -185,6 +185,7 @@ AstrBot をよく使うチャットプラットフォームに接続できます
 | Coze | LLMOps プラットフォーム |
 | OpenAI Whisper | 音声認識サービス |
 | SenseVoice | 音声認識サービス |
+| Xiaomi MiMo Omni | 音声認識サービス |
 | OpenAI TTS | 音声合成サービス |
 | Gemini TTS | 音声合成サービス |
 | GPT-Sovits-Inference | 音声合成サービス |
@@ -194,6 +195,7 @@ AstrBot をよく使うチャットプラットフォームに接続できます
 | Alibaba Cloud 百炼 TTS | 音声合成サービス |
 | Azure TTS | 音声合成サービス |
 | Minimax TTS | 音声合成サービス |
+| Xiaomi MiMo TTS | 音声合成サービス |
 | Volcano Engine TTS | 音声合成サービス |
 
 ## ❤️ コントリビューション

README_ru.md

@@ -184,6 +184,7 @@ yay -S astrbot-git
 | Coze | Платформы LLMOps |
 | OpenAI Whisper | Сервисы распознавания речи |
 | SenseVoice | Сервисы распознавания речи |
+| Xiaomi MiMo Omni | Сервисы распознавания речи |
 | OpenAI TTS | Сервисы синтеза речи |
 | Gemini TTS | Сервисы синтеза речи |
 | GPT-Sovits-Inference | Сервисы синтеза речи |
@@ -193,6 +194,7 @@ yay -S astrbot-git
 | Alibaba Cloud Bailian TTS | Сервисы синтеза речи |
 | Azure TTS | Сервисы синтеза речи |
 | Minimax TTS | Сервисы синтеза речи |
+| Xiaomi MiMo TTS | Сервисы синтеза речи |
 | Volcano Engine TTS | Сервисы синтеза речи |
 
 ## ❤️ Вклад в проект

README_zh-TW.md

@@ -184,6 +184,7 @@ yay -S astrbot-git
 | Coze | LLMOps 平台 |
 | OpenAI Whisper | 語音轉文字服務 |
 | SenseVoice | 語音轉文字服務 |
+| Xiaomi MiMo Omni | 語音轉文字服務 |
 | OpenAI TTS | 文字轉語音服務 |
 | Gemini TTS | 文字轉語音服務 |
 | GPT-Sovits-Inference | 文字轉語音服務 |
@@ -193,6 +194,7 @@ yay -S astrbot-git
 | 阿里雲百煉 TTS | 文字轉語音服務 |
 | Azure TTS | 文字轉語音服務 |
 | Minimax TTS | 文字轉語音服務 |
+| Xiaomi MiMo TTS | 文字轉語音服務 |
 | 火山引擎 TTS | 文字轉語音服務 |
 
 ## ❤️ 貢獻

README_zh.md

@@ -185,6 +185,7 @@ yay -S astrbot-git
 | Coze | LLMOps 平台 |
 | OpenAI Whisper | 语音转文本 |
 | SenseVoice | 语音转文本 |
+| Xiaomi MiMo Omni | 语音转文本 |
 | OpenAI TTS | 文本转语音 |
 | Gemini TTS | 文本转语音 |
 | GPT-Sovits-Inference | 文本转语音 |
@@ -194,6 +195,7 @@ yay -S astrbot-git
 | 阿里云百炼 TTS | 文本转语音 |
 | Azure TTS | 文本转语音 |
 | Minimax TTS | 文本转语音 |
+| Xiaomi MiMo TTS | 文本转语音 |
 | 火山引擎 TTS | 文本转语音 |
 
 ## ❤️ 贡献

astrbot/core/config/default.py

@@ -4040,9 +4040,9 @@ class ChatProviderTemplate(TypedDict):
                         "hint": "时区设置。请填写 IANA 时区名称, 如 Asia/Shanghai, 为空时使用系统默认时区。所有时区请查看: https://data.iana.org/time-zones/tzdb-2021a/zone1970.tab",
                     },
                     "http_proxy": {
-                        "description": "HTTP 代理",
+                        "description": "代理",
                         "type": "string",
-                        "hint": "启用后，会以添加环境变量的方式设置代理。格式为 `http://ip:port`",
+                        "hint": "启用后，会以添加环境变量的方式设置代理。支持 http://、https://、socks5:// 格式，例如：http://127.0.0.1:7890 或 socks5://127.0.0.1:7891",
                     },
                     "no_proxy": {
                         "description": "直连地址列表",

astrbot/core/knowledge_base/kb_helper.py

@@ -108,6 +108,7 @@ async def _repair_and_translate_chunk_with_retry(
 class KBHelper:
     vec_db: BaseVecDB
     kb: KnowledgeBase
+    init_error: str | None
 
     def __init__(
         self,
@@ -122,6 +123,7 @@ def __init__(
         self.prov_mgr = provider_manager
         self.kb_root_dir = kb_root_dir
         self.chunker = chunker
+        self.init_error = None
 
         self.kb_dir = Path(self.kb_root_dir) / self.kb.kb_id
         self.kb_medias_dir = Path(self.kb_dir) / "medias" / self.kb.kb_id
@@ -148,21 +150,28 @@ async def get_ep(self) -> EmbeddingProvider:
     async def get_rp(self) -> RerankProvider | None:
         if not self.kb.rerank_provider_id:
             return None
-        rp: RerankProvider = await self.prov_mgr.get_provider_by_id(
+        rp: RerankProvider | None = await self.prov_mgr.get_provider_by_id(
             self.kb.rerank_provider_id,
         )  # type: ignore
         if not rp:
-            raise ValueError(
-                f"无法找到 ID 为 {self.kb.rerank_provider_id} 的 Rerank Provider",
+            logger.warning(
+                f"知识库 {self.kb.kb_name}({self.kb.kb_id}) 的 Rerank Provider({self.kb.rerank_provider_id}) 不可用，将跳过重排序。",
             )
+            return None
         return rp
 
     async def _ensure_vec_db(self) -> FaissVecDB:
         if not self.kb.embedding_provider_id:
             raise ValueError(f"知识库 {self.kb.kb_name} 未配置 Embedding Provider")
 
         ep = await self.get_ep()
-        rp = await self.get_rp()
+        rp: RerankProvider | None = None
+        try:
+            rp = await self.get_rp()
+        except Exception as e:
+            logger.warning(
+                f"知识库 {self.kb.kb_name}({self.kb.kb_id}) 初始化重排序能力失败，将跳过重排序: {e}",
+            )
 
         vec_db = FaissVecDB(
             doc_store_path=str(self.kb_dir / "doc.db"),
@@ -172,6 +181,8 @@ async def _ensure_vec_db(self) -> FaissVecDB:
         )
         await vec_db.initialize()
         self.vec_db = vec_db
+        # Clear stale init_error once initialization succeeds.
+        self.init_error = None
         return vec_db
 
     async def delete_vec_db(self) -> None:
@@ -183,7 +194,7 @@ async def delete_vec_db(self) -> None:
             shutil.rmtree(self.kb_dir)
 
     async def terminate(self) -> None:
-        if self.vec_db:
+        if hasattr(self, "vec_db") and self.vec_db:
             await self.vec_db.close()
 
     async def upload_document(

astrbot/core/knowledge_base/kb_mgr.py

@@ -1,4 +1,3 @@
-import traceback
 from pathlib import Path
 
 from astrbot.core import logger
@@ -56,8 +55,7 @@ async def initialize(self) -> None:
             logger.error(f"知识库模块导入失败: {e}")
             logger.warning("请确保已安装所需依赖: pypdf, aiofiles, Pillow, rank-bm25")
         except Exception as e:
-            logger.error(f"知识库模块初始化失败: {e}")
-            logger.error(traceback.format_exc())
+            logger.error(f"知识库模块初始化失败: {e}", exc_info=True)
 
     async def _init_kb_database(self) -> None:
         self.kb_db = KBSQLiteDatabase(DB_PATH.as_posix())
@@ -76,7 +74,14 @@ async def load_kbs(self) -> None:
                 kb_root_dir=FILES_PATH,
                 chunker=CHUNKER,
             )
-            await kb_helper.initialize()
+            try:
+                await kb_helper.initialize()
+            except Exception as e:
+                kb_helper.init_error = str(e)
+                logger.error(
+                    f"知识库 {record.kb_name}({record.kb_id}) 初始化失败: {e}",
+                    exc_info=True,
+                )
             self.kb_insts[record.kb_id] = kb_helper
 
     async def create_kb(
@@ -179,6 +184,20 @@ async def update_kb(
             return None
 
         kb = kb_helper.kb
+        previous_state = {
+            "kb_name": kb.kb_name,
+            "description": kb.description,
+            "emoji": kb.emoji,
+            "embedding_provider_id": kb.embedding_provider_id,
+            "rerank_provider_id": kb.rerank_provider_id,
+            "chunk_size": kb.chunk_size,
+            "chunk_overlap": kb.chunk_overlap,
+            "top_k_dense": kb.top_k_dense,
+            "top_k_sparse": kb.top_k_sparse,
+            "top_m_final": kb.top_m_final,
+        }
+        previous_init_error = kb_helper.init_error
+
         if kb_name is not None:
             kb.kb_name = kb_name
         if description is not None:
@@ -198,12 +217,47 @@ async def update_kb(
             kb.top_k_sparse = top_k_sparse
         if top_m_final is not None:
             kb.top_m_final = top_m_final
+
+        # Build a new helper first. Keep current vec_db alive until new init succeeds.
+        new_helper = KBHelper(
+            kb_db=self.kb_db,
+            kb=kb,
+            provider_manager=self.provider_manager,
+            kb_root_dir=FILES_PATH,
+            chunker=CHUNKER,
+        )
+
+        try:
+            await new_helper.initialize()
+        except Exception as e:
+            # Roll back in-memory settings and keep current helper available.
+            kb.kb_name = previous_state["kb_name"]
+            kb.description = previous_state["description"]
+            kb.emoji = previous_state["emoji"]
+            kb.embedding_provider_id = previous_state["embedding_provider_id"]
+            kb.rerank_provider_id = previous_state["rerank_provider_id"]
+            kb.chunk_size = previous_state["chunk_size"]
+            kb.chunk_overlap = previous_state["chunk_overlap"]
+            kb.top_k_dense = previous_state["top_k_dense"]
+            kb.top_k_sparse = previous_state["top_k_sparse"]
+            kb.top_m_final = previous_state["top_m_final"]
+            kb_helper.init_error = previous_init_error
+            logger.error(
+                f"知识库 {kb.kb_name}({kb.kb_id}) 重新初始化失败，继续使用旧实例: {e}",
+                exc_info=True,
+            )
+            return kb_helper
+
         async with self.kb_db.get_db() as session:
             session.add(kb)
             await session.commit()
             await session.refresh(kb)
 
-        return kb_helper
+        old_helper = kb_helper
+        self.kb_insts[kb_id] = new_helper
+        await old_helper.terminate()
+        new_helper.init_error = None
+        return new_helper
 
     async def retrieve(
         self,
@@ -215,11 +269,21 @@ async def retrieve(
         """从指定知识库中检索相关内容"""
         kb_ids = []
         kb_id_helper_map = {}
+        unavailable_kbs = []
         for kb_name in kb_names:
             if kb_helper := await self.get_kb_by_name(kb_name):
+                if kb_helper.init_error:
+                    unavailable_kbs.append((kb_name, kb_helper.init_error))
+                    logger.warning(f"知识库 {kb_name} 不可用: {kb_helper.init_error}")
+                    continue
                 kb_ids.append(kb_helper.kb.kb_id)
                 kb_id_helper_map[kb_helper.kb.kb_id] = kb_helper
 
+        # all requested KBs are unavailable
+        if not kb_ids and unavailable_kbs:
+            errors = "; ".join(f"{n}: {e}" for n, e in unavailable_kbs)
+            raise ValueError(f"所有请求的知识库均不可用: {errors}")
+
         if not kb_ids:
             return {}