AstrBot 4.13.2__py3-none-any.whl → 4.14.1__py3-none-any.whl

astrbot/builtin_stars/astrbot/main.py

@@ -7,7 +7,6 @@ from astrbot.api.provider import LLMResponse, ProviderRequest
 from astrbot.core import logger
 
 from .long_term_memory import LongTermMemory
-from .process_llm_request import ProcessLLMRequest
 
 
 class Main(star.Star):
@@ -19,8 +18,6 @@ class Main(star.Star):
         except BaseException as e:
             logger.error(f"聊天增强 err: {e}")
 
-        self.proc_llm_req = ProcessLLMRequest(self.context)
-
     def ltm_enabled(self, event: AstrMessageEvent):
         ltmse = self.context.get_config(umo=event.unified_msg_origin)[
             "provider_ltm_settings"
@@ -80,7 +77,6 @@ class Main(star.Star):
 
                     yield event.request_llm(
                         prompt=prompt,
-                        func_tool_manager=self.context.get_llm_tool_manager(),
                         session_id=event.session_id,
                         conversation=conv,
                     )
@@ -91,8 +87,6 @@ class Main(star.Star):
     @filter.on_llm_request()
     async def decorate_llm_req(self, event: AstrMessageEvent, req: ProviderRequest):
         """在请求 LLM 前注入人格信息、Identifier、时间、回复内容等 System Prompt"""
-        await self.proc_llm_req.process_llm_request(event, req)
-
         if self.ltm and self.ltm_enabled(event):
             try:
                 await self.ltm.on_req_llm(event, req)

astrbot/builtin_stars/session_controller/main.py

@@ -49,7 +49,7 @@ class Main(Star):
                     if p_settings.get("empty_mention_waiting_need_reply", True):
                         try:
                             # 尝试使用 LLM 生成更生动的回复
-                            func_tools_mgr = self.context.get_llm_tool_manager()
+                            # func_tools_mgr = self.context.get_llm_tool_manager()
 
                             # 获取用户当前的对话信息
                             curr_cid = await self.context.conversation_manager.get_curr_conversation_id(
@@ -76,7 +76,6 @@ class Main(Star):
                                     "你友好地询问用户想要聊些什么或者需要什么帮助，回复要符合人设，不要太过机械化。"
                                     "请注意，你仅需要输出要回复用户的内容，不要输出其他任何东西"
                                 ),
-                                func_tool_manager=func_tools_mgr,
                                 session_id=curr_cid,
                                 contexts=[],
                                 system_prompt="",

astrbot/cli/__init__.py

@@ -1 +1 @@
-__version__ = "4.13.2"
+__version__ = "4.14.1"

astrbot/core/agent/agent.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Generic
+from typing import Any, Generic
 
 from .hooks import BaseAgentRunHooks
 from .run_context import TContext
@@ -12,3 +12,4 @@ class Agent(Generic[TContext]):
     instructions: str | None = None
     tools: list[str | FunctionTool] | None = None
     run_hooks: BaseAgentRunHooks[TContext] | None = None
+    begin_dialogs: list[Any] | None = None

astrbot/core/agent/handoff.py

@@ -12,16 +12,29 @@ class HandoffTool(FunctionTool, Generic[TContext]):
         self,
         agent: Agent[TContext],
         parameters: dict | None = None,
+        tool_description: str | None = None,
         **kwargs,
     ):
         self.agent = agent
+
+        # Avoid passing duplicate `description` to the FunctionTool dataclass.
+        # Some call sites (e.g. SubAgentOrchestrator) pass `description` via kwargs
+        # to override what the main agent sees, while we also compute a default
+        # description here.
+        # `tool_description` is the public description shown to the main LLM.
+        # Keep a separate kwarg to avoid conflicting with FunctionTool's `description`.
+        description = tool_description or self.default_description(agent.name)
         super().__init__(
             name=f"transfer_to_{agent.name}",
             parameters=parameters or self.default_parameters(),
-            description=agent.instructions or self.default_description(agent.name),
+            description=description,
             **kwargs,
         )
 
+        # Optional provider override for this subagent. When set, the handoff
+        # execution will use this chat provider id instead of the global/default.
+        self.provider_id: str | None = None
+
     def default_parameters(self) -> dict:
         return {
             "type": "object",

astrbot/core/agent/runners/tool_loop_agent_runner.py

@@ -111,10 +111,12 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
         # See #4681
         self.tool_schema_mode = tool_schema_mode
         self._tool_schema_param_set = None
+        self._skill_like_raw_tool_set = None
         if tool_schema_mode == "skills_like":
             tool_set = self.req.func_tool
             if not tool_set:
                 return
+            self._skill_like_raw_tool_set = tool_set
             light_set = tool_set.get_light_tool_set()
             self._tool_schema_param_set = tool_set.get_param_only_tool_set()
             # MODIFIE the req.func_tool to use light tool schemas
@@ -379,7 +381,17 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
             try:
                 if not req.func_tool:
                     return
-                func_tool = req.func_tool.get_tool(func_tool_name)
+
+                if (
+                    self.tool_schema_mode == "skills_like"
+                    and self._skill_like_raw_tool_set
+                ):
+                    # in 'skills_like' mode, raw.func_tool is light schema, does not have handler
+                    # so we need to get the tool from the raw tool set
+                    func_tool = self._skill_like_raw_tool_set.get_tool(func_tool_name)
+                else:
+                    func_tool = req.func_tool.get_tool(func_tool_name)
+
                 logger.info(f"使用工具：{func_tool_name}，参数：{func_tool_args}")
 
                 if not func_tool:
@@ -557,6 +569,7 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
                     )
                 ],
             )
+            logger.info(f"Tool `{func_tool_name}` Result: {last_tcr_content}")
 
         # 处理函数调用响应
         if tool_call_result_blocks:

astrbot/core/agent/tool.py

@@ -58,6 +58,11 @@ class FunctionTool(ToolSchema, Generic[TContext]):
     Whether the tool is active. This field is a special field for AstrBot.
     You can ignore it when integrating with other frameworks.
     """
+    is_background_task: bool = False
+    """
+    Declare this tool as a background task. Background tasks return immediately
+    with a task identifier while the real work continues asynchronously.
+    """
 
     def __repr__(self):
         return f"FuncTool(name={self.name}, parameters={self.parameters}, description={self.description})"

astrbot/core/astr_agent_run_util.py

@@ -54,6 +54,14 @@ async def run_agent(
                     return
                 if resp.type == "tool_call_result":
                     msg_chain = resp.data["chain"]
+
+                    astr_event.trace.record(
+                        "agent_tool_result",
+                        tool_result=msg_chain.get_plain_text(
+                            with_other_comps_mark=True
+                        ),
+                    )
+
                     if msg_chain.type == "tool_direct_result":
                         # tool_direct_result 用于标记 llm tool 需要直接发送给用户的内容
                         await astr_event.send(msg_chain)
@@ -67,12 +75,22 @@ async def run_agent(
                         # 用来标记流式响应需要分节
                         yield MessageChain(chain=[], type="break")
 
+                    tool_info = None
+
+                    if resp.data["chain"].chain:
+                        json_comp = resp.data["chain"].chain[0]
+                        if isinstance(json_comp, Json):
+                            tool_info = json_comp.data
+                        astr_event.trace.record(
+                            "agent_tool_call",
+                            tool_name=tool_info if tool_info else "unknown",
+                        )
+
                     if astr_event.get_platform_name() == "webchat":
                         await astr_event.send(resp.data["chain"])
                     elif show_tool_use:
-                        json_comp = resp.data["chain"].chain[0]
-                        if isinstance(json_comp, Json):
-                            m = f"🔨 调用工具: {json_comp.data.get('name')}"
+                        if tool_info:
+                            m = f"🔨 调用工具: {tool_info.get('name', 'unknown')}"
                         else:
                             m = "🔨 调用工具..."
                         chain = MessageChain(type="tool_call").message(m)

astrbot/core/astr_agent_tool_exec.py

@@ -1,23 +1,34 @@
 import asyncio
 import inspect
+import json
 import traceback
 import typing as T
+import uuid
 
 import mcp
 
 from astrbot import logger
 from astrbot.core.agent.handoff import HandoffTool
 from astrbot.core.agent.mcp_client import MCPTool
+from astrbot.core.agent.message import Message
 from astrbot.core.agent.run_context import ContextWrapper
 from astrbot.core.agent.tool import FunctionTool, ToolSet
 from astrbot.core.agent.tool_executor import BaseFunctionToolExecutor
 from astrbot.core.astr_agent_context import AstrAgentContext
+from astrbot.core.astr_main_agent_resources import (
+    BACKGROUND_TASK_RESULT_WOKE_SYSTEM_PROMPT,
+    SEND_MESSAGE_TO_USER_TOOL,
+)
+from astrbot.core.cron.events import CronMessageEvent
 from astrbot.core.message.message_event_result import (
     CommandResult,
     MessageChain,
     MessageEventResult,
 )
+from astrbot.core.platform.message_session import MessageSession
+from astrbot.core.provider.entites import ProviderRequest
 from astrbot.core.provider.register import llm_tools
+from astrbot.core.utils.history_saver import persist_agent_history
 
 
 class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
@@ -43,6 +54,31 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
                 yield r
             return
 
+        elif tool.is_background_task:
+            task_id = uuid.uuid4().hex
+
+            async def _run_in_background():
+                try:
+                    await cls._execute_background(
+                        tool=tool,
+                        run_context=run_context,
+                        task_id=task_id,
+                        **tool_args,
+                    )
+                except Exception as e:  # noqa: BLE001
+                    logger.error(
+                        f"Background task {task_id} failed: {e!s}",
+                        exc_info=True,
+                    )
+
+            asyncio.create_task(_run_in_background())
+            text_content = mcp.types.TextContent(
+                type="text",
+                text=f"Background task submitted. task_id={task_id}",
+            )
+            yield mcp.types.CallToolResult(content=[text_content])
+
+            return
         else:
             async for r in cls._execute_local(tool, run_context, **tool_args):
                 yield r
@@ -74,13 +110,35 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
         ctx = run_context.context.context
         event = run_context.context.event
         umo = event.unified_msg_origin
-        prov_id = await ctx.get_current_chat_provider_id(umo)
+
+        # Use per-subagent provider override if configured; otherwise fall back
+        # to the current/default provider resolution.
+        prov_id = getattr(
+            tool, "provider_id", None
+        ) or await ctx.get_current_chat_provider_id(umo)
+
+        # prepare begin dialogs
+        contexts = None
+        dialogs = tool.agent.begin_dialogs
+        if dialogs:
+            contexts = []
+            for dialog in dialogs:
+                try:
+                    contexts.append(
+                        dialog
+                        if isinstance(dialog, Message)
+                        else Message.model_validate(dialog)
+                    )
+                except Exception:
+                    continue
+
         llm_resp = await ctx.tool_loop_agent(
             event=event,
             chat_provider_id=prov_id,
             prompt=input_,
             system_prompt=tool.agent.instructions,
             tools=toolset,
+            contexts=contexts,
             max_steps=30,
             run_hooks=tool.agent.run_hooks,
         )
@@ -88,11 +146,128 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
             content=[mcp.types.TextContent(type="text", text=llm_resp.completion_text)]
         )
 
+    @classmethod
+    async def _execute_background(
+        cls,
+        tool: FunctionTool,
+        run_context: ContextWrapper[AstrAgentContext],
+        task_id: str,
+        **tool_args,
+    ):
+        from astrbot.core.astr_main_agent import (
+            MainAgentBuildConfig,
+            _get_session_conv,
+            build_main_agent,
+        )
+
+        # run the tool
+        result_text = ""
+        try:
+            async for r in cls._execute_local(
+                tool, run_context, tool_call_timeout=3600, **tool_args
+            ):
+                # collect results, currently we just collect the text results
+                if isinstance(r, mcp.types.CallToolResult):
+                    result_text = ""
+                    for content in r.content:
+                        if isinstance(content, mcp.types.TextContent):
+                            result_text += content.text + "\n"
+        except Exception as e:
+            result_text = (
+                f"error: Background task execution failed, internal error: {e!s}"
+            )
+
+        event = run_context.context.event
+        ctx = run_context.context.context
+
+        note = (
+            event.get_extra("background_note")
+            or f"Background task {tool.name} finished."
+        )
+        extras = {
+            "background_task_result": {
+                "task_id": task_id,
+                "tool_name": tool.name,
+                "result": result_text or "",
+                "tool_args": tool_args,
+            }
+        }
+        session = MessageSession.from_str(event.unified_msg_origin)
+        cron_event = CronMessageEvent(
+            context=ctx,
+            session=session,
+            message=note,
+            extras=extras,
+            message_type=session.message_type,
+        )
+        cron_event.role = event.role
+        config = MainAgentBuildConfig(tool_call_timeout=3600)
+
+        req = ProviderRequest()
+        conv = await _get_session_conv(event=cron_event, plugin_context=ctx)
+        req.conversation = conv
+        context = json.loads(conv.history)
+        if context:
+            req.contexts = context
+            context_dump = req._print_friendly_context()
+            req.contexts = []
+            req.system_prompt += (
+                "\n\nBellow is you and user previous conversation history:\n"
+                f"{context_dump}"
+            )
+
+        bg = json.dumps(extras["background_task_result"], ensure_ascii=False)
+        req.system_prompt += BACKGROUND_TASK_RESULT_WOKE_SYSTEM_PROMPT.format(
+            background_task_result=bg
+        )
+        req.prompt = (
+            "Proceed according to your system instructions. "
+            "Output using same language as previous conversation."
+            " After completing your task, summarize and output your actions and results."
+        )
+        if not req.func_tool:
+            req.func_tool = ToolSet()
+        req.func_tool.add_tool(SEND_MESSAGE_TO_USER_TOOL)
+
+        result = await build_main_agent(
+            event=cron_event, plugin_context=ctx, config=config, req=req
+        )
+        if not result:
+            logger.error("Failed to build main agent for background task job.")
+            return
+
+        runner = result.agent_runner
+        async for _ in runner.step_until_done(30):
+            # agent will send message to user via using tools
+            pass
+        llm_resp = runner.get_final_llm_resp()
+        task_meta = extras.get("background_task_result", {})
+        summary_note = (
+            f"[BackgroundTask] {task_meta.get('tool_name', tool.name)} "
+            f"(task_id={task_meta.get('task_id', task_id)}) finished. "
+            f"Result: {task_meta.get('result') or result_text or 'no content'}"
+        )
+        if llm_resp and llm_resp.completion_text:
+            summary_note += (
+                f"I finished the task, here is the result: {llm_resp.completion_text}"
+            )
+        await persist_agent_history(
+            ctx.conversation_manager,
+            event=cron_event,
+            req=req,
+            summary_note=summary_note,
+        )
+        if not llm_resp:
+            logger.warning("background task agent got no response")
+            return
+
     @classmethod
     async def _execute_local(
         cls,
         tool: FunctionTool,
         run_context: ContextWrapper[AstrAgentContext],
+        *,
+        tool_call_timeout: int | None = None,
         **tool_args,
     ):
         event = run_context.context.event
@@ -133,7 +308,7 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
             try:
                 resp = await asyncio.wait_for(
                     anext(wrapper),
-                    timeout=run_context.tool_call_timeout,
+                    timeout=tool_call_timeout or run_context.tool_call_timeout,
                 )
                 if resp is not None:
                     if isinstance(resp, mcp.types.CallToolResult):
@@ -165,7 +340,7 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
                     yield None
             except asyncio.TimeoutError:
                 raise Exception(
-                    f"tool {tool.name} execution timeout after {run_context.tool_call_timeout} seconds.",
+                    f"tool {tool.name} execution timeout after {tool_call_timeout or run_context.tool_call_timeout} seconds.",
                 )
             except StopAsyncIteration:
                 break