Rebuild AgentScope Studio

src/agentscope/_init.py

@@ -13,6 +13,7 @@
 from .models import read_model_configs
 from .constants import _DEFAULT_DIR
 from .constants import _DEFAULT_LOG_LEVEL
+from .web.client import HttpClient
 
 # init setting
 _INIT_SETTINGS = {}
@@ -30,6 +31,7 @@ def init(
     logger_level: LOG_LEVEL = _DEFAULT_LOG_LEVEL,
     runtime_id: Optional[str] = None,
     agent_configs: Optional[Union[str, list, dict]] = None,
+    studio_url: Optional[str] = None,
 ) -> Sequence[AgentBase]:
     """A unified entry to initialize the package, including model configs,
     runtime names, saving directories and logging settings.
@@ -65,6 +67,8 @@ def init(
             which can be loaded by json.loads(). One agent config should
             cover the required arguments to initialize a specific agent
             object, otherwise the default values will be used.
+        studio_url (`Optional[str]`, defaults to `None`):
+            The url of the studio.
     """
     init_process(
         model_configs=model_configs,
@@ -76,6 +80,7 @@ def init(
         save_log=save_log,
         use_monitor=use_monitor,
         logger_level=logger_level,
+        studio_url=studio_url,
     )
 
     # save init settings for subprocess
@@ -128,6 +133,7 @@ def init_process(
     save_log: bool = False,
     use_monitor: bool = True,
     logger_level: LOG_LEVEL = _DEFAULT_LOG_LEVEL,
+    studio_url: Optional[str] = None,
 ) -> None:
     """An entry to initialize the package in a process.
 
@@ -157,10 +163,16 @@ def init_process(
     # Init the runtime
     if project is not None:
         _runtime.project = project
+    else:
+        project = _runtime.project
     if name is not None:
         _runtime.name = name
+    else:
+        name = _runtime.name
     if runtime_id is not None:
         _runtime.runtime_id = runtime_id
+    else:
+        runtime_id = _runtime.runtime_id
 
     # Init logger
     dir_log = str(file_manager.dir_log) if save_log else None
@@ -178,3 +190,14 @@ def init_process(
         db_path=file_manager.path_db,
         impl_type="sqlite" if use_monitor else "dummy",
     )
+    if studio_url is not None:
+        client = HttpClient(
+            studio_url=studio_url,
+            run_id=_runtime.runtime_id,
+        )
+        client.register_run(
+            project=project,
+            name=name,
+            run_dir=file_manager.dir,
+        )
+        _runtime.studio_client = client

src/agentscope/_runtime.py

@@ -5,6 +5,7 @@
 
 from agentscope.utils.tools import _get_timestamp
 from agentscope.utils.tools import _generate_random_code
+from agentscope.web.client import HttpClient
 
 _RUNTIME_ID_FORMAT = "run_%Y%m%d-%H%M%S_{}"
 _RUNTIME_TIMESTAMP_FORMAT = "%Y-%m-%d %H:%M:%S"
@@ -24,6 +25,9 @@ class _Runtime:
     """The id for runtime, which is used to identify the this runtime and
         name the saving directory."""
 
+    studio_client: HttpClient = None
+    """The client of AgentScope Studio."""
+
     _timestamp: datetime = datetime.now()
     """The timestamp of when the runtime is initialized."""
 

src/agentscope/agents/agent.py

@@ -14,6 +14,7 @@
 from agentscope.agents.operator import Operator
 from agentscope.models import load_model_by_config_name
 from agentscope.memory import TemporaryMemory
+from agentscope._runtime import _runtime
 
 
 class _AgentMeta(ABCMeta):
@@ -305,6 +306,19 @@ def speak(
     ) -> None:
         """Speak out the content generated by the agent."""
         logger.chat(content)
+        if _runtime.studio_client is not None:
+            if isinstance(content, dict):
+                _runtime.studio_client.send_message(
+                    name=content.get("name", ""),
+                    content=content.get("content", ""),
+                    url=content.get("url", ""),
+                )
+            else:
+                _runtime.studio_client.send_message(
+                    name="",
+                    content=str(content),
+                    url="",
+                )
 
     def observe(self, x: Union[dict, Sequence[dict]]) -> None:
         """Observe the input, store it in memory without response to it.

src/agentscope/agents/user_agent.py

@@ -6,6 +6,7 @@
 from loguru import logger
 
 from agentscope.agents import AgentBase
+from agentscope._runtime import _runtime
 from agentscope.message import Msg
 from agentscope.web.studio.utils import user_input
 
@@ -29,6 +30,14 @@ def __init__(self, name: str = "User", require_url: bool = False) -> None:
 
         self.name = name
         self.require_url = require_url
+        if _runtime.studio_client is not None:
+            self.input_client = (
+                _runtime.studio_client.generate_user_input_client(
+                    self.agent_id,
+                )
+            )
+        else:
+            self.input_client = None
 
     def reply(
         self,
@@ -67,23 +76,29 @@ def reply(
 
         # TODO: To avoid order confusion, because `input` print much quicker
         #  than logger.chat
-        time.sleep(0.5)
-        content = user_input(timeout=timeout)
-
-        kwargs = {}
-        if required_keys is not None:
-            if isinstance(required_keys, str):
-                required_keys = [required_keys]
-
-            for key in required_keys:
-                kwargs[key] = input(f"{key}: ")
-
-        # Input url of file, image, video, audio or website
-        url = None
-        if self.require_url:
-            url = input("URL (or Enter to skip): ")
-            if url == "":
-                url = None
+        if self.input_client:
+            logger.info(
+                f"Waiting for input from {self.input_client.studio_url}...",
+            )
+            raw_input = self.input_client.get_user_input()
+            content = raw_input["content"]
+        else:
+            time.sleep(0.5)
+            content = user_input(timeout=timeout)
+            kwargs = {}
+            if required_keys is not None:
+                if isinstance(required_keys, str):
+                    required_keys = [required_keys]
+
+                for key in required_keys:
+                    kwargs[key] = input(f"{key}: ")
+
+            # Input url of file, image, video, audio or website
+            url = None
+            if self.require_url:
+                url = input("URL (or Enter to skip): ")
+                if url == "":
+                    url = None
 
         # Add additional keys
         msg = Msg(
@@ -101,10 +116,3 @@ def reply(
             self.memory.add(msg)
 
         return msg
-
-    def speak(
-        self,
-        content: Union[str, dict],
-    ) -> None:
-        """Speak the content to the audience."""
-        logger.chat(content, disable_studio=True)