Fix/drift search reduce (#1948)

* Fix Reduce Response for non streaming calls

* Semver
This commit is contained in:
Alonso Guevara 2025-05-23 08:07:09 -06:00 committed by GitHub
parent 7fba9522d4
commit f1e2041f07
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 1 deletions

View File

@ -0,0 +1,4 @@
{
"type": "patch",
"description": "Fix Drift Reduce Response for non streaming calls"
}

View File

@ -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