@didim365/agent-cli-core 0.1.0

package/dist/src/core/loggingContentGenerator.js

@@ -0,0 +1,254 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ApiRequestEvent, ApiResponseEvent, ApiErrorEvent, } from '../telemetry/types.js';
+import { logApiError, logApiRequest, logApiResponse, } from '../telemetry/loggers.js';
+import { debugLogger } from '../utils/debugLogger.js';
+import { CodeAssistServer } from '../code_assist/server.js';
+import { toContents } from '../code_assist/converter.js';
+import { isStructuredError } from '../utils/quotaErrorDetection.js';
+import { runInDevTraceSpan } from '../telemetry/trace.js';
+/**
+ * A decorator that wraps a ContentGenerator to add logging to API calls.
+ */
+export class LoggingContentGenerator {
+    wrapped;
+    config;
+    constructor(wrapped, config) {
+        this.wrapped = wrapped;
+        this.config = config;
+    }
+    getWrapped() {
+        return this.wrapped;
+    }
+    get userTier() {
+        return this.wrapped.userTier;
+    }
+    get userTierName() {
+        return this.wrapped.userTierName;
+    }
+    get providerName() {
+        return this.wrapped.providerName;
+    }
+    logApiRequest(contents, model, promptId, generationConfig, serverDetails) {
+        const requestText = JSON.stringify(contents);
+        logApiRequest(this.config, new ApiRequestEvent(model, {
+            prompt_id: promptId,
+            contents,
+            generate_content_config: generationConfig,
+            server: serverDetails,
+        }, requestText));
+    }
+    _getEndpointUrl(req, method) {
+        // Case 1: Authenticated with a Google account (`gcloud auth login`).
+        // Requests are routed through the internal CodeAssistServer.
+        if (this.wrapped instanceof CodeAssistServer) {
+            const url = new URL(this.wrapped.getMethodUrl(method));
+            const port = url.port
+                ? parseInt(url.port, 10)
+                : url.protocol === 'https:'
+                    ? 443
+                    : 80;
+            return { address: url.hostname, port };
+        }
+        const genConfig = this.config.getContentGeneratorConfig();
+        // Case 2: Using an API key for Vertex AI.
+        if (genConfig?.vertexai) {
+            const location = process.env['GOOGLE_CLOUD_LOCATION'];
+            if (location) {
+                return { address: `${location}-aiplatform.googleapis.com`, port: 443 };
+            }
+            else {
+                return { address: 'unknown', port: 0 };
+            }
+        }
+        // Case 3: Default to the public Gemini API endpoint.
+        // This is used when an API key is provided but not for Vertex AI.
+        return { address: `generativelanguage.googleapis.com`, port: 443 };
+    }
+    _logApiResponse(requestContents, durationMs, model, prompt_id, responseId, responseCandidates, usageMetadata, responseText, generationConfig, serverDetails) {
+        logApiResponse(this.config, new ApiResponseEvent(model, durationMs, {
+            prompt_id,
+            contents: requestContents,
+            generate_content_config: generationConfig,
+            server: serverDetails,
+        }, {
+            candidates: responseCandidates,
+            response_id: responseId,
+        }, this.config.getContentGeneratorConfig()?.authType, usageMetadata, responseText));
+    }
+    _logApiError(durationMs, error, model, prompt_id, requestContents, generationConfig, serverDetails) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        const errorType = error instanceof Error ? error.name : 'unknown';
+        logApiError(this.config, new ApiErrorEvent(model, errorMessage, durationMs, {
+            prompt_id,
+            contents: requestContents,
+            generate_content_config: generationConfig,
+            server: serverDetails,
+        }, this.config.getContentGeneratorConfig()?.authType, errorType, isStructuredError(error)
+            ? error.status
+            : undefined));
+    }
+    async generateContent(req, userPromptId) {
+        return runInDevTraceSpan({
+            name: 'generateContent',
+        }, async ({ metadata: spanMetadata }) => {
+            spanMetadata.input = { request: req, userPromptId, model: req.model };
+            const startTime = Date.now();
+            const contents = toContents(req.contents);
+            const serverDetails = this._getEndpointUrl(req, 'generateContent');
+            this.logApiRequest(contents, req.model, userPromptId, req.config, serverDetails);
+            try {
+                const response = await this.wrapped.generateContent(req, userPromptId);
+                spanMetadata.output = {
+                    response,
+                    usageMetadata: response.usageMetadata,
+                };
+                const durationMs = Date.now() - startTime;
+                this._logApiResponse(contents, durationMs, response.modelVersion || req.model, userPromptId, response.responseId, response.candidates, response.usageMetadata, JSON.stringify({
+                    candidates: response.candidates,
+                    usageMetadata: response.usageMetadata,
+                    responseId: response.responseId,
+                    modelVersion: response.modelVersion,
+                    promptFeedback: response.promptFeedback,
+                }), req.config, serverDetails);
+                return response;
+            }
+            catch (error) {
+                const durationMs = Date.now() - startTime;
+                this._logApiError(durationMs, error, req.model, userPromptId, contents, req.config, serverDetails);
+                throw error;
+            }
+        });
+    }
+    async generateContentStream(req, userPromptId) {
+        return runInDevTraceSpan({
+            name: 'generateContentStream',
+            noAutoEnd: true,
+        }, async ({ metadata: spanMetadata, endSpan }) => {
+            spanMetadata.input = { request: req, userPromptId, model: req.model };
+            const startTime = Date.now();
+            const serverDetails = this._getEndpointUrl(req, 'generateContentStream');
+            // For debugging: Capture the latest main agent request payload.
+            // Main agent prompt IDs end with exactly 8 hashes and a turn counter (e.g. "...########1")
+            if (/########\d+$/.test(userPromptId)) {
+                this.config.setLatestApiRequest(req);
+            }
+            this.logApiRequest(toContents(req.contents), req.model, userPromptId, req.config, serverDetails);
+            let stream;
+            try {
+                stream = await this.wrapped.generateContentStream(req, userPromptId);
+            }
+            catch (error) {
+                const durationMs = Date.now() - startTime;
+                this._logApiError(durationMs, error, req.model, userPromptId, toContents(req.contents), req.config, serverDetails);
+                throw error;
+            }
+            return this.loggingStreamWrapper(req, stream, startTime, userPromptId, spanMetadata, endSpan);
+        });
+    }
+    async *loggingStreamWrapper(req, stream, startTime, userPromptId, spanMetadata, endSpan) {
+        const responses = [];
+        let lastUsageMetadata;
+        const serverDetails = this._getEndpointUrl(req, 'generateContentStream');
+        const requestContents = toContents(req.contents);
+        try {
+            for await (const response of stream) {
+                responses.push(response);
+                if (response.usageMetadata) {
+                    lastUsageMetadata = response.usageMetadata;
+                }
+                yield response;
+            }
+            // Only log successful API response if no error occurred
+            const durationMs = Date.now() - startTime;
+            this._logApiResponse(requestContents, durationMs, responses[0]?.modelVersion || req.model, userPromptId, responses[0]?.responseId, responses.flatMap((response) => response.candidates || []), lastUsageMetadata, JSON.stringify(responses.map((r) => ({
+                candidates: r.candidates,
+                usageMetadata: r.usageMetadata,
+                responseId: r.responseId,
+                modelVersion: r.modelVersion,
+                promptFeedback: r.promptFeedback,
+            }))), req.config, serverDetails);
+            spanMetadata.output = {
+                streamChunks: responses.map((r) => ({
+                    content: r.candidates?.[0]?.content ?? null,
+                })),
+                usageMetadata: lastUsageMetadata,
+                durationMs,
+            };
+        }
+        catch (error) {
+            spanMetadata.error = error;
+            const durationMs = Date.now() - startTime;
+            this._logApiError(durationMs, error, responses[0]?.modelVersion || req.model, userPromptId, requestContents, req.config, serverDetails);
+            throw error;
+        }
+        finally {
+            endSpan();
+        }
+    }
+    async countTokens(req) {
+        return this.wrapped.countTokens(req);
+    }
+    async embedContent(req) {
+        return runInDevTraceSpan({
+            name: 'embedContent',
+        }, async ({ metadata: spanMetadata }) => {
+            spanMetadata.input = { request: req };
+            const output = await this.wrapped.embedContent(req);
+            spanMetadata.output = output;
+            return output;
+        });
+    }
+    // Provider-independent methods
+    async llmGenerateContent(request, userPromptId, options) {
+        if (!this.wrapped.llmGenerateContent) {
+            throw new Error('Wrapped generator does not support provider-independent API');
+        }
+        const startTime = Date.now();
+        debugLogger.debug(`[LLM] generateContent model=${request.model} promptId=${userPromptId}`);
+        try {
+            const response = await this.wrapped.llmGenerateContent(request, userPromptId, options);
+            const durationMs = Date.now() - startTime;
+            debugLogger.debug(`[LLM] generateContent completed in ${durationMs}ms`);
+            return response;
+        }
+        catch (error) {
+            const durationMs = Date.now() - startTime;
+            debugLogger.debug(`[LLM] generateContent error after ${durationMs}ms: ${error}`);
+            throw error;
+        }
+    }
+    llmGenerateContentStream(request, userPromptId, options) {
+        if (!this.wrapped.llmGenerateContentStream) {
+            throw new Error('Wrapped generator does not support provider-independent API');
+        }
+        debugLogger.debug(`[LLM] generateContentStream model=${request.model} promptId=${userPromptId}`);
+        const stream = this.wrapped.llmGenerateContentStream(request, userPromptId, options);
+        return this.llmLoggingStreamWrapper(stream, request.model, userPromptId);
+    }
+    async *llmLoggingStreamWrapper(stream, model, userPromptId) {
+        const startTime = Date.now();
+        try {
+            for await (const event of stream) {
+                yield event;
+            }
+            const durationMs = Date.now() - startTime;
+            debugLogger.debug(`[LLM] generateContentStream completed in ${durationMs}ms`);
+        }
+        catch (error) {
+            const durationMs = Date.now() - startTime;
+            debugLogger.debug(`[LLM] generateContentStream error after ${durationMs}ms model=${model} promptId=${userPromptId}: ${error}`);
+            throw error;
+        }
+    }
+    async llmCountTokens(request) {
+        if (!this.wrapped.llmCountTokens) {
+            throw new Error('Wrapped generator does not support provider-independent API');
+        }
+        return this.wrapped.llmCountTokens(request);
+    }
+}
+//# sourceMappingURL=loggingContentGenerator.js.map

package/dist/src/core/loggingContentGenerator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loggingContentGenerator.js","sourceRoot":"","sources":["../../../src/core/loggingContentGenerator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgBH,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,aAAa,GACd,MAAM,uBAAuB,CAAC;AAO/B,OAAO,EACL,WAAW,EACX,aAAa,EACb,cAAc,GACf,MAAM,yBAAyB,CAAC;AAQjC,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AACpE,OAAO,EAAE,iBAAiB,EAAqB,MAAM,uBAAuB,CAAC;AAM7E;;GAEG;AACH,MAAM,OAAO,uBAAuB;IAEf;IACA;IAFnB,YACmB,OAAyB,EACzB,MAAc;QADd,YAAO,GAAP,OAAO,CAAkB;QACzB,WAAM,GAAN,MAAM,CAAQ;IAC9B,CAAC;IAEJ,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;IAC/B,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;IACnC,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;IACnC,CAAC;IAEO,aAAa,CACnB,QAA4B,EAC5B,KAAa,EACb,QAAgB,EAChB,gBAA0C,EAC1C,aAA6B;QAE7B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC7C,aAAa,CACX,IAAI,CAAC,MAAM,EACX,IAAI,eAAe,CACjB,KAAK,EACL;YACE,SAAS,EAAE,QAAQ;YACnB,QAAQ;YACR,uBAAuB,EAAE,gBAAgB;YACzC,MAAM,EAAE,aAAa;SACtB,EACD,WAAW,CACZ,CACF,CAAC;IACJ,CAAC;IAEO,eAAe,CACrB,GAA8B,EAC9B,MAAmD;QAEnD,qEAAqE;QACrE,6DAA6D;QAC7D,IAAI,IAAI,CAAC,OAAO,YAAY,gBAAgB,EAAE,CAAC;YAC7C,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;YACvD,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI;gBACnB,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;gBACxB,CAAC,CAAC,GAAG,CAAC,QAAQ,KAAK,QAAQ;oBACzB,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,EAAE,CAAC;YACT,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;QACzC,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,yBAAyB,EAAE,CAAC;QAE1D,0CAA0C;QAC1C,IAAI,SAAS,EAAE,QAAQ,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YACtD,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,EAAE,OAAO,EAAE,GAAG,QAAQ,4BAA4B,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;YACzE,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YACzC,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,kEAAkE;QAClE,OAAO,EAAE,OAAO,EAAE,mCAAmC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;IACrE,CAAC;IAEO,eAAe,CACrB,eAAmC,EACnC,UAAkB,EAClB,KAAa,EACb,SAAiB,EACjB,UAA8B,EAC9B,kBAAyC,EACzC,aAAsC,EACtC,YAAqB,EACrB,gBAA0C,EAC1C,aAA6B;QAE7B,cAAc,CACZ,IAAI,CAAC,MAAM,EACX,IAAI,gBAAgB,CAClB,KAAK,EACL,UAAU,EACV;YACE,SAAS;YACT,QAAQ,EAAE,eAAe;YACzB,uBAAuB,EAAE,gBAAgB;YACzC,MAAM,EAAE,aAAa;SACtB,EACD;YACE,UAAU,EAAE,kBAAkB;YAC9B,WAAW,EAAE,UAAU;SACxB,EACD,IAAI,CAAC,MAAM,CAAC,yBAAyB,EAAE,EAAE,QAAQ,EACjD,aAAa,EACb,YAAY,CACb,CACF,CAAC;IACJ,CAAC;IAEO,YAAY,CAClB,UAAkB,EAClB,KAAc,EACd,KAAa,EACb,SAAiB,EACjB,eAAmC,EACnC,gBAA0C,EAC1C,aAA6B;QAE7B,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,SAAS,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QAElE,WAAW,CACT,IAAI,CAAC,MAAM,EACX,IAAI,aAAa,CACf,KAAK,EACL,YAAY,EACZ,UAAU,EACV;YACE,SAAS;YACT,QAAQ,EAAE,eAAe;YACzB,uBAAuB,EAAE,gBAAgB;YACzC,MAAM,EAAE,aAAa;SACtB,EACD,IAAI,CAAC,MAAM,CAAC,yBAAyB,EAAE,EAAE,QAAQ,EACjD,SAAS,EACT,iBAAiB,CAAC,KAAK,CAAC;YACtB,CAAC,CAAE,KAAyB,CAAC,MAAM;YACnC,CAAC,CAAC,SAAS,CACd,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,GAA8B,EAC9B,YAAoB;QAEpB,OAAO,iBAAiB,CACtB;YACE,IAAI,EAAE,iBAAiB;SACxB,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;YACnC,YAAY,CAAC,KAAK,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC;YAEtE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAuB,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC9D,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;YACnE,IAAI,CAAC,aAAa,CAChB,QAAQ,EACR,GAAG,CAAC,KAAK,EACT,YAAY,EACZ,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CACjD,GAAG,EACH,YAAY,CACb,CAAC;gBACF,YAAY,CAAC,MAAM,GAAG;oBACpB,QAAQ;oBACR,aAAa,EAAE,QAAQ,CAAC,aAAa;iBACtC,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAC1C,IAAI,CAAC,eAAe,CAClB,QAAQ,EACR,UAAU,EACV,QAAQ,CAAC,YAAY,IAAI,GAAG,CAAC,KAAK,EAClC,YAAY,EACZ,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,UAAU,EACnB,QAAQ,CAAC,aAAa,EACtB,IAAI,CAAC,SAAS,CAAC;oBACb,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,aAAa,EAAE,QAAQ,CAAC,aAAa;oBACrC,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,YAAY,EAAE,QAAQ,CAAC,YAAY;oBACnC,cAAc,EAAE,QAAQ,CAAC,cAAc;iBACxC,CAAC,EACF,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;gBACF,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAC1C,IAAI,CAAC,YAAY,CACf,UAAU,EACV,KAAK,EACL,GAAG,CAAC,KAAK,EACT,YAAY,EACZ,QAAQ,EACR,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,GAA8B,EAC9B,YAAoB;QAEpB,OAAO,iBAAiB,CACtB;YACE,IAAI,EAAE,uBAAuB;YAC7B,SAAS,EAAE,IAAI;SAChB,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,EAAE,EAAE;YAC5C,YAAY,CAAC,KAAK,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC;YACtE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CACxC,GAAG,EACH,uBAAuB,CACxB,CAAC;YAEF,gEAAgE;YAChE,2FAA2F;YAC3F,IAAI,cAAc,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;YACvC,CAAC;YAED,IAAI,CAAC,aAAa,CAChB,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EACxB,GAAG,CAAC,KAAK,EACT,YAAY,EACZ,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;YAEF,IAAI,MAA+C,CAAC;YACpD,IAAI,CAAC;gBACH,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YACvE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAC1C,IAAI,CAAC,YAAY,CACf,UAAU,EACV,KAAK,EACL,GAAG,CAAC,KAAK,EACT,YAAY,EACZ,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EACxB,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;YAED,OAAO,IAAI,CAAC,oBAAoB,CAC9B,GAAG,EACH,MAAM,EACN,SAAS,EACT,YAAY,EACZ,YAAY,EACZ,OAAO,CACR,CAAC;QACJ,CAAC,CACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB,CACjC,GAA8B,EAC9B,MAA+C,EAC/C,SAAiB,EACjB,YAAoB,EACpB,YAA0B,EAC1B,OAAmB;QAEnB,MAAM,SAAS,GAA8B,EAAE,CAAC;QAEhD,IAAI,iBAAqD,CAAC;QAC1D,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,uBAAuB,CAAC,CAAC;QACzE,MAAM,eAAe,GAAuB,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACrE,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;gBACpC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACzB,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;oBAC3B,iBAAiB,GAAG,QAAQ,CAAC,aAAa,CAAC;gBAC7C,CAAC;gBACD,MAAM,QAAQ,CAAC;YACjB,CAAC;YACD,wDAAwD;YACxD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,IAAI,CAAC,eAAe,CAClB,eAAe,EACf,UAAU,EACV,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,IAAI,GAAG,CAAC,KAAK,EACvC,YAAY,EACZ,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EACxB,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,EAC1D,iBAAiB,EACjB,IAAI,CAAC,SAAS,CACZ,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpB,UAAU,EAAE,CAAC,CAAC,UAAU;gBACxB,aAAa,EAAE,CAAC,CAAC,aAAa;gBAC9B,UAAU,EAAE,CAAC,CAAC,UAAU;gBACxB,YAAY,EAAE,CAAC,CAAC,YAAY;gBAC5B,cAAc,EAAE,CAAC,CAAC,cAAc;aACjC,CAAC,CAAC,CACJ,EACD,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;YACF,YAAY,CAAC,MAAM,GAAG;gBACpB,YAAY,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAClC,OAAO,EAAE,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,IAAI;iBAC5C,CAAC,CAAC;gBACH,aAAa,EAAE,iBAAiB;gBAChC,UAAU;aACX,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,KAAK,GAAG,KAAK,CAAC;YAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,IAAI,CAAC,YAAY,CACf,UAAU,EACV,KAAK,EACL,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,IAAI,GAAG,CAAC,KAAK,EACvC,YAAY,EACZ,eAAe,EACf,GAAG,CAAC,MAAM,EACV,aAAa,CACd,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAA0B;QAC1C,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,GAA2B;QAE3B,OAAO,iBAAiB,CACtB;YACE,IAAI,EAAE,cAAc;SACrB,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE;YACnC,YAAY,CAAC,KAAK,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACpD,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC;YAC7B,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;IACJ,CAAC;IAED,+BAA+B;IAE/B,KAAK,CAAC,kBAAkB,CACtB,OAA2B,EAC3B,YAAoB,EACpB,OAAyB;QAEzB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,kBAAkB,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QACJ,CAAC;QACD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,WAAW,CAAC,KAAK,CACf,+BAA+B,OAAO,CAAC,KAAK,aAAa,YAAY,EAAE,CACxE,CAAC;QACF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,kBAAkB,CACpD,OAAO,EACP,YAAY,EACZ,OAAO,CACR,CAAC;YACF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,WAAW,CAAC,KAAK,CAAC,sCAAsC,UAAU,IAAI,CAAC,CAAC;YACxE,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,WAAW,CAAC,KAAK,CACf,qCAAqC,UAAU,OAAO,KAAK,EAAE,CAC9D,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,wBAAwB,CACtB,OAA2B,EAC3B,YAAoB,EACpB,OAAyB;QAEzB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,wBAAwB,EAAE,CAAC;YAC3C,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QACJ,CAAC;QACD,WAAW,CAAC,KAAK,CACf,qCAAqC,OAAO,CAAC,KAAK,aAAa,YAAY,EAAE,CAC9E,CAAC;QACF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,wBAAwB,CAClD,OAAO,EACP,YAAY,EACZ,OAAO,CACR,CAAC;QACF,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;IAC3E,CAAC;IAEO,KAAK,CAAC,CAAC,uBAAuB,CACpC,MAAsB,EACtB,KAAa,EACb,YAAoB;QAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,WAAW,CAAC,KAAK,CACf,4CAA4C,UAAU,IAAI,CAC3D,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC1C,WAAW,CAAC,KAAK,CACf,2CAA2C,UAAU,YAAY,KAAK,aAAa,YAAY,KAAK,KAAK,EAAE,CAC5G,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAA2B;QAC9C,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CACb,6DAA6D,CAC9D,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAC9C,CAAC;CACF"}

package/dist/src/core/loggingContentGenerator.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/core/loggingContentGenerator.test.js

@@ -0,0 +1,221 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const logApiRequest = vi.hoisted(() => vi.fn());
+const logApiResponse = vi.hoisted(() => vi.fn());
+const logApiError = vi.hoisted(() => vi.fn());
+vi.mock('../telemetry/loggers.js', () => ({
+    logApiRequest,
+    logApiResponse,
+    logApiError,
+}));
+const runInDevTraceSpan = vi.hoisted(() => vi.fn(async (meta, fn) => fn({ metadata: {}, endSpan: vi.fn() })));
+vi.mock('../telemetry/trace.js', () => ({
+    runInDevTraceSpan,
+}));
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { LoggingContentGenerator } from './loggingContentGenerator.js';
+import { ApiRequestEvent } from '../telemetry/types.js';
+import { UserTierId } from '../code_assist/types.js';
+describe('LoggingContentGenerator', () => {
+    let wrapped;
+    let config;
+    let loggingContentGenerator;
+    beforeEach(() => {
+        wrapped = {
+            generateContent: vi.fn(),
+            generateContentStream: vi.fn(),
+            countTokens: vi.fn(),
+            embedContent: vi.fn(),
+        };
+        config = {
+            getGoogleAIConfig: vi.fn(),
+            getVertexAIConfig: vi.fn(),
+            getContentGeneratorConfig: vi.fn().mockReturnValue({
+                authType: 'API_KEY',
+            }),
+        };
+        loggingContentGenerator = new LoggingContentGenerator(wrapped, config);
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+        vi.useRealTimers();
+    });
+    describe('generateContent', () => {
+        it('should log request and response on success', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [{ text: 'hello' }] }],
+                model: 'gemini-pro',
+            };
+            const userPromptId = 'prompt-123';
+            const response = {
+                candidates: [],
+                usageMetadata: {
+                    promptTokenCount: 1,
+                    candidatesTokenCount: 2,
+                    totalTokenCount: 3,
+                },
+                text: undefined,
+                functionCalls: undefined,
+                executableCode: undefined,
+                codeExecutionResult: undefined,
+                data: undefined,
+            };
+            vi.mocked(wrapped.generateContent).mockResolvedValue(response);
+            const startTime = new Date('2025-01-01T00:00:00.000Z');
+            vi.setSystemTime(startTime);
+            const promise = loggingContentGenerator.generateContent(req, userPromptId);
+            vi.advanceTimersByTime(1000);
+            await promise;
+            expect(wrapped.generateContent).toHaveBeenCalledWith(req, userPromptId);
+            expect(logApiRequest).toHaveBeenCalledWith(config, expect.any(ApiRequestEvent));
+            const responseEvent = vi.mocked(logApiResponse).mock.calls[0][1];
+            expect(responseEvent.duration_ms).toBe(1000);
+        });
+        it('should log error on failure', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [{ text: 'hello' }] }],
+                model: 'gemini-pro',
+            };
+            const userPromptId = 'prompt-123';
+            const error = new Error('test error');
+            vi.mocked(wrapped.generateContent).mockRejectedValue(error);
+            const startTime = new Date('2025-01-01T00:00:00.000Z');
+            vi.setSystemTime(startTime);
+            const promise = loggingContentGenerator.generateContent(req, userPromptId);
+            vi.advanceTimersByTime(1000);
+            await expect(promise).rejects.toThrow(error);
+            expect(logApiRequest).toHaveBeenCalledWith(config, expect.any(ApiRequestEvent));
+            const errorEvent = vi.mocked(logApiError).mock.calls[0][1];
+            expect(errorEvent.duration_ms).toBe(1000);
+        });
+    });
+    describe('generateContentStream', () => {
+        it('should log request and response on success', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [{ text: 'hello' }] }],
+                model: 'gemini-pro',
+            };
+            const userPromptId = 'prompt-123';
+            const response = {
+                candidates: [],
+                usageMetadata: {
+                    promptTokenCount: 1,
+                    candidatesTokenCount: 2,
+                    totalTokenCount: 3,
+                },
+            };
+            async function* createAsyncGenerator() {
+                yield response;
+            }
+            vi.mocked(wrapped.generateContentStream).mockResolvedValue(createAsyncGenerator());
+            const startTime = new Date('2025-01-01T00:00:00.000Z');
+            vi.setSystemTime(startTime);
+            const stream = await loggingContentGenerator.generateContentStream(req, userPromptId);
+            vi.advanceTimersByTime(1000);
+            for await (const _ of stream) {
+                // consume stream
+            }
+            expect(wrapped.generateContentStream).toHaveBeenCalledWith(req, userPromptId);
+            expect(logApiRequest).toHaveBeenCalledWith(config, expect.any(ApiRequestEvent));
+            const responseEvent = vi.mocked(logApiResponse).mock.calls[0][1];
+            expect(responseEvent.duration_ms).toBe(1000);
+        });
+        it('should log error on failure', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [{ text: 'hello' }] }],
+                model: 'gemini-pro',
+            };
+            const userPromptId = 'prompt-123';
+            const error = new Error('test error');
+            async function* createAsyncGenerator() {
+                yield Promise.reject(error);
+            }
+            vi.mocked(wrapped.generateContentStream).mockResolvedValue(createAsyncGenerator());
+            const startTime = new Date('2025-01-01T00:00:00.000Z');
+            vi.setSystemTime(startTime);
+            const stream = await loggingContentGenerator.generateContentStream(req, userPromptId);
+            vi.advanceTimersByTime(1000);
+            await expect(async () => {
+                for await (const _ of stream) {
+                    // do nothing
+                }
+            }).rejects.toThrow(error);
+            expect(logApiRequest).toHaveBeenCalledWith(config, expect.any(ApiRequestEvent));
+            const errorEvent = vi.mocked(logApiError).mock.calls[0][1];
+            expect(errorEvent.duration_ms).toBe(1000);
+        });
+        it('should set latest API request in config for main agent requests', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [{ text: 'hello' }] }],
+                model: 'gemini-pro',
+            };
+            // Main agent prompt IDs end with exactly 8 hashes and a turn counter
+            const mainAgentPromptId = 'session-uuid########1';
+            config.setLatestApiRequest = vi.fn();
+            async function* createAsyncGenerator() {
+                yield { candidates: [] };
+            }
+            vi.mocked(wrapped.generateContentStream).mockResolvedValue(createAsyncGenerator());
+            await loggingContentGenerator.generateContentStream(req, mainAgentPromptId);
+            expect(config.setLatestApiRequest).toHaveBeenCalledWith(req);
+        });
+        it('should NOT set latest API request in config for sub-agent requests', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [{ text: 'hello' }] }],
+                model: 'gemini-pro',
+            };
+            // Sub-agent prompt IDs contain fewer hashes, typically separating the agent name and ID
+            const subAgentPromptId = 'codebase_investigator#12345';
+            config.setLatestApiRequest = vi.fn();
+            async function* createAsyncGenerator() {
+                yield { candidates: [] };
+            }
+            vi.mocked(wrapped.generateContentStream).mockResolvedValue(createAsyncGenerator());
+            await loggingContentGenerator.generateContentStream(req, subAgentPromptId);
+            expect(config.setLatestApiRequest).not.toHaveBeenCalled();
+        });
+    });
+    describe('getWrapped', () => {
+        it('should return the wrapped content generator', () => {
+            expect(loggingContentGenerator.getWrapped()).toBe(wrapped);
+        });
+    });
+    describe('countTokens', () => {
+        it('should call the wrapped countTokens method', async () => {
+            const req = { contents: [], model: 'gemini-pro' };
+            const response = { totalTokens: 10 };
+            vi.mocked(wrapped.countTokens).mockResolvedValue(response);
+            const result = await loggingContentGenerator.countTokens(req);
+            expect(wrapped.countTokens).toHaveBeenCalledWith(req);
+            expect(result).toBe(response);
+        });
+    });
+    describe('embedContent', () => {
+        it('should call the wrapped embedContent method', async () => {
+            const req = {
+                contents: [{ role: 'user', parts: [] }],
+                model: 'gemini-pro',
+            };
+            const response = { embeddings: [{ values: [] }] };
+            vi.mocked(wrapped.embedContent).mockResolvedValue(response);
+            const result = await loggingContentGenerator.embedContent(req);
+            expect(wrapped.embedContent).toHaveBeenCalledWith(req);
+            expect(result).toBe(response);
+        });
+    });
+    describe('delegation', () => {
+        it('should delegate userTier to wrapped', () => {
+            wrapped.userTier = UserTierId.STANDARD;
+            expect(loggingContentGenerator.userTier).toBe(UserTierId.STANDARD);
+        });
+        it('should delegate userTierName to wrapped', () => {
+            wrapped.userTierName = 'Standard Tier';
+            expect(loggingContentGenerator.userTierName).toBe('Standard Tier');
+        });
+    });
+});
+//# sourceMappingURL=loggingContentGenerator.test.js.map

package/dist/src/core/loggingContentGenerator.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"loggingContentGenerator.test.js","sourceRoot":"","sources":["../../../src/core/loggingContentGenerator.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAChD,MAAM,cAAc,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AACjD,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;AAE9C,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,aAAa;IACb,cAAc;IACd,WAAW;CACZ,CAAC,CAAC,CAAC;AAEJ,MAAM,iBAAiB,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CACxC,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAClE,CAAC;AAEF,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,iBAAiB;CAClB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAMzE,OAAO,EAAE,uBAAuB,EAAE,MAAM,8BAA8B,CAAC;AAEvE,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAErD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,OAAyB,CAAC;IAC9B,IAAI,MAAc,CAAC;IACnB,IAAI,uBAAgD,CAAC;IAErD,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG;YACR,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC9B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;SACtB,CAAC;QACF,MAAM,GAAG;YACP,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACjD,QAAQ,EAAE,SAAS;aACpB,CAAC;SACkB,CAAC;QACvB,uBAAuB,GAAG,IAAI,uBAAuB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACvE,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACxD,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,MAAM,YAAY,GAAG,YAAY,CAAC;YAClC,MAAM,QAAQ,GAA4B;gBACxC,UAAU,EAAE,EAAE;gBACd,aAAa,EAAE;oBACb,gBAAgB,EAAE,CAAC;oBACnB,oBAAoB,EAAE,CAAC;oBACvB,eAAe,EAAE,CAAC;iBACnB;gBACD,IAAI,EAAE,SAAS;gBACf,aAAa,EAAE,SAAS;gBACxB,cAAc,EAAE,SAAS;gBACzB,mBAAmB,EAAE,SAAS;gBAC9B,IAAI,EAAE,SAAS;aAChB,CAAC;YACF,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAC/D,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvD,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAE5B,MAAM,OAAO,GAAG,uBAAuB,CAAC,eAAe,CACrD,GAAG,EACH,YAAY,CACb,CAAC;YAEF,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE7B,MAAM,OAAO,CAAC;YAEd,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YACxE,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,EACN,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAC5B,CAAC;YACF,MAAM,aAAa,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACxD,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,MAAM,YAAY,GAAG,YAAY,CAAC;YAClC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YACtC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvD,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAE5B,MAAM,OAAO,GAAG,uBAAuB,CAAC,eAAe,CACrD,GAAG,EACH,YAAY,CACb,CAAC;YAEF,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE7B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAE7C,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,EACN,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAC5B,CAAC;YACF,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACxD,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,MAAM,YAAY,GAAG,YAAY,CAAC;YAClC,MAAM,QAAQ,GAAG;gBACf,UAAU,EAAE,EAAE;gBACd,aAAa,EAAE;oBACb,gBAAgB,EAAE,CAAC;oBACnB,oBAAoB,EAAE,CAAC;oBACvB,eAAe,EAAE,CAAC;iBACnB;aACoC,CAAC;YAExC,KAAK,SAAS,CAAC,CAAC,oBAAoB;gBAClC,MAAM,QAAQ,CAAC;YACjB,CAAC;YAED,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CACxD,oBAAoB,EAAE,CACvB,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvD,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAE5B,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,qBAAqB,CAChE,GAAG,EACH,YAAY,CACb,CAAC;YAEF,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE7B,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBAC7B,iBAAiB;YACnB,CAAC;YAED,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CACxD,GAAG,EACH,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,EACN,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAC5B,CAAC;YACF,MAAM,aAAa,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACxD,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,MAAM,YAAY,GAAG,YAAY,CAAC;YAClC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YAEtC,KAAK,SAAS,CAAC,CAAC,oBAAoB;gBAClC,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC;YAED,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CACxD,oBAAoB,EAAE,CACvB,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvD,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAE5B,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,qBAAqB,CAChE,GAAG,EACH,YAAY,CACb,CAAC;YAEF,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAE7B,MAAM,MAAM,CAAC,KAAK,IAAI,EAAE;gBACtB,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;oBAC7B,aAAa;gBACf,CAAC;YACH,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAE1B,MAAM,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACxC,MAAM,EACN,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAC5B,CAAC;YACF,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACxD,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,qEAAqE;YACrE,MAAM,iBAAiB,GAAG,uBAAuB,CAAC;YAClD,MAAM,CAAC,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAErC,KAAK,SAAS,CAAC,CAAC,oBAAoB;gBAClC,MAAM,EAAE,UAAU,EAAE,EAAE,EAAwC,CAAC;YACjE,CAAC;YACD,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CACxD,oBAAoB,EAAE,CACvB,CAAC;YAEF,MAAM,uBAAuB,CAAC,qBAAqB,CACjD,GAAG,EACH,iBAAiB,CAClB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACxD,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,wFAAwF;YACxF,MAAM,gBAAgB,GAAG,6BAA6B,CAAC;YACvD,MAAM,CAAC,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAErC,KAAK,SAAS,CAAC,CAAC,oBAAoB;gBAClC,MAAM,EAAE,UAAU,EAAE,EAAE,EAAwC,CAAC;YACjE,CAAC;YACD,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,iBAAiB,CACxD,oBAAoB,EAAE,CACvB,CAAC;YAEF,MAAM,uBAAuB,CAAC,qBAAqB,CACjD,GAAG,EACH,gBAAgB,CACjB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,uBAAuB,CAAC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,GAAG,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;YAClD,MAAM,QAAQ,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;YACrC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE3D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YAE9D,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,GAAG,GAAG;gBACV,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;gBACvC,KAAK,EAAE,YAAY;aACpB,CAAC;YACF,MAAM,QAAQ,GAAyB,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YACxE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YAE/D,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,OAAO,CAAC,QAAQ,GAAG,UAAU,CAAC,QAAQ,CAAC;YACvC,MAAM,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,OAAO,CAAC,YAAY,GAAG,eAAe,CAAC;YACvC,MAAM,CAAC,uBAAuB,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/prompts-substitution.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/core/prompts-substitution.test.js

@@ -0,0 +1,101 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { getCoreSystemPrompt } from './prompts.js';
+import fs from 'node:fs';
+import * as toolNames from '../tools/tool-names.js';
+vi.mock('node:fs');
+vi.mock('../utils/gitUtils', () => ({
+    isGitRepository: vi.fn().mockReturnValue(false),
+}));
+describe('Core System Prompt Substitution', () => {
+    let mockConfig;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.stubEnv('GEMINI_SYSTEM_MD', 'true');
+        mockConfig = {
+            getToolRegistry: vi.fn().mockReturnValue({
+                getAllToolNames: vi
+                    .fn()
+                    .mockReturnValue([
+                    toolNames.WRITE_FILE_TOOL_NAME,
+                    toolNames.READ_FILE_TOOL_NAME,
+                ]),
+            }),
+            getEnableShellOutputEfficiency: vi.fn().mockReturnValue(true),
+            storage: {
+                getProjectTempDir: vi.fn().mockReturnValue('/tmp/project-temp'),
+            },
+            isInteractive: vi.fn().mockReturnValue(true),
+            isInteractiveShellEnabled: vi.fn().mockReturnValue(true),
+            isAgentsEnabled: vi.fn().mockReturnValue(false),
+            getModel: vi.fn().mockReturnValue('auto'),
+            getActiveModel: vi.fn().mockReturnValue('gemini-1.5-pro'),
+            getPreviewFeatures: vi.fn().mockReturnValue(false),
+            getAgentRegistry: vi.fn().mockReturnValue({
+                getDirectoryContext: vi.fn().mockReturnValue('Mock Agent Directory'),
+            }),
+            getSkillManager: vi.fn().mockReturnValue({
+                getSkills: vi.fn().mockReturnValue([]),
+            }),
+        };
+    });
+    it('should substitute ${AgentSkills} in custom system prompt', () => {
+        const skills = [
+            {
+                name: 'test-skill',
+                description: 'A test skill description',
+                location: '/path/to/test-skill/SKILL.md',
+                body: 'Skill content',
+            },
+        ];
+        vi.mocked(mockConfig.getSkillManager().getSkills).mockReturnValue(skills);
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('Skills go here: ${AgentSkills}');
+        const prompt = getCoreSystemPrompt(mockConfig);
+        expect(prompt).toContain('Skills go here:');
+        expect(prompt).toContain('<available_skills>');
+        expect(prompt).toContain('<name>test-skill</name>');
+        expect(prompt).not.toContain('${AgentSkills}');
+    });
+    it('should substitute ${SubAgents} in custom system prompt', () => {
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('Agents: ${SubAgents}');
+        vi.mocked(mockConfig.getAgentRegistry().getDirectoryContext).mockReturnValue('Actual Agent Directory');
+        const prompt = getCoreSystemPrompt(mockConfig);
+        expect(prompt).toContain('Agents: Actual Agent Directory');
+        expect(prompt).not.toContain('${SubAgents}');
+    });
+    it('should substitute ${AvailableTools} in custom system prompt', () => {
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('Tools:\n${AvailableTools}');
+        const prompt = getCoreSystemPrompt(mockConfig);
+        expect(prompt).toContain(`Tools:\n- ${toolNames.WRITE_FILE_TOOL_NAME}\n- ${toolNames.READ_FILE_TOOL_NAME}`);
+        expect(prompt).not.toContain('${AvailableTools}');
+    });
+    it('should substitute tool names using the ${toolName}_ToolName pattern', () => {
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('Use ${write_file_ToolName} and ${read_file_ToolName}.');
+        const prompt = getCoreSystemPrompt(mockConfig);
+        expect(prompt).toContain(`Use ${toolNames.WRITE_FILE_TOOL_NAME} and ${toolNames.READ_FILE_TOOL_NAME}.`);
+        expect(prompt).not.toContain('${write_file_ToolName}');
+        expect(prompt).not.toContain('${read_file_ToolName}');
+    });
+    it('should not substitute old patterns', () => {
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('${WriteFileToolName} and ${WRITE_FILE_TOOL_NAME}');
+        const prompt = getCoreSystemPrompt(mockConfig);
+        expect(prompt).toBe('${WriteFileToolName} and ${WRITE_FILE_TOOL_NAME}');
+    });
+    it('should not substitute disabled tool names', () => {
+        vi.mocked(mockConfig.getToolRegistry().getAllToolNames).mockReturnValue([]);
+        vi.mocked(fs.existsSync).mockReturnValue(true);
+        vi.mocked(fs.readFileSync).mockReturnValue('Use ${write_file_ToolName}.');
+        const prompt = getCoreSystemPrompt(mockConfig);
+        expect(prompt).toBe('Use ${write_file_ToolName}.');
+    });
+});
+//# sourceMappingURL=prompts-substitution.test.js.map

package/dist/src/core/prompts-substitution.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts-substitution.test.js","sourceRoot":"","sources":["../../../src/core/prompts-substitution.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,MAAM,SAAS,CAAC;AAEzB,OAAO,KAAK,SAAS,MAAM,wBAAwB,CAAC;AAEpD,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACnB,EAAE,CAAC,IAAI,CAAC,mBAAmB,EAAE,GAAG,EAAE,CAAC,CAAC;IAClC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;CAChD,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,IAAI,UAAkB,CAAC;IACvB,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,OAAO,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;QACvC,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACvC,eAAe,EAAE,EAAE;qBAChB,EAAE,EAAE;qBACJ,eAAe,CAAC;oBACf,SAAS,CAAC,oBAAoB;oBAC9B,SAAS,CAAC,mBAAmB;iBAC9B,CAAC;aACL,CAAC;YACF,8BAA8B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC7D,OAAO,EAAE;gBACP,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC;aAChE;YACD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5C,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YACxD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;YAC/C,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC;YACzC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;YACzD,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;YAClD,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACxC,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,sBAAsB,CAAC;aACrE,CAAC;YACF,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACvC,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;aACvC,CAAC;SACkB,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,MAAM,GAAG;YACb;gBACE,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,0BAA0B;gBACvC,QAAQ,EAAE,8BAA8B;gBACxC,IAAI,EAAE,eAAe;aACtB;SACF,CAAC;QACF,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC1E,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CACxC,gCAAgC,CACjC,CAAC;QAEF,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,sBAAsB,CAAC,CAAC;QACnE,EAAE,CAAC,MAAM,CACP,UAAU,CAAC,gBAAgB,EAAE,CAAC,mBAAmB,CAClD,CAAC,eAAe,CAAC,wBAAwB,CAAC,CAAC;QAE5C,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,2BAA2B,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,aAAa,SAAS,CAAC,oBAAoB,OAAO,SAAS,CAAC,mBAAmB,EAAE,CAClF,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CACxC,uDAAuD,CACxD,CAAC;QAEF,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CACtB,OAAO,SAAS,CAAC,oBAAoB,QAAQ,SAAS,CAAC,mBAAmB,GAAG,CAC9E,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CACxC,kDAAkD,CACnD,CAAC;QAEF,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,CAAC,eAAe,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAC5E,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,6BAA6B,CAAC,CAAC;QAE1E,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}