Upload 5 files

examples/callback/mcp.json

@@ -0,0 +1,85 @@
+{
+  "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": {
+            "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/callback/run-test.py

@@ -0,0 +1,122 @@
+# 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.agent.llm_agent import Agent
+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
+
+#import aworld.tools.examples.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="现在纽约、上海、北京的天气怎么样？这里是三个城市，希望大模型识别调用工具的时候返回三个工具",
+        # 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/callback/simple_callback.py

@@ -0,0 +1,59 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+
+"""
+Simple tool callback example, demonstrating the basic functionality of callback registration and execution.
+"""
+
+# Import business package, its __init__.py will automatically import and register callback functions
+import business
+from aworld.runners.callback.decorator import reg_callback, CallbackRegistry
+
+
+# Import CallbackRegistry
+
+
+@reg_callback("mcp_server__action")
+def simple_callback(content):
+    """Simple callback function, prints content and returns it
+
+    Args:
+        content: Content to print
+
+    Returns:
+        The input content
+    """
+    print(f"Callback function received content: {content}")
+    return content
+
+def main():
+    """Main function, demonstrating how to get and execute callback functions"""
+    # List all registered callback functions
+    # print("\n===== Registered Callback Functions =====")
+    # business.list_all_callbacks()
+    
+    # Get and execute print_content callback function
+    print("\n===== Execute print_content Callback Function =====")
+    callback_func = CallbackRegistry.get("mcp_server__action")
+    
+    if callback_func:
+        print("Callback function found, executing...")
+        result = callback_func("Hello, Callback!!!!!")
+        print(f"Callback function execution result: {result}")
+    else:
+        print("print_content callback function not found")
+
+    # # Get and execute uppercase_content callback function
+    # print("\n===== Execute uppercase_content Callback Function =====")
+    # uppercase_func = CallbackRegistry.get("uppercase_content")
+    #
+    # if uppercase_func:
+    #     print("Callback function found, executing...")
+    #     result = uppercase_func("Hello, Uppercase Callback!")
+    #     print(f"Callback function execution result: {result}")
+    # else:
+    #     print("uppercase_content callback function not found")
+
+
+if __name__ == "__main__":
+    main() 

examples/callback/test_env.py

@@ -0,0 +1,12 @@
+import os
+
+
+def main():
+    from dotenv import load_dotenv
+    load_dotenv()
+    print(os.environ)
+    uid = os.getenv('AWORLD_SEARCH_UID')
+    print(uid)
+
+if __name__ == "__main__":
+    main()