mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-07-09 17:08:29 +00:00
bcf2634000
Load active-project AGENTS.md path-chain guidance into the protocol area without duplicating the project root instructions. Move the AGENTS.md protocol wording into a prompt template, reuse existing file/path helpers, and cover direct-path discovery plus prompt assembly with focused tests.
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
from typing import Any
|
|
|
|
from helpers.extension import Extension, extensible
|
|
from helpers import projects
|
|
from agent import Agent, LoopData
|
|
|
|
|
|
class ProjectPrompt(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, loop_data=loop_data)
|
|
if prompt:
|
|
system_prompt.append(prompt)
|
|
|
|
|
|
@extensible
|
|
async def build_prompt(agent: Agent, loop_data: LoopData | None = None) -> str:
|
|
result = agent.read_prompt("agent.system.projects.main.md")
|
|
project_name = agent.context.get_data(projects.CONTEXT_DATA_KEY_PROJECT)
|
|
if loop_data:
|
|
loop_data.protocol_persistent.pop("agents_md_instructions", None)
|
|
loop_data.protocol_persistent.pop("project_instructions", None)
|
|
if project_name:
|
|
project_vars = projects.build_system_prompt_vars(project_name)
|
|
if loop_data and project_vars.get("include_agents_md", True):
|
|
agents_md_protocol = projects.build_agents_md_protocol(project_name)
|
|
if agents_md_protocol:
|
|
loop_data.protocol_persistent["agents_md_instructions"] = (
|
|
agents_md_protocol
|
|
)
|
|
if loop_data and project_vars.get("project_instructions"):
|
|
loop_data.protocol_persistent["project_instructions"] = agent.read_prompt(
|
|
"agent.protocol.projects.instructions.md",
|
|
**project_vars,
|
|
)
|
|
result += "\n\n" + agent.read_prompt(
|
|
"agent.system.projects.active.md", **project_vars
|
|
)
|
|
else:
|
|
result += "\n\n" + agent.read_prompt("agent.system.projects.inactive.md")
|
|
return result
|