diff --git a/backend/onyx/context/search/retrieval/search_runner.py b/backend/onyx/context/search/retrieval/search_runner.py index 6c77167ad..d8cdd112b 100644 --- a/backend/onyx/context/search/retrieval/search_runner.py +++ b/backend/onyx/context/search/retrieval/search_runner.py @@ -256,9 +256,15 @@ def retrieve_chunks( continue simplified_queries.add(simplified_rephrase) - q_copy = query.copy(update={"query": rephrase}, deep=True) - q_copy.precomputed_query_embedding = ( - None # need to recompute for each rephrase + q_copy = query.model_copy( + update={ + "query": rephrase, + # need to recompute for each rephrase + # note that `SearchQuery` is a frozen model, so we can't update + # it below + "precomputed_query_embedding": None, + }, + deep=True, ) run_queries.append( (