Losita/mai-bot
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix：纠正语义修改

Browse Source
This commit is contained in:
DawnARC
2026-05-09 01:04:04 +08:00
parent 5fc6551f57
commit 5be2245830
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/A_memorix/core/utils/search_execution_service.py
View File

@@ -287,8 +287,11 @@ class SearchExecutionService:
) )
async def _executor() -> Dict[str, Any]: async def _executor() -> Dict[str, Any]:
original_ppr = bool(retriever.config.enable_ppr) retriever_config = getattr(retriever, "config", None)
retriever.config.enable_ppr = bool(request.enable_ppr) has_runtime_ppr_switch = retriever_config is not None and hasattr(retriever_config, "enable_ppr")
original_ppr = bool(retriever_config.enable_ppr) if has_runtime_ppr_switch else None
if has_runtime_ppr_switch:
retriever_config.enable_ppr = bool(request.enable_ppr)
started_at = time.time() started_at = time.time()
try: try:
retrieved = await retriever.retrieve( retrieved = await retriever.retrieve(
@@ -381,7 +384,8 @@ class SearchExecutionService:
elapsed_ms = (time.time() - started_at) * 1000.0 elapsed_ms = (time.time() - started_at) * 1000.0
return {"results": retrieved, "elapsed_ms": elapsed_ms} return {"results": retrieved, "elapsed_ms": elapsed_ms}
finally: finally:
retriever.config.enable_ppr = original_ppr if has_runtime_ppr_switch:
retriever_config.enable_ppr = bool(original_ppr)
dedup_hit = False dedup_hit = False
try: try: