diff --git a/owl/run_mcp.py b/owl/run_mcp.py index 67849ee..a535658 100644 --- a/owl/run_mcp.py +++ b/owl/run_mcp.py @@ -102,7 +102,7 @@ from camel.types import ModelPlatformType, ModelType from camel.logger import set_log_level from camel.toolkits import MCPToolkit -from utils.enhanced_role_playing import OwlRolePlaying, run_society +from utils.enhanced_role_playing import OwlRolePlaying, arun_society load_dotenv() @@ -169,7 +169,7 @@ async def main(): # Connect to all MCP toolkits tools = [*mcp_toolkit.get_tools()] society = await construct_society(question, tools) - answer, chat_history, token_count = await run_society(society) + answer, chat_history, token_count = await arun_society(society) print(f"\033[94mAnswer: {answer}\033[0m") finally: diff --git a/owl/utils/__init__.py b/owl/utils/__init__.py index 3f6267c..ba242a9 100644 --- a/owl/utils/__init__.py +++ b/owl/utils/__init__.py @@ -13,7 +13,12 @@ # ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. ========= from .common import extract_pattern -from .enhanced_role_playing import OwlRolePlaying, OwlGAIARolePlaying, run_society +from .enhanced_role_playing import ( + OwlRolePlaying, + OwlGAIARolePlaying, + run_society, + arun_society, +) from .gaia import GAIABenchmark from .document_toolkit import DocumentProcessingToolkit @@ -22,6 +27,7 @@ __all__ = [ "OwlRolePlaying", "OwlGAIARolePlaying", "run_society", + "arun_society", "GAIABenchmark", "DocumentProcessingToolkit", ]