This commit is contained in:
Anjali Sridhar 2025-07-23 06:49:27 -07:00
parent 7c3a84075d
commit 1de246236b
1 changed files with 4 additions and 12 deletions

View File

@ -26,10 +26,6 @@ import {
logApiResponse,
logApiError,
} from '../telemetry/loggers.js';
import {
getStructuredResponse,
getStructuredResponseFromParts,
} from '../utils/generateContentResponseUtilities.js';
import {
ApiErrorEvent,
ApiRequestEvent,
@ -142,11 +138,7 @@ export class GeminiChat {
}
private _getRequestTextFromContents(contents: Content[]): string {
return contents
.flatMap((content) => content.parts ?? [])
.map((part) => part.text)
.filter(Boolean)
.join('');
return JSON.stringify(contents);
}
private async _logApiRequest(
@ -318,7 +310,7 @@ export class GeminiChat {
durationMs,
prompt_id,
response.usageMetadata,
getStructuredResponse(response),
JSON.stringify(response),
);
this.sendPromise = (async () => {
@ -554,12 +546,12 @@ export class GeminiChat {
allParts.push(...content.parts);
}
}
const fullText = getStructuredResponseFromParts(allParts);
await this._logApiResponse(
durationMs,
prompt_id,
this.getFinalUsageMetadata(chunks),
fullText,
JSON.stringify(chunks),
);
}
this.recordHistory(inputContent, outputContent);