fix redis pool

modelscope · Sep 9, 2024 · ae2ae9a · ae2ae9a
1 parent b10f426
commit ae2ae9a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/agentscope/server/async_result_pool.py b/src/agentscope/server/async_result_pool.py
@@ -118,8 +118,10 @@ def prepare(self) -> int:
         return self._get_object_id()
 
     def set(self, key: int, value: bytes) -> None:
-        self.pool.set(key, value, ex=self.max_timeout)
-        self.pool.rpush(RedisPool.TASK_QUEUE_PREFIX + str(key), key)
+        pipe = self.pool.pipeline()
+        pipe.set(key, value, ex=self.max_timeout)
+        pipe.rpush(RedisPool.TASK_QUEUE_PREFIX + str(key), key)
+        pipe.execute()
 
     def get(self, key: int) -> bytes:
         result = self.pool.get(key)