Spaces:
Paused
Paused
File size: 854 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 | from typing import Any
from helpers.extension import Extension, extensible
from helpers.mcp_handler import MCPConfig
from agent import Agent, LoopData
class MCPToolsPrompt(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)
if prompt:
system_prompt.append(prompt)
@extensible
async def build_prompt(agent: Agent) -> str:
mcp_config = MCPConfig.get_instance()
if not mcp_config.servers:
return ""
pre_progress = agent.context.log.progress
agent.context.log.set_progress("Collecting MCP tools")
tools = mcp_config.get_tools_prompt()
agent.context.log.set_progress(pre_progress)
return tools
|