Labsco
ognis1205 logo

Unity Catalog MCP Server

โ˜… 15

from ognis1205

An MCP server that allows LLM agents to seamlessly execute functions within Unity Catalog.

๐Ÿ”ฅ๐Ÿ”ฅ๐Ÿ”ฅโœ“ VerifiedAccount requiredNeeds API keys

mcp-server-unitycatalog: An Unity Catalog MCP server

Overview

A Model Context Protocol server for Unity Catalog. This server provides Unity Catalog Functions as MCP tools.

Tools

You can use all Unity Catalog Functions registered in Unity Catalog alongside the following predefined Unity Catalog AI tools:

  1. uc_list_functions

    • Lists functions within the specified parent catalog and schema.
    • Returns: A list of functions retrieved from Unity Catalog.
  2. uc_get_function

    • Gets a function within a parent catalog and schema.
    • Input:
      • name (string): The name of the function (not fully-qualified).
    • Returns: A function details retrieved from Unity Catalog.
  3. uc_create_function

    • Creates a function within a parent catalog and schema. WARNING: This API is experimental and will change in future versions.
    • Input:
      • name (string): The name of the function (not fully-qualified).
      • script (string): The Python script including the function to be registered.
    • Returns: A function details created within Unity Catalog.
  4. uc_delete_function

    • Deletes a function within a parent catalog and schema.
    • Input:
      • name (string): The name of the function (not fully-qualified).
    • Returns: None.

Building

Docker:

docker build -t mcp/unitycatalog .   

Future Plans

  • Implement support for list_functions.
  • Implement support for get_function.
  • Implement support for create_python_function.
  • Implement support for execute_function.
  • Implement support for delete_function.
  • Implement semantic catalog explorer tools.
  • Add Docker image.
  • Implement use_xxx methods. In the current implementation, catalog and schema need to be defined when starting the server. However, they will be implemented as use_catalog and use_schema functions, dynamically updating the list of available functions when the use_xxx is executed.