Skip to content

datarobot_genai.drmcp.test_utils.utils

utils

format_tool_call 

format_tool_call(tool_call: dict[str, Any]) -> str

Format a single tool call in a readable way.

Source code in datarobot_genai/drmcp/test_utils/utils.py 
31
32
33
34
35
36
37
def format_tool_call(tool_call: dict[str, Any]) -> str:
    """Format a single tool call in a readable way."""
    return (
        f"Tool: {tool_call['tool_name']}\n"
        f"Parameters: {json.dumps(tool_call['parameters'], indent=2)}\n"
        f"Reasoning: {tool_call['reasoning']}"
    )

format_response 

format_response(response: LLMResponse) -> str

Format the LLM response in a readable way.

Source code in datarobot_genai/drmcp/test_utils/utils.py 
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
def format_response(response: "LLMResponse") -> str:
    """Format the LLM response in a readable way."""
    formatted_parts = []

    # Format the main content
    formatted_parts.append("=== LLM Response ===\n")
    formatted_parts.append(response.content)

    # Format tool calls if any
    if response.tool_calls:
        formatted_parts.append("\n=== Tools Used ===")
        for i, tool_call in enumerate(response.tool_calls, 1):
            formatted_parts.append(f"\nTool Call #{i}:")
            formatted_parts.append(
                format_tool_call(
                    {
                        "tool_name": tool_call.tool_name,
                        "parameters": tool_call.parameters,
                        "reasoning": tool_call.reasoning,
                    }
                )
            )

    # Format tool results if any
    if response.tool_results:
        formatted_parts.append("\n=== Tool Results ===")
        for i, result in enumerate(response.tool_results, 1):
            formatted_parts.append(f"\nResult #{i}:")
            formatted_parts.append(result)

    return "\n".join(formatted_parts)

save_response_to_file 

save_response_to_file(response: LLMResponse, name: str | None = None) -> None

Save the response to a file in a readable format.

Parameters:

Name Type Description Default
response LLMResponse

The LLM response to save

 required
name str | None

Optional name to use in the filename. If not provided, will use a timestamp only.

 None
Source code in datarobot_genai/drmcp/test_utils/utils.py 
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
def save_response_to_file(response: "LLMResponse", name: str | None = None) -> None:
    """Save the response to a file in a readable format.

    Args:
        response: The LLM response to save
        name: Optional name to use in the filename. If not provided,
              will use a timestamp only.
    """
    # Create responses directory with timestamp
    timestamp = datetime.datetime.now().strftime("%Y%m%d")
    dir_path = "test_results/drmcp/.ete_responses/" + timestamp
    os.makedirs(dir_path, exist_ok=True)

    # Save both raw JSON and formatted text
    base_name = f"{name}" if name else "response"

    # Save formatted text
    with open(f"{dir_path}/{base_name}.txt", "w") as f:
        f.write(format_response(response))