fix: address review feedback — L2 consistency, reshape, tempfile, platform guard

Author: irmia2026

app/astrbot/core/db/vec_db/faiss_impl/embedding_storage.py

@@ -7,6 +7,7 @@
 import os
 import shutil
 import tempfile
+import uuid
 
 import numpy as np
 
@@ -27,27 +28,46 @@ def _safe_temp_dir() -> str:
     1. %SystemRoot%\\Temp（Windows 系统临时目录，如 C:\\WINDOWS\\TEMP）
     2. tempfile.gettempdir()（当其为纯 ASCII 时）
     3. 当前工作目录
+    4. 非 Windows 平台使用 tempfile.gettempdir()
     """
-    candidates = []
-    # C:\Windows\Temp — 所有 Windows 安装的纯 ASCII 路径
-    root = os.environ.get("SystemRoot", r"C:\Windows")
-    candidates.append(os.path.join(root, "Temp"))
-    # 标准 tempdir（可能是 ASCII 也可能不是）
-    candidates.append(tempfile.gettempdir())
-    # 兜底：当前目录
-    try:
-        candidates.append(os.getcwd())
-    except OSError:
-        pass
-
-    for d in candidates:
-        if d.isascii() and os.path.isdir(d) and os.access(d, os.W_OK):
-            return d
-
-    # 终极兜底：创建新目录
-    fallback = r"C:\Windows\Temp"
-    os.makedirs(fallback, exist_ok=True)
-    return fallback
+    # Windows 专属硬编码
+    if os.name == "nt":
+        candidates = []
+        root = os.environ.get("SystemRoot", r"C:\Windows")
+        candidates.append(os.path.join(root, "Temp"))
+        candidates.append(tempfile.gettempdir())
+        try:
+            candidates.append(os.getcwd())
+        except OSError:
+            pass
+
+        for d in candidates:
+            if d.isascii() and os.path.isdir(d) and os.access(d, os.W_OK):
+                return d
+
+        # 所有候选都不行时抛异常，不再静默兜底
+        raise OSError(
+            f"_safe_temp_dir: 无法找到可写的纯 ASCII 临时目录。"
+            f"检查过: {candidates}"
+        )
+
+    # 非 Windows（Linux / macOS）：tempfile 足够
+    return tempfile.gettempdir()
+
+
+def _make_temp_file(prefix: str) -> str:
+    """创建用于 Faiss 桥接的唯一临时文件，返回路径。
+
+    使用 tempfile.mkstemp + UUID 保证多线程/多协程并发安全。
+    """
+    safe_dir = _safe_temp_dir()
+    fd, path = tempfile.mkstemp(
+        prefix=f"{prefix}_{uuid.uuid4().hex[:8]}_",
+        suffix=".faiss",
+        dir=safe_dir,
+    )
+    os.close(fd)
+    return path
 
 
 class EmbeddingStorage:
@@ -74,15 +94,16 @@ def _read_index(path: str) -> "faiss.Index":
         except RuntimeError:
             pass  # 不吞其他异常类型
 
-        tmp = os.path.join(_safe_temp_dir(), f"_faiss_read_{os.getpid()}.faiss")
+        tmp = _make_temp_file("_faiss_read")
         try:
             shutil.copy2(path, tmp)
             return faiss.read_index(tmp)
         finally:
-            try:
-                os.remove(tmp)
-            except OSError:
-                pass
+            if os.path.exists(tmp):
+                try:
+                    os.remove(tmp)
+                except OSError:
+                    pass
 
     @staticmethod
     def _write_index(index: "faiss.Index", path: str) -> None:
@@ -94,18 +115,21 @@ def _write_index(index: "faiss.Index", path: str) -> None:
 
         写入前先确保目标目录存在，防止 shutil.move 时目录缺失。
         """
-        os.makedirs(os.path.dirname(path), exist_ok=True)
+        dirname = os.path.dirname(path)
+        if dirname:
+            os.makedirs(dirname, exist_ok=True)
 
-        tmp = os.path.join(_safe_temp_dir(), f"_faiss_write_{os.getpid()}.faiss")
+        tmp = _make_temp_file("_faiss_write")
         try:
             faiss.write_index(index, tmp)
             # Windows 同盘 move 是原子 rename，跨盘则走 copy+delete
             shutil.move(tmp, path)
         finally:
-            try:
-                os.remove(tmp)
-            except OSError:
-                pass
+            if os.path.exists(tmp):
+                try:
+                    os.remove(tmp)
+                except OSError:
+                    pass
 
     async def insert(self, vector: np.ndarray, id: int) -> None:
         """插入向量
@@ -122,7 +146,7 @@ async def insert(self, vector: np.ndarray, id: int) -> None:
             raise ValueError(
                 f"向量维度不匹配, 期望: {self.dimension}, 实际: {vector.shape[0]}",
             )
-        self.index.add_with_ids(vector.reshape(1, -1), np.array([id]))
+        self.index.add_with_ids(vector.reshape(1, -1), np.array([id], dtype=np.int64))
         await self.save_index()
 
     async def insert_batch(self, vectors: np.ndarray, ids: list[int]) -> None:
@@ -140,7 +164,7 @@ async def insert_batch(self, vectors: np.ndarray, ids: list[int]) -> None:
             raise ValueError(
                 f"向量维度不匹配, 期望: {self.dimension}, 实际: {vectors.shape[1]}",
             )
-        self.index.add_with_ids(vectors, np.array(ids))
+        self.index.add_with_ids(vectors, np.array(ids, dtype=np.int64))
         await self.save_index()
 
     async def search(self, vector: np.ndarray, k: int) -> tuple:
@@ -154,8 +178,9 @@ async def search(self, vector: np.ndarray, k: int) -> tuple:
 
         """
         assert self.index is not None, "FAISS index is not initialized."
-        faiss.normalize_L2(vector)
-        distances, indices = self.index.search(vector, k)
+        # IndexFlatL2 是欧氏距离索引，不进行归一化，
+        # 确保与 insert/insert_batch 的一致性
+        distances, indices = self.index.search(vector.reshape(1, -1), k)
         return distances, indices
 
     async def delete(self, ids: list[int]) -> None:
@@ -172,6 +197,6 @@ async def delete(self, ids: list[int]) -> None:
 
     async def save_index(self) -> None:
         """保存索引（兼容含非 ASCII 字符的 Windows 路径）"""
-        if self.index is None:
+        if self.index is None or not self.path:
             return
         self._write_index(self.index, self.path)