add ImportErrorReporter

src/agentscope/agents/rpc_agent.py

@@ -9,7 +9,7 @@
 import base64
 import traceback
 import asyncio
-from typing import Any, Type, Optional, Union, Sequence
+from typing import Type, Optional, Union, Sequence
 from concurrent import futures
 from loguru import logger
 
@@ -19,10 +19,12 @@
     from grpc import ServicerContext
     from expiringdict import ExpiringDict
 except ImportError:
-    dill = None
-    grpc = None
-    ServicerContext = Any
-    ExpiringDict = None
+    from agentscope.utils.tools import ImportErrorReporter
+
+    dill = ImportErrorReporter("dill", "distribute")
+    grpc = ImportErrorReporter("grpcio", "distribute")
+    ServicerContext = ImportErrorReporter("grpcio", "distribute")
+    ExpiringDict = ImportErrorReporter("expiringdict", "distribute")
 
 from agentscope._init import init_process, _INIT_SETTINGS
 from agentscope.agents.agent import AgentBase

src/agentscope/models/dashscope_model.py

@@ -11,7 +11,7 @@
 
 try:
     import dashscope
-except ModuleNotFoundError:
+except ImportError:
     dashscope = None
 
 from .model import ModelWrapperBase, ModelResponse

src/agentscope/rpc/__init__.py

@@ -5,16 +5,19 @@
 
 try:
     from .rpc_agent_pb2 import RpcMsg  # pylint: disable=E0611
-except ModuleNotFoundError:
-    RpcMsg = Any  # type: ignore[misc]
-try:
     from .rpc_agent_pb2_grpc import RpcAgentServicer
     from .rpc_agent_pb2_grpc import RpcAgentStub
     from .rpc_agent_pb2_grpc import add_RpcAgentServicer_to_server
 except ImportError:
-    RpcAgentServicer = object
-    RpcAgentStub = Any
-    add_RpcAgentServicer_to_server = Any
+    from agentscope.utils.tools import ImportErrorReporter
+
+    RpcMsg = ImportErrorReporter("protobuf", "distribute")  # type: ignore[misc]
+    RpcAgentServicer = ImportErrorReporter("grpcio", "distribute")
+    RpcAgentStub = ImportErrorReporter("grpcio", "distribute")
+    add_RpcAgentServicer_to_server = ImportErrorReporter(
+        "grpcio",
+        "distribute",
+    )
 
 
 __all__ = [

src/agentscope/rpc/rpc_agent_client.py

@@ -3,24 +3,23 @@
 
 import threading
 import base64
-from typing import Any, Optional
+from typing import Optional
 from loguru import logger
 
 try:
     import dill
     import grpc
     from grpc import RpcError
-except ImportError:
-    dill = None
-    grpc = None
-    RpcError = None
-
-try:
     from agentscope.rpc.rpc_agent_pb2 import RpcMsg  # pylint: disable=E0611
     from agentscope.rpc.rpc_agent_pb2_grpc import RpcAgentStub
-except ModuleNotFoundError:
-    RpcMsg = Any  # type: ignore[misc]
-    RpcAgentStub = Any
+except ImportError:
+    from agentscope.utils.tools import ImportErrorReporter
+
+    dill = ImportErrorReporter("dill", "distribute")
+    grpc = ImportErrorReporter("grpcio", "distribute")
+    RpcMsg = ImportErrorReporter("protobuf", "distribute")
+    RpcAgentStub = ImportErrorReporter("grpcio", "distribute")
+    RpcError = ImportError
 
 
 class RpcAgentClient:

src/agentscope/rpc/rpc_agent_pb2_grpc.py

@@ -4,7 +4,9 @@
 try:
     import grpc
 except ImportError:
-    grpc = None
+    from agentscope.utils.tools import ImportErrorReporter
+
+    grpc = ImportErrorReporter("grpcio", "distribute")
 
 import agentscope.rpc.rpc_agent_pb2 as rpc__agent__pb2
 

src/agentscope/utils/tools.py

@@ -296,3 +296,36 @@ def _join_str_with_comma_and(elements: List[str]) -> str:
         return " and ".join(elements)
     else:
         return ", ".join(elements[:-1]) + f", and {elements[-1]}"
+
+
+class ImportErrorReporter:
+    """Used as a placeholder for missing packages.
+    When called, an ImportError will be raised, prompting the user to install
+    the specified extras requirement.
+    """
+
+    def __init__(self, package_name: str, extras_require: str = None) -> None:
+        """Init the ImportErrorReporter.
+
+        Args:
+            package_name (`str`): the name of the package to be imported.
+            extras_require (`str`): the extras requirement.
+        """
+        self.package_name = package_name
+        self.extras_require = extras_require
+
+    def raise_error(self) -> Any:
+        """Raise an ImportError."""
+        msg = f"Failed to import {self.package_name}."
+        if self.extras_require is not None:
+            msg += (
+                f" Please install [{self.extras_require}] version of"
+                " agentscope."
+            )
+        raise ImportError(msg)
+
+    def __call__(self, *args: Any, **kwds: Any) -> Any:
+        return self.raise_error()
+
+    def __getattr__(self, name: str) -> Any:
+        return self.raise_error()