From 174363c7ae90926b96026c24e6e3419db6bc5e54 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 24 Feb 2025 11:36:51 +0100 Subject: [PATCH] Update app.py --- lollms/app.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lollms/app.py b/lollms/app.py index 51383b3..71fd372 100644 --- a/lollms/app.py +++ b/lollms/app.py @@ -1171,6 +1171,7 @@ Answer directly with the reformulation of the last prompt. show_progress=True, callback=self.personality.sink ) + query = self.personality.remove_thinking_blocks(query) self.personality.step_end("Building vector store query") self.personality.step(f"Query: {query}") else: @@ -1276,6 +1277,7 @@ Answer directly with the reformulation of the last prompt. f"{self.ai_custom_header('websearch query')}" ]) query = self.personality.fast_gen(q, max_generation_size=256, show_progress=True, callback=self.personality.sink) + query = self.personality.remove_thinking_blocks(query) query = query.replace("\"","") self.personality.step_end("Crafting internet search query") self.personality.step(f"web search query: {query}")