mcp_server.py

from mcp.server.fastmcp import FastMCP
from src.securty.prompt_injection import check_prompt_injection
mcp = FastMCP("Youtube Service")

@mcp.tool()
def say_hello(name: str) -> str:
  """
    Returns a greeting message for the given name.

    Args:
        name (str): The name to greet.
        
    Returns:
        str: A greeting message.
  """
  return f"Hello, {name}!"

mcp.add_tool(check_prompt_injection)

if __name__ == "__main__":
    mcp.run()