Function run_agent

astrbot/core/astr_agent_run_util.py:116–340

(
    agent_runner: AgentRunner,
    max_step: int = 30,
    show_tool_use: bool = True,
    show_tool_call_result: bool = False,
    stream_to_general: bool = False,
    show_reasoning: bool = False,
    buffer_intermediate_messages: bool = False,
)

async def run_agent(
    agent_runner: AgentRunner,
    max_step: int = 30,
    show_tool_use: bool = True,
    show_tool_call_result: bool = False,
    stream_to_general: bool = False,
    show_reasoning: bool = False,
    buffer_intermediate_messages: bool = False,
) -> AsyncGenerator[MessageChain | None, None]:
    step_idx = 0
    astr_event = agent_runner.run_context.context.event
    tool_name_by_call_id: dict[str, str] = {}
    buffered_llm_chains: list[MessageChain] = []
    can_buffer_llm_result = _should_buffer_llm_result(
        buffer_intermediate_messages,
        stream_to_general,
        agent_runner,
    )
    while step_idx < max_step + 1:
        step_idx += 1

        if step_idx == max_step + 1:
            logger.warning(
                f"Agent reached max steps ({max_step}), forcing a final response."
            )
            if not agent_runner.done():
                # 拔掉所有工具
                if agent_runner.req:
                    agent_runner.req.func_tool = None
                # 注入提示词
                agent_runner.run_context.messages.append(
                    Message(
                        role="user",
                        content="工具调用次数已达到上限，请停止使用工具，并根据已经收集到的信息，对你的任务和发现进行总结，然后直接回复用户。",
                    )
                )

        stop_watcher = asyncio.create_task(
            _watch_agent_stop_signal(agent_runner, astr_event),
        )
        try:
            async for resp in agent_runner.step():
                if _should_stop_agent(astr_event):
                    agent_runner.request_stop()

                if resp.type == "aborted":
                    if can_buffer_llm_result:
                        merged_chain = _merge_buffered_llm_chains(buffered_llm_chains)
                        if merged_chain:
                            astr_event.set_result(
                                MessageEventResult(
                                    chain=merged_chain.chain,
                                    result_content_type=ResultContentType.LLM_RESULT,
                                ),
                            )
                            yield merged_chain
                            astr_event.clear_result()
                    if not stop_watcher.done():