🐞 fix: tiny fix

hcd233 · hcd233 · commit 2349bf903f50 · 2025-07-25T14:24:04.000+08:00
diff --git a/src/engine/reranker/infinity.py b/src/engine/reranker/infinity.py
@@ -26,8 +26,7 @@ def from_config(cls, config: InfinityRerankerConfig) -> "InfinityRerankerEngine"
         )
 
         # 创建 AsyncEmbeddingEngine
-        engine_array = AsyncEmbeddingEngine.from_args([engine_args])
-        engine = engine_array[0]  # 获取第一个引擎
+        engine = AsyncEmbeddingEngine.from_args(engine_args)
 
         logger.success(f"[InfinityRerankerEngine] load model from {config.path}")
         return cls(engine=engine, **config.model_dump())

Original file line number	Diff line number	Diff line change
`@@ -26,8 +26,7 @@ def from_config(cls, config: InfinityRerankerConfig) -> "InfinityRerankerEngine"`
`26`	`26`	`)`
`27`	`27`
`28`	`28`	`# 创建 AsyncEmbeddingEngine`
`29`		`- engine_array = AsyncEmbeddingEngine.from_args([engine_args])`
`30`		`- engine = engine_array[0] # 获取第一个引擎`
	`29`	`+ engine = AsyncEmbeddingEngine.from_args(engine_args)`
`31`	`30`
`32`	`31`	`logger.success(f"[InfinityRerankerEngine] load model from {config.path}")`
`33`	`32`	`return cls(engine=engine, **config.model_dump())`