From a3330b6ab704707a850b1508231db34ec0a08567 Mon Sep 17 00:00:00 2001
From: "panxuchen.pxc" <panxuchen.pxc@alibaba-inc.com>
Date: Tue, 21 May 2024 20:26:12 +0800
Subject: [PATCH] add psutils to rpc dependency

---
 setup.py                          | 1 +
 src/agentscope/server/servicer.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index ee22169f4..3e52dcc20 100644
--- a/setup.py
+++ b/setup.py
@@ -23,6 +23,7 @@
     "protobuf==4.25.0",
     "expiringdict",
     "dill",
+    "psutil",
 ]
 
 service_requires = [
diff --git a/src/agentscope/server/servicer.py b/src/agentscope/server/servicer.py
index 42bf3b3f9..156d4bbf4 100644
--- a/src/agentscope/server/servicer.py
+++ b/src/agentscope/server/servicer.py
@@ -6,10 +6,10 @@
 import json
 from concurrent import futures
 from loguru import logger
-import psutil
 
 try:
     import dill
+    import psutil
     import grpc
     from grpc import ServicerContext
     from google.protobuf.empty_pb2 import Empty
@@ -20,6 +20,7 @@
     from agentscope.utils.tools import ImportErrorReporter
 
     dill = ImportErrorReporter(import_error, "distribute")
+    psutil = ImportErrorReporter(import_error, "distribute")
     grpc = ImportErrorReporter(import_error, "distribute")
     Empty = ImportErrorReporter(  # type: ignore[misc]
         import_error,