Spaces:
Paused
Paused
File size: 1,766 Bytes
3dfb537 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | import os
from typing import Any
from helpers.extension import Extension, extensible
from helpers import files, subagents
from helpers.print_style import PrintStyle
from agent import Agent, LoopData
TOOL_KWARGS_KEY = "_tool_prompt_kwargs"
class ToolsPrompt(Extension):
async def execute(
self,
system_prompt: list[str] = [],
loop_data: LoopData = LoopData(),
**kwargs: Any,
):
if not self.agent:
return
prompt = await build_prompt(self.agent)
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")
tool_files = files.get_unique_filenames_in_dirs(
prompt_dirs, "agent.system.tool.*.md"
)
# per-file kwargs registered by plugin config extensions (e.g. _09_text_editor_config)
all_tool_kwargs: dict[str, dict[str, Any]] = agent.get_data(TOOL_KWARGS_KEY) or {}
tools: list[str] = []
for tool_file in tool_files:
try:
basename = os.path.basename(tool_file)
extra = all_tool_kwargs.get(basename, {})
tool = agent.read_prompt(basename, **extra)
tools.append(tool)
except Exception as e:
PrintStyle().error(f"Error loading tool '{tool_file}': {e}")
tools_str = "\n\n".join(tools)
prompt = agent.read_prompt("agent.system.tools.md", tools=tools_str)
# vision support
from plugins._model_config.helpers.model_config import get_chat_model_config
chat_cfg = get_chat_model_config(agent)
if chat_cfg.get("vision", False):
prompt += "\n\n" + agent.read_prompt("agent.system.tools_vision.md")
return prompt
|