From d0a432457607f05ca586d71e8e8d369161019d65 Mon Sep 17 00:00:00 2001
From: Antonino Lorenzo <94693967+antoninoLorenzo@users.noreply.github.com>
Date: Tue, 16 Jul 2024 11:44:20 +0200
Subject: [PATCH] llm.py: removed stream parameter from query method

---
 src/agent/llm/llm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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