summaryrefslogtreecommitdiff
path: root/packages/core/src/code_assist/converter.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/core/src/code_assist/converter.ts')
-rw-r--r--packages/core/src/code_assist/converter.ts3
1 files changed, 3 insertions, 0 deletions
diff --git a/packages/core/src/code_assist/converter.ts b/packages/core/src/code_assist/converter.ts
index 8340cfc1..ffd471da 100644
--- a/packages/core/src/code_assist/converter.ts
+++ b/packages/core/src/code_assist/converter.ts
@@ -32,6 +32,7 @@ import {
export interface CAGenerateContentRequest {
model: string;
project?: string;
+ user_prompt_id?: string;
request: VertexGenerateContentRequest;
}
@@ -115,12 +116,14 @@ export function fromCountTokenResponse(
export function toGenerateContentRequest(
req: GenerateContentParameters,
+ userPromptId: string,
project?: string,
sessionId?: string,
): CAGenerateContentRequest {
return {
model: req.model,
project,
+ user_prompt_id: userPromptId,
request: toVertexGenerateContentRequest(req, sessionId),
};
}