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

dec's updates

Browse source
This commit is contained in:
frdel 2025-05-22 10:02:39 +02:00
commit a00c433658
36 changed files with 5215 additions and 136 deletions
Download patch file Download diff file Expand all files Collapse all files

15
python/extensions/system_prompt/_10_system_prompt.py
View file

@ -1,17 +1,22 @@
from datetime import datetime, timezone
from datetime import datetime
from typing import Any, Optional
from python.helpers.extension import Extension
from python.helpers.mcp_handler import MCPConfig
from agent import Agent, LoopData
from python.helpers.localization import Localization
class SystemPrompt(Extension):
async def execute(self, system_prompt: list[str]=[], loop_data: LoopData = LoopData(), **kwargs):
async def execute(self, system_prompt: list[str] = [], loop_data: LoopData = LoopData(), **kwargs: Any):
# append main system prompt and tools
main = get_main_prompt(self.agent)
tools = get_tools_prompt(self.agent)
mcp_tools = get_mcp_tools_prompt(self.agent)
system_prompt.append(main)
system_prompt.append(tools)
system_prompt.append(mcp_tools)
def get_main_prompt(agent: Agent):
@ -22,4 +27,8 @@ def get_tools_prompt(agent: Agent):
prompt = agent.read_prompt("agent.system.tools.md")
if agent.config.chat_model.vision:
prompt += '\n' + agent.read_prompt("agent.system.tools_vision.md")
return prompt
return prompt
def get_mcp_tools_prompt(agent: Agent):
return MCPConfig.get_instance().get_tools_prompt()