mcp-server-tester-sse-http-stdio/dist/commands/evals/providers/llm-provider.js

MCP Server Tester with SSE support - Test MCP servers using HTTP, SSE, and STDIO transports

/**
 * LLM Provider interface for evaluation tests
 * Using Vercel AI SDK unified types
 */
export class LlmProvider {
    /**
     * Extract tool calls from AI SDK messages
     */
    extractToolCalls(messages) {
        const toolCalls = [];
        for (const message of messages) {
            if (message.role === 'assistant') {
                if (typeof message.content === 'string') {
                    // Skip text-only messages
                    continue;
                }
                // Handle array content with tool calls
                if (Array.isArray(message.content)) {
                    for (const part of message.content) {
                        if (part.type === 'tool-call') {
                            toolCalls.push({
                                toolCallId: part.toolCallId,
                                toolName: part.toolName,
                                args: part.args || {},
                            });
                        }
                    }
                }
            }
        }
        return toolCalls;
    }
    /**
     * Extract tool results from AI SDK messages
     */
    extractToolResults(messages) {
        const toolResults = [];
        for (const message of messages) {
            if (message.role === 'tool' && Array.isArray(message.content)) {
                for (const part of message.content) {
                    if (part.type === 'tool-result') {
                        toolResults.push({
                            toolCallId: part.toolCallId,
                            toolName: part.toolName,
                            args: {}, // Tool args not available in result
                            result: part.result,
                        });
                    }
                }
            }
        }
        return toolResults;
    }
}