Update search.py (#1457)

intelligencecompany · web-flow · commit 756f5c38a77d · 2024-12-04T16:52:08.000-06:00
Missing query in astream_search
diff --git a/graphrag/query/structured_search/global_search/search.py b/graphrag/query/structured_search/global_search/search.py
@@ -110,6 +110,7 @@ async def astream_search(
     ) -> AsyncGenerator:
         """Stream the global search response."""
         context_result = await self.context_builder.build_context(
+            query=query,
             conversation_history=conversation_history,
             **self.context_builder_params,
         )

Original file line number	Diff line number	Diff line change
`@@ -110,6 +110,7 @@ async def astream_search(`
`110`	`110`	`) -> AsyncGenerator:`
`111`	`111`	`"""Stream the global search response."""`
`112`	`112`	`context_result = await self.context_builder.build_context(`
	`113`	`+ query=query,`
`113`	`114`	`conversation_history=conversation_history,`
`114`	`115`	`**self.context_builder_params,`
`115`	`116`	`)`