feat: improve compression/web agent memory management

sources/agents/agent.py

@@ -100,7 +100,7 @@ class Agent():
         """
         description = ""
         for name in self.get_tools_name():
-            description += f"{tool}: {self.tools[name].description}\n"
+            description += f"{name}: {self.tools[name].description}\n"
         return description
     
     def load_prompt(self, file_path: str) -> str:

sources/agents/browser_agent.py

@@ -243,6 +243,14 @@ class BrowserAgent(Agent):
         self.logger.warning("No link selected.")
         return None
     
+    def get_page_text(self, compression = False) -> str:
+        """Get the text content of the current page."""
+        page_text = self.browser.get_text()
+        if compression:
+            #page_text = self.memory.compress_text_to_max_ctx(page_text)
+            page_text = self.memory.trim_text_to_max_ctx(page_text)
+        return page_text
+    
     def conclude_prompt(self, user_query: str) -> str:
         annotated_notes = [f"{i+1}: {note.lower()}" for i, note in enumerate(self.notes)]
         search_note = '\n'.join(annotated_notes)
@@ -357,13 +365,13 @@ class BrowserAgent(Agent):
                 self.status_message = "Filling web form..."
                 pretty_print(f"Filling inputs form...", color="status")
                 fill_success = self.browser.fill_form(extracted_form)
-                page_text = self.browser.get_text()
+                page_text = self.get_page_text()
                 answer = self.handle_update_prompt(user_prompt, page_text, fill_success)
                 answer, reasoning = await self.llm_decide(prompt)
 
             if Action.FORM_FILLED.value in answer:
                 pretty_print(f"Filled form. Handling page update.", color="status")
-                page_text = self.browser.get_text()
+                page_text = self.get_page_text()
                 self.navigable_links = self.browser.get_navigable()
                 prompt = self.make_navigation_prompt(user_prompt, page_text)
                 continue
@@ -399,7 +407,7 @@ class BrowserAgent(Agent):
                 prompt = self.make_newsearch_prompt(user_prompt, unvisited)
                 continue
             self.current_page = link
-            page_text = self.browser.get_text()
+            page_text = self.get_page_text()
             self.navigable_links = self.browser.get_navigable()
             prompt = self.make_navigation_prompt(user_prompt, page_text)
             self.status_message = "Navigating..."