Upload 5 files

examples/sandbox/mcp.json

@@ -0,0 +1,88 @@
+{
+  "mcpServers": {
+    "amap-amap-sse": {
+      "type": "sse",
+      "url": "https://mcp.amap.com/sse?key=${AMAP_AMAP_SSE_KEY}",
+      "timeout": 5.0,
+      "sse_read_timeout": 300.0
+    },
+    "tavily-mcp": {
+      "type": "stdio",
+      "command": "npx",
+      "args": ["-y", "[email protected]"],
+      "env": {
+        "TAVILY_API_KEY": "tvly-dev-"
+      }
+    },
+    "aworldsearch_server": {
+        "type": "function_tool"
+    },
+    "aworldsearch_server1": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.aworldsearch_server"
+            ],
+            "env": {
+              "AWORLD_SEARCH_URL": "${AWORLD_SEARCH_URL}",
+              "AWORLD_SEARCH_TOTAL_NUM": "${AWORLD_SEARCH_TOTAL_NUM}",
+              "AWORLD_SEARCH_SLICE_NUM": "${AWORLD_SEARCH_SLICE_NUM}",
+              "AWORLD_SEARCH_DOMAIN": "${AWORLD_SEARCH_DOMAIN}",
+              "AWORLD_SEARCH_SEARCHMODE": "${AWORLD_SEARCH_SEARCHMODE}",
+              "AWORLD_SEARCH_SOURCE": "${AWORLD_SEARCH_SOURCE}",
+              "AWORLD_SEARCH_UID": "${AWORLD_SEARCH_UID}"
+            }
+        },
+    "picsearch_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.picsearch_server"
+            ],
+            "env": {
+              "PIC_SEARCH_URL": "${PIC_SEARCH_URL}",
+              "PIC_SEARCH_TOTAL_NUM": "${PIC_SEARCH_TOTAL_NUM}",
+              "PIC_SEARCH_SLICE_NUM": "${PIC_SEARCH_SLICE_NUM}",
+              "PIC_SEARCH_DOMAIN": "${PIC_SEARCH_DOMAIN}",
+              "PIC_SEARCH_SEARCHMODE": "${PIC_SEARCH_SEARCHMODE}",
+              "PIC_SEARCH_SOURCE": "${PIC_SEARCH_SOURCE}"
+            }
+        },
+    "gen_audio_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.gen_audio_server"
+            ],
+            "env": {
+              "AUDIO_TASK_URL": "${AUDIO_TASK_URL}",
+              "AUDIO_QUERY_URL": "${AUDIO_QUERY_URL}",
+              "AUDIO_APP_KEY": "${AUDIO_APP_KEY}",
+              "AUDIO_SECRET": "${AUDIO_SECRET}",
+              "AUDIO_SAMPLE_RATE": "${AUDIO_SAMPLE_RATE}",
+              "AUDIO_AUDIO_FORMAT": "${AUDIO_AUDIO_FORMAT}",
+              "AUDIO_TTS_VOICE": "${AUDIO_TTS_VOICE}",
+              "AUDIO_TTS_SPEECH_RATE": "${AUDIO_TTS_SPEECH_RATE}",
+              "AUDIO_TTS_VOLUME": "${AUDIO_TTS_VOLUME}",
+              "AUDIO_TTS_PITCH": "${AUDIO_TTS_PITCH}",
+              "AUDIO_VOICE_TYPE": "${AUDIO_VOICE_TYPE}"
+            }
+        },
+    "gen_video_server": {
+            "command": "python",
+            "args": [
+                "-m",
+                "mcp_servers.gen_video_server"
+            ],
+            "env": {
+              "DASHSCOPE_API_KEY": "${DASHSCOPE_API_KEY}",
+              "DASHSCOPE_VIDEO_SUBMIT_URL": "${DASHSCOPE_VIDEO_SUBMIT_URL}",
+              "DASHSCOPE_QUERY_BASE_URL": "${DASHSCOPE_QUERY_BASE_URL}",
+              "DASHSCOPE_VIDEO_MODEL": "${DASHSCOPE_VIDEO_MODEL}",
+              "DASHSCOPE_VIDEO_SIZE": "${DASHSCOPE_VIDEO_SIZE}",
+              "DASHSCOPE_VIDEO_SLEEP_TIME": "${DASHSCOPE_VIDEO_SLEEP_TIME}",
+              "DASHSCOPE_VIDEO_RETRY_TIMES": "${DASHSCOPE_VIDEO_RETRY_TIMES}"
+            }
+        }
+  }
+}

examples/sandbox/run-inner-tool-test.py

@@ -0,0 +1,120 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+import asyncio
+import json
+import logging
+import os
+
+from dotenv import load_dotenv
+
+from aworld.agents.llm_agent import Agent
+from aworld.config.conf import AgentConfig, TaskConfig
+from aworld.core.task import Task
+
+from aworld.runner import Runners
+from aworld.runners.callback.decorator import reg_callback
+from aworld.tools.mcp_tool import async_mcp_tool
+from examples.sandbox.inner_tools import aworldsearch_function_tools
+
+
+
+
+@reg_callback("print_content")
+def simple_callback(content):
+    """Simple callback function, prints content and returns it
+
+        Args:
+            content: Content to print
+
+        Returns:
+            The input content
+        """
+    print(f"callback content: {content}")
+    return content
+
+async def run():
+    load_dotenv()
+    llm_provider = os.getenv("LLM_PROVIDER_WEATHER", "openai")
+    llm_model_name = os.getenv("LLM_MODEL_NAME_WEATHER")
+    llm_api_key = os.getenv("LLM_API_KEY_WEATHER")
+    llm_base_url = os.getenv("LLM_BASE_URL_WEATHER")
+    llm_temperature = os.getenv("LLM_TEMPERATURE_WEATHER", 0.0)
+
+    agent_config = AgentConfig(
+        llm_provider=llm_provider,
+        llm_model_name=llm_model_name,
+        llm_api_key=llm_api_key,
+        llm_base_url=llm_base_url,
+        llm_temperature=llm_temperature,
+    )
+    #mcp_servers = ["filewrite_server", "fileread_server"]
+    #mcp_servers = ["amap-amap-sse","filewrite_server", "fileread_server"]
+    #mcp_servers = ["file_server"]
+    #mcp_servers = ["amap-amap-sse"]
+    mcp_servers = ["aworldsearch_server"]
+    #mcp_servers = ["gen_video_server"]
+   # mcp_servers = ["picsearch_server"]
+    #mcp_servers = ["gen_audio_server"]
+    #mcp_servers = ["playwright"]
+    #mcp_servers = ["tavily-mcp"]
+
+    path_cwd = os.path.dirname(os.path.abspath(__file__))
+    mcp_path = os.path.join(path_cwd, "mcp.json")
+    with open(mcp_path, "r") as f:
+        mcp_config = json.load(f)
+
+    print("-------------------mcp_config--------------",mcp_config)
+
+    #sand_box = Sandbox(mcp_servers=mcp_servers,mcp_config=mcp_config)
+    # You can specify sandbox
+    #sand_box = Sandbox(mcp_servers=mcp_servers, mcp_config=mcp_config,env_type=SandboxEnvType.K8S)
+    #sand_box = Sandbox(mcp_servers=mcp_servers, mcp_config=mcp_config,env_type=SandboxEnvType.SUPERCOMPUTER)
+
+    search_sys_prompt = "You are a versatile assistant"
+    search = Agent(
+        conf=agent_config,
+        name="search_agent",
+        system_prompt=search_sys_prompt,
+        mcp_config=mcp_config,
+        mcp_servers=mcp_servers,
+        #sandbox=sand_box,
+    )
+
+    # Run agent
+    # Runners.sync_run(input="Use tavily-mcp to check what tourist attractions are in Hangzhou", agent=search)
+    task = Task(
+        # input="Use tavily-mcp to check what tourist attractions are in Hangzhou",
+        # input="Use the file_server tool to analyze this audio link: https://amap-aibox-data.oss-cn-zhangjiakou.aliyuncs.com/.mp3",
+        # input="Use the amap-amap-sse tool to find hotels within one kilometer of West Lake in Hangzhou",
+        input="Use the aworldsearch_server tool to search for the origin of the Dragon Boat Festival",
+        # input="Use the picsearch_server tool to search for Captain America",
+        # input="Make sure to use the human_confirm tool to let the user confirm this message: 'Do you want to make a payment to this customer'",
+        # input="Use the gen_audio_server tool to convert this sentence to audio: 'Nice to meet you'",
+        #input="Use the gen_video_server tool to generate a video of this description: 'A cat walking alone on a snowy day'",
+        #input="How's the weather in New York, Shanghai, and Beijing right now? These are three cities, I hope the large model returns three tools when it identifies tool calls",
+        # input="First call the filewrite_server tool, then call the fileread_server tool",
+        # input="Use the playwright tool, with Google browser, search for the latest news about the Trump administration on www.baidu.com",
+        # input="Use tavily-mcp",
+        agent=search,
+        conf=TaskConfig(),
+        event_driven=True
+    )
+
+    #result = Runners.sync_run_task(task)
+    #result = Runners.sync_run_task(task)
+    #result = await Runners.streamed_run_task(task)
+    # result = await Runners.run_task(task)
+    # print(
+    #     "----------------------------------------------------------------------------------------------"
+    # )
+    # print(result)
+    # async for chunk in Runners.streamed_run_task(task).stream_events():
+    #     print(chunk, end="", flush=True)
+
+    async for output in Runners.streamed_run_task(task).stream_events():
+        print(f"Agent Ouput: {output}")
+
+
+
+if __name__ == "__main__":
+    asyncio.run(run())

examples/sandbox/run-test.py

@@ -0,0 +1,59 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+import asyncio
+import json
+import os
+
+from dotenv import load_dotenv
+
+from aworld.config.conf import AgentConfig, TaskConfig
+from aworld.agents.llm_agent import Agent
+from aworld.core.task import Task
+from aworld.runner import Runners
+
+
+async def run():
+    load_dotenv()
+    llm_provider = os.getenv("LLM_PROVIDER_WEATHER", "openai")
+    llm_model_name = os.getenv("LLM_MODEL_NAME_WEATHER")
+    llm_api_key = os.getenv("LLM_API_KEY_WEATHER")
+    llm_base_url = os.getenv("LLM_BASE_URL_WEATHER")
+    llm_temperature = os.getenv("LLM_TEMPERATURE_WEATHER", 0.0)
+
+    agent_config = AgentConfig(
+        llm_provider=llm_provider,
+        llm_model_name=llm_model_name,
+        llm_api_key=llm_api_key,
+        llm_base_url=llm_base_url,
+        llm_temperature=llm_temperature,
+    )
+    mcp_servers = ["tavily-mcp"]
+
+    path_cwd = os.path.dirname(os.path.abspath(__file__))
+    mcp_path = os.path.join(path_cwd, "mcp.json")
+    with open(mcp_path, "r") as f:
+        mcp_config = json.load(f)
+
+    search_sys_prompt = "You are a versatile assistant"
+    search = Agent(
+        conf=agent_config,
+        name="search_agent",
+        system_prompt=search_sys_prompt,
+        mcp_config=mcp_config,
+        mcp_servers=mcp_servers,
+    )
+
+    # Run agent
+    task = Task(
+        input="Use tavily-mcp to check what tourist attractions are in Hangzhou",
+        agent=search,
+        conf=TaskConfig(),
+    )
+
+    result = Runners.sync_run_task(task)
+    print( "----------------------------------------------------------------------------------------------")
+    print(result)
+
+
+if __name__ == "__main__":
+    asyncio.run(run())

examples/sandbox/test_inner_tool.py

@@ -0,0 +1,76 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+
+import logging
+from aworld.tools import get_function_tools, list_function_tools
+from examples.sandbox.inner_tools import aworldsearch_function_tools, another_function_tools
+
+aworldsearch_server = get_function_tools("aworldsearch_server")
+
+print(aworldsearch_server.list_tools())
+res = aworldsearch_server.call_tool("search", {"query_list": ["Tencent financial report", "Baidu financial report", "Alibaba financial report"],})
+print(res)
+
+another_server = get_function_tools("another-server")
+print(another_server.list_tools())
+
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+
+# Step 1: Import different modules, which will automatically register their respective FunctionTools instances
+print("=== Step 1: Import modules, automatically register FunctionTools instances ===")
+# Import aworldsearch_function_tools module, which registers "aworldsearch-server"
+print("Imported aworldsearch_function_tools module")
+
+# Import another_function_tools module, which registers "another-server"
+
+print("Imported another_function_tools module")
+
+# Step 2: Get FunctionTools instances by name
+print("\n=== Step 2: Get FunctionTools instances by name ===")
+from aworld.tools import get_function_tools, list_function_tools
+
+# List all registered FunctionTools servers
+print(f"All registered servers: {list_function_tools()}")
+
+# Get server instance by specific name
+aworldsearch_server = get_function_tools("aworldsearch-server")
+print(f"Retrieved server: {aworldsearch_server.name}")
+print(f"Server description: {aworldsearch_server.description}")
+
+another_server = get_function_tools("another-server")
+print(f"Retrieved server: {another_server.name}")
+print(f"Server description: {another_server.description}")
+
+# Step 3: Use the retrieved instances to call methods
+print("\n=== Step 3: Use the retrieved instances to call methods ===")
+# List all tools of aworldsearch server
+print("aworldsearch-server tool list:")
+for tool in aworldsearch_server.list_tools():
+    print(f"  - {tool.name}: {tool.description}")
+
+# List all tools of another server
+print("\nanother-server tool list:")
+for tool in another_server.list_tools():
+    print(f"  - {tool.name}: {tool.description}")
+
+# Step 4: Call tools
+print("\n=== Step 4: Call tool examples ===")
+# Call aworldsearch server's tool
+if "demo_search" in [tool.name for tool in aworldsearch_server.list_tools()]:
+    print("Calling demo_search tool:")
+    result = aworldsearch_server.call_tool("demo_search", {"query_list": ["Test query"]})
+    print(result)
+
+# Call another server's tool
+if "get_weather" in [tool.name for tool in another_server.list_tools()]:
+    print("\nCalling get_weather tool:")
+    result = another_server.call_tool("get_weather", {"city": "Beijing"})
+    print(result)
+
+if __name__ == "__main__":
+    pass  # Main logic has already been executed at the module level