Removed stream parameter for query, unused

antoninoLorenzo · Jul 4, 2024 · 6561def · 6561def
1 parent f38d68a
commit 6561def
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/src/agent/agent.py b/src/agent/agent.py
@@ -35,7 +35,7 @@ def __init__(self, model: str,
         # Start Ollama
         self._startup_ollama()
 
-    def query(self, sid: int, user_in: str, rag=True, stream=True):
+    def query(self, sid: int, user_in: str, rag=True):
         """Performs a query to the Large Language Model,
         set `rag=True` to leverage Retrieval Augmented Generation."""
         if rag:

diff --git a/static/acceptance_tests_plots.py → static/inference_times_plots.py b/static/acceptance_tests_plots.py → static/inference_times_plots.py
diff --git a/test/tests/results/inference_times.json b/test/tests/results/inference_times.json