From db2a8fbbcfbafeef7ef1b98282f25ef5f863b56f Mon Sep 17 00:00:00 2001 From: Maximilian Winter Date: Sun, 26 May 2024 13:30:05 +0200 Subject: [PATCH] Update tool.py --- src/llama_cpp_agent/tools/web_search/tool.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llama_cpp_agent/tools/web_search/tool.py b/src/llama_cpp_agent/tools/web_search/tool.py index fbb7485..a4ae3a3 100644 --- a/src/llama_cpp_agent/tools/web_search/tool.py +++ b/src/llama_cpp_agent/tools/web_search/tool.py @@ -65,13 +65,13 @@ def search_web(self, search_query: str): if self.max_tokens_search_results < len(tokens): remove_chars = len(tokens) - self.max_tokens_search_results while True: - tokens = self.llm_provider.tokenize(res[:remove_chars]) + tokens = self.llm_provider.tokenize(res[:-remove_chars]) if self.max_tokens_search_results >= len(tokens): break else: remove_chars += 100 - return "\nResults of searching the web:\n\n" + res[:remove_chars] + return "\nResults of searching the web:\n\n" + res[:-remove_chars] def get_tool(self): return self.search_web