init

fahmiaziz98 · fahmiaziz98 · commit 64b5451342a4 · 2025-11-02T22:09:14.000+07:00
diff --git a/src/api/routers/rerank.py b/src/api/routers/rerank.py
@@ -63,7 +63,7 @@ async def rerank_documents(
         start = time.time()
         if config.type == "rerank":
             scores = model.rank_document(
-                request.query, [doc for _, doc in valid_docs], request.top_k, **kwargs
+                request.query, request.documents, request.top_k, **kwargs
             )
             processing_time = time.time() - start
 

Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ async def rerank_documents(`
`63`	`63`	`start = time.time()`
`64`	`64`	`if config.type == "rerank":`
`65`	`65`	`scores = model.rank_document(`
`66`		`- request.query, [doc for _, doc in valid_docs], request.top_k, **kwargs`
	`66`	`+ request.query, request.documents, request.top_k, **kwargs`
`67`	`67`	`)`
`68`	`68`	`processing_time = time.time() - start`
`69`	`69`