fz-mirrors/AstrBot
1
0
Fork 0
mirror of https://github.com/AstrBotDevs/AstrBot synced 2026-07-15 17:30:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct dimension typo and enhance API registration logic

Browse Source
This commit is contained in:
Soulter
2025-05-30 11:42:39 +08:00
parent 1e0cf7c112
commit af934c5d09
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
astrbot/core/db/vec_db/faiss_impl/embedding_storage.py
View File

@@ -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

8
astrbot/core/star/context.py
View File

@@ -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))