完成混合检索

2026-01-13 02:23:27 +08:00
parent 9190fee16f
commit d5ee00d404
3 changed files with 63 additions and 11 deletions
--- a/backend/services/crawler_service.py
+++ b/backend/services/crawler_service.py
@@ -142,9 +142,9 @@ class CrawlerService:
        
        return {"msg": "Batch processed", "count": processed}

-    def search(self, query: str, task_id: int, limit: int):
+    def search(self, query: str, task_id, limit: int):
        vector = llm_service.get_embedding(query)
        if not vector: return {"msg": "Embedding failed", "results": []}
-        return data_service.search(vector, task_id, limit)
+        return data_service.search(query_text=query, query_vector=vector, task_id=task_id, limit=limit)

 crawler_service = CrawlerService()