feat: refactor rerank

earayu · earayu · commit c9b42625fc2e · 2025-04-11T22:13:39.000+08:00
diff --git a/aperag/rank/reranker.py b/aperag/rank/reranker.py
@@ -75,7 +75,7 @@ async def rank(self, query, results: List[DocumentWithScore]):
         }
         headers = {
             "Content-Type": "application/json",
-            "Authorization": f"Bearer ${self.auth_token}"
+            "Authorization": f"Bearer {self.auth_token}"
         }
         async with aiohttp.ClientSession() as session:
             async with session.post(self.url, headers=headers, json=body) as resp:

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ async def rank(self, query, results: List[DocumentWithScore]):`
`75`	`75`	`}`
`76`	`76`	`headers = {`
`77`	`77`	`"Content-Type": "application/json",`
`78`		`- "Authorization": f"Bearer ${self.auth_token}"`
	`78`	`+ "Authorization": f"Bearer {self.auth_token}"`
`79`	`79`	`}`
`80`	`80`	`async with aiohttp.ClientSession() as session:`
`81`	`81`	`async with session.post(self.url, headers=headers, json=body) as resp:`