mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-15 00:21:09 -07:00
29 lines
932 B
TypeScript
29 lines
932 B
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { AsyncLocalStorage } from 'node:async_hooks';
|
|
import { debugLogger } from './debugLogger.js';
|
|
|
|
export const promptIdContext = new AsyncLocalStorage<string>();
|
|
|
|
/**
|
|
* Retrieves the prompt ID from the context, or generates a fallback if not found.
|
|
* @param componentName The name of the component requesting the ID (used for the fallback prefix).
|
|
* @returns The retrieved or generated prompt ID.
|
|
*/
|
|
export function getPromptIdWithFallback(componentName: string): string {
|
|
const promptId = promptIdContext.getStore();
|
|
if (promptId) {
|
|
return promptId;
|
|
}
|
|
|
|
const fallbackId = `${componentName}-fallback-${Date.now()}-${Math.random().toString(16).slice(2)}`;
|
|
debugLogger.warn(
|
|
`Could not find promptId in context for ${componentName}. This is unexpected. Using a fallback ID: ${fallbackId}`,
|
|
);
|
|
return fallbackId;
|
|
}
|