aaaa

backend/main.py

@@ -16,7 +16,8 @@ async def register(req: RegisterRequest):
 @app.post("/add_urls")
 async def add_urls(req: AddUrlsRequest):
     try:
-        data = crawler_service.add_urls(req.task_id, req.urls_obj)
+        urls = req.urls_obj["urls"]
+        data = crawler_service.add_urls(req.task_id, urls=urls)
         return make_response(1, "Success", data)
     except Exception as e:
         return make_response(0, str(e))
@@ -46,7 +47,9 @@ async def search(req: SearchRequest):
     """
     try:
         # 1. 基础校验：确保向量不为空且维度正确（阿里 v4 模型通常为 1536）
-        if not req.query_embedding or len(req.query_embedding) != 1536:
+        vector = req.query_embedding['vector']
+
+        if not vector or len(vector) != 1536:
             return make_response(
                 code=2, 
                 msg=f"向量维度错误。期望 1536, 实际收到 {len(req.query_embedding) if req.query_embedding else 0}",
@@ -55,7 +58,7 @@ async def search(req: SearchRequest):
 
         # 2. 调用业务类执行搜索
         data = crawler_service.search_knowledge(
-            query_embedding=req.query_embedding,
+            query_embedding=vector,
             task_id=req.task_id,
             limit=req.limit
         )