添加search方法

backend/main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from .service import crawler_service
-from .schemas import RegisterRequest, PendingRequest, SaveResultsRequest, AddUrlsRequest
+from .schemas import RegisterRequest, PendingRequest, SaveResultsRequest, AddUrlsRequest, SearchRequest
 from .utils import make_response
 
 app = FastAPI(title="Wiki Crawler API")
@@ -37,7 +37,41 @@ async def save_results(req: SaveResultsRequest):
         return make_response(1, "Success", data)
     except Exception as e:
         return make_response(0, str(e))
-    
+
+@app.post("/search")
+async def search(req: SearchRequest):
+    """
+    通用搜索接口：
+    支持基于 task_id 的局部搜索，也支持不传 task_id 的全库搜索。
+    """
+    try:
+        # 1. 基础校验：确保向量不为空且维度正确（阿里 v4 模型通常为 1536）
+        if not req.query_embedding or len(req.query_embedding) != 1536:
+            return make_response(
+                code=2, 
+                msg=f"向量维度错误。期望 1536, 实际收到 {len(req.query_embedding) if req.query_embedding else 0}",
+                data=None
+            )
+
+        # 2. 调用业务类执行搜索
+        data = crawler_service.search_knowledge(
+            query_embedding=req.query_embedding,
+            task_id=req.task_id,
+            limit=req.limit
+        )
+
+        # 3. 统一返回
+        return make_response(
+            code=1, 
+            msg="搜索完成", 
+            data=data
+        )
+
+    except Exception as e:
+        # 记录日志并返回失败信息
+        print(f"搜索接口异常: {str(e)}")
+        return make_response(code=0, msg=f"搜索失败: {str(e)}")
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000, reload=True)