diff --git a/sources/agents/browser_agent.py b/sources/agents/browser_agent.py index 5a86f4e..20dc7c9 100644 --- a/sources/agents/browser_agent.py +++ b/sources/agents/browser_agent.py @@ -243,10 +243,10 @@ class BrowserAgent(Agent): self.logger.warning("No link selected.") return None - def get_page_text(self, compression = False) -> str: + def get_page_text(self, limit_to_model_ctx = False) -> str: """Get the text content of the current page.""" page_text = self.browser.get_text() - if compression: + if limit_to_model_ctx: #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 @@ -365,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.get_page_text() + page_text = self.get_page_text(limit_to_model_ctx=True) 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.get_page_text() + page_text = self.get_page_text(limit_to_model_ctx=True) self.navigable_links = self.browser.get_navigable() prompt = self.make_navigation_prompt(user_prompt, page_text) continue @@ -407,7 +407,7 @@ class BrowserAgent(Agent): prompt = self.make_newsearch_prompt(user_prompt, unvisited) continue self.current_page = link - page_text = self.get_page_text() + page_text = self.get_page_text(limit_to_model_ctx=True) self.navigable_links = self.browser.get_navigable() prompt = self.make_navigation_prompt(user_prompt, page_text) self.status_message = "Navigating..."