llm.py: removed stream parameter from query method

antoninoLorenzo · Jul 16, 2024 · d0a4324 · d0a4324
1 parent 1b241ee
commit d0a4324
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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