Update tool.py

Maximilian-Winter · May 26, 2024 · ca90694 · ca90694
1 parent e26f7fd
commit ca90694
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/llama_cpp_agent/tools/web_search/tool.py b/src/llama_cpp_agent/tools/web_search/tool.py
@@ -70,8 +70,8 @@ def search_web(self, search_query: str):
                 if self.max_tokens_search_results >= len(tokens):
                     break
                 else:
-                    remove_char_count += 25
-                    tokens = self.llm_provider.tokenize(res[:-remove_char_count])
+                    remove_char_count += 50
+                    tokens = self.llm_provider.tokenize(res[:remove_char_count])
         if not has_remove_char:
             return "\nResults of searching the web:\n\n" + res
         return "\nResults of searching the web:\n\n" + res[:remove_char_count]