genkit-cli/dist/mcp/trace.js

CLI for interacting with the Google Genkit AI framework

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.defineTraceTools = defineTraceTools;
const utils_1 = require("@genkit-ai/tools-common/utils");
const zod_1 = __importDefault(require("zod"));
const analytics_js_1 = require("./analytics.js");
function defineTraceTools(server, manager) {
    server.registerTool('get_trace', {
        title: 'Get Genkit Trace',
        description: 'Returns the trace details',
        inputSchema: {
            traceId: zod_1.default
                .string()
                .describe('trace id (typically returned after running a flow or other actions)'),
        },
    }, async ({ traceId }) => {
        await (0, utils_1.record)(new analytics_js_1.McpRunToolEvent('get_trace'));
        try {
            const response = await manager.getTrace({ traceId });
            return {
                content: [
                    { type: 'text', text: JSON.stringify(response, undefined, 2) },
                ],
            };
        }
        catch (e) {
            return {
                content: [{ type: 'text', text: `Error: ${e}` }],
            };
        }
    });
}
//# sourceMappingURL=trace.js.map