feat(genkit-tools): support trace uploads via tools router (#3735)

Author: MichaelDoyle

genkit-tools/common/src/manager/manager.ts

@@ -324,6 +324,17 @@ export class RuntimeManager {
     return response.data as TraceData;
   }
 
+  /**
+   * Adds a trace to the trace store
+   */
+  async addTrace(input: TraceData): Promise<void> {
+    await axios
+      .post(`${this.telemetryServerUrl}/api/traces/`, input)
+      .catch((err) =>
+        this.httpErrorHandler(err, 'Error writing trace to store.')
+      );
+  }
+
   /**
    * Notifies the runtime of dependencies it may need (e.g. telemetry server URL).
    */

genkit-tools/common/src/server/router.ts

@@ -23,6 +23,7 @@ import {
 } from '../eval';
 import type { RuntimeManager } from '../manager/manager';
 import { GenkitToolsError, type RuntimeInfo } from '../manager/types';
+import { TraceDataSchema } from '../types';
 import type { Action } from '../types/action';
 import * as apis from '../types/apis';
 import type { EnvironmentVariable } from '../types/env';
@@ -164,6 +165,14 @@ export const TOOLS_SERVER_ROUTER = (manager: RuntimeManager) =>
         return manager.getTrace(input);
       }),
 
+    /** Adds a trace to the trace store */
+    addTrace: loggedProcedure
+      .input(TraceDataSchema)
+      .output(z.void())
+      .mutation(async ({ input }) => {
+        return manager.addTrace(input);
+      }),
+
     /** Retrieves all eval run keys */
     listEvalRunKeys: loggedProcedure
       .input(apis.ListEvalKeysRequestSchema)