Update app.py

app.py

@@ -12,6 +12,7 @@ from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
 import threading
+import math
 
 # 创建安全缓存目录
 CACHE_DIR = "/home/user/cache"
@@ -136,6 +137,24 @@ threading.Thread(target=refresh_index, daemon=True).start()
 # 确保资源在API调用前加载
 load_resources()
 
+def sanitize_floats(obj):
+    if isinstance(obj, float):
+        if math.isnan(obj) or math.isinf(obj):
+            return 0.0  # 替换非法值为默认值
+        return obj
+    elif isinstance(obj, dict):
+        return {k: sanitize_floats(v) for k, v in obj.items()}
+    elif isinstance(obj, list):
+        return [sanitize_floats(x) for x in obj]
+    else:
+        return obj
+
+    # 在返回结果前调用清理器
+    return {
+        "status": "success",
+        "results": sanitize_floats(results)  # 深度清理
+    }
+
 def predict(vector):
     try:
         print(f"接收向量: {vector[:3]}... (长度: {len(vector)})")
@@ -164,56 +183,45 @@ def predict(vector):
         for i in range(k):
             try:
                 idx = I[0][i]
-                
-                # # 安全处理距离值
-                # distance = D[0][i]
-                # if not np.isfinite(distance) or distance < 0:
-                #     print(f"警告: 无效距离值 {distance}，使用默认值0")
-                #     distance = 0.0
-                
-                # # 安全处理置信度值
-                # try:
-                #     confidence = 1 / (1 + distance)
-                # except ZeroDivisionError:
-                #     confidence = 1.0
-                
-                # if not np.isfinite(confidence) or confidence < 0:
-                #     print(f"警告: 无效置信度 {confidence}，使用默认值0.5")
-                #     confidence = 0.5
-
                 distance = D[0][i]
     
-                # 修复非法浮点数
-                if not np.isfinite(distance):
-                    distance = 100.0  # 设置为最大距离阈值
-                    
-                # 确保confidence在[0,1]范围内
-                confidence = max(0.0, min(1.0, 1/(1+distance)))
-                
+                # 修复1：处理非法浮点数
+                if not np.isfinite(distance) or distance < 0:
+                    distance = 100.0  # 设置为安全阈值
+    
+                # 修复2：安全计算置信度 (0-1范围)
+                confidence = 1 / (1 + distance)
+                confidence = max(0.0, min(1.0, confidence))  # 钳制到[0,1]
+    
+                # 修复3：强制转换为合法浮点
+                distance = float(distance)
+                confidence = float(confidence)
+    
                 result = {
                     "source": metadata.iloc[idx]["source"],
                     "content": metadata.iloc[idx].get("content", ""),
-                    "confidence": float(confidence),
-                    "distance": float(distance)
+                    "confidence": confidence,
+                    "distance": distance
                 }
                 results.append(result)
             except Exception as e:
-                print(f"结果处理错误: {str(e)}")
+                # 确保异常结果也符合JSON规范
                 results.append({
-                    "error": f"结果 {i+1}: 数据获取失败",
+                    "error": str(e),
                     "confidence": 0.5,
                     "distance": 0.0
                 })
 
         return {
             "status": "success",
-            "results": results
+            "results": sanitize_floats(results)
         }
     except Exception as e:
         # 返回错误响应
         return {
             "status": "error",
-            "message": f"服务器内部错误: {str(e)}"
+            "message": f"服务器内部错误: {str(e)}",
+            "details": sanitize_floats({"trace": traceback.format_exc()})
         }
         
 # 创建FastAPI应用