imdb-mcp-server

Instructions

Implementation Reference

• src/imdb_mcp_server/tools.py:248-260 (handler)

  The main execution logic for the 'get_top_250_tv_shows' tool. It constructs the API URL, fetches data using the make_imdb_request helper, handles empty responses, paginates the results using paginated_response helper, and returns formatted JSON.

  @mcp.tool()
  async def get_top_250_tv_shows(start: int, ctx: Context) -> str:
      """Get the top 250 TV shows from IMDb with pagination.
      Args:
          start: The starting index (0-based) to retrieve TV shows from.
      Returns:
          JSON object containing 5 top TV shows starting from the specified index.
      """
      top_250_tv_shows_url = f"{BASE_URL}/top250-tv"
      top_250_tv_shows_data = await make_imdb_request(top_250_tv_shows_url, {}, ctx)
      if not top_250_tv_shows_data:
          return "Unable to fetch top 250 TV shows data."
      return json.dumps(paginated_response(top_250_tv_shows_data, start, len(top_250_tv_shows_data)), indent=4)

• src/imdb_mcp_server/api.py:58-77 (helper)

  Supporting utility function used by the tool to create a standardized paginated JSON response structure with a fixed page size of 5 items.

  def paginated_response(items, start, total_count=None):
      """Format a paginated response with a fixed page size of 5."""
      if total_count is None:
          total_count = len(items)
      
      # Validate starting index
      start = max(0, min(total_count - 1 if total_count > 0 else 0, start))
      
      # Fixed page size of 5
      page_size = 5
      end = min(start + page_size, total_count)
      
      return {
          "items": items[start:end],
          "start": start,
          "count": end - start,
          "totalCount": total_count,
          "hasMore": end < total_count,
          "nextStart": end if end < total_count else None
      }

• src/imdb_mcp_server/api.py:14-55 (helper)

  Key supporting utility that handles API requests to the IMDb backend, including caching via cache_manager, authentication with RapidAPI key, error handling, and response parsing. Called by the handler with the specific top250-tv URL.

  async def make_imdb_request(url: str, querystring: dict[str, Any], ctx: Optional[Context] = None) -> Optional[Dict[str, Any]]:
      """Make a request to the IMDb API with proper error handling and caching."""
      
      # Check if it's time to clean the cache
      cache_manager.cleanup_if_needed()
      
      # Create a cache key from the URL and querystring
      cache_key = f"{url}_{str(querystring)}"
      
      # Try to get from cache first
      cached_data = cache_manager.cache.get(cache_key)
      if cached_data:
          return cached_data
      
      # Get API key from session config or fallback to environment variable
      api_key = None
      if ctx and hasattr(ctx, 'session_config') and ctx.session_config:
          api_key = ctx.session_config.rapidApiKeyImdb
      
      if not api_key:
          api_key = os.getenv("RAPID_API_KEY_IMDB")
      
      # Not in cache, make the request
      headers = {
          "x-rapidapi-key": api_key,
          "x-rapidapi-host": "imdb236.p.rapidapi.com",
      }
      
      if not api_key:
          raise ValueError("API key not found. Please set the RAPID_API_KEY_IMDB environment variable or provide rapidApiKeyImdb in the request.")
      
      try:
          response = requests.get(url, headers=headers, params=querystring, timeout=30.0)
          response.raise_for_status()
          data = response.json()
          
          # Cache the response
          cache_manager.cache.set(cache_key, data)
              
          return data
      except Exception as e:
          raise ValueError(f"Unable to fetch data from IMDb. Please try again later. Error: {e}")

• src/imdb_mcp_server/main.py:22-23 (registration)

  Invocation of register_tools(server) which defines the nested tool functions (including get_top_250_tv_shows) decorated with @mcp.tool() to register them on the FastMCP server instance. Similar call exists at line 61 for stdio mode.

  # Register all tools with the server
  register_tools(server)