Skip to content

Commit b0ba327

Browse files
authored
chore: Flip experimental realtime telemetry flag (#3989)
1 parent a4b330c commit b0ba327

File tree

4 files changed

+18
-17
lines changed

4 files changed

+18
-17
lines changed

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ interface RunOptions {
2626
noui?: boolean;
2727
port?: string;
2828
open?: boolean;
29-
experimentalRealtimeTelemetry?: boolean;
29+
disableRealtimeTelemetry?: boolean;
3030
}
3131

3232
/** Command to run code in dev mode and/or the Dev UI. */
@@ -36,8 +36,8 @@ export const start = new Command('start')
3636
.option('-p, --port <port>', 'port for the Dev UI')
3737
.option('-o, --open', 'Open the browser on UI start up')
3838
.option(
39-
'--experimental-realtime-telemetry',
40-
'Enable real-time telemetry streaming (experimental)'
39+
'--disable-realtime-telemetry',
40+
'Disable real-time telemetry streaming'
4141
)
4242
.action(async (options: RunOptions) => {
4343
const projectRoot = await findProjectRoot();
@@ -55,7 +55,7 @@ export const start = new Command('start')
5555
projectRoot,
5656
start.args[0],
5757
start.args.slice(1),
58-
{ enableRealtimeTelemetry: options.experimentalRealtimeTelemetry }
58+
{ disableRealtimeTelemetry: options.disableRealtimeTelemetry }
5959
);
6060
manager = result.manager;
6161
processPromise = result.processPromise;

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ export async function startManager(
6767
}
6868

6969
export interface DevProcessManagerOptions {
70-
enableRealtimeTelemetry?: boolean;
70+
disableRealtimeTelemetry?: boolean;
7171
}
7272

7373
export async function startDevProcessManager(
@@ -77,11 +77,12 @@ export async function startDevProcessManager(
7777
options?: DevProcessManagerOptions
7878
): Promise<{ manager: RuntimeManager; processPromise: Promise<void> }> {
7979
const telemetryServerUrl = await resolveTelemetryServer(projectRoot);
80+
const disableRealtimeTelemetry = options?.disableRealtimeTelemetry ?? false;
8081
const envVars: Record<string, string> = {
8182
GENKIT_TELEMETRY_SERVER: telemetryServerUrl,
8283
GENKIT_ENV: 'dev',
8384
};
84-
if (options?.enableRealtimeTelemetry) {
85+
if (!disableRealtimeTelemetry) {
8586
envVars.GENKIT_ENABLE_REALTIME_TELEMETRY = 'true';
8687
}
8788
const processManager = new ProcessManager(command, args, envVars);
@@ -90,7 +91,7 @@ export async function startDevProcessManager(
9091
manageHealth: true,
9192
projectRoot,
9293
processManager,
93-
enableRealtimeTelemetry: options?.enableRealtimeTelemetry,
94+
disableRealtimeTelemetry,
9495
});
9596
const processPromise = processManager.start();
9697
return { manager, processPromise };

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

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -60,13 +60,13 @@ interface RuntimeManagerOptions {
6060
projectRoot: string;
6161
/** An optional process manager for the main application process. */
6262
processManager?: ProcessManager;
63-
/** Whether realtime telemetry streaming is enabled. */
64-
enableRealtimeTelemetry?: boolean;
63+
/** Whether to disable realtime telemetry streaming. Defaults to false. */
64+
disableRealtimeTelemetry?: boolean;
6565
}
6666

6767
export class RuntimeManager {
6868
readonly processManager?: ProcessManager;
69-
readonly enableRealtimeTelemetry: boolean;
69+
readonly disableRealtimeTelemetry: boolean;
7070
private filenameToRuntimeMap: Record<string, RuntimeInfo> = {};
7171
private filenameToDevUiMap: Record<string, DevToolsInfo> = {};
7272
private idToFileMap: Record<string, string> = {};
@@ -79,10 +79,10 @@ export class RuntimeManager {
7979
private manageHealth: boolean,
8080
readonly projectRoot: string,
8181
processManager?: ProcessManager,
82-
enableRealtimeTelemetry?: boolean
82+
disableRealtimeTelemetry?: boolean
8383
) {
8484
this.processManager = processManager;
85-
this.enableRealtimeTelemetry = enableRealtimeTelemetry ?? false;
85+
this.disableRealtimeTelemetry = disableRealtimeTelemetry ?? false;
8686
}
8787

8888
/**
@@ -94,7 +94,7 @@ export class RuntimeManager {
9494
options.manageHealth ?? true,
9595
options.projectRoot,
9696
options.processManager,
97-
options.enableRealtimeTelemetry
97+
options.disableRealtimeTelemetry
9898
);
9999
await manager.setupRuntimesWatcher();
100100
await manager.setupDevUiWatcher();

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -80,12 +80,12 @@ export function startServer(manager: RuntimeManager, port: number) {
8080

8181
// When realtime telemetry is disabled, flush headers immediately.
8282
// The trace ID will be available in the response body.
83-
if (!manager.enableRealtimeTelemetry) {
83+
if (manager.disableRealtimeTelemetry) {
8484
res.flushHeaders();
8585
}
8686

8787
try {
88-
const onTraceIdCallback = manager.enableRealtimeTelemetry
88+
const onTraceIdCallback = !manager.disableRealtimeTelemetry
8989
? (traceId: string) => {
9090
// Set trace ID header and flush - this fires before response body
9191
res.setHeader('X-Genkit-Trace-Id', traceId);
@@ -127,12 +127,12 @@ export function startServer(manager: RuntimeManager, port: number) {
127127

128128
// When realtime telemetry is disabled, flush headers immediately.
129129
// The trace ID will be available in the response body.
130-
if (!manager.enableRealtimeTelemetry) {
130+
if (manager.disableRealtimeTelemetry) {
131131
res.flushHeaders();
132132
}
133133

134134
try {
135-
const onTraceIdCallback = manager.enableRealtimeTelemetry
135+
const onTraceIdCallback = !manager.disableRealtimeTelemetry
136136
? (traceId: string) => {
137137
// Set trace ID header and flush - this fires before first chunk
138138
res.setHeader('X-Genkit-Trace-Id', traceId);

0 commit comments

Comments
 (0)