@genkit-ai/telemetry-server
Version:
Genkit AI telemetry server
97 lines • 3.69 kB
JavaScript
;
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.stopTelemetryApi = exports.startTelemetryServer = exports.TraceQuerySchema = exports.LocalFileTraceStore = void 0;
const tools_common_1 = require("@genkit-ai/tools-common");
const utils_1 = require("@genkit-ai/tools-common/utils");
const express_1 = __importDefault(require("express"));
var localFileTraceStore_js_1 = require("./localFileTraceStore.js");
Object.defineProperty(exports, "LocalFileTraceStore", { enumerable: true, get: function () { return localFileTraceStore_js_1.LocalFileTraceStore; } });
var types_1 = require("./types");
Object.defineProperty(exports, "TraceQuerySchema", { enumerable: true, get: function () { return types_1.TraceQuerySchema; } });
let server;
async function startTelemetryServer(params) {
await params.traceStore.init();
const api = (0, express_1.default)();
api.use(express_1.default.json({ limit: params.maxRequestBodySize ?? '30mb' }));
api.get('/api/__health', async (_, response) => {
response.status(200).send('OK');
});
api.get('/api/traces/:traceId', async (request, response, next) => {
try {
const { traceId } = request.params;
response.json(await params.traceStore.load(traceId));
}
catch (e) {
next(e);
}
});
api.post('/api/traces', async (request, response, next) => {
try {
const traceData = tools_common_1.TraceDataSchema.parse(request.body);
await params.traceStore.save(traceData.traceId, traceData);
response.status(200).send('OK');
}
catch (e) {
next(e);
}
});
api.get('/api/traces', async (request, response, next) => {
try {
const { limit, continuationToken, filter } = request.query;
response.json(await params.traceStore.list({
limit: limit ? Number.parseInt(limit.toString()) : 10,
continuationToken: continuationToken
? continuationToken.toString()
: undefined,
filter: filter
? tools_common_1.TraceQueryFilterSchema.parse(JSON.parse(filter))
: undefined,
}));
}
catch (e) {
next(e);
}
});
api.use((err, req, res, next) => {
utils_1.logger.error(err.stack);
const error = err;
const { message, stack } = error;
const errorResponse = {
code: 13,
message,
details: {
stack,
traceId: err.traceId,
},
};
res.status(500).json(errorResponse);
});
server = api.listen(params.port, () => {
utils_1.logger.info(`Telemetry API running on http://localhost:${params.port}`);
});
server.on('error', (error) => {
utils_1.logger.error(error);
});
process.on('SIGTERM', async () => await stopTelemetryApi());
}
exports.startTelemetryServer = startTelemetryServer;
async function stopTelemetryApi() {
await Promise.all([
new Promise((resolve) => {
if (server) {
server.close(() => {
utils_1.logger.debug('Telemetry API has succesfully shut down.');
resolve();
});
}
else {
resolve();
}
}),
]);
}
exports.stopTelemetryApi = stopTelemetryApi;
//# sourceMappingURL=index.js.map