From dafc7092adbd0401cc1fbba13d734f7f59c277d9 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 28 May 2026 13:18:27 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/rag/query_decompose.py | 4 +--- studio/backend/routes/rag.py | 4 +--- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/studio/backend/core/rag/query_decompose.py b/studio/backend/core/rag/query_decompose.py index f6a020f8db..6602827fd6 100644 --- a/studio/backend/core/rag/query_decompose.py +++ b/studio/backend/core/rag/query_decompose.py @@ -108,9 +108,7 @@ def decompose_query(query: str) -> list[str]: response = client.post(endpoint, json = payload) response.raise_for_status() data = response.json() - content = ( - data.get("choices", [{}])[0].get("message", {}).get("content", "") - ) + content = data.get("choices", [{}])[0].get("message", {}).get("content", "") queries = _parse_queries(content if isinstance(content, str) else "", q) logger.info("RAG query-decompose: produced queries", n = len(queries)) return queries diff --git a/studio/backend/routes/rag.py b/studio/backend/routes/rag.py index 28f39e4d43..ef69fcfed3 100644 --- a/studio/backend/routes/rag.py +++ b/studio/backend/routes/rag.py @@ -1638,9 +1638,7 @@ def _execute_search( ) -> list[SearchHit]: """Run one retrieval against an already-resolved scope. Shared by the /search and /prefetch endpoints.""" - candidate_k = ( - max(top_k, RAG_RERANK_CANDIDATE_K) if enable_rerank else top_k - ) + candidate_k = max(top_k, RAG_RERANK_CANDIDATE_K) if enable_rerank else top_k if mode == "bm25": hits = retrieval.retrieve_bm25(scope, query, candidate_k)