Feat/jaeger like spans

.changeset/free-stamps-lie.md

@@ -0,0 +1,10 @@
+---
+"@kopai/clickhouse-datasource": minor
+"@kopai/sqlite-datasource": minor
+"@kopai/core": minor
+"@kopai/api": minor
+"@kopai/sdk": minor
+"@kopai/ui": minor
+---
+
+Add new trace-related API methods

packages/api/src/index.test.ts

@@ -23,20 +23,38 @@ describe("apiRoutes", () => {
   let discoverMetricsSpy: ReturnType<
     typeof vi.fn<datasource.ReadMetricsDatasource["discoverMetrics"]>
   >;
+  let getServicesSpy: ReturnType<
+    typeof vi.fn<datasource.ReadTracesMetaDatasource["getServices"]>
+  >;
+  let getOperationsSpy: ReturnType<
+    typeof vi.fn<datasource.ReadTracesMetaDatasource["getOperations"]>
+  >;
+  let getTraceSummariesSpy: ReturnType<
+    typeof vi.fn<datasource.ReadTracesMetaDatasource["getTraceSummaries"]>
+  >;
 
   beforeEach(async () => {
     getTracesSpy = vi.fn<datasource.ReadTracesDatasource["getTraces"]>();
     getLogsSpy = vi.fn<datasource.ReadLogsDatasource["getLogs"]>();
     getMetricsSpy = vi.fn<datasource.ReadMetricsDatasource["getMetrics"]>();
     discoverMetricsSpy =
       vi.fn<datasource.ReadMetricsDatasource["discoverMetrics"]>();
+    getServicesSpy =
+      vi.fn<datasource.ReadTracesMetaDatasource["getServices"]>();
+    getOperationsSpy =
+      vi.fn<datasource.ReadTracesMetaDatasource["getOperations"]>();
+    getTraceSummariesSpy =
+      vi.fn<datasource.ReadTracesMetaDatasource["getTraceSummaries"]>();
     server = Fastify();
     await server.register(signalsRoutes, {
       readTelemetryDatasource: {
         getTraces: getTracesSpy,
         getLogs: getLogsSpy,
         getMetrics: getMetricsSpy,
         discoverMetrics: discoverMetricsSpy,
+        getServices: getServicesSpy,
+        getOperations: getOperationsSpy,
+        getTraceSummaries: getTraceSummariesSpy,
       },
     });
     await server.ready();

packages/api/src/routes/dashboards.ts

@@ -16,11 +16,8 @@ export const dashboardsRoutes: FastifyPluginAsyncZod<{
         "Get UI tree schema as markdown prompt instructions for AI agents",
       produces: ["text/markdown"],
       response: {
-        200: { type: "string", description: "Markdown prompt instructions" },
-        404: {
-          type: "string",
-          description: "Prompt instructions not configured",
-        },
+        200: z.string().describe("Markdown prompt instructions"),
+        404: z.string().describe("Prompt instructions not configured"),
       },
     },
     handler: async (_req, reply) => {

packages/api/src/routes/traces.ts

@@ -8,7 +8,8 @@ import {
 import { problemDetailsSchema } from "./error-schema-zod.js";
 
 export const tracesRoutes: FastifyPluginAsyncZod<{
-  readTracesDatasource: datasource.ReadTracesDatasource;
+  readTracesDatasource: datasource.ReadTracesDatasource &
+    datasource.ReadTracesMetaDatasource;
 }> = async function (fastify, opts) {
   fastify.route({
     method: "GET",
@@ -58,4 +59,68 @@ export const tracesRoutes: FastifyPluginAsyncZod<{
       res.send(result);
     },
   });
+
+  fastify.route({
+    method: "GET",
+    url: "/signals/services",
+    schema: {
+      description: "List distinct service names",
+      response: {
+        200: z.object({ services: z.array(z.string()) }),
+        "4xx": problemDetailsSchema,
+        "5xx": problemDetailsSchema,
+      },
+    },
+    handler: async (req, res) => {
+      const result = await opts.readTracesDatasource.getServices({
+        requestContext: req.requestContext,
+      });
+      res.send(result);
+    },
+  });
+
+  fastify.route({
+    method: "GET",
+    url: "/signals/traces/operations",
+    schema: {
+      description: "List distinct operations for a service",
+      querystring: z.object({ serviceName: z.string() }),
+      response: {
+        200: z.object({ operations: z.array(z.string()) }),
+        "4xx": problemDetailsSchema,
+        "5xx": problemDetailsSchema,
+      },
+    },
+    handler: async (req, res) => {
+      const result = await opts.readTracesDatasource.getOperations({
+        serviceName: req.query.serviceName,
+        requestContext: req.requestContext,
+      });
+      res.send(result);
+    },
+  });
+
+  fastify.route({
+    method: "POST",
+    url: "/signals/traces/summaries",
+    schema: {
+      description: "Search trace summaries",
+      body: dataFilterSchemas.traceSummariesFilterSchema,
+      response: {
+        200: z.object({
+          data: z.array(dataFilterSchemas.traceSummaryRowSchema),
+          nextCursor: z.string().nullable(),
+        }),
+        "4xx": problemDetailsSchema,
+        "5xx": problemDetailsSchema,
+      },
+    },
+    handler: async (req, res) => {
+      const result = await opts.readTracesDatasource.getTraceSummaries({
+        ...req.body,
+        requestContext: req.requestContext,
+      });
+      res.send(result);
+    },
+  });
 };

packages/api/src/signals.test.ts

@@ -23,20 +23,38 @@ describe("signalsRoutes", () => {
   let discoverMetricsSpy: ReturnType<
     typeof vi.fn<datasource.ReadMetricsDatasource["discoverMetrics"]>
   >;
+  let getServicesSpy: ReturnType<
+    typeof vi.fn<datasource.ReadTracesMetaDatasource["getServices"]>
+  >;
+  let getOperationsSpy: ReturnType<
+    typeof vi.fn<datasource.ReadTracesMetaDatasource["getOperations"]>
+  >;
+  let getTraceSummariesSpy: ReturnType<
+    typeof vi.fn<datasource.ReadTracesMetaDatasource["getTraceSummaries"]>
+  >;
 
   beforeEach(async () => {
     getTracesSpy = vi.fn<datasource.ReadTracesDatasource["getTraces"]>();
     getLogsSpy = vi.fn<datasource.ReadLogsDatasource["getLogs"]>();
     getMetricsSpy = vi.fn<datasource.ReadMetricsDatasource["getMetrics"]>();
     discoverMetricsSpy =
       vi.fn<datasource.ReadMetricsDatasource["discoverMetrics"]>();
+    getServicesSpy =
+      vi.fn<datasource.ReadTracesMetaDatasource["getServices"]>();
+    getOperationsSpy =
+      vi.fn<datasource.ReadTracesMetaDatasource["getOperations"]>();
+    getTraceSummariesSpy =
+      vi.fn<datasource.ReadTracesMetaDatasource["getTraceSummaries"]>();
     server = Fastify();
     await server.register(signalsRoutes, {
       readTelemetryDatasource: {
         getTraces: getTracesSpy,
         getLogs: getLogsSpy,
         getMetrics: getMetricsSpy,
         discoverMetrics: discoverMetricsSpy,
+        getServices: getServicesSpy,
+        getOperations: getOperationsSpy,
+        getTraceSummaries: getTraceSummariesSpy,
       },
     });
     await server.ready();