Model Context protocol (mcp)

class MCPAgent:
    def __init__(self, server_type="filesystem"):
        self.llm = get_llm()
        self.mcp_client = MCPClient(server_type=server_type)
        self.available_tools = self.mcp_client.list_tools()

    def process_query(self, query):
        # Show available tools to the LLM
        tools_desc = self.get_tools_description()
        prompt = f"""
        Available tools: {tools_desc}
        Query: {query}
        Which tool should I use? Format: TOOL: name, INPUT: data
        """
        response = self.llm.generate(prompt).content

        # Parse and execute the tool call
        tool_name, input_data = self.parse_tool_call(response)
        if tool_name:
            result = self.mcp_client.execute_tool(tool_name, input_data)
            # Generate final response using tool result
            final = self.llm.generate(
                f"Tool result: {result}\nAnswer the query: {query}"
            ).content
            return final
        return response