From af934c5d09e66473b550e49bd6cce2aec607df85 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Fri, 30 May 2025 11:42:39 +0800 Subject: [PATCH] fix: correct dimension typo and enhance API registration logic --- astrbot/core/db/vec_db/faiss_impl/embedding_storage.py | 4 ++-- astrbot/core/star/context.py | 8 +++++++- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/astrbot/core/db/vec_db/faiss_impl/embedding_storage.py b/astrbot/core/db/vec_db/faiss_impl/embedding_storage.py index 262a459e3..18ee3189c 100644 --- a/astrbot/core/db/vec_db/faiss_impl/embedding_storage.py +++ b/astrbot/core/db/vec_db/faiss_impl/embedding_storage.py @@ -29,9 +29,9 @@ class EmbeddingStorage: Raises: ValueError: 如果向量的维度与存储的维度不匹配 """ - if vector.shape[0] != self.dimention: + if vector.shape[0] != self.dimension: raise ValueError( - f"向量维度不匹配, 期望: {self.dimention}, 实际: {vector.shape[0]}" + f"向量维度不匹配, 期望: {self.dimension}, 实际: {vector.shape[0]}" ) self.index.add_with_ids(vector.reshape(1, -1), np.array([id])) self.storage[id] = vector diff --git a/astrbot/core/star/context.py b/astrbot/core/star/context.py index 7cb3ffd1c..996b8ae5e 100644 --- a/astrbot/core/star/context.py +++ b/astrbot/core/star/context.py @@ -301,5 +301,11 @@ class Context: """ self._register_tasks.append(task) - def register_web_api(self, route: str, view_handler: Awaitable, methods: list, desc: str): + def register_web_api( + self, route: str, view_handler: Awaitable, methods: list, desc: str + ): + for idx, api in enumerate(self.registered_web_apis): + if api[0] == route and methods == api[2]: + self.registered_web_apis[idx] = (route, view_handler, methods, desc) + return self.registered_web_apis.append((route, view_handler, methods, desc))