feat: Add auth_type to StartSessionEvent OTel logging (#12034)

Co-authored-by: Shnatu <snatu@google.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Shardul Natu
2025-10-29 12:28:28 -07:00
committed by GitHub
parent c60d8ef5a8
commit 66e981ed1e
2 changed files with 4 additions and 0 deletions

View File

@@ -231,6 +231,7 @@ describe('loggers', () => {
output_format: 'json',
extension_ids: '',
extensions_count: 0,
auth_type: 'vertex-ai',
},
});
});

View File

@@ -68,6 +68,7 @@ export class StartSessionEvent implements BaseTelemetryEvent {
output_format: OutputFormat;
extensions_count: number;
extension_ids: string;
auth_type?: string;
constructor(config: Config, toolRegistry?: ToolRegistry) {
const generatorConfig = config.getContentGeneratorConfig();
@@ -102,6 +103,7 @@ export class StartSessionEvent implements BaseTelemetryEvent {
const extensions = config.getExtensions();
this.extensions_count = extensions.length;
this.extension_ids = extensions.map((e) => e.id).join(',');
this.auth_type = generatorConfig?.authType;
if (toolRegistry) {
const mcpTools = toolRegistry
.getAllTools()
@@ -135,6 +137,7 @@ export class StartSessionEvent implements BaseTelemetryEvent {
output_format: this.output_format,
extensions_count: this.extensions_count,
extension_ids: this.extension_ids,
auth_type: this.auth_type,
};
}