Update tool.py

Maximilian-Winter · May 26, 2024 · db2a8fb · db2a8fb
1 parent 55ef26d
commit db2a8fb
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
@@ -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