[Do not merge] add agentchat

.pre-commit-config.yaml

@@ -1,25 +1,21 @@
 repos:
--   repo: https://github.com/psf/black.git
+-   repo: https://githubfast.com/psf/black.git
     rev: 23.3.0
     hooks:
     -   id: black
         files: (.*\.(py|pyi|bzl)|BUILD|.*\.BUILD|WORKSPACE)$
--   repo: https://github.com/pycqa/isort
-    rev: 5.11.5
-    hooks:
-    -   id: isort
--   repo: https://github.com/PyCQA/flake8
+-   repo: https://githubfast.com/PyCQA/flake8
     rev: 5.0.4
     hooks:
     -   id: flake8
         args: [--config=.flake8]
--   repo: https://github.com/pre-commit/mirrors-mypy
+-   repo: https://githubfast.com/pre-commit/mirrors-mypy
     rev: v1.6.1
     hooks:
     -   id: mypy
         exclude: ^(setup\.py|.*tests.*)$
         additional_dependencies: ['types-requests', 'types-PyYAML']
--   repo: https://github.com/pre-commit/pre-commit-hooks
+-   repo: https://githubfast.com/pre-commit/pre-commit-hooks
     rev: a11d9314b22d8f8c7556443875b731ef05965464
     hooks:
     -   id: check-merge-conflict
@@ -28,7 +24,7 @@ repos:
     -   id: trailing-whitespace
     -   id: detect-private-key
     -   id: check-added-large-files
--   repo: https://github.com/Lucas-C/pre-commit-hooks
+-   repo: https://githubfast.com/Lucas-C/pre-commit-hooks
     rev: v1.0.1
     hooks:
     -   id: forbid-crlf

erniebot-agent/erniebot_agent/agents/agentchat.py

@@ -0,0 +1,84 @@
+from typing import Dict, List, Optional, Union
+
+
+class AgentChat:
+    """(In preview) An abstract class for AI agent.
+
+    An agent can communicate with other agents and perform actions.
+    Different agents can differ in what actions they perform in the `receive` method.
+    """
+
+    def __init__(self, name: str):
+        """
+        Args:
+            name (str): name of the agent.
+        """
+        # a dictionary of conversations, default value is list
+        self._name = name
+
+    @property
+    def name(self):
+        """Get the name of the agent."""
+        return self._name
+
+    def send(self, message: Union[Dict, str], recipient: "AgentChat", request_reply: Optional[bool] = None):
+        """(Abstract method) Send a message to another agent."""
+
+    async def a_send(
+        self, message: Union[Dict, str], recipient: "AgentChat", request_reply: Optional[bool] = None
+    ):
+        """(Abstract async method) Send a message to another agent."""
+
+    def receive(
+        self,
+        message: Union[Dict, str],
+        sender: "AgentChat",
+        request_reply: Optional[bool] = None,
+        silent: Optional[bool] = None,
+    ):
+        """(Abstract method) Receive a message from another agent."""
+
+    async def a_receive(
+        self,
+        message: Union[Dict, str],
+        sender: "AgentChat",
+        request_reply: Optional[bool] = None,
+        silent: Optional[bool] = None,
+    ):
+        """(Abstract async method) Receive a message from another agent."""
+
+    def reset(self):
+        """(Abstract method) Reset the agent."""
+
+    def can_execute_function(self, name: str):
+        """Determine whether the function can be executed"""
+
+    def generate_reply(
+        self,
+        messages: Optional[List[Dict]] = None,
+        sender: Optional["AgentChat"] = None,
+        **kwargs,
+    ) -> Union[str, Dict, None]:
+        """(Abstract method) Generate a reply based on the received messages.
+
+        Args:
+            messages (list[dict]): a list of messages received.
+            sender: sender of an Agent instance.
+        Returns:
+            str or dict or None: the generated reply. If None, no reply is generated.
+        """
+
+    async def a_generate_reply(
+        self,
+        messages: Optional[List[Dict]] = None,
+        sender: Optional["AgentChat"] = None,
+        **kwargs,
+    ) -> Union[str, Dict, None]:
+        """(Abstract async method) Generate a reply based on the received messages.
+
+        Args:
+            messages (list[dict]): a list of messages received.
+            sender: sender of an Agent instance.
+        Returns:
+            str or dict or None: the generated reply. If None, no reply is generated.
+        """

erniebot-agent/erniebot_agent/agents/assistant_agent.py

@@ -0,0 +1,31 @@
+from .conversable_agent import ConversableAgent
+from typing import Callable, Dict, Optional
+
+
+class AssistantAgent(ConversableAgent):
+    DEFAULT_SYSTEM_MESSAGE = """您是一位有用的人工智能助手。请利用您的语言技能解决任务。
+    如果需要，请逐步解决任务。如果未提供计划，请先解释您的计划。
+    当你找到答案时，请仔细验证答案。
+    如果可能，请在您的回复中包含可验证的证据。
+    当一切完成后，最后回复'终止'
+    """
+
+    def __init__(
+        self,
+        name: str,
+        llm_config: Dict,
+        system_message: str = DEFAULT_SYSTEM_MESSAGE,
+        is_termination_msg: Optional[Callable[[Dict], bool]] = None,
+        max_consecutive_auto_reply: Optional[int] = None,
+        human_input_mode: Optional[str] = "NEVER",
+        **kwargs,
+    ):
+        super().__init__(
+            name,
+            llm_config,
+            system_message,
+            is_termination_msg,
+            max_consecutive_auto_reply,
+            human_input_mode,
+            **kwargs,
+        )