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)