airbyte_ops_mcp.mcp.prompts

 1# Copyright (c) 2025 Airbyte, Inc., all rights reserved.
 2"""MCP prompt definitions for the Airbyte Ops MCP server.
 3
 4This module defines prompts that can be invoked by MCP clients to perform
 5common workflows.
 6
 7## MCP reference
 8
 9.. include:: ../../../docs/mcp-generated/prompts.md
10    :start-line: 2
11"""
12
13from __future__ import annotations
14
15__all__: list[str] = []
16
17from typing import Annotated
18
19from fastmcp import FastMCP
20from fastmcp_extensions import mcp_prompt, register_mcp_prompts
21from pydantic import Field
22
23from airbyte_ops_mcp.mcp._guidance import TEST_MY_TOOLS_GUIDANCE
24
25
26@mcp_prompt(
27    name="test-my-tools",
28    description="Test all available MCP tools to confirm they are working properly",
29)
30def test_my_tools_prompt(
31    scope: Annotated[
32        str | None,
33        Field(
34            description=(
35                "Optional free-form text to focus or constrain testing. "
36                "This can be a single word, a sentence, or a paragraph "
37                "describing the desired scope or constraints."
38            ),
39        ),
40    ] = None,
41) -> list[dict[str, str]]:
42    """Generate a prompt that instructs the agent to test all available tools.
43
44    Returns:
45        List containing a single message dict with the guidance text
46    """
47    content = TEST_MY_TOOLS_GUIDANCE
48
49    if scope:
50        content = f"{content}\n\n---\n\nAdditional scope or constraints:\n{scope}"
51
52    return [
53        {
54            "role": "user",
55            "content": content,
56        }
57    ]
58
59
60def register_prompts(app: FastMCP) -> None:
61    """Register all prompts with the FastMCP app.
62
63    Args:
64        app: FastMCP application instance
65    """
66    register_mcp_prompts(app, mcp_module=__name__)