diff --git a/.semversioner/next-release/patch-20250522234310308717.json b/.semversioner/next-release/patch-20250522234310308717.json new file mode 100644 index 00000000..2c36e56f --- /dev/null +++ b/.semversioner/next-release/patch-20250522234310308717.json @@ -0,0 +1,4 @@ +{ + "type": "patch", + "description": "Fix Drift Reduce Response for non streaming calls" +} diff --git a/graphrag/query/structured_search/drift_search/search.py b/graphrag/query/structured_search/drift_search/search.py index 10d62343..a452cf1d 100644 --- a/graphrag/query/structured_search/drift_search/search.py +++ b/graphrag/query/structured_search/drift_search/search.py @@ -386,7 +386,7 @@ class DRIFTSearch(BaseSearch[DRIFTSearchContextBuilder]): model_response = await self.model.achat( prompt=query, history=search_messages, - model_parameters=llm_kwargs, + model_parameters=llm_kwargs.get("model_params", {}), ) reduced_response = model_response.output.content