init

AlgoriAgent/examples/distributed_parallel_optimization/answerer_agent.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+"""Answerer Agent."""
+from typing import Optional, Union, Sequence
+
+from agentscope.message import Msg
+from agentscope.agents import AgentBase
+from agentscope.service import load_web
+
+
+class AnswererAgent(AgentBase):
+    """An agent with web digest tool."""
+
+    def __init__(
+        self,
+        name: str,
+        model_config_name: str = None,
+    ) -> None:
+        super().__init__(
+            name=name,
+            sys_prompt="You are an AI assistant. You need to find answers to "
+            "user questions based on specified web content.",
+            model_config_name=model_config_name,
+            use_memory=False,
+        )
+
+    def reply(self, x: Optional[Union[Msg, Sequence[Msg]]] = None) -> Msg:
+        response = load_web(
+            url=x.url,
+            keep_raw=False,
+            html_selected_tags=["p", "div", "h1", "li"],
+            timeout=5,
+        ).content
+        if (
+            "html_to_text" not in response
+            or len(response["html_to_text"]) == 0
+        ):
+            return Msg(
+                self.name,
+                content=f"Unable to load web page [{x.url}].",
+                url=x.url,
+            )
+        # prepare prompt
+        prompt = self.model.format(
+            Msg(name="system", role="system", content=self.sys_prompt),
+            Msg(
+                name="user",
+                role="user",
+                content=f"Please answer my question based on the content of"
+                " the following web page:\n\n"
+                f"{response['html_to_text']}"
+                f"\n\nBased on the above web page,"
+                f" please answer my question\n{x.query}",
+            ),
+        )
+        # call llm and generate response
+        response = self.model(prompt).text
+        msg = Msg(self.name, content=response, url=x.url)
+
+        self.speak(msg)
+
+        return msg