diff --git a/src/agent/llm/llm.py b/src/agent/llm/llm.py
index 85b6f23..fe57aa9 100644
--- a/src/agent/llm/llm.py
+++ b/src/agent/llm/llm.py
@@ -132,7 +132,7 @@ def __post_init__(self):
             api_key=self.api_key
         )
 
-    def query(self, messages: list, stream=True):
+    def query(self, messages: list):
         """Generator that returns response chunks."""
-        for chunk in self.provider.query(messages, stream=stream):
+        for chunk in self.provider.query(messages):
             yield chunk