feat: Propagate traceId from code assist to response metadata (Fixes … (#11360)

Co-authored-by: owenofbrien <86964623+owenofbrien@users.noreply.github.com>
This commit is contained in:
Paweł Dec
2025-10-20 22:00:24 +02:00
committed by GitHub
parent 085e5b1f4d
commit 36de686224
7 changed files with 187 additions and 6 deletions

View File

@@ -310,6 +310,27 @@ describe('converter', () => {
const genaiRes = fromGenerateContentResponse(codeAssistRes);
expect(genaiRes.modelVersion).toEqual('gemini-2.5-pro');
});
it('should handle traceId', () => {
const codeAssistRes: CaGenerateContentResponse = {
response: {
candidates: [],
},
traceId: 'my-trace-id',
};
const genaiRes = fromGenerateContentResponse(codeAssistRes);
expect(genaiRes.responseId).toEqual('my-trace-id');
});
it('should handle missing traceId', () => {
const codeAssistRes: CaGenerateContentResponse = {
response: {
candidates: [],
},
};
const genaiRes = fromGenerateContentResponse(codeAssistRes);
expect(genaiRes.responseId).toBeUndefined();
});
});
describe('toContents', () => {

View File

@@ -73,6 +73,7 @@ interface VertexGenerationConfig {
export interface CaGenerateContentResponse {
response: VertexGenerateContentResponse;
traceId?: string;
}
interface VertexGenerateContentResponse {
@@ -139,6 +140,7 @@ export function fromGenerateContentResponse(
out.promptFeedback = inres.promptFeedback;
out.usageMetadata = inres.usageMetadata;
out.modelVersion = inres.modelVersion;
out.responseId = res.traceId;
return out;
}