From efec27c0b8df84cd13492d95e5b110eca2430845 Mon Sep 17 00:00:00 2001 From: Yannick Stephan Date: Sun, 16 Feb 2025 20:10:28 +0100 Subject: [PATCH] fixed stream --- lightrag/api/lightrag_server.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index 756148fe..29330fdb 100644 --- a/lightrag/api/lightrag_server.py +++ b/lightrag/api/lightrag_server.py @@ -739,7 +739,9 @@ class QueryRequest(BaseModel): request_data = self.model_dump(exclude_none=True, exclude={"query"}) # Ensure `mode` and `stream` are set explicitly - return QueryParam(**request_data, stream=is_stream) + param = QueryParam(**request_data) + param.stream = is_stream + return param class QueryResponse(BaseModel):