Teradata MCP Server

Instructions

Implementation Reference

• src/teradata_mcp_server/tools/dba/dba_tools.py:152-203 (handler)

  Handler function implementing the dba_databaseSpace tool. Queries DBC.DiskSpaceV to compute space metrics (allocated, used, free in GB, percent used) for a specific database or all databases.

  def handle_dba_databaseSpace(conn: TeradataConnection, database_name: str | None | None, *args, **kwargs):
      """
      Get database space if database name is provided, otherwise get all databases space allocations.
  
      Arguments:
        database_name - database name
  
      Returns:
        ResponseType: formatted response with query results + metadata
      """
      logger.debug(f"Tool: handle_dba_databaseSpace: Args: database_name: {database_name}")
  
      database_name_filter = f"AND objectdatabasename = '{database_name}'" if database_name else ""
  
      with conn.cursor() as cur:
          if not database_name:
              logger.debug("No database name provided, returning all databases and space information.")
              rows = cur.execute("""
                  SELECT
                      DatabaseName,
                      CAST(SUM(MaxPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceAllocated_GB,
                      CAST(SUM(CurrentPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceUsed_GB,
                      CAST((SUM(MaxPerm) - SUM(CurrentPerm))/1024/1024/1024 AS DECIMAL(10,2)) AS FreeSpace_GB,
                      CAST((SUM(CurrentPerm) * 100.0 / NULLIF(SUM(MaxPerm),0)) AS DECIMAL(10,2)) AS PercentUsed
                  FROM DBC.DiskSpaceV
                  WHERE MaxPerm > 0
                  GROUP BY 1
                  ORDER BY 5 DESC;
              """)
          else:
              logger.debug(f"Database name: {database_name}, returning space information for this database.")
              rows = cur.execute(f"""
                  SELECT
                      DatabaseName,
                      CAST(SUM(MaxPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceAllocated_GB,
                      CAST(SUM(CurrentPerm)/1024/1024/1024 AS DECIMAL(10,2)) AS SpaceUsed_GB,
                      CAST((SUM(MaxPerm) - SUM(CurrentPerm))/1024/1024/1024 AS DECIMAL(10,2)) AS FreeSpace_GB,
                      CAST((SUM(CurrentPerm) * 100.0 / NULLIF(SUM(MaxPerm),0)) AS DECIMAL(10,2)) AS PercentUsed
                  FROM DBC.DiskSpaceV
                  WHERE MaxPerm > 0
                  AND DatabaseName = '{database_name}'
                  GROUP BY 1;
              """)
  
          data = rows_to_json(cur.description, rows.fetchall())
          metadata = {
              "tool_name": "dba_databaseSpace",
              "database_name": database_name,
              "total_databases": len(data)
          }
          logger.debug(f"Tool: handle_dba_databaseSpace: metadata: {metadata}")
          return create_response(data, metadata)

• src/teradata_mcp_server/app.py:461-486 (registration)

  Dynamic registration code that discovers handle_* functions via module_loader, wraps them for MCP (injecting DB connection, QueryBand, etc.), and registers as tools if matching profile patterns. Registers handle_dba_databaseSpace as dba_databaseSpace.

  # Register code tools via module loader
  module_loader = td.initialize_module_loader(config)
  if module_loader:
      all_functions = module_loader.get_all_functions()
      for name, func in all_functions.items():
          if not (inspect.isfunction(func) and name.startswith("handle_")):
              continue
          tool_name = name[len("handle_"):]
          if not any(re.match(p, tool_name) for p in config.get('tool', [])):
              continue
          # Skip template tools (used for developer reference only)
          if tool_name.startswith("tmpl_"):
              logger.debug(f"Skipping template tool: {tool_name}")
              continue
          # Skip BAR tools if BAR functionality is disabled
          if tool_name.startswith("bar_") and not enableBAR:
              logger.info(f"Skipping BAR tool: {tool_name} (BAR functionality disabled)")
              continue
          # Skip chat completion tools if chat completion functionality is disabled
          if tool_name.startswith("chat_") and not enableChat:
              logger.info(f"Skipping chat completion tool: {tool_name} (chat completion functionality disabled)")
              continue
          wrapped = make_tool_wrapper(func)
          mcp.tool(name=tool_name, description=wrapped.__doc__)(wrapped)
          logger.info(f"Created tool: {tool_name}")
          logger.debug(f"Tool Docstring: {wrapped.__doc__}")

• src/teradata_mcp_server/tools/dba/__init__.py:1-2 (registration)

  Imports dba_tools.py into the dba module, making its functions available for dynamic discovery and loading by the module_loader.

  from .dba_resources import *
  from .dba_tools import *

• src/teradata_mcp_server/tools/dba/dba_tools.py:1-7 (helper)

  Imports and logger setup used by the dba_databaseSpace handler.

  import logging
  
  from teradatasql import TeradataConnection
  
  from teradata_mcp_server.tools.utils import create_response, rows_to_json
  
  logger = logging.getLogger("teradata_mcp_server")