zrem

Instructions

Implementation Reference

• src/tools/sorted_set.py:59-79 (handler)

  The zrem tool handler: removes a member from a Redis sorted set using r.zrem() and returns a confirmation or error message.

  @mcp.tool()
  async def zrem(key: str, member: str) -> str:
      """Remove a member from a Redis sorted set.
  
      Args:
          key (str): The sorted set key.
          member (str): The member to remove.
  
      Returns:
          str: Confirmation message or an error message.
      """
      try:
          r = RedisConnectionManager.get_connection()
          result = r.zrem(key, member)
          return (
              f"Successfully removed {member} from {key}"
              if result
              else f"Member {member} not found in {key}"
          )
      except RedisError as e:
          return f"Error removing from sorted set {key}: {str(e)}"

• src/tools/sorted_set.py:59-60 (registration)

  The @mcp.tool() decorator registers the zrem function as an MCP tool.

  @mcp.tool()
  async def zrem(key: str, member: str) -> str:

• src/tools/sorted_set.py:1-80 (helper)

  File imports: RedisConnectionManager for DB connection, RedisError for error handling, and the mcp server for tool registration.

  from typing import Optional
  
  from redis.exceptions import RedisError
  
  from src.common.connection import RedisConnectionManager
  from src.common.server import mcp
  
  
  @mcp.tool()
  async def zadd(
      key: str, score: float, member: str, expiration: Optional[int] = None
  ) -> str:
      """Add a member to a Redis sorted set with an optional expiration time.
  
      Args:
          key (str): The sorted set key.
          score (float): The score of the member.
          member (str): The member to add.
          expiration (int, optional): Expiration time in seconds.
  
      Returns:
          str: Confirmation message or an error message.
      """
      try:
          r = RedisConnectionManager.get_connection()
          r.zadd(key, {member: score})
          if expiration:
              r.expire(key, expiration)
          return f"Successfully added {member} to {key} with score {score}" + (
              f" and expiration {expiration} seconds" if expiration else ""
          )
      except RedisError as e:
          return f"Error adding to sorted set {key}: {str(e)}"
  
  
  @mcp.tool()
  async def zrange(key: str, start: int, end: int, with_scores: bool = False) -> str:
      """Retrieve a range of members from a Redis sorted set.
  
      Args:
          key (str): The sorted set key.
          start (int): The starting index.
          end (int): The ending index.
          with_scores (bool, optional): Whether to include scores in the result.
  
      Returns:
          str: The sorted set members in the given range or an error message.
      """
      try:
          r = RedisConnectionManager.get_connection()
          members = r.zrange(key, start, end, withscores=with_scores)
          return (
              str(members) if members else f"Sorted set {key} is empty or does not exist"
          )
      except RedisError as e:
          return f"Error retrieving sorted set {key}: {str(e)}"
  
  
  @mcp.tool()
  async def zrem(key: str, member: str) -> str:
      """Remove a member from a Redis sorted set.
  
      Args:
          key (str): The sorted set key.
          member (str): The member to remove.
  
      Returns:
          str: Confirmation message or an error message.
      """
      try:
          r = RedisConnectionManager.get_connection()
          result = r.zrem(key, member)
          return (
              f"Successfully removed {member} from {key}"
              if result
              else f"Member {member} not found in {key}"
          )
      except RedisError as e:
          return f"Error removing from sorted set {key}: {str(e)}"