Add realtime telemetry flag

Author: huangjeff5

genkit-tools/cli/src/commands/start.ts

@@ -26,6 +26,7 @@ interface RunOptions {
   noui?: boolean;
   port?: string;
   open?: boolean;
+  experimentalRealtimeTelemetry?: boolean;
 }
 
 /** Command to run code in dev mode and/or the Dev UI. */
@@ -34,6 +35,10 @@ export const start = new Command('start')
   .option('-n, --noui', 'do not start the Dev UI', false)
   .option('-p, --port <port>', 'port for the Dev UI')
   .option('-o, --open', 'Open the browser on UI start up')
+  .option(
+    '--experimental-realtime-telemetry',
+    'Enable real-time telemetry streaming (experimental)'
+  )
   .action(async (options: RunOptions) => {
     const projectRoot = await findProjectRoot();
     if (projectRoot.includes('/.Trash/')) {
@@ -49,7 +54,8 @@ export const start = new Command('start')
       const result = await startDevProcessManager(
         projectRoot,
         start.args[0],
-        start.args.slice(1)
+        start.args.slice(1),
+        { enableRealtimeTelemetry: options.experimentalRealtimeTelemetry }
       );
       manager = result.manager;
       processPromise = result.processPromise;

genkit-tools/cli/src/utils/manager-utils.ts

@@ -66,16 +66,25 @@ export async function startManager(
   return manager;
 }
 
+export interface DevProcessManagerOptions {
+  enableRealtimeTelemetry?: boolean;
+}
+
 export async function startDevProcessManager(
   projectRoot: string,
   command: string,
-  args: string[]
+  args: string[],
+  options?: DevProcessManagerOptions
 ): Promise<{ manager: RuntimeManager; processPromise: Promise<void> }> {
   const telemetryServerUrl = await resolveTelemetryServer(projectRoot);
-  const processManager = new ProcessManager(command, args, {
+  const envVars: Record<string, string> = {
     GENKIT_TELEMETRY_SERVER: telemetryServerUrl,
     GENKIT_ENV: 'dev',
-  });
+  };
+  if (options?.enableRealtimeTelemetry) {
+    envVars.GENKIT_ENABLE_REALTIME_TELEMETRY = 'true';
+  }
+  const processManager = new ProcessManager(command, args, envVars);
   const manager = await RuntimeManager.create({
     telemetryServerUrl,
     manageHealth: true,