vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-05-05 15:31:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

refactor: use @extensible decorator on build_prompt functions

Browse source
This commit is contained in:
linuztx 2026-03-20 12:48:26 +08:00
parent 2566ee134d
commit 8847af089a
6 changed files with 18 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

7
extensions/python/system_prompt/_11_tools_prompt.py
View file

@ -1,7 +1,7 @@
import os
from typing import Any
from helpers.extension import Extension, call_extensions_async
from helpers.extension import Extension, extensible
from helpers import files, subagents
from helpers.print_style import PrintStyle
from agent import Agent, LoopData
@ -24,6 +24,7 @@ class ToolsPrompt(Extension):
system_prompt.append(prompt)
@extensible
async def build_prompt(agent: Agent) -> str:
# collect tool files from all prompt directories
prompt_dirs = subagents.get_paths(agent, "prompts")
@ -54,6 +55,4 @@ async def build_prompt(agent: Agent) -> str:
if chat_cfg.get("vision", False):
prompt += "\n\n" + agent.read_prompt("agent.system.tools_vision.md")
data: dict[str, Any] = {"prompt": prompt}
await call_extensions_async("system_prompt_tools", agent=agent, data=data)
return data["prompt"]
return prompt