apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add browser view to memory after executing browser tool

Browse source
This commit is contained in:
shenchucheng 2024-06-25 21:34:49 +08:00
parent bd675d5178
commit 1dbe88b96d
2 changed files with 11 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
metagpt/roles/di/role_zero.py
View file

@ -2,6 +2,7 @@ from __future__ import annotations
import inspect
import json
import re
import traceback
from typing import Callable, Literal, Tuple
@ -144,7 +145,11 @@ class RoleZero(Role):
)
memory = self.rc.memory.get(self.memory_k)
if not self.browser.is_empty_page:
memory.append(UserMessage(cause_by="browser", content=await self.browser.view()))
pattern = re.compile(r"Command Browser\.(\w+) executed")
for index, msg in zip(range(len(memory), 0, -1), memory[::-1]):
if pattern.match(msg.content):
memory.insert(index, UserMessage(cause_by="browser", content=await self.browser.view()))
break
context = self.llm.format_msg(memory + [UserMessage(content=prompt)])
# print(*context, sep="\n" + "*" * 5 + "\n")
async with ThoughtReporter(enable_llm_stream=True):

10
metagpt/tools/libs/browser.py
View file

@ -3,7 +3,7 @@ from __future__ import annotations
import time
from typing import Literal, Optional
from playwright.async_api import Browser as _Browser
from playwright.async_api import Browser as Browser_
from playwright.async_api import (
BrowserContext,
Frame,
@ -67,7 +67,7 @@ class Browser:
def __init__(self):
self.playwright: Optional[Playwright] = None
self.browser: Optional[_Browser] = None
self.browser_instance: Optional[Browser_] = None
self.browser_ctx: Optional[BrowserContext] = None
self.page: Optional[Page] = None
self.accessibility_tree: list = []
@ -80,7 +80,7 @@ class Browser:
"""Starts Playwright and launches a browser"""
if self.playwright is None:
self.playwright = playwright = await async_playwright().start()
browser = self.browser = await playwright.chromium.launch(headless=self.headless, proxy=self.proxy)
browser = self.browser_instance = await playwright.chromium.launch(headless=self.headless, proxy=self.proxy)
browser_ctx = self.browser_ctx = await browser.new_context()
self.page = await browser_ctx.new_page()
@ -88,7 +88,7 @@ class Browser:
if self.playwright:
playwright = self.playwright
self.playwright = None
self.browser = None
self.browser_instance = None
self.browser_ctx = None
await playwright.stop()
@ -196,7 +196,7 @@ class Browser:
async def view(self):
observation = parse_accessibility_tree(self.accessibility_tree)
return f"Current _Browser Viewer\n URL: {self.page.url}\nOBSERVATION:\n{observation[0]}\n"
return f"Current Browser Viewer\n URL: {self.page.url}\nOBSERVATION:\n{observation[0]}\n"
async def __aenter__(self):
await self.start()