代码

cookbook/agent_concepts/user_control_flows/external_tool_execution_async.py
import asyncio
import subprocess

from agno.agent import Agent
from agno.models.openai import OpenAIChat
from agno.tools import tool
from agno.utils import pprint

@tool(external_execution=True)
def execute_shell_command(command: str) -> str:
    """执行一个 shell 命令。

    Args:
        command (str): 要执行的 shell 命令

    Returns:
        str: shell 命令的输出
    """
    if command.startswith("ls"):
        return subprocess.check_output(command, shell=True).decode("utf-8")
    else:
        raise Exception(f"不支持的命令: {command}")

agent = Agent(
    model=OpenAIChat(id="gpt-4o-mini"),
    tools=[execute_shell_command],
    markdown=True,
)

run_response = asyncio.run(agent.arun("我当前目录下有什么文件?"))
if run_response.is_paused:
    for tool in run_response.tools_awaiting_external_execution:
        if tool.tool_name == execute_shell_command.name:
            print(f"正在外部执行 {tool.tool_name},参数为 {tool.tool_args}")
            # 我们自己执行工具。您也可以在这里执行完全外部的东西。
            result = execute_shell_command.entrypoint(**tool.tool_args)
            # 我们必须在工具执行对象上设置结果,以便代理可以继续
            tool.result = result

    run_response = asyncio.run(agent.acontinue_run(run_response=run_response))
    pprint.pprint_run_response(run_response)

用法

1

创建虚拟环境

打开 Terminal 并创建一个 python 虚拟环境。

python3 -m venv .venv
source .venv/bin/activate
2

设置你的 API 密钥

export OPENAI_API_KEY=xxx
3

安装库

pip install -U agno openai
4

运行示例

python cookbook/agent_concepts/user_control_flows/external_tool_execution_async.py

主要特性

  • 使用 agent.arun() 进行异步代理执行
  • 实现 agent.acontinue_run() 进行异步继续执行
  • 保持与同步版本相同的外部工具执行流程
  • 演示了如何处理外部工具的异步执行

用例

  • 非阻塞式外部工具执行
  • 需要异步执行的高性能应用程序
  • 带有外部服务调用的 Web 应用程序
  • 带有外部工具的长时间运行操作