@anthropic-field/core 0.1.0

package/dist/src/core/loggingContentGenerator.js

@@ -0,0 +1,202 @@
+/**
+ * @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 { 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;
+    }
+    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;
+        });
+    }
+}
+//# 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;AAeH,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,aAAa,GACd,MAAM,uBAAuB,CAAC;AAG/B,OAAO,EACL,WAAW,EACX,aAAa,EACb,cAAc,GACf,MAAM,yBAAyB,CAAC;AAEjC,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;IAEO,aAAa,CACnB,QAAmB,EACnB,KAAa,EACb,QAAgB,EAChB,gBAAwC,EACxC,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,eAA0B,EAC1B,UAAkB,EAClB,KAAa,EACb,SAAiB,EACjB,UAA8B,EAC9B,kBAAgC,EAChC,aAAoD,EACpD,YAAqB,EACrB,gBAAwC,EACxC,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,eAA0B,EAC1B,gBAAwC,EACxC,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,GAAc,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACrD,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,iBAAmE,CAAC;QACxE,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,uBAAuB,CAAC,CAAC;QACzE,MAAM,eAAe,GAAc,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5D,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;CACF"}

package/dist/src/core/prompts.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+/**
+ * Resolves a path or switch value from an environment variable.
+ * @deprecated Use resolvePathFromEnv from @anthropic-field/core/prompts/utils instead.
+ */
+export declare function resolvePathFromEnv(envVar?: string): import("../prompts/utils.js").ResolvedPath;
+/**
+ * Returns the core system prompt for the agent.
+ */
+export declare function getCoreSystemPrompt(config: Config, userMemory?: string, interactiveOverride?: boolean): string;
+/**
+ * Provides the system prompt for the history compression process.
+ */
+export declare function getCompressionPrompt(): string;

package/dist/src/core/prompts.js

@@ -0,0 +1,27 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { PromptProvider } from '../prompts/promptProvider.js';
+import { resolvePathFromEnv as resolvePathFromEnvImpl } from '../prompts/utils.js';
+/**
+ * Resolves a path or switch value from an environment variable.
+ * @deprecated Use resolvePathFromEnv from @anthropic-field/core/prompts/utils instead.
+ */
+export function resolvePathFromEnv(envVar) {
+    return resolvePathFromEnvImpl(envVar);
+}
+/**
+ * Returns the core system prompt for the agent.
+ */
+export function getCoreSystemPrompt(config, userMemory, interactiveOverride) {
+    return new PromptProvider().getCoreSystemPrompt(config, userMemory, interactiveOverride);
+}
+/**
+ * Provides the system prompt for the history compression process.
+ */
+export function getCompressionPrompt() {
+    return new PromptProvider().getCompressionPrompt();
+}
+//# sourceMappingURL=prompts.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../../src/core/prompts.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,kBAAkB,IAAI,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAEnF;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAAe;IAChD,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAC;AACxC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,UAAmB,EACnB,mBAA6B;IAE7B,OAAO,IAAI,cAAc,EAAE,CAAC,mBAAmB,CAC7C,MAAM,EACN,UAAU,EACV,mBAAmB,CACpB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,OAAO,IAAI,cAAc,EAAE,CAAC,oBAAoB,EAAE,CAAC;AACrD,CAAC"}

package/dist/src/core/recordingContentGenerator.d.ts

@@ -0,0 +1,19 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { CountTokensResponse, GenerateContentParameters, GenerateContentResponse, CountTokensParameters, EmbedContentResponse, EmbedContentParameters } from '@google/genai';
+import type { ContentGenerator } from './contentGenerator.js';
+import type { UserTierId } from '../code_assist/types.js';
+export declare class RecordingContentGenerator implements ContentGenerator {
+    private readonly realGenerator;
+    private readonly filePath;
+    constructor(realGenerator: ContentGenerator, filePath: string);
+    get userTier(): UserTierId | undefined;
+    get userTierName(): string | undefined;
+    generateContent(request: GenerateContentParameters, userPromptId: string): Promise<GenerateContentResponse>;
+    generateContentStream(request: GenerateContentParameters, userPromptId: string): Promise<AsyncGenerator<GenerateContentResponse>>;
+    countTokens(request: CountTokensParameters): Promise<CountTokensResponse>;
+    embedContent(request: EmbedContentParameters): Promise<EmbedContentResponse>;
+}

package/dist/src/core/recordingContentGenerator.js

@@ -0,0 +1,82 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { appendFileSync } from 'node:fs';
+import { safeJsonStringify } from '../utils/safeJsonStringify.js';
+// A ContentGenerator that wraps another content generator and records all the
+// responses, with the ability to write them out to a file. These files are
+// intended to be consumed later on by a FakeContentGenerator, given the
+// `--fake-responses` CLI argument.
+//
+// Note that only the "interesting" bits of the responses are actually kept.
+export class RecordingContentGenerator {
+    realGenerator;
+    filePath;
+    constructor(realGenerator, filePath) {
+        this.realGenerator = realGenerator;
+        this.filePath = filePath;
+    }
+    get userTier() {
+        return this.realGenerator.userTier;
+    }
+    get userTierName() {
+        return this.realGenerator.userTierName;
+    }
+    async generateContent(request, userPromptId) {
+        const response = await this.realGenerator.generateContent(request, userPromptId);
+        const recordedResponse = {
+            method: 'generateContent',
+            response: {
+                candidates: response.candidates,
+                usageMetadata: response.usageMetadata,
+            },
+        };
+        appendFileSync(this.filePath, `${safeJsonStringify(recordedResponse)}\n`);
+        return response;
+    }
+    async generateContentStream(request, userPromptId) {
+        const recordedResponse = {
+            method: 'generateContentStream',
+            response: [],
+        };
+        const realResponses = await this.realGenerator.generateContentStream(request, userPromptId);
+        async function* stream(filePath) {
+            for await (const response of realResponses) {
+                recordedResponse.response.push({
+                    candidates: response.candidates,
+                    usageMetadata: response.usageMetadata,
+                });
+                yield response;
+            }
+            appendFileSync(filePath, `${safeJsonStringify(recordedResponse)}\n`);
+        }
+        return Promise.resolve(stream(this.filePath));
+    }
+    async countTokens(request) {
+        const response = await this.realGenerator.countTokens(request);
+        const recordedResponse = {
+            method: 'countTokens',
+            response: {
+                totalTokens: response.totalTokens,
+                cachedContentTokenCount: response.cachedContentTokenCount,
+            },
+        };
+        appendFileSync(this.filePath, `${safeJsonStringify(recordedResponse)}\n`);
+        return response;
+    }
+    async embedContent(request) {
+        const response = await this.realGenerator.embedContent(request);
+        const recordedResponse = {
+            method: 'embedContent',
+            response: {
+                embeddings: response.embeddings,
+                metadata: response.metadata,
+            },
+        };
+        appendFileSync(this.filePath, `${safeJsonStringify(recordedResponse)}\n`);
+        return response;
+    }
+}
+//# sourceMappingURL=recordingContentGenerator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"recordingContentGenerator.js","sourceRoot":"","sources":["../../../src/core/recordingContentGenerator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAUH,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAIzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,+BAA+B,CAAC;AAElE,8EAA8E;AAC9E,2EAA2E;AAC3E,wEAAwE;AACxE,mCAAmC;AACnC,EAAE;AACF,4EAA4E;AAC5E,MAAM,OAAO,yBAAyB;IAEjB;IACA;IAFnB,YACmB,aAA+B,EAC/B,QAAgB;QADhB,kBAAa,GAAb,aAAa,CAAkB;QAC/B,aAAQ,GAAR,QAAQ,CAAQ;IAChC,CAAC;IAEJ,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC;IACrC,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,OAAkC,EAClC,YAAoB;QAEpB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,eAAe,CACvD,OAAO,EACP,YAAY,CACb,CAAC;QACF,MAAM,gBAAgB,GAAiB;YACrC,MAAM,EAAE,iBAAiB;YACzB,QAAQ,EAAE;gBACR,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,aAAa,EAAE,QAAQ,CAAC,aAAa;aACX;SAC7B,CAAC;QACF,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC1E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,OAAkC,EAClC,YAAoB;QAEpB,MAAM,gBAAgB,GAAiB;YACrC,MAAM,EAAE,uBAAuB;YAC/B,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,qBAAqB,CAClE,OAAO,EACP,YAAY,CACb,CAAC;QAEF,KAAK,SAAS,CAAC,CAAC,MAAM,CAAC,QAAgB;YACrC,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;gBAC1C,gBAAgB,CAAC,QAAsC,CAAC,IAAI,CAAC;oBAC5D,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,aAAa,EAAE,QAAQ,CAAC,aAAa;iBACX,CAAC,CAAC;gBAC9B,MAAM,QAAQ,CAAC;YACjB,CAAC;YACD,cAAc,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACvE,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,WAAW,CACf,OAA8B;QAE9B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC/D,MAAM,gBAAgB,GAAiB;YACrC,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE;gBACR,WAAW,EAAE,QAAQ,CAAC,WAAW;gBACjC,uBAAuB,EAAE,QAAQ,CAAC,uBAAuB;aAC1D;SACF,CAAC;QACF,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC1E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,OAA+B;QAE/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAEhE,MAAM,gBAAgB,GAAiB;YACrC,MAAM,EAAE,cAAc;YACtB,QAAQ,EAAE;gBACR,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;aAC5B;SACF,CAAC;QACF,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC1E,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/src/core/tokenLimits.d.ts

@@ -0,0 +1,10 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+type Model = string;
+type TokenCount = number;
+export declare const DEFAULT_TOKEN_LIMIT = 1048576;
+export declare function tokenLimit(model: Model): TokenCount;
+export {};

package/dist/src/core/tokenLimits.js

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DEFAULT_GEMINI_FLASH_LITE_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_FLASH_MODEL, PREVIEW_GEMINI_MODEL, } from '../config/models.js';
+export const DEFAULT_TOKEN_LIMIT = 1_048_576;
+export function tokenLimit(model) {
+    // Add other models as they become relevant or if specified by config
+    // Pulled from https://ai.google.dev/gemini-api/docs/models
+    switch (model) {
+        case PREVIEW_GEMINI_MODEL:
+        case PREVIEW_GEMINI_FLASH_MODEL:
+        case DEFAULT_GEMINI_MODEL:
+        case DEFAULT_GEMINI_FLASH_MODEL:
+        case DEFAULT_GEMINI_FLASH_LITE_MODEL:
+            return 1_048_576;
+        default:
+            return DEFAULT_TOKEN_LIMIT;
+    }
+}
+//# sourceMappingURL=tokenLimits.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tokenLimits.js","sourceRoot":"","sources":["../../../src/core/tokenLimits.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,+BAA+B,EAC/B,0BAA0B,EAC1B,oBAAoB,EACpB,0BAA0B,EAC1B,oBAAoB,GACrB,MAAM,qBAAqB,CAAC;AAK7B,MAAM,CAAC,MAAM,mBAAmB,GAAG,SAAS,CAAC;AAE7C,MAAM,UAAU,UAAU,CAAC,KAAY;IACrC,qEAAqE;IACrE,2DAA2D;IAC3D,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,oBAAoB,CAAC;QAC1B,KAAK,0BAA0B,CAAC;QAChC,KAAK,oBAAoB,CAAC;QAC1B,KAAK,0BAA0B,CAAC;QAChC,KAAK,+BAA+B;YAClC,OAAO,SAAS,CAAC;QACnB;YACE,OAAO,mBAAmB,CAAC;IAC/B,CAAC;AACH,CAAC"}

package/dist/src/core/turn.d.ts

@@ -0,0 +1,167 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { PartListUnion, GenerateContentResponse, FunctionDeclaration, FinishReason, GenerateContentResponseUsageMetadata } from '@google/genai';
+import type { ToolCallConfirmationDetails, ToolResult } from '../tools/tools.js';
+import type { GeminiChat } from './geminiChat.js';
+import { type ThoughtSummary } from '../utils/thoughtUtils.js';
+import type { ModelConfigKey } from '../services/modelConfigService.js';
+import { type ToolCallRequestInfo, type ToolCallResponseInfo } from '../scheduler/types.js';
+export interface ServerTool {
+    name: string;
+    schema: FunctionDeclaration;
+    execute(params: Record<string, unknown>, signal?: AbortSignal): Promise<ToolResult>;
+    shouldConfirmExecute(params: Record<string, unknown>, abortSignal: AbortSignal): Promise<ToolCallConfirmationDetails | false>;
+}
+export declare enum GeminiEventType {
+    Content = "content",
+    ToolCallRequest = "tool_call_request",
+    ToolCallResponse = "tool_call_response",
+    ToolCallConfirmation = "tool_call_confirmation",
+    UserCancelled = "user_cancelled",
+    Error = "error",
+    ChatCompressed = "chat_compressed",
+    Thought = "thought",
+    MaxSessionTurns = "max_session_turns",
+    Finished = "finished",
+    LoopDetected = "loop_detected",
+    Citation = "citation",
+    Retry = "retry",
+    ContextWindowWillOverflow = "context_window_will_overflow",
+    InvalidStream = "invalid_stream",
+    ModelInfo = "model_info",
+    AgentExecutionStopped = "agent_execution_stopped",
+    AgentExecutionBlocked = "agent_execution_blocked"
+}
+export type ServerGeminiRetryEvent = {
+    type: GeminiEventType.Retry;
+};
+export type ServerGeminiAgentExecutionStoppedEvent = {
+    type: GeminiEventType.AgentExecutionStopped;
+    value: {
+        reason: string;
+        systemMessage?: string;
+        contextCleared?: boolean;
+    };
+};
+export type ServerGeminiAgentExecutionBlockedEvent = {
+    type: GeminiEventType.AgentExecutionBlocked;
+    value: {
+        reason: string;
+        systemMessage?: string;
+        contextCleared?: boolean;
+    };
+};
+export type ServerGeminiContextWindowWillOverflowEvent = {
+    type: GeminiEventType.ContextWindowWillOverflow;
+    value: {
+        estimatedRequestTokenCount: number;
+        remainingTokenCount: number;
+    };
+};
+export type ServerGeminiInvalidStreamEvent = {
+    type: GeminiEventType.InvalidStream;
+};
+export type ServerGeminiModelInfoEvent = {
+    type: GeminiEventType.ModelInfo;
+    value: string;
+};
+export interface StructuredError {
+    message: string;
+    status?: number;
+}
+export interface GeminiErrorEventValue {
+    error: StructuredError;
+}
+export interface GeminiFinishedEventValue {
+    reason: FinishReason | undefined;
+    usageMetadata: GenerateContentResponseUsageMetadata | undefined;
+}
+export interface ServerToolCallConfirmationDetails {
+    request: ToolCallRequestInfo;
+    details: ToolCallConfirmationDetails;
+}
+export type ServerGeminiContentEvent = {
+    type: GeminiEventType.Content;
+    value: string;
+    traceId?: string;
+};
+export type ServerGeminiThoughtEvent = {
+    type: GeminiEventType.Thought;
+    value: ThoughtSummary;
+    traceId?: string;
+};
+export type ServerGeminiToolCallRequestEvent = {
+    type: GeminiEventType.ToolCallRequest;
+    value: ToolCallRequestInfo;
+};
+export type ServerGeminiToolCallResponseEvent = {
+    type: GeminiEventType.ToolCallResponse;
+    value: ToolCallResponseInfo;
+};
+export type ServerGeminiToolCallConfirmationEvent = {
+    type: GeminiEventType.ToolCallConfirmation;
+    value: ServerToolCallConfirmationDetails;
+};
+export type ServerGeminiUserCancelledEvent = {
+    type: GeminiEventType.UserCancelled;
+};
+export type ServerGeminiErrorEvent = {
+    type: GeminiEventType.Error;
+    value: GeminiErrorEventValue;
+};
+export declare enum CompressionStatus {
+    /** The compression was successful */
+    COMPRESSED = 1,
+    /** The compression failed due to the compression inflating the token count */
+    COMPRESSION_FAILED_INFLATED_TOKEN_COUNT = 2,
+    /** The compression failed due to an error counting tokens */
+    COMPRESSION_FAILED_TOKEN_COUNT_ERROR = 3,
+    /** The compression failed because the summary was empty */
+    COMPRESSION_FAILED_EMPTY_SUMMARY = 4,
+    /** The compression was not necessary and no action was taken */
+    NOOP = 5
+}
+export interface ChatCompressionInfo {
+    originalTokenCount: number;
+    newTokenCount: number;
+    compressionStatus: CompressionStatus;
+}
+export type ServerGeminiChatCompressedEvent = {
+    type: GeminiEventType.ChatCompressed;
+    value: ChatCompressionInfo | null;
+};
+export type ServerGeminiMaxSessionTurnsEvent = {
+    type: GeminiEventType.MaxSessionTurns;
+};
+export type ServerGeminiFinishedEvent = {
+    type: GeminiEventType.Finished;
+    value: GeminiFinishedEventValue;
+};
+export type ServerGeminiLoopDetectedEvent = {
+    type: GeminiEventType.LoopDetected;
+};
+export type ServerGeminiCitationEvent = {
+    type: GeminiEventType.Citation;
+    value: string;
+};
+export type ServerGeminiStreamEvent = ServerGeminiChatCompressedEvent | ServerGeminiCitationEvent | ServerGeminiContentEvent | ServerGeminiErrorEvent | ServerGeminiFinishedEvent | ServerGeminiLoopDetectedEvent | ServerGeminiMaxSessionTurnsEvent | ServerGeminiThoughtEvent | ServerGeminiToolCallConfirmationEvent | ServerGeminiToolCallRequestEvent | ServerGeminiToolCallResponseEvent | ServerGeminiUserCancelledEvent | ServerGeminiRetryEvent | ServerGeminiContextWindowWillOverflowEvent | ServerGeminiInvalidStreamEvent | ServerGeminiModelInfoEvent | ServerGeminiAgentExecutionStoppedEvent | ServerGeminiAgentExecutionBlockedEvent;
+export declare class Turn {
+    private readonly chat;
+    private readonly prompt_id;
+    readonly pendingToolCalls: ToolCallRequestInfo[];
+    private debugResponses;
+    private pendingCitations;
+    finishReason: FinishReason | undefined;
+    constructor(chat: GeminiChat, prompt_id: string);
+    run(modelConfigKey: ModelConfigKey, req: PartListUnion, signal: AbortSignal, displayContent?: PartListUnion): AsyncGenerator<ServerGeminiStreamEvent>;
+    private handlePendingFunctionCall;
+    getDebugResponses(): GenerateContentResponse[];
+    /**
+     * Get the concatenated response text from all responses in this turn.
+     * This extracts and joins all text content from the model's responses.
+     */
+    getResponseText(): string;
+}