from amfs import AgentMemory
from amfs_branching.sdk import extend_with_branching
mem = AgentMemory(agent_id="deploy-agent")
extend_with_branching(mem)
# Create and switch to a branch
mem.create_branch("experiment/retry-v3")
mem.switch_branch("experiment/retry-v3")
mem.write("checkout-service", "retry-pattern", {"max_retries": 5})
# Grant access to a specific API key
mem.grant_branch_access("experiment/retry-v3", "api_key", "amfs_sk_bob", "read")
# Diff and merge
diffs = mem.diff_branch("experiment/retry-v3")
result = mem.merge_branch("experiment/retry-v3")