Skip to content

datarobot_genai.drmcp.test_utils.stubs.prompt_stubs

prompt_stubs

Stub prompt templates and versions for integration tests (test_dynamic_prompts).

get_stub_prompt_templates 

get_stub_prompt_templates() -> list[SimpleNamespace]

Return stub prompt templates for register_prompts_from_datarobot_prompt_management.

Source code in datarobot_genai/drmcp/test_utils/stubs/prompt_stubs.py 
58
59
60
61
62
63
64
65
66
def get_stub_prompt_templates() -> list[SimpleNamespace]:
    """Return stub prompt templates for register_prompts_from_datarobot_prompt_management."""
    # Exclude STUB_PROMPT_WITHOUT_VERSION so it is not registered (test expects it not in list).
    return [
        _template("stub-pt-1", STUB_PROMPT_VERSION_NO_VARS),
        _template("stub-pt-2", STUB_PROMPT_VERSION_WITH_VARS),
        _template("stub-pt-dup-1", STUB_PROMPT_DUPLICATE_NAME),
        _template("stub-pt-dup-2", STUB_PROMPT_DUPLICATE_NAME),
    ]

get_stub_prompt_template_versions 

get_stub_prompt_template_versions(prompt_template_ids: list[str]) -> dict[str, list[SimpleNamespace]]

Return stub versions for each template id (for stub mode).

Source code in datarobot_genai/drmcp/test_utils/stubs/prompt_stubs.py 
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
def get_stub_prompt_template_versions(
    prompt_template_ids: list[str],
) -> dict[str, list[SimpleNamespace]]:
    """Return stub versions for each template id (for stub mode)."""
    result: dict[str, list[SimpleNamespace]] = {}
    if "stub-pt-1" in prompt_template_ids:
        result["stub-pt-1"] = [
            _version(
                "stub-v-1",
                "stub-pt-1",
                STUB_PROMPT_VERSION_NO_VARS_TEXT,
                [],
            )
        ]
    if "stub-pt-2" in prompt_template_ids:
        result["stub-pt-2"] = [
            _version(
                "stub-v-2",
                "stub-pt-2",
                STUB_PROMPT_VERSION_WITH_VARS_TEXT,
                [_var("name"), _var("sentences")],
            )
        ]
    if "stub-pt-dup-1" in prompt_template_ids:
        result["stub-pt-dup-1"] = [
            _version("stub-v-dup-1", "stub-pt-dup-1", "Duplicate prompt 1.", [])
        ]
    if "stub-pt-dup-2" in prompt_template_ids:
        result["stub-pt-dup-2"] = [
            _version("stub-v-dup-2", "stub-pt-dup-2", "Duplicate prompt 2.", [])
        ]
    return result