add general environment and refactor environment's w/r funcs

metagpt/environment/__init__.py

@@ -0,0 +1,3 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Desc   :

metagpt/environment/environment.py

@@ -0,0 +1,79 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2023/5/11 22:12
+@Author  : alexanderwu
+@File    : environment.py
+"""
+import asyncio
+from typing import Iterable
+
+from pydantic import BaseModel, Field
+
+from metagpt.memory import Memory
+from metagpt.roles import Role
+from metagpt.schema import Message
+
+
+class Environment(BaseModel):
+    """环境，承载一批角色，角色可以向环境发布消息，可以被其他角色观察到
+       Environment, hosting a batch of roles, roles can publish messages to the environment,
+       and can be observed by other roles
+    """
+
+    roles: dict[str, Role] = Field(default_factory=dict)
+    memory: Memory = Field(default_factory=Memory)
+    history: str = Field(default='')
+
+    class Config:
+        arbitrary_types_allowed = True
+
+    def add_role(self, role: Role):
+        """增加一个在当前环境的角色
+           Add a role in the current environment
+        """
+        role.set_env(self)
+        self.roles[role.profile] = role
+
+    def add_roles(self, roles: Iterable[Role]):
+        """增加一批在当前环境的角色
+            Add a batch of characters in the current environment
+        """
+        for role in roles:
+            self.add_role(role)
+
+    def publish_message(self, message: Message):
+        """向当前环境发布信息
+          Post information to the current environment
+        """
+        # self.message_queue.put(message)
+        self.memory.add(message)
+        self.history += f"\n{message}"
+
+    async def run(self, k=1):
+        """处理一次所有信息的运行
+        Process all Role runs at once
+        """
+        # while not self.message_queue.empty():
+        # message = self.message_queue.get()
+        # rsp = await self.manager.handle(message, self)
+        # self.message_queue.put(rsp)
+        for _ in range(k):
+            futures = []
+            for role in self.roles.values():
+                future = role.run()
+                futures.append(future)
+
+            await asyncio.gather(*futures)
+
+    def get_roles(self) -> dict[str, Role]:
+        """获得环境内的所有角色
+           Process all Role runs at once
+        """
+        return self.roles
+
+    def get_role(self, name: str) -> Role:
+        """获得环境内的指定角色
+           get all the environment roles
+        """
+        return self.roles.get(name, None)

metagpt/environment/general_environment.py

@@ -0,0 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Desc   :
+
+from typing import Callable
+
+from pydantic import Field
+
+from metagpt.environment.environment import Environment
+
+
+class GeneralEnvironment(Environment):
+    """
+    A GeneralEnvironment for interfacing with games, etc. It create a registration mechanism to register
+    custom methods when operating with the particular environment.
+    """
+    name: str = Field(default="")
+    registered_funcs: dict[str, Callable] = Field(default={})
+
+    def register_func(self, func_name: str, func: Callable):
+        if func_name not in self.registered_funcs:
+            self.registered_funcs[func_name] = func
+
+    def call_func(self, func_name: str, *args, **kwargs):
+        assert func_name in self.registered_funcs
+
+        func = self.registered_funcs.get(func_name)
+        return func(*args, **kwargs)
+
+    @staticmethod
+    def init_register_funcs(self):
+        raise NotImplementedError()

metagpt/environment/gym_environment.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# @Desc   : RL environment about Gymnasium(forked from openai gym)
+
+from typing import Callable
+
+import gymnasium as gym
+
+from metagpt.logs import logger
+from metagpt.environment.general_environment import GeneralEnvironment
+
+
+class GymEnvironment(GeneralEnvironment):
+
+    def init_register_funcs(self):
+        env = gym.make(self.name)
+        logger.info(f"init gym environment: {self.name}")
+        self.register_func("reset", env.reset)
+        self.register_func("sample_action", env.action_space.sample)
+        self.register_func("step", env.step)
+        self.register_func("close", env.close)