Update env & test code

This commit is contained in:
didi 2024-02-28 17:00:51 +08:00 committed by better629
parent 26bbdef6c6
commit ba2fdb222e
11 changed files with 98 additions and 74 deletions

View file

@ -9,10 +9,10 @@ from typing import Any, Optional
from pydantic import Field
from metagpt.const import ADB_EXEC_FAIL
from metagpt.environment.base_env import Env, ExtEnv, mark_as_readable, mark_as_writeable
from metagpt.environment.base_env import Environment, ExtEnv, mark_as_readable, mark_as_writeable
class AndroidExtEnv(Env, ExtEnv):
class AndroidExtEnv(Environment, ExtEnv):
device_id: Optional[str] = Field(default=None)
screenshot_dir: Optional[Path] = Field(default=None)
xml_dir: Optional[Path] = Field(default=None)

View file

@ -45,11 +45,10 @@ def mark_as_readable(func):
def mark_as_writeable(func):
"""mark function as a writeable one in ExtEnv, it does something to ExtEnv"""
"""mark functionn as a writeable one in ExtEnv, it does something to ExtEnv"""
env_write_api_registry[func.__name__] = get_function_schema(func)
return func
class ExtEnv(BaseModel):
"""External Env to integrate actual game environment"""