support async func in custom_build_fn (#195)

support async func in `custom_build_fn`

Author: rayrayraykk

src/agentscope_runtime/engine/agents/agentscope_agent.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # pylint:disable=too-many-nested-blocks, too-many-branches, too-many-statements
 # pylint:disable=line-too-long, protected-access
+import inspect
 import copy
 import logging
 import json
@@ -250,7 +251,12 @@ async def run_async(
         # We should always build a new agent since the state is manage outside
         # the agent
         if self._attr.get("custom_build_fn"):
-            _agent = self._attr["custom_build_fn"](as_context, **kwargs)
+            build_fn = self._attr["custom_build_fn"]
+
+            if inspect.iscoroutinefunction(build_fn):
+                _agent = await build_fn(as_context, **kwargs)
+            else:
+                _agent = build_fn(as_context, **kwargs)
         else:
             _agent = self.build(as_context)
 

src/agentscope_runtime/engine/agents/agno_agent.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 # pylint:disable=too-many-nested-blocks, too-many-branches, too-many-statements
+import inspect
 import json
 from typing import Optional, Type, Callable
 
@@ -164,7 +165,12 @@ async def run_async(
         # We should always build a new agent since the state is manage outside
         # the agent
         if self._attr.get("custom_build_fn"):
-            _agent = self._attr["custom_build_fn"](ag_context, **kwargs)
+            build_fn = self._attr["custom_build_fn"]
+
+            if inspect.iscoroutinefunction(build_fn):
+                _agent = await build_fn(ag_context, **kwargs)
+            else:
+                _agent = build_fn(ag_context, **kwargs)
         else:
             _agent = self.build(ag_context)
 

src/agentscope_runtime/engine/agents/autogen_agent.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+import inspect
 from typing import Optional, Type, Callable
 
 from autogen_core.models import ChatCompletionClient
@@ -169,7 +170,12 @@ async def run_async(
         # We should always build a new agent since the state is manage outside
         # the agent
         if self._attr.get("custom_build_fn"):
-            _agent = self._attr["custom_build_fn"](ag_context, **kwargs)
+            build_fn = self._attr["custom_build_fn"]
+
+            if inspect.iscoroutinefunction(build_fn):
+                _agent = await build_fn(ag_context, **kwargs)
+            else:
+                _agent = build_fn(ag_context, **kwargs)
         else:
             _agent = self.build(ag_context)
 

src/agentscope_runtime/engine/deployers/__init__.py

@@ -7,9 +7,13 @@
 from .modelstudio_deployer import (
     ModelstudioDeployManager,
 )
-from .agentrun_deployer import (
-    AgentRunDeployManager,
-)
+
+try:
+    from .agentrun_deployer import (
+        AgentRunDeployManager,
+    )
+except ImportError:
+    AgentRunDeployManager = None  # type: ignore
 
 __all__ = [
     "DeployManager",