Fix/drift search reduce (#1948)

AlonsoGuevara · web-flow · commit f1e2041f077c · 2025-05-23T08:07:09.000-06:00
* Fix Reduce Response for non streaming calls

* Semver
diff --git a/.semversioner/next-release/patch-20250522234310308717.json 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
@@ -386,7 +386,7 @@ async def _reduce_response(
         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

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "type": "patch",
 +  "description": "Fix Drift Reduce Response for non streaming calls"
 +}
Original file line number	Diff line number	Diff line change
`@@ -386,7 +386,7 @@ async def _reduce_response(`
`386`	`386`	`model_response = await self.model.achat(`
`387`	`387`	`prompt=query,`
`388`	`388`	`history=search_messages,`
`389`		`- model_parameters=llm_kwargs,`
	`389`	`+ model_parameters=llm_kwargs.get("model_params", {}),`
`390`	`390`	`)`
`391`	`391`
`392`	`392`	`reduced_response = model_response.output.content`